mugit/internal/git/gitservice/gitservice.go(view raw)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
package gitservice
import (
"bytes"
"fmt"
"io"
"os/exec"
"strings"
"syscall"
)
// Thanks https://git.icyphox.sh/legit/blob/master/git/service/service.go
// InfoRefs executes git-upload-pack --advertise-refs for smart-HTTP discovery.
func InfoRefs(dir string, out io.Writer) error {
cmd := exec.Command("git",
"upload-pack",
"--stateless-rpc",
"--advertise-refs",
".")
cmd.Dir = dir
cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
stdoutPipe, _ := cmd.StdoutPipe()
cmd.Stderr = cmd.Stdout
defer stdoutPipe.Close()
if err := cmd.Start(); err != nil {
return fmt.Errorf("start git-upload-pack: %w", err)
}
if err := packLine(out, "# service=git-upload-pack\n"); err != nil {
return fmt.Errorf("write pack line: %w", err)
}
if err := packFlush(out); err != nil {
return fmt.Errorf("flush pack: %w", err)
}
var buf bytes.Buffer
if _, err := io.Copy(&buf, stdoutPipe); err != nil {
return fmt.Errorf("copy stdout to buffer: %w", err)
}
if err := cmd.Wait(); err != nil {
var out strings.Builder
io.Copy(&out, &buf)
return fmt.Errorf("git-upload-pack: %w: %s", err, out.String())
}
if _, err := io.Copy(out, &buf); err != nil {
return fmt.Errorf("copy buffer to output: %w", err)
}
return nil
}
// UploadPack executes git-upload-pack for smart-HTTP git fetch/clone.
func UploadPack(dir string, in io.Reader, out io.Writer) error {
cmd := exec.Command("git",
"-c", "uploadpack.allowFilter=true",
"upload-pack",
"--stateless-rpc",
".")
cmd.Dir = dir
cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
stdoutPipe, _ := cmd.StdoutPipe()
cmd.Stderr = cmd.Stdout
defer stdoutPipe.Close()
stdinPipe, err := cmd.StdinPipe()
if err != nil {
return fmt.Errorf("stdin pipe: %w", err)
}
defer stdinPipe.Close()
if err := cmd.Start(); err != nil {
return fmt.Errorf("start git-upload-pack: %w", err)
}
if _, err := io.Copy(stdinPipe, in); err != nil {
return fmt.Errorf("copy to stdin: %w", err)
}
stdinPipe.Close()
if _, err := io.Copy(out, stdoutPipe); err != nil {
return fmt.Errorf("copy stdout: %w", err)
}
if err := cmd.Wait(); err != nil {
return fmt.Errorf("git-upload-pack: %w", err)
}
return nil
}
func packLine(w io.Writer, s string) error {
_, err := fmt.Fprintf(w, "%04x%s", len(s)+4, s)
return err
}
func packFlush(w io.Writer) error {
_, err := fmt.Fprint(w, "0000")
return err
}
|