main.go

// Package main provides entry point for enum generator
package main

import (
	"flag"
	"fmt"
	"os"
	"runtime/debug"

	"github.com/go-pkgz/enum/internal/generator"
)

// allow mocking os.Exit in tests
var osExit = os.Exit

func main() {
	typeFlag := flag.String("type", "", "type name (must be lowercase)")
	pathFlag := flag.String("path", "", "output directory path (default: same as source)")
	lowerFlag := flag.Bool("lower", false, "use lowercase for string representation (e.g., 'active' instead of 'Active')")
	getterFlag := flag.Bool("getter", false, "generate GetByID function to retrieve enum by integer value (requires unique IDs)")
	// optional integrations (all disabled by default to avoid extra deps)
	sqlFlag := flag.Bool("sql", false, "generate SQL support (database/sql/driver.Valuer and sql.Scanner)")
	bsonFlag := flag.Bool("bson", false, "generate MongoDB BSON support (MarshalBSONValue/UnmarshalBSONValue)")
	yamlFlag := flag.Bool("yaml", false, "generate YAML support (gopkg.in/yaml.v3 Marshaler/Unmarshaler)")
	helpFlag := flag.Bool("help", false, "show usage")
	versionFlag := flag.Bool("version", false, "print version")
	flag.Parse()

	// collect build info (version), new in go 1.24
	buildInfo := "dev"
	if info, ok := debug.ReadBuildInfo(); ok {
		if info.Main.Version != "" {
			buildInfo = info.Main.Version
		}
	}

	if *helpFlag {
		showUsage()
		osExit(0)
		return
	}
	if *versionFlag {
		fmt.Printf("enum generator %s\n", buildInfo)
		osExit(0)
		return
	}

	gen, err := generator.New(*typeFlag, *pathFlag)
	if err != nil {
		fmt.Printf("%v\n", err)
		showUsage()
		osExit(1)
		return
	}

	gen.SetLowerCase(*lowerFlag)
	gen.SetGenerateGetter(*getterFlag)
	gen.SetGenerateSQL(*sqlFlag)
	gen.SetGenerateBSON(*bsonFlag)
	gen.SetGenerateYAML(*yamlFlag)

	if err := gen.Parse("."); err != nil {
		fmt.Printf("%v\n", err)
		osExit(1)
		return
	}

	if err := gen.Generate(); err != nil {
		fmt.Printf("%v\n", err)
		osExit(1)
		return
	}
}

func showUsage() {
	fmt.Printf("usage: enum [flags]\n\n")
	fmt.Printf("Flags:\n")
	flag.PrintDefaults()
}