Merge pull request #431 from depot/luke/jj-tqurkrplwmqz

feat: `status` and `logs` commands

Author: lukevmorris

pkg/api/ci.go

@@ -11,6 +11,43 @@ import (
 
 var baseURLFunc = getBaseURL
 
+func newCIServiceClient() civ1connect.CIServiceClient {
+	baseURL := baseURLFunc()
+	return civ1connect.NewCIServiceClient(getHTTPClient(baseURL), baseURL, WithUserAgent())
+}
+
+// CIGetRunStatus returns the current status of a CI run including its workflows, jobs, and attempts.
+func CIGetRunStatus(ctx context.Context, token, runID string) (*civ1.GetRunStatusResponse, error) {
+	client := newCIServiceClient()
+	resp, err := client.GetRunStatus(ctx, WithAuthentication(connect.NewRequest(&civ1.GetRunStatusRequest{RunId: runID}), token))
+	if err != nil {
+		return nil, err
+	}
+	return resp.Msg, nil
+}
+
+// CIGetJobAttemptLogs returns all log lines for a job attempt, paginating through all pages.
+func CIGetJobAttemptLogs(ctx context.Context, token, attemptID string) ([]*civ1.LogLine, error) {
+	client := newCIServiceClient()
+	var allLines []*civ1.LogLine
+	var pageToken string
+
+	for {
+		req := &civ1.GetJobAttemptLogsRequest{AttemptId: attemptID, PageToken: pageToken}
+		resp, err := client.GetJobAttemptLogs(ctx, WithAuthentication(connect.NewRequest(req), token))
+		if err != nil {
+			return nil, err
+		}
+		allLines = append(allLines, resp.Msg.Lines...)
+		if resp.Msg.NextPageToken == "" {
+			break
+		}
+		pageToken = resp.Msg.NextPageToken
+	}
+
+	return allLines, nil
+}
+
 func newCISecretServiceClient() civ1connect.SecretServiceClient {
 	baseURL := baseURLFunc()
 	return civ1connect.NewSecretServiceClient(getHTTPClient(baseURL), baseURL, WithUserAgent())

pkg/cmd/ci/ci.go

@@ -12,8 +12,10 @@ func NewCmdCI() *cobra.Command {
 		},
 	}
 
+	cmd.AddCommand(NewCmdLogs())
 	cmd.AddCommand(NewCmdMigrate())
 	cmd.AddCommand(NewCmdSecrets())
+	cmd.AddCommand(NewCmdStatus())
 	cmd.AddCommand(NewCmdVars())
 
 	return cmd

pkg/cmd/ci/logs.go

@@ -0,0 +1,50 @@
+package ci
+
+import (
+	"fmt"
+
+	"github.com/depot/cli/pkg/api"
+	"github.com/depot/cli/pkg/helpers"
+	"github.com/spf13/cobra"
+)
+
+func NewCmdLogs() *cobra.Command {
+	var token string
+
+	cmd := &cobra.Command{
+		Use:   "logs <attempt-id>",
+		Short: "Fetch logs for a CI job attempt [beta]",
+		Long:  "Fetch and display log output for a CI job attempt.\n\nThis command is in beta and subject to change.",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if len(args) == 0 {
+				return cmd.Help()
+			}
+
+			ctx := cmd.Context()
+			attemptID := args[0]
+
+			tokenVal, err := helpers.ResolveOrgAuth(ctx, token)
+			if err != nil {
+				return err
+			}
+			if tokenVal == "" {
+				return fmt.Errorf("missing API token, please run `depot login`")
+			}
+
+			lines, err := api.CIGetJobAttemptLogs(ctx, tokenVal, attemptID)
+			if err != nil {
+				return fmt.Errorf("failed to get job attempt logs: %w", err)
+			}
+
+			for _, line := range lines {
+				fmt.Println(line.Body)
+			}
+
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVar(&token, "token", "", "Depot API token")
+
+	return cmd
+}

pkg/cmd/ci/status.go

@@ -0,0 +1,65 @@
+package ci
+
+import (
+	"fmt"
+
+	"github.com/depot/cli/pkg/api"
+	"github.com/depot/cli/pkg/helpers"
+	"github.com/spf13/cobra"
+)
+
+func NewCmdStatus() *cobra.Command {
+	var token string
+
+	cmd := &cobra.Command{
+		Use:   "status <run-id>",
+		Short: "Look up the status of a CI run [beta]",
+		Long:  "Look up the status of a CI run, including its workflows, jobs, and attempts.\n\nThis command is in beta and subject to change.",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if len(args) == 0 {
+				return cmd.Help()
+			}
+
+			ctx := cmd.Context()
+			runID := args[0]
+
+			tokenVal, err := helpers.ResolveOrgAuth(ctx, token)
+			if err != nil {
+				return err
+			}
+			if tokenVal == "" {
+				return fmt.Errorf("missing API token, please run `depot login`")
+			}
+
+			resp, err := api.CIGetRunStatus(ctx, tokenVal, runID)
+			if err != nil {
+				return fmt.Errorf("failed to get run status: %w", err)
+			}
+
+			fmt.Printf("Org: %s\n", resp.OrgId)
+			fmt.Printf("Run: %s (%s)\n", resp.RunId, resp.Status)
+
+			for _, workflow := range resp.Workflows {
+				fmt.Println()
+				fmt.Printf("  Workflow: %s (%s)\n", workflow.WorkflowId, workflow.Status)
+				if workflow.WorkflowPath != "" {
+					fmt.Printf("    Path: %s\n", workflow.WorkflowPath)
+				}
+
+				for _, job := range workflow.Jobs {
+					fmt.Printf("    Job: %s [%s] (%s)\n", job.JobId, job.JobKey, job.Status)
+
+					for _, attempt := range job.Attempts {
+						fmt.Printf("      Attempt: %s #%d (%s)\n", attempt.AttemptId, attempt.Attempt, attempt.Status)
+					}
+				}
+			}
+
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVar(&token, "token", "", "Depot API token")
+
+	return cmd
+}