Prompt to create directory when using -C/--cwd with non-existent path (#6266)

* Initial plan

* Initial planning for cwd directory creation prompt

Co-authored-by: vhvb1989 <[email protected]>

* Add directory creation prompt for -C/--cwd flag

Co-authored-by: vhvb1989 <[email protected]>

* Complete directory creation feature for -C/--cwd flag

Co-authored-by: vhvb1989 <[email protected]>

---------

Co-authored-by: copilot-swe-agent[bot] <[email protected]>
Co-authored-by: vhvb1989 <[email protected]>

Author: Copilot

.gitignore

@@ -67,3 +67,4 @@ cli/azd/extensions/microsoft.azd.ai.builder/microsoft.azd.ai.builder
 cli/azd/extensions/microsoft.azd.ai.builder/microsoft.azd.ai.builder.exe
 cli/azd/extensions/microsoft.azd.demo/microsoft.azd.demo
 cli/azd/extensions/microsoft.azd.demo/microsoft.azd.demo.exe
+cli/azd/azd-test

cli/azd/cmd/root.go

@@ -26,6 +26,7 @@ import (
 	"github.com/azure/azure-dev/cli/azd/internal/cmd/show"
 	"github.com/azure/azure-dev/cli/azd/internal/telemetry"
 	"github.com/azure/azure-dev/cli/azd/pkg/output"
+	"github.com/azure/azure-dev/cli/azd/pkg/ux"
 	"github.com/spf13/cobra"
 )
 
@@ -80,6 +81,44 @@ func NewRootCmd(
 
 				prevDir = current
 
+				// Check if the directory exists
+				if _, err := os.Stat(opts.Cwd); os.IsNotExist(err) {
+					// Directory doesn't exist, prompt user to create it
+					shouldCreate := true // Default to Yes
+
+					if !opts.NoPrompt {
+						// Prompt the user
+						defaultValue := true
+						confirm := ux.NewConfirm(&ux.ConfirmOptions{
+							Message: fmt.Sprintf(
+								"Directory '%s' does not exist. Would you like to create it?",
+								opts.Cwd,
+							),
+							DefaultValue: &defaultValue,
+						})
+
+						result, err := confirm.Ask(cmd.Context())
+						if err != nil {
+							return fmt.Errorf("failed to prompt for directory creation: %w", err)
+						}
+
+						if result == nil {
+							return fmt.Errorf("no response received for directory creation prompt")
+						}
+
+						shouldCreate = *result
+					}
+
+					if !shouldCreate {
+						return fmt.Errorf("directory '%s' does not exist and creation was declined", opts.Cwd)
+					}
+
+					// Create the directory
+					if err := os.MkdirAll(opts.Cwd, 0755); err != nil {
+						return fmt.Errorf("failed to create directory '%s': %w", opts.Cwd, err)
+					}
+				}
+
 				if err := os.Chdir(opts.Cwd); err != nil {
 					return fmt.Errorf("failed to change directory to %s: %w", opts.Cwd, err)
 				}

cli/azd/test/functional/init_test.go

@@ -291,6 +291,56 @@ func Test_CLI_Init_CanUseTemplate(t *testing.T) {
 	require.FileExists(t, filepath.Join(dir, "README.md"))
 }
 
+// Test_CLI_Init_WithCwdAutoCreate tests the automatic directory creation when using -C/--cwd flag.
+func Test_CLI_Init_WithCwdAutoCreate(t *testing.T) {
+	tests := []struct {
+		name   string
+		subDir string // subdirectory to create within temp dir (using -C flag)
+		args   []string
+	}{
+		{
+			name:   "single level directory",
+			subDir: "new-project",
+			args:   []string{"init", "-t", "azure-samples/todo-nodejs-mongo", "--no-prompt"},
+		},
+		{
+			name:   "nested directory",
+			subDir: "parent/child/project",
+			args:   []string{"init", "-t", "azure-samples/todo-nodejs-mongo", "--no-prompt"},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			ctx, cancel := newTestContext(t)
+			defer cancel()
+
+			// Create a parent temp directory
+			parentDir := tempDirWithDiagnostics(t)
+			cli := azdcli.NewCLI(t)
+			cli.WorkingDirectory = parentDir
+
+			// Add -C flag with the subdirectory path
+			targetDir := filepath.Join(parentDir, tt.subDir)
+			args := append([]string{"-C", tt.subDir}, tt.args...)
+
+			// Directory should not exist before running the command
+			require.NoDirExists(t, targetDir)
+
+			// Run the command
+			// Note: We expect an error because --no-prompt will fail on environment name prompt
+			// but the directory creation should succeed before that
+			cli.RunCommand(ctx, args...)
+
+			// Verify the directory was created
+			require.DirExists(t, targetDir)
+
+			// Verify that the template was initialized in the created directory
+			require.FileExists(t, filepath.Join(targetDir, azdcontext.ProjectFileName))
+		})
+	}
+}
+
 // verifyEnvInitialized is a helper function that returns a verification function.
 // This avoids duplicating the verification logic in the test table.
 func verifyEnvInitialized(envName string) func(t *testing.T, dir string) {