Something went wrong. Try again.
Monorepo for Tangled
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492package nixery
import ( "bufio" "context" "errors" "fmt" "io" "log/slog" "path" "runtime" "sync" "time"
"github.com/docker/docker/api/types/container" "github.com/docker/docker/api/types/image" "github.com/docker/docker/api/types/mount" "github.com/docker/docker/api/types/network" "github.com/docker/docker/client" "github.com/docker/docker/pkg/stdcopy" "gopkg.in/yaml.v3" "tangled.org/core/api/tangled" "tangled.org/core/log" "tangled.org/core/spindle/config" "tangled.org/core/spindle/engine" "tangled.org/core/spindle/models" "tangled.org/core/spindle/secrets")
const ( workspaceDir = "/tangled/workspace" homeDir = "/tangled/home")
type cleanupFunc func(context.Context) error
type Engine struct { docker client.APIClient l *slog.Logger cfg *config.Config
cleanupMu sync.Mutex cleanup map[string][]cleanupFunc}
type Step struct { name string kind models.StepKind command string environment map[string]string}
func (s Step) Name() string { return s.name}
func (s Step) Command() string { return s.command}
func (s Step) Kind() models.StepKind { return s.kind}
// setupSteps get added to start of Stepstype setupSteps []models.Step
// addStep adds a step to the beginning of the workflow's steps.func (ss *setupSteps) addStep(step models.Step) { *ss = append(*ss, step)}
type addlFields struct { image string container string mounts []mount.Mount}
func (e *Engine) InitWorkflow(twf tangled.Pipeline_Workflow, tpl tangled.Pipeline) (*models.Workflow, error) { swf := &models.Workflow{} addl := addlFields{}
dwf := &struct { Steps []struct { Command string `yaml:"command"` Name string `yaml:"name"` Environment map[string]string `yaml:"environment"` } `yaml:"steps"` Dependencies map[string][]string `yaml:"dependencies"` Environment map[string]string `yaml:"environment"` }{} err := yaml.Unmarshal([]byte(twf.Raw), &dwf) if err != nil { return nil, err }
for _, dstep := range dwf.Steps { sstep := Step{} sstep.environment = dstep.Environment sstep.command = dstep.Command sstep.name = dstep.Name sstep.kind = models.StepKindUser swf.Steps = append(swf.Steps, sstep) } swf.Name = twf.Name swf.Environment = dwf.Environment addl.image = workflowImage(dwf.Dependencies, e.cfg.NixeryPipelines.Nixery)
if sock := e.cfg.Server.DockerSocket; sock != "" { addl.mounts = append(addl.mounts, mount.Mount{ Type: mount.TypeBind, Source: sock, Target: sock, ReadOnly: false, }) } setup := &setupSteps{}
setup.addStep(nixConfStep()) setup.addStep(models.BuildCloneStep(twf, *tpl.TriggerMetadata, e.cfg.Server.Dev)) // this step could be empty if s := dependencyStep(dwf.Dependencies); s != nil { setup.addStep(*s) }
// append setup steps in order to the start of workflow steps swf.Steps = append(*setup, swf.Steps...) swf.Data = addl
return swf, nil}
func (e *Engine) WorkflowTimeout() time.Duration { workflowTimeoutStr := e.cfg.NixeryPipelines.WorkflowTimeout workflowTimeout, err := time.ParseDuration(workflowTimeoutStr) if err != nil { e.l.Error("failed to parse workflow timeout", "error", err, "timeout", workflowTimeoutStr) workflowTimeout = 5 * time.Minute }
return workflowTimeout}
func workflowImage(deps map[string][]string, nixery string) string { var dependencies string for reg, ds := range deps { if reg == "nixpkgs" { dependencies = path.Join(ds...) } }
// load defaults from somewhere else dependencies = path.Join(dependencies, "bash", "git", "coreutils", "nix")
if runtime.GOARCH == "arm64" { dependencies = path.Join("arm64", dependencies) }
return path.Join(nixery, dependencies)}
func New(ctx context.Context, cfg *config.Config) (*Engine, error) { dcli, err := client.NewClientWithOpts(client.FromEnv, client.WithAPIVersionNegotiation()) if err != nil { return nil, err }
l := log.FromContext(ctx).With("component", "spindle")
e := &Engine{ docker: dcli, l: l, cfg: cfg, }
e.cleanup = make(map[string][]cleanupFunc)
return e, nil}
func (e *Engine) SetupWorkflow(ctx context.Context, wid models.WorkflowId, wf *models.Workflow, wfLogger models.WorkflowLogger) error { /// -------------------------INITIAL SETUP------------------------------------------ l := e.l.With("workflow", wid) l.Info("setting up workflow")
setupStep := Step{ name: "Pull image from Nixery", kind: models.StepKindSystem, } setupStepIdx := -1
wfLogger.ControlWriter(setupStepIdx, setupStep, models.StepStatusStart).Write([]byte{0}) defer wfLogger.ControlWriter(setupStepIdx, setupStep, models.StepStatusEnd).Write([]byte{0})
/// -------------------------NETWORK CREATION--------------------------------------- _, err := e.docker.NetworkCreate(ctx, networkName(wid), network.CreateOptions{ Driver: "bridge", }) if err != nil { return err }
e.registerCleanup(wid, func(ctx context.Context) error { if err := e.docker.NetworkRemove(ctx, networkName(wid)); err != nil { return fmt.Errorf("removing network: %w", err) } return nil })
/// -------------------------IMAGE PULL--------------------------------------------- addl := wf.Data.(addlFields) l.Info("pulling image", "image", addl.image) fmt.Fprintf( wfLogger.DataWriter(setupStepIdx, "stdout"), "Pulling image: %s", addl.image, )
reader, err := e.docker.ImagePull(ctx, addl.image, image.PullOptions{}) if err != nil { l.Error("pipeline image pull failed!", "error", err.Error()) fmt.Fprintf(wfLogger.DataWriter(setupStepIdx, "stderr"), "image pull failed: %s", err) return fmt.Errorf("pulling image: %w", err) } defer reader.Close()
scanner := bufio.NewScanner(reader) for scanner.Scan() { line := scanner.Text() wfLogger.DataWriter(setupStepIdx, "stdout").Write([]byte(line)) l.Info("image pull progress", "stdout", line) }
/// -------------------------CONTAINER CREATION------------------------------------- l.Info("creating container") wfLogger.DataWriter(setupStepIdx, "stdout").Write([]byte("Creating container..."))
resp, err := e.docker.ContainerCreate(ctx, &container.Config{ Image: addl.image, Cmd: []string{"cat"}, OpenStdin: true, // so cat stays alive :3 Tty: false, Hostname: "spindle", WorkingDir: workspaceDir, Labels: map[string]string{ "sh.tangled.pipeline/workflow_id": wid.String(), }, // TODO(winter): investigate whether environment variables passed here // get propagated to ContainerExec processes }, &container.HostConfig{ Mounts: append([]mount.Mount{ { Type: mount.TypeTmpfs, Target: "/tmp", ReadOnly: false, TmpfsOptions: &mount.TmpfsOptions{ Mode: 0o1777, // world-writable sticky bit Options: [][]string{ {"exec"}, }, }, }, }, addl.mounts...), ReadonlyRootfs: false, CapDrop: []string{"ALL"}, CapAdd: []string{"CAP_DAC_OVERRIDE", "CAP_CHOWN", "CAP_FOWNER", "CAP_SETUID", "CAP_SETGID"}, SecurityOpt: []string{"no-new-privileges"}, ExtraHosts: []string{"host.docker.internal:host-gateway"}, Resources: container.Resources{ Memory: e.cfg.NixeryPipelines.MaxJobMemoryMB * 1024 * 1024, }, }, nil, nil, "") if err != nil { fmt.Fprintf( wfLogger.DataWriter(setupStepIdx, "stderr"), "container creation failed: %s", err, ) return fmt.Errorf("creating container: %w", err) }
e.registerCleanup(wid, func(ctx context.Context) error { if err := e.docker.ContainerStop(ctx, resp.ID, container.StopOptions{}); err != nil { return fmt.Errorf("stopping container: %w", err) }
err := e.docker.ContainerRemove(ctx, resp.ID, container.RemoveOptions{ RemoveVolumes: true, RemoveLinks: false, Force: false, }) if err != nil { return fmt.Errorf("removing container: %w", err) }
return nil })
/// -------------------------CONTAINER START---------------------------------------- wfLogger.DataWriter(setupStepIdx, "stdout").Write([]byte("Starting container...")) if err := e.docker.ContainerStart(ctx, resp.ID, container.StartOptions{}); err != nil { return fmt.Errorf("starting container: %w", err) }
mkExecResp, err := e.docker.ContainerExecCreate(ctx, resp.ID, container.ExecOptions{ Cmd: []string{"mkdir", "-p", workspaceDir, homeDir}, AttachStdout: true, // NOTE(winter): pretty sure this will make it so that when stdout read is done below, mkdir is done. maybe?? AttachStderr: true, // for good measure, backed up by docker/cli ("If -d is not set, attach to everything by default") }) if err != nil { return err }
// This actually *starts* the command. Thanks, Docker! execResp, err := e.docker.ContainerExecAttach(ctx, mkExecResp.ID, container.ExecAttachOptions{}) if err != nil { return err } defer execResp.Close()
// This is apparently best way to wait for the command to complete. _, err = io.ReadAll(execResp.Reader) if err != nil { return err }
/// -----------------------------------FINISH--------------------------------------- execInspectResp, err := e.docker.ContainerExecInspect(ctx, mkExecResp.ID) if err != nil { return err }
if execInspectResp.ExitCode != 0 { return fmt.Errorf("mkdir exited with exit code %d", execInspectResp.ExitCode) } else if execInspectResp.Running { return errors.New("mkdir is somehow still running??") }
addl.container = resp.ID wf.Data = addl
return nil}
func (e *Engine) RunStep(ctx context.Context, wid models.WorkflowId, w *models.Workflow, idx int, secrets []secrets.UnlockedSecret, wfLogger models.WorkflowLogger) error { addl := w.Data.(addlFields) workflowEnvs := ConstructEnvs(w.Environment) // TODO(winter): should SetupWorkflow also have secret access? // IMO yes, but probably worth thinking on. for _, s := range secrets { workflowEnvs.AddEnv(s.Key, s.Value) }
step := w.Steps[idx]
select { case <-ctx.Done(): return ctx.Err() default: }
envs := append(EnvVars(nil), workflowEnvs...) if nixStep, ok := step.(Step); ok { for k, v := range nixStep.environment { envs.AddEnv(k, v) } }
envs.AddEnv("HOME", homeDir) existingPath := "/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin" envs.AddEnv("PATH", fmt.Sprintf("%s/.nix-profile/bin:/nix/var/nix/profiles/default/bin:%s", homeDir, existingPath)) if sock := e.cfg.Server.DockerSocket; sock != "" { envs.AddEnv("DOCKER_HOST", fmt.Sprintf("unix://%s", sock)) }
mkExecResp, err := e.docker.ContainerExecCreate(ctx, addl.container, container.ExecOptions{ Cmd: []string{"bash", "-c", step.Command()}, AttachStdout: true, AttachStderr: true, Env: envs, }) if err != nil { return fmt.Errorf("creating exec: %w", err) }
// start tailing logs in background tailDone := make(chan error, 1) go func() { tailDone <- e.tailStep(ctx, wfLogger, mkExecResp.ID, idx) }()
select { case <-tailDone:
case <-ctx.Done(): // cleanup will be handled by DestroyWorkflow, since // Docker doesn't provide an API to kill an exec run // (sure, we could grab the PID and kill it ourselves, // but that's wasted effort) e.l.Warn("step timed out", "step", step.Name())
<-tailDone
return engine.ErrTimedOut }
select { case <-ctx.Done(): return ctx.Err() default: }
execInspectResp, err := e.docker.ContainerExecInspect(ctx, mkExecResp.ID) if err != nil { return err }
if execInspectResp.ExitCode != 0 { inspectResp, err := e.docker.ContainerInspect(ctx, addl.container) if err != nil { return err }
e.l.Error("workflow failed!", "workflow_id", wid.String(), "exit_code", execInspectResp.ExitCode, "oom_killed", inspectResp.State.OOMKilled)
if inspectResp.State.OOMKilled { return ErrOOMKilled } return engine.ErrWorkflowFailed }
return nil}
func (e *Engine) tailStep(ctx context.Context, wfLogger models.WorkflowLogger, execID string, stepIdx int) error { if wfLogger == nil { return nil }
// This actually *starts* the command. Thanks, Docker! logs, err := e.docker.ContainerExecAttach(ctx, execID, container.ExecAttachOptions{}) if err != nil { return err } defer logs.Close()
_, err = stdcopy.StdCopy( wfLogger.DataWriter(stepIdx, "stdout"), wfLogger.DataWriter(stepIdx, "stderr"), logs.Reader, ) if err != nil && err != io.EOF && !errors.Is(err, context.DeadlineExceeded) { return fmt.Errorf("failed to copy logs: %w", err) }
return nil}
func (e *Engine) DestroyWorkflow(ctx context.Context, wid models.WorkflowId) error { fns := e.drainCleanups(wid)
for _, fn := range fns { if err := fn(ctx); err != nil { e.l.Error("failed to cleanup workflow resource", "workflowId", wid, "error", err) } } return nil}
func (e *Engine) registerCleanup(wid models.WorkflowId, fn cleanupFunc) { e.cleanupMu.Lock() defer e.cleanupMu.Unlock()
key := wid.String() e.cleanup[key] = append(e.cleanup[key], fn)}
func (e *Engine) drainCleanups(wid models.WorkflowId) []cleanupFunc { e.cleanupMu.Lock() key := wid.String()
fns := e.cleanup[key] delete(e.cleanup, key) e.cleanupMu.Unlock()
return fns}
func networkName(wid models.WorkflowId) string { return fmt.Sprintf("workflow-network-%s", wid)}