Something went wrong. Try again.
Monorepo for Tangled
Something went wrong. Try again.
2.7 kB · 140 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141package service
import ( "bytes" "fmt" "io" "log" "net/http" "os/exec" "strings" "sync" "syscall"
"tangled.org/core/knotserver/sandbox")
// Mostly from charmbracelet/soft-serve and sosedoff/gitkit.
type ServiceCommand struct { GitProtocol string Dir string Stdin io.Reader Stdout http.ResponseWriter Sandbox sandbox.Backend}
func (c *ServiceCommand) RunService(cmd *exec.Cmd) error { cmd.Env = append(cmd.Env, fmt.Sprintf("GIT_PROTOCOL=%s", c.GitProtocol))
if c.Sandbox != nil { var wrapErr error cmd, wrapErr = c.Sandbox.Wrap(c.Dir, cmd) if wrapErr != nil { return fmt.Errorf("sandbox wrap: %w", wrapErr) } } else { cmd.Dir = c.Dir }
if cmd.SysProcAttr == nil { cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true} }
var stderr bytes.Buffer cmd.Stderr = &stderr
stdoutPipe, err := cmd.StdoutPipe() if err != nil { return fmt.Errorf("failed to create stdout pipe: %w", err) }
stdinPipe, err := cmd.StdinPipe() if err != nil { return fmt.Errorf("failed to create stdin pipe: %w", err) }
if err := cmd.Start(); err != nil { return fmt.Errorf("failed to start '%s': %w", cmd.String(), err) }
var wg sync.WaitGroup
if c.Stdin != nil { wg.Add(1) go func() { defer wg.Done() defer stdinPipe.Close() io.Copy(stdinPipe, c.Stdin) }() }
if c.Stdout != nil { wg.Add(1) go func() { defer wg.Done() io.Copy(newWriteFlusher(c.Stdout), stdoutPipe) stdoutPipe.Close() }() }
wg.Wait()
if err := cmd.Wait(); err != nil { return fmt.Errorf("'%s' failed: %w, stderr: %s", cmd.String(), err, stderr.String()) }
return nil}
func (c *ServiceCommand) InfoRefs() error { cmd := exec.Command("git", []string{ "upload-pack", "--stateless-rpc", "--http-backend-info-refs", ".", }...)
if !strings.Contains(c.GitProtocol, "version=2") { if err := packLine(c.Stdout, "# service=git-upload-pack\n"); err != nil { log.Printf("git: failed to write pack line: %s", err) return err }
if err := packFlush(c.Stdout); err != nil { log.Printf("git: failed to flush pack: %s", err) return err } }
return c.RunService(cmd)}
func (c *ServiceCommand) UploadArchive() error { cmd := exec.Command("git", []string{ "upload-archive", ".", }...)
return c.RunService(cmd)}
func (c *ServiceCommand) UploadPack() error { cmd := exec.Command("git", []string{ "upload-pack", "--stateless-rpc", ".", }...)
return c.RunService(cmd)}
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}