Merge pull request #1 from Klass-lk/develop

feat: separated the db implementations

Author: Danushka96

cmd/new.go

@@ -12,6 +12,7 @@ import (
 var (
 	projectName string
 	moduleName  string
+	dbType      string
 )
 
 var newCmd = &cobra.Command{
@@ -31,17 +32,62 @@ var newCmd = &cobra.Command{
 			moduleName = fmt.Sprintf("github.com/%s/%s", os.Getenv("USER"), projectName)
 		}
 
+		// Handle database type prompting
+		if dbType == "" {
+			var setupDb string
+			fmt.Print("Do you want to set up a database in this project? (y/n): ")
+			_, err := fmt.Scanln(&setupDb)
+			if err != nil || (setupDb != "y" && setupDb != "yes" && setupDb != "Y" && setupDb != "YES") {
+				dbType = "none"
+			} else {
+				fmt.Println("\nCurrently supported databases:")
+				fmt.Println("  1) MongoDB")
+				fmt.Println("  2) PostgreSQL")
+				fmt.Println("  3) MySQL")
+				fmt.Println("  4) DynamoDB")
+				for {
+					var choice int
+					fmt.Print("Select a database (1-4): ")
+					_, err := fmt.Scanln(&choice)
+					if err != nil || choice < 1 || choice > 4 {
+						fmt.Println("Invalid selection. Please enter a number between 1 and 4.")
+						// Clear standard input if needed or handle retry
+						continue
+					}
+					switch choice {
+					case 1:
+						dbType = "mongodb"
+					case 2:
+						dbType = "postgres"
+					case 3:
+						dbType = "mysql"
+					case 4:
+						dbType = "dynamodb"
+					}
+					break
+				}
+			}
+		} else {
+			// Validate flag
+			switch dbType {
+			case "none", "mongodb", "postgres", "mysql", "dynamodb":
+				// valid
+			default:
+				return fmt.Errorf("invalid database type '%s': must be one of none, mongodb, postgres, mysql, dynamodb", dbType)
+			}
+		}
+
 		projectPath := filepath.Join(".", projectName)
 		if err := os.MkdirAll(projectPath, 0755); err != nil {
 			return fmt.Errorf("failed to create project directory: %w", err)
 		}
 
-		gen := generator.NewProjectGenerator(projectPath, projectName, moduleName)
+		gen := generator.NewProjectGenerator(projectPath, projectName, moduleName, dbType)
 		if err := gen.Generate(); err != nil {
 			return fmt.Errorf("failed to generate project: %w", err)
 		}
 
-		fmt.Printf("Successfully created project '%s' at %s\n", projectName, projectPath)
+		fmt.Printf("Successfully created project '%s' at %s (Database: %s)\n", projectName, projectPath, dbType)
 		fmt.Println("\nNext steps:")
 		fmt.Printf("  cd %s\n", projectName)
 		fmt.Println("  go mod tidy")
@@ -60,4 +106,5 @@ func isValidProjectName(name string) bool {
 
 func init() {
 	newCmd.Flags().StringVar(&moduleName, "module", "", "Go module name (default: github.com/username/project-name)")
+	newCmd.Flags().StringVar(&dbType, "db", "", "Database type: none, mongodb, postgres, mysql, dynamodb")
 }

internal/generator/generator.go

@@ -8,16 +8,18 @@ import (
 )
 
 type ProjectGenerator struct {
-	ProjectPath string
-	ProjectName string
-	ModuleName  string
+	ProjectPath  string
+	ProjectName  string
+	ModuleName   string
+	DatabaseType string
 }
 
-func NewProjectGenerator(projectPath, projectName, moduleName string) *ProjectGenerator {
+func NewProjectGenerator(projectPath, projectName, moduleName, databaseType string) *ProjectGenerator {
 	return &ProjectGenerator{
-		ProjectPath: projectPath,
-		ProjectName: projectName,
-		ModuleName:  moduleName,
+		ProjectPath:  projectPath,
+		ProjectName:  projectName,
+		ModuleName:   moduleName,
+		DatabaseType: databaseType,
 	}
 }
 
@@ -36,14 +38,53 @@ func (g *ProjectGenerator) Generate() error {
 		}
 	}
 
+	// Select templates based on database choice
+	var mainTmpl, goModTmpl, dockerComposeTmpl, userModelTmpl, userRepoTmpl string
+
+	switch g.DatabaseType {
+	case "mongodb":
+		mainTmpl = mainMongoTemplate
+		goModTmpl = goModMongoTemplate
+		dockerComposeTmpl = dockerComposeMongoTemplate
+		userModelTmpl = userModelMongoTemplate
+		userRepoTmpl = userRepositoryMongoTemplate
+	case "postgres":
+		mainTmpl = mainPostgresTemplate
+		goModTmpl = goModPostgresTemplate
+		dockerComposeTmpl = dockerComposePostgresTemplate
+		userModelTmpl = userModelPostgresTemplate
+		userRepoTmpl = userRepositoryPostgresTemplate
+	case "mysql":
+		mainTmpl = mainMysqlTemplate
+		goModTmpl = goModMysqlTemplate
+		dockerComposeTmpl = dockerComposeMysqlTemplate
+		userModelTmpl = userModelMysqlTemplate
+		userRepoTmpl = userRepositoryMysqlTemplate
+	case "dynamodb":
+		mainTmpl = mainDynamodbTemplate
+		goModTmpl = goModDynamodbTemplate
+		dockerComposeTmpl = dockerComposeDynamodbTemplate
+		userModelTmpl = userModelDynamodbTemplate
+		userRepoTmpl = userRepositoryDynamodbTemplate
+	default: // "none"
+		mainTmpl = mainNoneTemplate
+		goModTmpl = goModNoneTemplate
+		dockerComposeTmpl = dockerComposeNoneTemplate
+		userModelTmpl = userModelNoneTemplate
+		userRepoTmpl = userRepositoryNoneTemplate
+	}
+
 	// Generate files
 	files := map[string]string{
-		"main.go":            mainTemplate,
-		"go.mod":             goModTemplate,
-		"Makefile":           makefileTemplate,
-		"template.yaml":      templateYamlTemplate,
-		"Dockerfile":         dockerfileTemplate,
-		"docker-compose.yml": dockerComposeTemplate,
+		"main.go":       mainTmpl,
+		"go.mod":        goModTmpl,
+		"Makefile":      makefileTemplate,
+		"template.yaml": templateYamlTemplate,
+		"Dockerfile":    dockerfileTemplate,
+	}
+
+	if dockerComposeTmpl != "" {
+		files["docker-compose.yml"] = dockerComposeTmpl
 	}
 
 	for filename, tmpl := range files {
@@ -55,8 +96,8 @@ func (g *ProjectGenerator) Generate() error {
 	// Generate internal package files
 	internalFiles := map[string]string{
 		"internal/controller/user_controller.go": userControllerTemplate,
-		"internal/model/user.go":                 userModelTemplate,
-		"internal/repository/user_repository.go": userRepositoryTemplate,
+		"internal/model/user.go":                 userModelTmpl,
+		"internal/repository/user_repository.go": userRepoTmpl,
 		"internal/service/user_service.go":       userServiceTemplate,
 	}