main.go

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"strings"

	"github.com/go-easygen/cli"
	"github.com/marusama/go-enum/generator"
)

type rootT struct {
	cli.Helper
	FileNames []string `cli:"*f,file" usage:"The file(s) to generate enums.  Use more than one flag for more files."`
	NoPrefix  bool     `cli:"noprefix" usage:"Prevents the constants generated from having the Enum as a prefix."`
	Lowercase bool     `cli:"lower" usage:"Adds lowercase variants of the enum strings for lookup."`
	Marshal   bool     `cli:"marshal" usage:"Adds text marshalling functions."`
	JSON      bool     `cli:"json" usage:"Adds json marshalling functions."`
	SQL       bool     `cli:"sql" usage:"Adds SQL database scan and value functions."`
	Flag      bool     `cli:"flag" usage:"Adds golang flag functions."`
	Prefix    string   `cli:"prefix" usage:"Replaces the prefix with a user one."`
	Names     bool     `cli:"names" usage:"Generates a 'Names() []string' function, and adds the possible enum values in the error response during parsing"`
}

func main() {
	cli.Run(new(rootT), func(ctx *cli.Context) error {
		argv := ctx.Argv().(*rootT)

		for _, fileName := range argv.FileNames {

			g := generator.NewGenerator()

			if argv.NoPrefix {
				g.WithNoPrefix()
			}
			if argv.Lowercase {
				g.WithLowercaseVariant()
			}
			if argv.Marshal {
				g.WithMarshal()
			}
			if argv.JSON {
				g.WithJSON()
			}
			if argv.SQL {
				g.WithSQLDriver()
			}
			if argv.Flag {
				g.WithFlag()
			}
			if argv.Names {
				g.WithNames()
			}

			originalName := fileName

			ctx.String("go-enum started. file: %s\n", ctx.Color().Cyan(originalName))
			fileName, _ = filepath.Abs(fileName)
			outFilePath := fmt.Sprintf("%s_enum.go", strings.TrimSuffix(fileName, filepath.Ext(fileName)))

			// Parse the file given in arguments
			raw, err := g.GenerateFromFile(fileName)
			if err != nil {
				return fmt.Errorf("failed generating enums\nInputFile=%s\nError=%s", ctx.Color().Cyan(fileName), ctx.Color().RedBg(err))
			}

			mode := int(0644)
			err = ioutil.WriteFile(outFilePath, raw, os.FileMode(mode))
			if err != nil {
				return fmt.Errorf("failed writing to file %s: %s", ctx.Color().Cyan(outFilePath), ctx.Color().Red(err))
			}
			ctx.String("go-enum finished. file: %s\n", ctx.Color().Cyan(originalName))
		}

		return nil
	})
}