args.go

package main

import (
	"strings"
	"sync"
	
	"github.com/jasonlabz/gentol/configx"
	"github.com/pborman/getopt/v2"
)

var once sync.Once

type Gentol struct {
	CurrentPkg string
}

func argHandler() {
	exist := IsExist("./conf/table.yaml")
	if !exist {
		exist = IsExist("./table.yaml")
	}
	if exist {
		configx.Init()
	} else {
		var (
			dbType = getopt.StringLong("db_type", 0, "postgres", "database type such as [mysql, sqlserver, postgres, oracle, greenplum etc. ]")

			dsn = getopt.StringLong("dsn", 0, "", "option database connection string, or provide host and port ...")

			host     = getopt.StringLong("host", 'h', "", "db host, if there is a dsn, ignore it")
			port     = getopt.IntLong("port", 'p', 0, "db port, if there is a dsn, ignore it")
			username = getopt.StringLong("username", 'U', "", "db username, if there is a dsn, ignore it")
			password = getopt.StringLong("password", 'P', "", "db password, if there is a dsn, ignore it")
			database = getopt.StringLong("database", 'd', "", "database to for db table")

			module = getopt.StringLong("module", 'm', "", "module name for go project")
			schema = getopt.StringLong("schema", 's', "", "schema to for db table")
			table  = getopt.StringLong("table", 't', "", "table name to build struct from")
			//templateDir = getopt.StringLong("template_dir", 0, "./template", "Template Dir")

			modelPath   = getopt.StringLong("model", 0, "dal/db/model", "name to set for model package")
			daoPath     = getopt.StringLong("dao", 0, "dal/db/dao", "name to set for dao package")
			servicePath = getopt.StringLong("service", 0, "server/service", "name to set for service package")
			//grpcPath    = getopt.StringLong("grpc", 0, "./grpc", "name to set for grpc package")
			//outDir      = getopt.StringLong("out", 0, ".", "output dir")

			jsonNameFormat  = getopt.StringLong("json_format", 0, "snake", "json name format [snake | upper_camel | lower_camel]")
			protoNameFormat = getopt.StringLong("protobuf_format", 0, "snake", "proto name format [snake | upper_camel | lower_camel]")
			//gogoProtoImport = getopt.StringLong("gogoproto", 0, "", "location of gogo import ")

			onlyModel             = getopt.BoolLong("only_model", 0, "overwrite existing files (default)", "disable overwriting files")
			useHook               = getopt.BoolLong("gen_hook", 0, "disable gorm hook file (default)", "gorm hook file")
			useSQLNullable        = getopt.BoolLong("use_sql_nullable", 0, "use sql.Null if use_sql_nullable true, default use guregu")
			addProtobufAnnotation = getopt.BoolLong("proto", 0, "add protobuf annotations (tags)", "")
			runGoFmt              = getopt.BoolLong("rungofmt", 0, "run gofmt on output dir", "")
			DefaultDBName         = "_default_db_"
		)
		// check args
		getopt.Lookup("db_type").SetGroup("check")
		getopt.RequiredGroup("check")
		getopt.ParseV2()

		// fill args
		configx.TableConfigs.AddProtobufAnnotation = *addProtobufAnnotation
		configx.TableConfigs.RunGoFmt = *runGoFmt
		configx.TableConfigs.JsonFormat = *jsonNameFormat
		configx.TableConfigs.ProtobufFormat = *protoNameFormat
		configx.TableConfigs.GoModule = *module
		databaseConfig := &configx.DBTableInfo{
			DBName:         DefaultDBName,
			DBType:         *dbType,
			DSN:            *dsn,
			OnlyModel:      *onlyModel,
			GenHook:        *useHook,
			ModelPath:      *modelPath,
			DaoPath:        *daoPath,
			ServicePath:    *servicePath,
			Host:           *host,
			Port:           *port,
			User:           *username,
			Password:       *password,
			Database:       *database,
			UseSQLNullable: *useSQLNullable,
			Tables: []*configx.TableInfo{
				{
					SchemaName: *schema,
					TableList: func() []string {
						if *table != "" {
							return strings.Split(*table, ",")
						}
						return []string{}
					}(),
				},
			},
		}
		databaseConfig.GenDSN()
		configx.TableConfigs.Configs = []*configx.DBTableInfo{
			databaseConfig,
		}
	}
	//handleDB()
}