Add saucectl imagerunner logs command

cmd/saucectl/saucectl.go

@@ -13,6 +13,7 @@ import (
 	"github.com/saucelabs/saucectl/internal/cmd/completion"
 	"github.com/saucelabs/saucectl/internal/cmd/configure"
 	"github.com/saucelabs/saucectl/internal/cmd/doctor"
+	"github.com/saucelabs/saucectl/internal/cmd/imagerunner"
 	"github.com/saucelabs/saucectl/internal/cmd/ini"
 	"github.com/saucelabs/saucectl/internal/cmd/jobs"
 	"github.com/saucelabs/saucectl/internal/cmd/new"
@@ -71,6 +72,7 @@ func main() {
 		storage.Command(cmd.PersistentPreRun),
 		artifacts.Command(cmd.PersistentPreRun),
 		jobs.Command(cmd.PersistentPreRun),
+		imagerunner.Command(cmd.PersistentPreRun),
 	)
 
 	if err := cmd.Execute(); err != nil {

internal/cmd/imagerunner/cmd.go

@@ -0,0 +1,48 @@
+package imagerunner
+
+import (
+	"errors"
+	"time"
+
+	"github.com/saucelabs/saucectl/internal/credentials"
+	"github.com/saucelabs/saucectl/internal/http"
+	"github.com/saucelabs/saucectl/internal/region"
+	"github.com/spf13/cobra"
+)
+
+var (
+	imagerunnerClient http.ImageRunner
+)
+
+func Command(preRun func(cmd *cobra.Command, args []string)) *cobra.Command {
+	var regio string
+
+	cmd := &cobra.Command{
+		Use:   "imagerunner",
+		Short: "Commands for interacting with imagerunner runs",
+		PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
+			if preRun != nil {
+				preRun(cmd, args)
+			}
+
+			if region.FromString(regio) == region.None {
+				return errors.New("invalid region")
+			}
+
+			creds := credentials.Get()
+			url := region.FromString(regio).APIBaseURL()
+
+			imagerunnerClient = http.NewImageRunner(url, creds, 15*time.Minute)
+
+			return nil
+		},
+	}
+
+	flags := cmd.PersistentFlags()
+	flags.StringVarP(&regio, "region", "r", "us-west-1", "The Sauce Labs region. Options: us-west-1, eu-central-1.")
+
+	cmd.AddCommand(
+		LogsCommand(),
+	)
+	return cmd
+}

internal/cmd/imagerunner/logs.go

@@ -0,0 +1,49 @@
+package imagerunner
+
+import (
+	"context"
+	"fmt"
+
+	cmds "github.com/saucelabs/saucectl/internal/cmd"
+	imgrunner "github.com/saucelabs/saucectl/internal/imagerunner"
+	"github.com/saucelabs/saucectl/internal/segment"
+	"github.com/saucelabs/saucectl/internal/usage"
+	"github.com/spf13/cobra"
+	"golang.org/x/text/cases"
+	"golang.org/x/text/language"
+)
+
+func LogsCommand() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "logs <runID>",
+		Short: "Fetch the logs for an imagerunner run",
+		PreRun: func(cmd *cobra.Command, args []string) {
+			tracker := segment.DefaultTracker
+
+			go func() {
+				tracker.Collect(
+					cases.Title(language.English).String(cmds.FullName(cmd)),
+					usage.Properties{}.SetFlags(cmd.Flags()),
+				)
+				_ = tracker.Close()
+			}()
+		},
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return exec(args[0])
+		},
+	}
+
+	return cmd
+}
+
+func exec(runID string) error {
+	log, err := imagerunnerClient.GetLogs(context.Background(), runID)
+	if err != nil {
+		if err == imgrunner.ErrResourceNotFound {
+			return fmt.Errorf("could not find log URL for run with ID (%s): %w", runID, err)
+		}
+		return err
+	}
+	fmt.Println(log)
+	return nil
+}

internal/cmd/storage/cmd.go

@@ -19,7 +19,7 @@ func Command(preRun func(cmd *cobra.Command, args []string)) *cobra.Command {
 
 	cmd := &cobra.Command{
 		Use:              "storage",
-		Short:            "Interact with Sauce Storage.",
+		Short:            "Interact with Sauce Storage",
 		TraverseChildren: true,
 		PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
 			if preRun != nil {

internal/http/imagerunner.go

@@ -169,6 +169,9 @@ func (c *ImageRunner) GetLogs(ctx context.Context, id string) (string, error) {
 	}
 	defer resp.Body.Close()
 
+	if resp.StatusCode == http.StatusNotFound {
+		return "", imagerunner.ErrResourceNotFound
+	}
 	if resp.StatusCode != http.StatusOK {
 		b, _ := io.ReadAll(resp.Body)
 		return "", fmt.Errorf("unexpected server response (%d): %s", resp.StatusCode, b)