add cache with reset subcommand

Author: kylegalbraith

pkg/cmd/cache/cache.go

@@ -0,0 +1,21 @@
+package init
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+)
+
+func NewCmdCache() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "cache",
+		Short: "Operations for the Depot project cache",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			return fmt.Errorf("missing subcommand, please run `depot cache --help`")
+		},
+	}
+
+	cmd.AddCommand(NewCmdResetCache())
+
+	return cmd
+}

pkg/cmd/cache/reset.go

@@ -0,0 +1,70 @@
+package init
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"path/filepath"
+
+	"github.com/bufbuild/connect-go"
+	"github.com/depot/cli/pkg/api"
+	"github.com/depot/cli/pkg/config"
+	"github.com/depot/cli/pkg/project"
+	cliv1beta1 "github.com/depot/cli/pkg/proto/depot/cli/v1beta1"
+	"github.com/docker/cli/cli"
+	"github.com/pkg/errors"
+	"github.com/spf13/cobra"
+)
+
+func NewCmdResetCache() *cobra.Command {
+	var projectID string
+	var token string
+
+	cmd := &cobra.Command{
+		Use:   "reset",
+		Short: "Reset the cache for a project",
+		Args:  cli.RequiresMaxArgs(1),
+		RunE: func(cmd *cobra.Command, args []string) error {
+			if projectID == "" {
+				projectID = os.Getenv("DEPOT_PROJECT_ID")
+			}
+			if projectID == "" {
+				cwd, _ := filepath.Abs(args[0])
+				config, _, err := project.ReadConfig(cwd)
+				if err == nil {
+					projectID = config.ID
+				}
+			}
+			if projectID == "" {
+				return errors.Errorf("unknown project ID (run `depot init` or use --project or $DEPOT_PROJECT_ID)")
+			}
+
+			// TODO: make this a helper
+			if token == "" {
+				token = os.Getenv("DEPOT_TOKEN")
+			}
+			if token == "" {
+				token = config.GetApiToken()
+			}
+			if token == "" {
+				return fmt.Errorf("missing API token, please run `depot login`")
+			}
+
+			client := api.NewProjectsClient()
+			req := cliv1beta1.ResetProjectCacheRequest{ProjectId: projectID}
+			resp, err := client.ResetProjectCache(context.TODO(), api.WithAuthentication(connect.NewRequest(&req), token))
+			if err != nil {
+				return err
+			}
+
+			fmt.Printf("Cache reset for %s (%s)\n", resp.Msg.Name, resp.Msg.OrgName)
+
+			return nil
+		},
+	}
+
+	cmd.Flags().StringVar(&projectID, "project", "", "Depot project ID for the cache to reset")
+	cmd.Flags().StringVar(&token, "token", "", "Depot API token")
+
+	return cmd
+}

pkg/cmd/root/root.go

@@ -4,6 +4,7 @@ import (
 	"github.com/spf13/cobra"
 
 	buildCmd "github.com/depot/cli/pkg/cmd/build"
+	cacheCmd "github.com/depot/cli/pkg/cmd/cache"
 	initCmd "github.com/depot/cli/pkg/cmd/init"
 	loginCmd "github.com/depot/cli/pkg/cmd/login"
 	versionCmd "github.com/depot/cli/pkg/cmd/version"
@@ -31,6 +32,7 @@ func NewCmdRoot(version, buildDate string) *cobra.Command {
 
 	// Child commands
 	cmd.AddCommand(buildCmd.NewCmdBuild())
+	cmd.AddCommand(cacheCmd.NewCmdCache())
 	cmd.AddCommand(initCmd.NewCmdInit())
 	cmd.AddCommand(loginCmd.NewCmdLogin())
 	cmd.AddCommand(versionCmd.NewCmdVersion(version, buildDate))