handle errors with fmt.Errorf + code qol changes

Author: andrearcaina

cmd/generate.go

@@ -4,6 +4,8 @@ Copyright © 2026 NAME HERE <EMAIL ADDRESS>
 package cmd
 
 import (
+	"strings"
+
 	"github.com/andrearcaina/goforge/internal/goforge"
 	"github.com/andrearcaina/goforge/internal/spec"
 	"github.com/spf13/cobra"
@@ -31,10 +33,18 @@ goforge generate --some-flag "Developer"`,
 func init() {
 	rootCmd.AddCommand(generateCmd)
 
+	// general flags
 	generateCmd.Flags().StringVarP(&cfg.OutputPath, "path", "p", ".", "The directory to write the generated file to")
 	generateCmd.Flags().BoolVarP(&cfg.Default, "default", "d", false, "Use default configuration values")
 
+	// flags for form fields
 	generateCmd.Flags().StringVarP(&cfg.Form.Name, "name", "n", "", "The name for the go.mod module")
 	generateCmd.Flags().StringVarP((*string)(&cfg.Form.ServerTypeFlag), "server", "s", "", "Type of server to generate (rest/grpc/graphql)")
-	generateCmd.Flags().BoolVar(&cfg.Form.DatabaseFlag, "database", false, "Generate database files")
+	generateCmd.Flags().BoolVar(&cfg.Form.DatabaseFlag, "database", false, "Generate database files (if flag is set, set to true)")
+
+	// normalize server type flag to lowercase
+	generateCmd.PreRun = func(cmd *cobra.Command, args []string) {
+		// first convert to string, then lowercase the string, then convert back to ServerTypeFlag
+		cfg.Form.ServerTypeFlag = spec.ServerTypeFlag(strings.ToLower(string(cfg.Form.ServerTypeFlag)))
+	}
 }

internal/goforge/forge.go

@@ -1,7 +1,6 @@
 package goforge
 
 import (
-	"errors"
 	"fmt"
 
 	"github.com/andrearcaina/goforge/internal/spec"
@@ -18,7 +17,7 @@ func Forge(cfg *spec.Config) error {
 		}
 	} else {
 		if err := ui.Run(cfg); err != nil {
-			return errors.New("failed to run ui form")
+			return fmt.Errorf("failed to run UI: %w", err)
 		}
 	}
 

internal/goforge/generator.go

@@ -1,7 +1,6 @@
 package goforge
 
 import (
-	"errors"
 	"fmt"
 	"os"
 	"path/filepath"
@@ -13,11 +12,13 @@ import (
 )
 
 func Generate(cfg *spec.Config) error {
-	if cfg.Form.ServerTypeFlag == "rest" {
+	if cfg.Form.ServerTypeFlag == spec.REST {
 		return generateRESTServer(cfg)
 	}
 
-	return errors.New("unknown server type")
+	// TODO: add support for grpc and graphql
+
+	return fmt.Errorf("unsupported server type: %s", cfg.Form.ServerTypeFlag)
 }
 
 func generateRESTServer(cfg *spec.Config) error {

internal/ui/form.go

@@ -1,7 +1,7 @@
 package ui
 
 import (
-	"errors"
+	"fmt"
 
 	"github.com/andrearcaina/goforge/internal/spec"
 	"github.com/charmbracelet/huh"
@@ -37,7 +37,7 @@ func createForm(cfg *spec.Config) *huh.Form {
 				Value(&cfg.Form.Name).
 				Validate(func(s string) error {
 					if s == "" {
-						return errors.New("this field cannot be empty")
+						return fmt.Errorf("project name cannot be empty")
 					}
 					return nil
 				}),

internal/ui/success.go

@@ -11,19 +11,21 @@ func OutputSuccess(cfg *spec.Config) {
 	successStyle := lipgloss.NewStyle().
 		Foreground(lipgloss.Color("205")).
 		Bold(true).
-		Padding(1, 0).
-		Render("🔥 Generated successfully!")
+		Padding(1, 0)
 
 	cmdStyle := lipgloss.NewStyle().
 		Foreground(lipgloss.Color("63")).
 		PaddingLeft(2)
 
-	fmt.Println(successStyle)
+	fmt.Println(successStyle.Render("🔥 Generated successfully!"))
+
+	// next step
 	fmt.Println("Make sure to run the following commands:")
 	fmt.Println(cmdStyle.Render(fmt.Sprintf("cd %s", cfg.OutputPath)))
 	fmt.Println(cmdStyle.Render("go mod tidy"))
 	fmt.Println(cmdStyle.Render("swag init -g ./cmd/server/main.go (need to install swag first)"))
 
+	// additional commands if database is included
 	if cfg.Form.DatabaseFlag {
 		fmt.Println(cmdStyle.Render("sqlc generate ./... (you need to install sqlc first)"))
 		fmt.Println(cmdStyle.Render("go run ./cmd/server/main.go (you need to run a database first and set up the .env file)"))