diff --git a/go.mod b/go.mod index db9aa93..6ef6e62 100644 --- a/go.mod +++ b/go.mod @@ -26,6 +26,7 @@ require ( github.com/goccy/go-json v0.10.0 // indirect github.com/golang/protobuf v1.5.3 // indirect github.com/gopherjs/gopherjs v1.17.2 // indirect + github.com/inconshreveable/mousetrap v1.1.0 // indirect github.com/jonboulle/clockwork v0.3.0 // indirect github.com/json-iterator/go v1.1.12 // indirect github.com/jtolds/gls v4.20.0+incompatible // indirect @@ -50,8 +51,14 @@ require ( github.com/richardlehane/mscfb v1.0.4 // indirect github.com/richardlehane/msoleps v1.0.3 // indirect github.com/smartystreets/assertions v1.13.1 // indirect + github.com/spf13/cobra v1.7.0 // indirect + github.com/spf13/pflag v1.0.5 // indirect + github.com/tealeg/xlsx v1.0.5 // indirect github.com/templexxx/cpu v0.0.9 // indirect github.com/templexxx/xorsimd v0.4.1 // indirect + github.com/tidwall/gjson v1.14.4 // indirect + github.com/tidwall/match v1.1.1 // indirect + github.com/tidwall/pretty v1.2.0 // indirect github.com/tjfoc/gmsm v1.4.1 // indirect github.com/twitchyliquid64/golang-asm v0.15.1 // indirect github.com/ugorji/go/codec v1.2.9 // indirect diff --git a/go.sum b/go.sum index 5b81de5..0c58604 100644 --- a/go.sum +++ b/go.sum @@ -13,6 +13,7 @@ github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311 h1:qSGYFH7+jGhD github.com/chenzhuoyu/base64x v0.0.0-20221115062448-fe3a3abad311/go.mod h1:b583jCggY9gE99b6G5LEC39OIiVsWj+R97kbl5odCEk= github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw= github.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc= +github.com/cpuguy83/go-md2man/v2 v2.0.2/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o= github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E= github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= @@ -62,6 +63,8 @@ github.com/gopherjs/gopherjs v1.17.2 h1:fQnZVsXk8uxXIStYb0N4bGk7jeyTalG/wsZjQ25d github.com/gopherjs/gopherjs v1.17.2/go.mod h1:pRRIvn/QzFLrKfvEz3qUuEhtE/zLCWfreZ6J5gM2i+k= github.com/gorilla/websocket v1.5.0 h1:PPwGk2jz7EePpoHN/+ClbZu8SPxiqlu12wZP/3sWmnc= github.com/gorilla/websocket v1.5.0/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE= +github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8= +github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw= github.com/jonboulle/clockwork v0.3.0 h1:9BSCMi8C+0qdApAp4auwX0RkLGUjs956h0EkuQymUhg= github.com/jonboulle/clockwork v0.3.0/go.mod h1:Pkfl5aHPm1nk2H9h0bjmnJD/BcgbGXUBGnn1kMkgxc8= github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= @@ -134,12 +137,17 @@ github.com/richardlehane/msoleps v1.0.3 h1:aznSZzrwYRl3rLKRT3gUk9am7T/mLNSnJINvN github.com/richardlehane/msoleps v1.0.3/go.mod h1:BWev5JBpU9Ko2WAgmZEuiz4/u3ZYTKbjLycmwiWUfWg= github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8= github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs= +github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM= github.com/smartystreets/assertions v1.13.1 h1:Ef7KhSmjZcK6AVf9YbJdvPYG9avaF0ZxudX+ThRdWfU= github.com/smartystreets/assertions v1.13.1/go.mod h1:cXr/IwVfSo/RbCSPhoAPv73p3hlSdrBH/b3SdnW/LMY= github.com/smartystreets/goconvey v1.8.0 h1:Oi49ha/2MURE0WexF052Z0m+BNSGirfjg5RL+JXWq3w= github.com/smartystreets/goconvey v1.8.0/go.mod h1:EdX8jtrTIj26jmjCOVNMVSIYAtgexqXKHOXW2Dx9JLg= github.com/sony/sonyflake v1.1.0 h1:wnrEcL3aOkWmPlhScLEGAXKkLAIslnBteNUq4Bw6MM4= github.com/sony/sonyflake v1.1.0/go.mod h1:LORtCywH/cq10ZbyfhKrHYgAUGH7mOBa76enV9txy/Y= +github.com/spf13/cobra v1.7.0 h1:hyqWnYt1ZQShIddO5kBpj3vu05/++x6tJ6dg8EC572I= +github.com/spf13/cobra v1.7.0/go.mod h1:uLxZILRyS/50WlhOIKD7W6V5bgeIt+4sICxh6uRMrb0= +github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA= +github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg= github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= @@ -151,12 +159,20 @@ github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/ github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk= github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/tealeg/xlsx v1.0.5 h1:+f8oFmvY8Gw1iUXzPk+kz+4GpbDZPK1FhPiQRd+ypgE= +github.com/tealeg/xlsx v1.0.5/go.mod h1:btRS8dz54TDnvKNosuAqxrM1QgN1udgk9O34bDCnORM= github.com/templexxx/cpu v0.0.1/go.mod h1:w7Tb+7qgcAlIyX4NhLuDKt78AHA5SzPmq0Wj6HiEnnk= github.com/templexxx/cpu v0.0.7/go.mod h1:w7Tb+7qgcAlIyX4NhLuDKt78AHA5SzPmq0Wj6HiEnnk= github.com/templexxx/cpu v0.0.9 h1:cGGLK8twbc1J1S/fHnZW7BylXYaFP+0fR2s+nzsFDiU= github.com/templexxx/cpu v0.0.9/go.mod h1:w7Tb+7qgcAlIyX4NhLuDKt78AHA5SzPmq0Wj6HiEnnk= github.com/templexxx/xorsimd v0.4.1 h1:iUZcywbOYDRAZUasAs2eSCUW8eobuZDy0I9FJiORkVg= github.com/templexxx/xorsimd v0.4.1/go.mod h1:W+ffZz8jJMH2SXwuKu9WhygqBMbFnp14G2fqEr8qaNo= +github.com/tidwall/gjson v1.14.4 h1:uo0p8EbA09J7RQaflQ1aBRffTR7xedD2bcIVSYxLnkM= +github.com/tidwall/gjson v1.14.4/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk= +github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA= +github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM= +github.com/tidwall/pretty v1.2.0 h1:RWIZEg2iJ8/g6fDDYzMpobmaoGh5OLl4AXtGUGPcqCs= +github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU= github.com/tjfoc/gmsm v1.3.2/go.mod h1:HaUcFuY0auTiaHB9MHFGCPx5IaLhTUd2atbCFBQXn9w= github.com/tjfoc/gmsm v1.4.1 h1:aMe1GlZb+0bLjn+cKTPEvvn9oUEBlJitaZiiBwsbgho= github.com/tjfoc/gmsm v1.4.1/go.mod h1:j4INPkHWMrhJb38G+J6W4Tw0AbuN8Thu3PbdVYhVcTE= @@ -322,6 +338,7 @@ google.golang.org/protobuf v1.28.1 h1:d0NfwRgPtno5B1Wa6L2DAG+KivqkdutMf1UhdNx175 google.golang.org/protobuf v1.28.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= gopkg.in/natefinch/lumberjack.v2 v2.0.0/go.mod h1:l0ndWWf7gzL7RNwBG7wST/UCcT4T24xpD6X8LsfU/+k= gopkg.in/natefinch/lumberjack.v2 v2.2.1 h1:bBRl1b0OH9s/DuPhuXpNl+VtCaJXFZ5/uEFST95x9zc= diff --git a/planner/configexport/configexport.go b/planner/configexport/configexport.go new file mode 100644 index 0000000..c192f10 --- /dev/null +++ b/planner/configexport/configexport.go @@ -0,0 +1,98 @@ +package main + +import ( + "fmt" + "github.com/kercylan98/minotaur/planner/configexport/internal" + "github.com/kercylan98/minotaur/utils/file" + "github.com/kercylan98/minotaur/utils/log" + "github.com/kercylan98/minotaur/utils/str" + "github.com/tealeg/xlsx" + "go.uber.org/zap" + "os" + "os/exec" + "path/filepath" + "runtime/debug" + "sync" +) + +func New(xlsxPath string) *ConfigExport { + ce := &ConfigExport{xlsxPath: xlsxPath} + xlsxFile, err := xlsx.OpenFile(xlsxPath) + if err != nil { + panic(err) + } + for i := 0; i < len(xlsxFile.Sheets); i++ { + ce.configs = append(ce.configs, internal.NewConfig(xlsxFile.Sheets[i])) + } + return ce +} + +type ConfigExport struct { + xlsxPath string + configs []*internal.Config +} + +func (slf *ConfigExport) ExportJSON(outputDir string) { + var errors []func() + var wait sync.WaitGroup + for _, config := range slf.configs { + config := config + go func() { + wait.Add(1) + defer func() { + if err := recover(); err != nil { + errors = append(errors, func() { + log.Error("导出失败", zap.String("名称", slf.xlsxPath), zap.String("Sheet", config.GetName()), zap.Any("err", err)) + fmt.Println(debug.Stack()) + }) + } + }() + if err := file.WriterFile(filepath.Join(outputDir, fmt.Sprintf("%s.json", config.GetName())), config.GetJSON()); err != nil { + panic(err) + } + wait.Done() + }() + } + + wait.Wait() + + for _, f := range errors { + f() + } +} + +func (slf *ConfigExport) ExportGo(packageName string, outputDir string) { + var vars string + var varsMake string + var types string + var varsReplace string + for _, config := range slf.configs { + v := config.GetVariable() + vars += fmt.Sprintf("var %s %s\nvar _%sReady %s\n", str.FirstUpper(config.GetName()), v, str.FirstUpper(config.GetName()), v) + varsMake += fmt.Sprintf("_%sReady = make(%s)"+` + if err := handle("%s.json", &_%sReady); err != nil { + panic(err) + } +`, str.FirstUpper(config.GetName()), v, str.FirstUpper(config.GetName()), str.FirstUpper(config.GetName())) + types += fmt.Sprintf("%s\n", config.GetStruct()) + varsReplace += fmt.Sprintf("%s = _%sReady", str.FirstUpper(config.GetName()), str.FirstUpper(config.GetName())) + } + + _ = os.MkdirAll(outputDir, 0666) + if err := file.WriterFile(filepath.Join(outputDir, "config.struct.go"), []byte(fmt.Sprintf(internal.TemplateStructGo, packageName, types))); err != nil { + panic(err) + } + if err := file.WriterFile(filepath.Join(outputDir, "config.go"), []byte(fmt.Sprintf(internal.TemplateGo, packageName, vars, varsMake, varsReplace))); err != nil { + panic(err) + } + cmd := exec.Command("gofmt", "-w", filepath.Join(outputDir, "config.struct.go")) + if err := cmd.Run(); err != nil { + fmt.Println(err) + } + + cmd = exec.Command("gofmt", "-w", filepath.Join(outputDir, "config.go")) + if err := cmd.Run(); err != nil { + fmt.Println(err) + } + +} diff --git a/planner/configexport/internal/config.go b/planner/configexport/internal/config.go new file mode 100644 index 0000000..0096fdd --- /dev/null +++ b/planner/configexport/internal/config.go @@ -0,0 +1,257 @@ +package internal + +import ( + "errors" + "fmt" + jsonIter "github.com/json-iterator/go" + "github.com/kercylan98/minotaur/utils/str" + "github.com/tealeg/xlsx" + "strings" +) + +func NewConfig(sheet *xlsx.Sheet) *Config { + config := &Config{ + Sheet: sheet, + } + + var ( + skipField = make(map[int]bool) + describeLine = config.Sheet.Rows[3] + nameLine = config.Sheet.Rows[4] + typeLine = config.Sheet.Rows[5] + exportParamLine = config.Sheet.Rows[6] + ) + // 分析数据 + { + for i := 1; i < len(describeLine.Cells); i++ { + describe := strings.TrimSpace(nameLine.Cells[i].String()) + if strings.HasPrefix(describe, "#") { + skipField[i] = true + continue + } + typ := strings.TrimSpace(typeLine.Cells[i].String()) + if strings.HasPrefix(typ, "#") || len(typ) == 0 { + skipField[i] = true + continue + } + exportParam := strings.TrimSpace(exportParamLine.Cells[i].String()) + if strings.HasPrefix(exportParam, "#") || len(exportParam) == 0 { + skipField[i] = true + continue + } + } + if len(nameLine.Cells)-1-len(skipField) < config.GetIndexCount() { + panic(errors.New("index count must greater or equal to field count")) + } + } + + config.skipField = skipField + config.describeLine = describeLine + config.nameLine = nameLine + config.typeLine = typeLine + config.exportParamLine = exportParamLine + + // 整理数据 + var ( + dataLine = make([]map[any]any, len(config.Sheet.Rows)) + ) + for i := 1; i < len(config.describeLine.Cells); i++ { + if skipField[i] { + continue + } + + var ( + //describe = strings.TrimSpace(describeLine.Cells[i].String()) + name = strings.TrimSpace(nameLine.Cells[i].String()) + //typ = strings.TrimSpace(typeLine.Cells[i].String()) + //exportParam = strings.TrimSpace(exportParamLine.Cells[i].String()) + ) + + for row := 7; row < len(config.Sheet.Rows); row++ { + //value := slf.Sheet.Rows[row].Cells[i].String() + var value = getValueWithType(typeLine.Cells[i].String(), config.Sheet.Rows[row].Cells[i].String()) + + line := dataLine[row] + if line == nil { + line = map[any]any{} + dataLine[row] = line + } + line[name] = value + } + } + + // 索引 + var dataSource = make(map[any]any) + var data = dataSource + var index = config.GetIndexCount() + var currentIndex = 0 + for row := 7; row < len(config.Sheet.Rows); row++ { + for i, cell := range config.Sheet.Rows[row].Cells { + if i == 0 || skipField[i] { + continue + } + var value = getValueWithType(typeLine.Cells[i].String(), cell.String()) + + if currentIndex < index { + currentIndex++ + m, exist := data[value] + if !exist { + if currentIndex == index { + data[value] = dataLine[row] + } else { + m = map[any]any{} + data[value] = m + } + } + if currentIndex < index { + data = m.(map[any]any) + } + } + } + data = dataSource + currentIndex = 0 + } + config.data = dataSource + config.dataLine = dataLine[len(dataLine)-1] + return config +} + +type Config struct { + *xlsx.Sheet + skipField map[int]bool + describeLine *xlsx.Row + nameLine *xlsx.Row + typeLine *xlsx.Row + exportParamLine *xlsx.Row + data map[any]any + dataLine map[any]any +} + +// GetDisplayName 获取显示名称 +func (slf *Config) GetDisplayName() string { + return slf.Name +} + +// GetName 获取配置名称 +func (slf *Config) GetName() string { + return slf.Sheet.Rows[0].Cells[1].String() +} + +// GetIndexCount 获取索引数量 +func (slf *Config) GetIndexCount() int { + index, err := slf.Sheet.Rows[1].Cells[1].Int() + if err != nil { + panic(err) + } + return index +} + +// GetData 获取数据 +func (slf *Config) GetData() map[any]any { + return slf.data +} + +// GetJSON 获取JSON类型数据 +func (slf *Config) GetJSON() []byte { + bytes, err := jsonIter.MarshalIndent(slf.GetData(), "", " ") + if err != nil { + panic(err) + } + return bytes +} + +func (slf *Config) GetVariable() string { + var index = slf.GetIndexCount() + var mapStr = "map[%s]%s" + for i := 1; i < len(slf.typeLine.Cells); i++ { + if slf.skipField[i] { + continue + } + typ := slf.typeLine.Cells[i].String() + if index > 0 { + index-- + if index == 0 { + mapStr = fmt.Sprintf(mapStr, typ, "%s") + } else { + mapStr = fmt.Sprintf(mapStr, typ, "map[%s]%s") + } + } else { + mapStr = fmt.Sprintf(mapStr, "*_"+str.FirstUpper(slf.GetName())) + break + } + } + return fmt.Sprintf("%s", mapStr) +} + +func (slf *Config) GetStruct() string { + var result string + for i := 1; i < len(slf.typeLine.Cells); i++ { + if slf.skipField[i] { + continue + } + typ := slf.typeLine.Cells[i].String() + name := slf.nameLine.Cells[i].String() + + name = str.FirstUpper(name) + result += fmt.Sprintf("%s %s\n", name, slf.GetType(typ)) + } + return fmt.Sprintf("type _%s struct{\n%s}", str.FirstUpper(slf.GetName()), result) +} + +func (slf *Config) GetType(fieldType string) string { + if name, exist := basicTypeName[fieldType]; exist { + return name + } else if strings.HasPrefix(fieldType, "[]") { + s := strings.TrimPrefix(fieldType, "[]") + if name, exist := basicTypeName[s]; exist { + return fmt.Sprintf("map[int]%s", name) + } else { + return slf.GetType(s) + } + } + + var s = strings.TrimSuffix(strings.TrimPrefix(fieldType, "{"), "}") + var fields []string + var field string + var leftBrackets []int + for i, c := range s { + switch c { + case ',': + if len(leftBrackets) == 0 { + fields = append(fields, field) + field = "" + } else { + field += string(c) + } + case '{': + leftBrackets = append(leftBrackets, i) + field += string(c) + case '}': + leftBrackets = leftBrackets[:len(leftBrackets)-1] + field += string(c) + if len(leftBrackets) == 0 { + fields = append(fields, field) + field = "" + } + default: + field += string(c) + } + } + if len(field) > 0 { + fields = append(fields, field) + } + + var result = "*struct {\n%s}" + var fieldStr string + for _, fieldInfo := range fields { + fieldName, fieldType := str.KV(strings.TrimSpace(fieldInfo), ":") + n, exist := basicTypeName[fieldType] + if exist { + fieldStr += fmt.Sprintf("%s %s\n", str.FirstUpper(fieldName), n) + } else { + fieldStr += fmt.Sprintf("%s %s\n", str.FirstUpper(fieldName), slf.GetType(fieldType)) + } + } + + return fmt.Sprintf(result, fieldStr) +} diff --git a/planner/configexport/internal/constants.go b/planner/configexport/internal/constants.go new file mode 100644 index 0000000..652e621 --- /dev/null +++ b/planner/configexport/internal/constants.go @@ -0,0 +1,42 @@ +package internal + +var TemplateGo = `// Code generated DO NOT EDIT. + +package %s + +import ( + jsonIter "github.com/json-iterator/go" + "os" +) + +var json = jsonIter.ConfigCompatibleWithStandardLibrary + +%s + +func LoadConfig(handle func(filename string, config any) error) { + %s +} + +func Replace() { + %s +} + + +func DefaultLoad(filepath string) { + LoadConfig(func(filename string, config any) error { + bytes, err := os.ReadFile(filepath) + if err != nil { + return err + } + return json.Unmarshal(bytes, &config) + }) +} + +` + +var TemplateStructGo = `// Code generated DO NOT EDIT. + +package %s + +%s +` diff --git a/planner/configexport/internal/export.go b/planner/configexport/internal/export.go new file mode 100644 index 0000000..008ab8d --- /dev/null +++ b/planner/configexport/internal/export.go @@ -0,0 +1,46 @@ +package internal + +import ( + "fmt" + "github.com/kercylan98/minotaur/utils/file" + "github.com/kercylan98/minotaur/utils/log" + "github.com/tealeg/xlsx" + "go.uber.org/zap" + "path/filepath" + "runtime/debug" + "sync" +) + +func ExportJSON(xlsxPath string, output string) { + xlsxFile, err := xlsx.OpenFile(xlsxPath) + if err != nil { + panic(err) + } + var errors []func() + var wait sync.WaitGroup + for _, sheet := range xlsxFile.Sheets { + sheet := sheet + go func() { + defer func() { + if err := recover(); err != nil { + errors = append(errors, func() { + log.Error("导出失败", zap.String("名称", xlsxPath), zap.String("Sheet", sheet.Name), zap.Any("err", err)) + fmt.Println(debug.Stack()) + }) + } + }() + wait.Add(1) + config := NewConfig(sheet) + if err := file.WriterFile(filepath.Join(output, fmt.Sprintf("%s.json", config.GetName())), config.GetJSON()); err != nil { + panic(err) + } + wait.Done() + }() + } + + wait.Wait() + + for _, f := range errors { + f() + } +} diff --git a/planner/configexport/internal/export_go.go b/planner/configexport/internal/export_go.go new file mode 100644 index 0000000..5bf0569 --- /dev/null +++ b/planner/configexport/internal/export_go.go @@ -0,0 +1 @@ +package internal diff --git a/planner/configexport/internal/internal/defines.go b/planner/configexport/internal/internal/defines.go new file mode 100644 index 0000000..5db08d7 --- /dev/null +++ b/planner/configexport/internal/internal/defines.go @@ -0,0 +1,22 @@ +package internal + +type XXConfig struct { + Id int + Count string + Award map[int]string + Info *struct { + Id int + Name string + Info *struct { + Lv int + Exp *struct { + Mux int + Count int + } + } + } + Other *struct { + Id int + Name string + } +} diff --git a/planner/configexport/internal/template.xlsx b/planner/configexport/internal/template.xlsx new file mode 100644 index 0000000..b21a7a5 Binary files /dev/null and b/planner/configexport/internal/template.xlsx differ diff --git a/planner/configexport/internal/types.go b/planner/configexport/internal/types.go new file mode 100644 index 0000000..2d0910c --- /dev/null +++ b/planner/configexport/internal/types.go @@ -0,0 +1,222 @@ +package internal + +import ( + "github.com/kercylan98/minotaur/utils/str" + "github.com/tidwall/gjson" + "strconv" + "strings" +) + +var basicType = map[string]func(fieldValue string) any{ + "string": withStringType, + "int": withIntType, + "int8": withInt8Type, + "int16": withInt16Type, + "int32": withInt32Type, + "int64": withInt64Type, + "uint": withUintType, + "uint8": withUint8Type, + "uint16": withUint16Type, + "uint32": withUint32Type, + "uint64": withUint64Type, + "float32": withFloat32Type, + "float64": withFloat64Type, + "float": withFloat64Type, + "double": withFloat64Type, + "number": withFloat64Type, + "byte": withByteType, + "rune": withRuneType, + "bool": withBoolType, + "boolean": withBoolType, +} + +var basicTypeName = map[string]string{ + "string": "string", + "int": "int", + "int8": "int8", + "int16": "int16", + "int32": "int32", + "int64": "int64", + "uint": "uint", + "uint8": "uint8", + "uint16": "uint16", + "uint32": "uint32", + "uint64": "uint64", + "float32": "float32", + "float64": "float64", + "float": "float64", + "double": "float64", + "number": "float64", + "byte": "byte", + "rune": "rune", + "bool": "bool", + "boolean": "bool", +} + +func getValueWithType(fieldType string, fieldValue string) any { + fieldType = strings.ToLower(strings.TrimSpace(fieldType)) + handle, exist := basicType[fieldType] + if exist { + return handle(fieldValue) + } else { + return withStructType(fieldType, fieldValue) + } +} + +func withStructType(fieldType string, fieldValue string) any { + // {id:int,name:string,info:{lv:int,exp:int}} + if strings.HasPrefix(fieldType, "[]") { + return withSliceType(fieldType, fieldValue) + } else if !strings.HasPrefix(fieldType, "{") || !strings.HasSuffix(fieldType, "}") { + return nil + } + var s = strings.TrimSuffix(strings.TrimPrefix(fieldType, "{"), "}") + var data = map[any]any{} + var fields []string + var field string + var leftBrackets []int + for i, c := range s { + switch c { + case ',': + if len(leftBrackets) == 0 { + fields = append(fields, field) + field = "" + } else { + field += string(c) + } + case '{': + leftBrackets = append(leftBrackets, i) + field += string(c) + case '}': + leftBrackets = leftBrackets[:len(leftBrackets)-1] + field += string(c) + if len(leftBrackets) == 0 { + fields = append(fields, field) + field = "" + } + default: + field += string(c) + } + } + if len(field) > 0 { + fields = append(fields, field) + } + + for _, fieldInfo := range fields { + fieldName, fieldType := str.KV(strings.TrimSpace(fieldInfo), ":") + handle, exist := basicType[fieldType] + if exist { + data[fieldName] = handle(gjson.Get(fieldValue, fieldName).String()) + } else { + data[fieldName] = withStructType(fieldType, gjson.Get(fieldValue, fieldName).String()) + } + } + + return data +} + +func withSliceType(fieldType string, fieldValue string) any { + if !strings.HasPrefix(fieldType, "[]") { + return nil + } + t := strings.TrimPrefix(fieldType, "[]") + var data = map[any]any{} + gjson.ForEachLine(fieldValue, func(line gjson.Result) bool { + line.ForEach(func(key, value gjson.Result) bool { + handle, exist := basicType[t] + if exist { + data[len(data)] = handle(value.String()) + } else { + data[len(data)] = withStructType(t, value.String()) + } + return true + }) + return true + }) + return data +} + +func withStringType(fieldValue string) any { + return fieldValue +} + +func withIntType(fieldValue string) any { + value, _ := strconv.Atoi(fieldValue) + return value +} + +func withInt8Type(fieldValue string) any { + value, _ := strconv.Atoi(fieldValue) + return int8(value) +} + +func withInt16Type(fieldValue string) any { + value, _ := strconv.Atoi(fieldValue) + return int16(value) +} + +func withInt32Type(fieldValue string) any { + value, _ := strconv.Atoi(fieldValue) + return int32(value) +} + +func withInt64Type(fieldValue string) any { + value, _ := strconv.ParseInt(fieldValue, 10, 64) + return value +} + +func withUintType(fieldValue string) any { + value, _ := strconv.Atoi(fieldValue) + return uint(value) +} + +func withUint8Type(fieldValue string) any { + value, _ := strconv.Atoi(fieldValue) + return uint8(value) +} + +func withUint16Type(fieldValue string) any { + value, _ := strconv.Atoi(fieldValue) + return uint16(value) +} + +func withUint32Type(fieldValue string) any { + value, _ := strconv.Atoi(fieldValue) + return uint32(value) +} + +func withUint64Type(fieldValue string) any { + value, _ := strconv.ParseInt(fieldValue, 10, 64) + return uint64(value) +} + +func withFloat32Type(fieldValue string) any { + value, _ := strconv.ParseFloat(fieldValue, 32) + return value +} + +func withFloat64Type(fieldValue string) any { + value, _ := strconv.ParseFloat(fieldValue, 64) + return value +} + +func withByteType(fieldValue string) any { + value, _ := strconv.Atoi(fieldValue) + return byte(value) +} + +func withRuneType(fieldValue string) any { + value, _ := strconv.Atoi(fieldValue) + return rune(value) +} + +func withBoolType(fieldValue string) any { + switch fieldValue { + case "0", "false", "!": + return false + case "1", "true", "&": + return true + default: + return false + } +} diff --git a/utils/str/str.go b/utils/str/str.go index df8f7e7..ddf83ee 100644 --- a/utils/str/str.go +++ b/utils/str/str.go @@ -1,2 +1,20 @@ package str +// FirstUpper 首字母大写 +func FirstUpper(str string) string { + var upperStr string + vv := []rune(str) // 后文有介绍 + for i := 0; i < len(vv); i++ { + if i == 0 { + if vv[i] >= 97 && vv[i] <= 122 { // 后文有介绍 + vv[i] -= 32 // string的码表相差32位 + upperStr += string(vv[i]) + } else { + return str + } + } else { + upperStr += string(vv[i]) + } + } + return upperStr +}