feat: add download to dev mode

Author: yjp20

pkg/cmd/dev.go

@@ -26,7 +26,11 @@ type BuildModel struct {
 	build       *stainless.BuildObject
 	branch      string
 	diagnostics []stainless.BuildDiagnosticListResponse
-	view        string
+	downloads   map[stainless.Target]struct {
+		status string
+		path   string
+	}
+	view string
 
 	cc          *apiCommandContext
 	ctx         context.Context
@@ -38,6 +42,7 @@ type tickMsg time.Time
 type fetchBuildMsg *stainless.BuildObject
 type fetchDiagnosticsMsg []stainless.BuildDiagnosticListResponse
 type errorMsg error
+type downloadMsg stainless.Target
 type triggerNewBuildMsg struct{}
 
 func NewBuildModel(cc *apiCommandContext, ctx context.Context, branch string, fn func() (*stainless.BuildObject, error)) BuildModel {
@@ -77,6 +82,11 @@ func (m BuildModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			cmds = append(cmds, tea.Quit)
 		}
 
+	case downloadMsg:
+		download := m.downloads[stainless.Target(msg)]
+		download.status = "completed"
+		m.downloads[stainless.Target(msg)] = download
+
 	case tickMsg:
 		if m.build != nil {
 			cmds = append(cmds, m.fetchBuildStatus())
@@ -89,6 +99,19 @@ func (m BuildModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	case fetchBuildMsg:
 		if m.build == nil {
 			m.build = msg
+			m.downloads = make(map[stainless.Target]struct {
+				status string
+				path   string
+			})
+			for targetName, targetConfig := range m.cc.workspaceConfig.Targets {
+				m.downloads[stainless.Target(targetName)] = struct {
+					status string
+					path   string
+				}{
+					status: "not started",
+					path:   targetConfig.OutputPath,
+				}
+			}
 			cmds = append(cmds, m.updateView("header"))
 		}
 
@@ -97,6 +120,25 @@ func (m BuildModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			m.isCompleted = true
 			cmds = append(cmds, m.fetchDiagnostics())
 		}
+		languages := getBuildLanguages(m.build)
+		for _, target := range languages {
+			buildTarget := getBuildTarget(m.build, target)
+			if buildTarget == nil {
+				continue
+			}
+			commitUnion := getStepUnion(buildTarget, "commit")
+			if commitUnion == nil {
+				continue
+			}
+			status, _, _ := extractStepInfo(commitUnion)
+			if status == "completed" {
+				if download, ok := m.downloads[target]; ok && download.status == "not started" {
+					download.status = "started"
+					cmds = append(cmds, m.downloadTarget(target))
+					m.downloads[target] = download
+				}
+			}
+		}
 
 	case fetchDiagnosticsMsg:
 		if m.diagnostics == nil {
@@ -111,6 +153,32 @@ func (m BuildModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	return m, tea.Sequence(cmds...)
 }
 
+func (m BuildModel) downloadTarget(target stainless.Target) tea.Cmd {
+	return func() tea.Msg {
+		if m.build == nil {
+			return errorMsg(fmt.Errorf("no current build to download target from"))
+		}
+		params := stainless.BuildTargetOutputGetParams{
+			BuildID: m.build.ID,
+			Target:  stainless.BuildTargetOutputGetParamsTarget(target),
+			Type:    "source",
+			Output:  "git",
+		}
+		outputRes, err := m.cc.client.Builds.TargetOutputs.Get(
+			context.TODO(),
+			params,
+		)
+		if err != nil {
+			return errorMsg(err)
+		}
+		err = pullOutput(outputRes.Output, outputRes.URL, outputRes.Ref, m.downloads[target].path, &Group{silent: true})
+		if err != nil {
+			return errorMsg(err)
+		}
+		return downloadMsg(target)
+	}
+}
+
 func (m BuildModel) fetchBuildStatus() tea.Cmd {
 	return func() tea.Msg {
 		if m.build == nil {

pkg/cmd/dev_view.go

@@ -136,7 +136,7 @@ var parts = []struct {
 				languages := getBuildLanguages(m.build)
 				// Target rows with colors
 				for _, target := range languages {
-					pipeline := ViewBuildPipeline(m.build, target)
+					pipeline := ViewBuildPipeline(m.build, target, m.downloads)
 					langStyle := lipgloss.NewStyle().Foreground(lipgloss.Color("15")).Bold(true)
 					s.WriteString(fmt.Sprintf("%s %s\n", langStyle.Render(fmt.Sprintf("%-13s", string(target))), pipeline))
 				}
@@ -169,7 +169,10 @@ var parts = []struct {
 	},
 }
 
-func ViewBuildPipeline(build *stainless.BuildObject, target stainless.Target) string {
+func ViewBuildPipeline(build *stainless.BuildObject, target stainless.Target, downloads map[stainless.Target]struct {
+	status string
+	path   string
+}) string {
 	buildTarget := getBuildTarget(build, target)
 	if buildTarget == nil {
 		return ""
@@ -191,6 +194,18 @@ func ViewBuildPipeline(build *stainless.BuildObject, target stainless.Target) st
 		pipeline.WriteString(symbol + " " + Hyperlink(url, step))
 	}
 
+	if download, ok := downloads[target]; ok {
+		if download.status == "not started" {
+			// do nothing
+		} else if download.status == "started" {
+			pipeline.WriteString(" → " + "downloading")
+		} else if download.status == "completed" {
+			pipeline.WriteString(" → " + "downloaded")
+		} else {
+			pipeline.WriteString(" → " + download.status)
+		}
+	}
+
 	return pipeline.String()
 }
 

pkg/cmd/print.go

@@ -16,6 +16,7 @@ import (
 type Group struct {
 	prefix string
 	indent int
+	silent bool
 }
 
 func Header(format string, args ...any) Group {
@@ -96,32 +97,44 @@ func SSuccess(indent int, format string, args ...any) string {
 }
 
 func (g Group) Info(format string, args ...any) Group {
-	fmt.Fprint(os.Stderr, SInfo(g.indent, format, args...))
+	if !g.silent {
+		fmt.Fprint(os.Stderr, SInfo(g.indent, format, args...))
+	}
 	return Group{prefix: "i", indent: g.indent + 1}
 }
 
 func (g Group) Property(key, msg string) Group {
-	fmt.Fprint(os.Stderr, SProperty(g.indent, key, msg))
+	if !g.silent {
+		fmt.Fprint(os.Stderr, SProperty(g.indent, key, msg))
+	}
 	return Group{prefix: "✱", indent: g.indent + 1}
 }
 
 func (g Group) Progress(format string, args ...any) Group {
-	fmt.Fprint(os.Stderr, SProgress(g.indent, format, args...))
+	if !g.silent {
+		fmt.Fprint(os.Stderr, SProgress(g.indent, format, args...))
+	}
 	return Group{prefix: "✱", indent: g.indent + 1}
 }
 
 func (g Group) Error(format string, args ...any) Group {
-	fmt.Fprint(os.Stderr, SError(g.indent, format, args...))
+	if !g.silent {
+		fmt.Fprint(os.Stderr, SError(g.indent, format, args...))
+	}
 	return Group{prefix: "✗", indent: g.indent + 1}
 }
 
 func (g Group) Warn(format string, args ...any) Group {
-	fmt.Fprint(os.Stderr, SWarn(g.indent, format, args...))
+	if !g.silent {
+		fmt.Fprint(os.Stderr, SWarn(g.indent, format, args...))
+	}
 	return Group{prefix: "!", indent: g.indent + 1}
 }
 
 func (g Group) Success(format string, args ...any) Group {
-	fmt.Fprint(os.Stderr, SSuccess(g.indent, format, args...))
+	if !g.silent {
+		fmt.Fprint(os.Stderr, SSuccess(g.indent, format, args...))
+	}
 	return Group{prefix: "✓", indent: g.indent + 1}
 }