feat: basalt 0.0.1

cmd/api.go

@@ -0,0 +1,178 @@
+package cmd
+
+import (
+	"fmt"
+	"io/fs"
+	"os"
+	"path"
+	"path/filepath"
+	"strings"
+
+	"gitea.fernandovideira.com/FernandoJVideira/keystone/assets"
+	"github.com/manifoldco/promptui"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	newCmd.AddCommand(apiCmd)
+}
+
+var apiCmd = &cobra.Command{
+	Use:   "api",
+	Short: "Create a new Keystone API project",
+	Long:  `Create a new Keystone API project with the necessary structure and files.`,
+	Run:   genProject,
+}
+
+func genProject(cmd *cobra.Command, args []string) {
+
+	projectNamePrompt := promptui.Prompt{
+		Label: "Project Name",
+	}
+
+	rawInput, err := projectNamePrompt.Run()
+	if err != nil {
+		fmt.Printf("Prompt failed %v\n", err)
+		return
+	}
+	moduleName := strings.TrimSpace(rawInput)
+	projectDir := path.Base(strings.TrimSuffix(moduleName, "/"))
+
+	apiType := promptui.Select{
+		Label: "Select API Type",
+		Items: []string{"HTTP (REST)", "gRPC"},
+	}
+
+	_, result, err := apiType.Run()
+
+	if err != nil {
+		fmt.Printf("Prompt failed %v\n", err)
+		return
+	}
+
+	// Logic to map the selection to a boolean
+	useGRPC := strings.Contains(result, "gRPC")
+
+	fmt.Printf("🚀 Scaffolding '%s' as %s...\n", moduleName, result)
+	if err := scaffoldProject(projectDir, moduleName, useGRPC); err != nil {
+		fmt.Printf("error scaffolding project: %v", err)
+		return
+	}
+
+	fmt.Println("📦 Installing tools...")
+	if err := installTools(useGRPC); err != nil {
+		fmt.Printf("error installing tools: %v", err)
+		return
+	}
+
+	fmt.Println("🧹 Tidying up Go modules...")
+	if err := runCommand(projectDir, "go", "mod", "tidy"); err != nil {
+		fmt.Printf("error running 'go mod tidy': %v", err)
+		return
+	}
+
+	fmt.Println("✅ Done!")
+}
+
+func scaffoldProject(projectDir, moduleName string, useGRPC bool) error {
+
+	err := os.MkdirAll(projectDir, 0755)
+	if err != nil {
+		return fmt.Errorf("error creating project directory: %v", err)
+	}
+
+	root := "embedded"
+	err = fs.WalkDir(assets.EmbedFS, root, func(path string, d fs.DirEntry, err error) error {
+
+		relPath, _ := filepath.Rel(root, path)
+
+		if !useGRPC {
+			// If User chose HTTP: Remove gRPC stuff
+			if strings.HasPrefix(relPath, "pb") ||
+				strings.HasPrefix(relPath, "proto") ||
+				strings.Contains(relPath, "handlers/grpc") ||
+				strings.HasPrefix(relPath, "docs/statik") ||
+				strings.HasPrefix(relPath, "docs/swagger") {
+				return nil
+			}
+		} else {
+			// If User chose gRPC: Remove HTTP stuff
+			if strings.Contains(relPath, "handlers/http") {
+				return nil
+			}
+		}
+
+		if strings.HasSuffix(relPath, "cmd/api/main.go") {
+			return nil
+		}
+
+		if d.IsDir() {
+			return nil
+		}
+
+		contentBytes, err := assets.EmbedFS.ReadFile(path)
+		if err != nil {
+			return fmt.Errorf("error reading embedded file %s: %v", path, err)
+		}
+		content := string(contentBytes)
+
+		targetPath := filepath.Join(projectDir, relPath)
+
+		targetPath = strings.TrimSuffix(targetPath, ".tmpl")
+
+		if err := os.MkdirAll(filepath.Dir(targetPath), 0755); err != nil {
+			return fmt.Errorf("error creating directory %s: %v", filepath.Dir(targetPath), err)
+		}
+
+		newContent := strings.ReplaceAll(content, oldModuleName, moduleName)
+		newContent = strings.ReplaceAll(newContent, "{{ .ModuleName }}", moduleName)
+
+		return os.WriteFile(targetPath, []byte(newContent), 0644)
+	})
+
+	if err != nil {
+		return fmt.Errorf("Error copying template files: %v\n", err)
+	}
+
+	// Generate main.go from template
+	return genFile(
+		"templates/main.go.tpl",
+		"main",
+		filepath.Join(projectDir, "cmd/api", "main.go"),
+		MainTemplateData{
+			ModuleName: moduleName,
+			UseGRPC:    useGRPC,
+		},
+	)
+}
+
+func installTools(useGrpc bool) error {
+	tools := []string{
+		"github.com/air-verse/air@latest",
+		"github.com/golangci/golangci-lint/cmd/golangci-lint@latest",
+		"go.uber.org/mock/mockgen@latest",
+		"github.com/sqlc-dev/sqlc/cmd/sqlc@latest",
+		"github.com/golang-migrate/migrate/v4/cmd/migrate@latest",
+	}
+
+	if useGrpc {
+		tools = append(tools,
+			"google.golang.org/protobuf/cmd/protoc-gen-go@latest",
+			"google.golang.org/grpc/cmd/protoc-gen-go-grpc@latest",
+			"github.com/grpc-ecosystem/grpc-gateway/v2/protoc-gen-grpc-gateway@latest",
+			"github.com/grpc-ecosystem/grpc-gateway/v2/protoc-gen-openapiv2@latest",
+		)
+	} else {
+		tools = append(tools,
+			"github.com/swaggo/swag/cmd/swag@latest",
+		)
+	}
+
+	for _, tool := range tools {
+		if err := runCommand("", "go", "install", tool); err != nil {
+			return fmt.Errorf("error installing tool %s: %v", tool, err)
+		}
+	}
+
+	return nil
+}