--init parameter on create command

Author: jasonrclark

cmd/create.go

@@ -17,6 +17,7 @@ type createCmdFlags struct {
 	EnvironmentVariables []string
 	Secrets              []string
 	RevisionName 	  	 string
+	Init                 bool
 }
 
 type createReq struct {
@@ -104,12 +105,21 @@ func init() {
 			}
 
 			fmt.Printf("App created: %s\n", response.AppUrl) // TODO pretty print details
+
+			if createCmdFlags.Init {
+				err = writeRuntimeConfig(createCmdFlags.app, "")
+				if err != nil {
+					fmt.Printf("Error initializing config: %v\n", err)
+					return
+				}
+			}
 		},
 	}
 
 	createCmd.Flags().StringVarP(&createCmdFlags.app, "app", "a", "", "The app to create")
 	createCmd.Flags().StringSliceVarP(&createCmdFlags.EnvironmentVariables, "env", "e", []string{}, "Environment variables to set on the app in the form 'key=value'")
 	createCmd.Flags().StringSliceVarP(&createCmdFlags.Secrets, "secret", "s", []string{}, "Secrets to set on the app in the form 'key=value'")
 	createCmd.Flags().StringVarP(&createCmdFlags.RevisionName, "revision-name", "r", "", "The revision name to use for the app")
+	createCmd.Flags().BoolVar(&createCmdFlags.Init, "init", false, "Initialize a runtime.config.json file in the current directory after creating the app")
 	rootCmd.AddCommand(createCmd)
 }

cmd/init.go

@@ -21,6 +21,39 @@ type appResponse struct {
 	AppUrl string `json:"app_url"`
 }
 
+// writeRuntimeConfig writes a runtime.config.json file for the given app.
+// If outPath is empty, it defaults to "runtime.config.json" in the current directory.
+func writeRuntimeConfig(app string, outPath string) error {
+	configStruct := config.RuntimeConfig{
+		App: app,
+	}
+
+	configPath := "runtime.config.json"
+	if outPath != "" {
+		configPath = outPath
+		outputDir := filepath.Dir(configPath)
+		if outputDir != "." {
+			err := os.MkdirAll(outputDir, 0755)
+			if err != nil {
+				return fmt.Errorf("error creating directory '%s': %v", outputDir, err)
+			}
+		}
+	}
+
+	configBytes, err := json.MarshalIndent(configStruct, "", "  ")
+	if err != nil {
+		return fmt.Errorf("error creating configuration: %v", err)
+	}
+
+	err = os.WriteFile(configPath, configBytes, 0644)
+	if err != nil {
+		return fmt.Errorf("error writing configuration file: %v", err)
+	}
+
+	fmt.Printf("Successfully initialized local project for Spark app '%s' at '%s'\n", app, configPath)
+	return nil
+}
+
 func init() {
 	initCmdFlags := initCmdFlags{}
 	initCmd := &cobra.Command{
@@ -47,7 +80,6 @@ func init() {
 				return fmt.Errorf("--app flag is required")
 			}
 
-			// Determine the identifier to use for the API call
 			identifier := initCmdFlags.app
 
 			getUrl := fmt.Sprintf("runtime/%s/deployment", identifier)
@@ -63,36 +95,7 @@ func init() {
 				return fmt.Errorf("app '%s' does not exist or is not accessible: %v", identifier, err)
 			}
 
-			// Create runtime config
-			configStruct := config.RuntimeConfig{
-				App: initCmdFlags.app,
-			}
-
-			configPath := "runtime.config.json"
-			if initCmdFlags.out != "" {
-				configPath = initCmdFlags.out
-				// Create directory if it doesn't exist
-				outputDir := filepath.Dir(configPath)
-				if outputDir != "." {
-					err = os.MkdirAll(outputDir, 0755)
-					if err != nil {
-						return fmt.Errorf("error creating directory '%s': %v", outputDir, err)
-					}
-				}
-			}
-
-			configBytes, err := json.MarshalIndent(configStruct, "", "  ")
-			if err != nil {
-				return fmt.Errorf("error creating configuration: %v", err)
-			}
-
-			err = os.WriteFile(configPath, configBytes, 0644)
-			if err != nil {
-				return fmt.Errorf("error writing configuration file: %v", err)
-			}
-
-			fmt.Printf("Successfully initialized local project for Spark app '%s' at '%s'\n", identifier, configPath)
-			return nil
+			return writeRuntimeConfig(initCmdFlags.app, initCmdFlags.out)
 		},
 	}