Something went wrong. Try again.
Monorepo for Tangled tangled.org
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107package gitutil
import ( "bytes" "context" "fmt" "os" "os/exec" "strings" "syscall" "time")
// git runs network work in child processes, so killing only git leaves them running
// WaitDelay stops waits on output held open by git's childrenconst WaitDelay = 10 * time.Second
// LC_ALL=C keeps git's messages in english, callers match on its stderrfunc Command(ctx context.Context, args ...string) *exec.Cmd { cmd := exec.CommandContext(ctx, "git", args...) cmd.Env = append(append(os.Environ(), httpHeaderEnv(ctx)...), "LC_ALL=C") return Contain(cmd)}
type httpHeaderKey struct{}
type httpHeader struct { url string header func() string}
// header runs per git command to keep a short lived token fresh.// env, not argv, because ps shows argvfunc WithHTTPHeader(ctx context.Context, url string, header func() string) context.Context { return context.WithValue(ctx, httpHeaderKey{}, httpHeader{url, header})}
func httpHeaderEnv(ctx context.Context) []string { h, ok := ctx.Value(httpHeaderKey{}).(httpHeader) if !ok { return nil } value := h.header() if value == "" { return nil } return []string{ "GIT_CONFIG_COUNT=1", "GIT_CONFIG_KEY_0=http." + h.url + ".extraHeader", "GIT_CONFIG_VALUE_0=" + value, }}
// Contain is for sandbox-rebuilt commands and direct git binariesfunc Contain(cmd *exec.Cmd) *exec.Cmd { if cmd.SysProcAttr == nil { cmd.SysProcAttr = &syscall.SysProcAttr{} } cmd.SysProcAttr.Setpgid = true cmd.WaitDelay = WaitDelay // exec rejects Cancel on commands not created with CommandContext if cmd.Cancel != nil { cmd.Cancel = func() error { if cmd.Process == nil { return nil } err := syscall.Kill(-cmd.Process.Pid, syscall.SIGKILL) if err == syscall.ESRCH { return nil } return err } } return cmd}
func Run(ctx context.Context, args ...string) error { var stderr bytes.Buffer cmd := Command(ctx, args...) cmd.Stderr = &stderr if err := cmd.Run(); err != nil { return fmt.Errorf("%w: %s", err, strings.TrimSpace(stderr.String())) } return nil}
func Output(ctx context.Context, args ...string) ([]byte, error) { return output(Command(ctx, args...))}
func OutputOffline(ctx context.Context, args ...string) ([]byte, error) { cmd := Command(ctx, args...) cmd.Env = append(cmd.Env, "GIT_NO_LAZY_FETCH=1") return output(cmd)}
func output(cmd *exec.Cmd) ([]byte, error) { var stdout, stderr bytes.Buffer cmd.Stdout = &stdout cmd.Stderr = &stderr if err := cmd.Run(); err != nil { return nil, fmt.Errorf("%w: %s", err, strings.TrimSpace(stderr.String())) } return stdout.Bytes(), nil}