Merge pull request #376 from depot/luke/support-jit-token

feat: support JIT tokens from Depot environments

Author: lukevmorris

pkg/cmd/cargo/cargo.go

@@ -53,8 +53,7 @@ func NewCmdCargo() *cobra.Command {
 			}
 
 			// Get authentication token
-			token := os.Getenv("DEPOT_CACHE_TOKEN")
-			token, err = helpers.ResolveToken(ctx, token)
+			token, err := helpers.ResolveToken(ctx, "")
 			if err != nil {
 				return fmt.Errorf("failed to resolve token: %w", err)
 			}

pkg/cmd/gocache/gocache.go

@@ -49,7 +49,6 @@ func NewCmdGoCache() *cobra.Command {
 				return err
 			}
 
-			token := os.Getenv("DEPOT_CACHE_TOKEN")
 			token, err = helpers.ResolveToken(ctx, token)
 			if err != nil {
 				return err

pkg/helpers/token.go

@@ -40,6 +40,10 @@ func ResolveToken(ctx context.Context, token string) (string, error) {
 		}
 	}
 
+	if token == "" {
+		token = resolveJITToken()
+	}
+
 	if token == "" && IsTerminal() {
 		return AuthorizeDevice(ctx)
 	}
@@ -61,3 +65,15 @@ func AuthorizeDevice(ctx context.Context) (string, error) {
 	}
 	return tokenResponse.Token, nil
 }
+
+func resolveJITToken() string {
+	if token := os.Getenv("DEPOT_JIT_TOKEN"); token != "" {
+		return token
+	}
+
+	if token := os.Getenv("DEPOT_CACHE_TOKEN"); token != "" {
+		return token
+	}
+
+	return ""
+}