mesh-tools
1572e50e0e
mesh/merge-gate head 1572e50e0ef0 merged into trunk 4024909a2960 — pass, with a note: builds mesh-sdk (+41 dependent(s)) → ace, g14, novox, shanks; no bus …
mesh/repo-check head 1572e50e0ef0 merged into trunk 4024909a2960 — pass: its merge-check.sh passed, each part in its toolchain: merge-check.sh (typescript)…
mesh/delivery delivered
94 lines
3.0 KiB
Go
94 lines
3.0 KiB
Go
package forgekey
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// Ran is what a command did: its output, its exit status, and why it never ran to an answer.
|
|
type Ran struct {
|
|
Stdout string
|
|
Stderr string
|
|
Status int
|
|
// Err is "ENOENT" when the program is not installed, or says it was ended for taking too long.
|
|
Err string
|
|
}
|
|
|
|
// Runner runs one command with extra environment words, so every tool can be tested without ssh.
|
|
type Runner func(ctx context.Context, env []string, name string, args ...string) Ran
|
|
|
|
// CallTimeout bounds one command of ExecRunner: below the runtime's thirty-second call limit. A variable so a
|
|
// test can lower it.
|
|
var CallTimeout = 20 * time.Second
|
|
|
|
// ExecRunner runs a command on this machine, with no terminal and nothing on its stdin, bounded by
|
|
// CallTimeout.
|
|
func ExecRunner(ctx context.Context, env []string, name string, args ...string) Ran {
|
|
return ExecRunnerFor(CallTimeout)(ctx, env, name, args...)
|
|
}
|
|
|
|
// ExecRunnerFor is ExecRunner with its own bound (issue 1497): a build module's long step passes 0, so only the
|
|
// step's own context ends a command, and a fetch may outlast a tool call's thirty seconds.
|
|
func ExecRunnerFor(bound time.Duration) Runner {
|
|
return func(ctx context.Context, env []string, name string, args ...string) Ran {
|
|
return execRun(ctx, bound, false, env, name, args...)
|
|
}
|
|
}
|
|
|
|
// ExecRunnerClean is ExecRunnerFor whose commands get exactly the env argument as their environment, and nothing of
|
|
// this process's (issue 1497; novox/hq ADR 0466 §4: no registry password in a fetch's environment). The caller
|
|
// gives PATH and HOME itself. A directory is the command's own business (`git -C`, `go -C`).
|
|
func ExecRunnerClean(bound time.Duration) Runner {
|
|
return func(ctx context.Context, env []string, name string, args ...string) Ran {
|
|
return execRun(ctx, bound, true, env, name, args...)
|
|
}
|
|
}
|
|
|
|
func execRun(ctx context.Context, bound time.Duration, clean bool, env []string, name string, args ...string) Ran {
|
|
if bound > 0 {
|
|
var cancel context.CancelFunc
|
|
ctx, cancel = context.WithTimeout(ctx, bound)
|
|
defer cancel()
|
|
}
|
|
cmd := exec.CommandContext(ctx, name, args...)
|
|
if clean {
|
|
cmd.Env = append([]string{}, env...) // non-nil, so an empty list is an empty environment
|
|
} else {
|
|
cmd.Env = append(os.Environ(), env...)
|
|
}
|
|
var out, errb bytes.Buffer
|
|
cmd.Stdout, cmd.Stderr = &out, &errb
|
|
err := cmd.Run()
|
|
r := Ran{Stdout: out.String(), Stderr: errb.String()}
|
|
var exitErr *exec.ExitError
|
|
switch {
|
|
case errors.Is(ctx.Err(), context.DeadlineExceeded):
|
|
r.Status, r.Err = 124, "ended by its step's deadline"
|
|
if bound > 0 {
|
|
r.Err = fmt.Sprintf("no answer within %d s", int(bound/time.Second))
|
|
}
|
|
case errors.Is(err, exec.ErrNotFound):
|
|
r.Status, r.Err = 127, "ENOENT"
|
|
case errors.As(err, &exitErr):
|
|
r.Status = exitErr.ExitCode()
|
|
case err != nil:
|
|
r.Status, r.Err = 1, err.Error()
|
|
}
|
|
return r
|
|
}
|
|
|
|
func firstLine(s string) string {
|
|
for _, l := range strings.Split(s, "\n") {
|
|
if l = strings.TrimSpace(l); l != "" {
|
|
return l
|
|
}
|
|
}
|
|
return ""
|
|
}
|