feat: add directory overwrite checks

Author: archandatta

cmd/create.go

@@ -27,10 +27,22 @@ func (c CreateCmd) Create(ctx context.Context, ci CreateInput) error {
 		return fmt.Errorf("failed to resolve app path: %w", err)
 	}
 
-	// TODO: handle overwrite gracefully (prompt user)
-	// Check if directory already exists
+	// Check if directory already exists and prompt for overwrite
 	if _, err := os.Stat(appPath); err == nil {
-		return fmt.Errorf("directory %s already exists", ci.Name)
+		overwrite, err := create.PromptForOverwrite(ci.Name)
+		if err != nil {
+			return fmt.Errorf("failed to prompt for overwrite: %w", err)
+		}
+
+		if !overwrite {
+			pterm.Warning.Println("Operation cancelled.")
+			return nil
+		}
+
+		// Remove existing directory
+		if err := os.RemoveAll(appPath); err != nil {
+			return fmt.Errorf("failed to remove existing directory: %w", err)
+		}
 	}
 
 	if err := os.MkdirAll(appPath, 0755); err != nil {

cmd/create_test.go

@@ -14,6 +14,11 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
+const (
+	DIR_PERM  = 0755 // rwxr-xr-x
+	FILE_PERM = 0644 // rw-r--r--
+)
+
 func TestCreateCommand(t *testing.T) {
 	tests := []struct {
 		name        string
@@ -145,7 +150,7 @@ func TestAllTemplatesCreation(t *testing.T) {
 			appPath := filepath.Join(tmpDir, appName)
 
 			// Create app directory
-			err := os.MkdirAll(appPath, 0755)
+			err := os.MkdirAll(appPath, DIR_PERM)
 			require.NoError(t, err, "failed to create app directory")
 
 			// Copy template files without installing dependencies
@@ -352,6 +357,61 @@ func TestCreateCommand_RequiredToolMissing(t *testing.T) {
 	}
 }
 
+// TestCreateCommand_DirectoryOverwrite tests that overwriting an existing directory
+// properly removes old content and creates new content
+func TestCreateCommand_DirectoryOverwrite(t *testing.T) {
+	tmpDir := t.TempDir()
+	appName := "test-app"
+	appPath := filepath.Join(tmpDir, appName)
+
+	orgDir, err := os.Getwd()
+	require.NoError(t, err)
+
+	err = os.Chdir(tmpDir)
+	require.NoError(t, err)
+
+	t.Cleanup(func() {
+		os.Chdir(orgDir)
+	})
+
+	// Initialize directory with some files
+	err = os.MkdirAll(appPath, DIR_PERM)
+	require.NoError(t, err, "failed to create initial directory")
+
+	// Create some initial files that should be removed after overwrite
+	oldFile1 := filepath.Join(appPath, "old-file-1.txt")
+	oldSubDir := filepath.Join(appPath, "old-subdir")
+
+	err = os.WriteFile(oldFile1, []byte("old content 1"), FILE_PERM)
+	require.NoError(t, err, "failed to create old file 1")
+
+	err = os.MkdirAll(oldSubDir, DIR_PERM)
+	require.NoError(t, err, "failed to create old subdirectory")
+
+	// Verify initial files exist
+	assert.FileExists(t, oldFile1, "old file 1 should exist before overwrite")
+	assert.DirExists(t, oldSubDir, "old subdirectory should exist before overwrite")
+
+	// Manually remove the directory and create the new app
+	err = os.RemoveAll(appPath)
+	require.NoError(t, err, "failed to remove existing directory")
+
+	c := CreateCmd{}
+	err = c.Create(context.Background(), CreateInput{
+		Name:     appName,
+		Language: create.LanguageTypeScript,
+		Template: "sample-app",
+	})
+	require.NoError(t, err, "failed to create new app")
+
+	// Verify old files are gone
+	assert.NoFileExists(t, oldFile1, "old file 1 should not exist after overwrite")
+	assert.NoDirExists(t, oldSubDir, "old subdirectory should not exist after overwrite")
+
+	// Verify new template files exist
+	assert.FileExists(t, filepath.Join(appPath, "index.ts"), "new index.ts should exist")
+}
+
 func getTemplateInfo() []struct {
 	name     string
 	language string

pkg/create/prompts.go

@@ -122,3 +122,16 @@ func PromptForTemplate(providedTemplate string, providedLanguage string) (string
 	pterm.Warning.Printfln("Template '%s' not found. Please select from available templates.\n", providedTemplate)
 	return handleTemplatePrompt(templateKVs)
 }
+
+// PromptForOverwrite prompts the user to confirm overwriting an existing directory.
+func PromptForOverwrite(dirName string) (bool, error) {
+	overwrite, err := pterm.DefaultInteractiveConfirm.
+		WithDefaultText(fmt.Sprintf("\nDirectory %s already exists. Overwrite?", dirName)).
+		WithDefaultValue(false).
+		Show()
+	if err != nil {
+		return false, fmt.Errorf("failed to prompt for overwrite: %w", err)
+	}
+
+	return overwrite, nil
+}