mirror of
https://github.com/go-home-admin/toolset.git
synced 2025-12-24 13:37:52 +08:00
386 lines
9.4 KiB
Go
386 lines
9.4 KiB
Go
package commands
|
||
|
||
import (
|
||
"github.com/ctfang/command"
|
||
"github.com/go-home-admin/toolset/parser"
|
||
"log"
|
||
"os"
|
||
"sort"
|
||
"strconv"
|
||
"strings"
|
||
)
|
||
|
||
// BeanCommand @Bean
|
||
type BeanCommand struct{}
|
||
|
||
func (BeanCommand) Configure() command.Configure {
|
||
return command.Configure{
|
||
Name: "make:bean",
|
||
Description: "生成依赖注入的声明源代码文件, 使用@Bean注解, 和inject引入",
|
||
Input: command.Argument{
|
||
Has: []command.ArgParam{
|
||
{
|
||
Name: "-f",
|
||
Description: "强制更新",
|
||
},
|
||
},
|
||
Option: []command.ArgParam{
|
||
{
|
||
Name: "name",
|
||
Description: "New函数别名, 如果兼容旧的项目可以设置",
|
||
Default: "New{name}",
|
||
},
|
||
{
|
||
Name: "scan",
|
||
Description: "扫码目录下的源码; shell(pwd)",
|
||
Default: "@root",
|
||
},
|
||
{
|
||
Name: "skip",
|
||
Description: "跳过目录",
|
||
Default: "@root/generate",
|
||
},
|
||
{
|
||
Name: "skip",
|
||
Description: "跳过目录",
|
||
Default: "@root/vendor",
|
||
},
|
||
},
|
||
},
|
||
}
|
||
}
|
||
|
||
var newName = "New{name}"
|
||
|
||
func (BeanCommand) Execute(input command.Input) {
|
||
newName = input.GetOption("name")
|
||
|
||
root := getRootPath()
|
||
scan := input.GetOption("scan")
|
||
scan = strings.Replace(scan, "@root", root, 1)
|
||
|
||
skip := make(map[string]bool)
|
||
for _, s := range input.GetOptions("skip") {
|
||
s = strings.Replace(s, "@root", root, 1)
|
||
skip[s] = true
|
||
}
|
||
|
||
fileList := parser.NewAst(scan, skip)
|
||
var keys []string
|
||
for s, _ := range fileList {
|
||
keys = append(keys, s)
|
||
}
|
||
sort.Strings(keys)
|
||
|
||
for _, dir := range keys {
|
||
fileParsers := fileList[dir]
|
||
isSkip := false
|
||
for s := range skip {
|
||
if strings.Index(dir, s) != -1 {
|
||
isSkip = true
|
||
break
|
||
}
|
||
}
|
||
if isSkip {
|
||
continue
|
||
}
|
||
|
||
bc := newBeanCache()
|
||
for _, fileParser := range fileParsers {
|
||
bc.name = fileParser.PackageName
|
||
keys := make([]string, 0)
|
||
for k := range fileParser.Types {
|
||
keys = append(keys, k)
|
||
}
|
||
sort.Slice(keys, func(i, j int) bool { return keys[i] < keys[j] })
|
||
// 排序后循环 type
|
||
for _, k := range keys {
|
||
goType := fileParser.Types[k]
|
||
for _, attr := range goType.Attrs {
|
||
if attr.HasTag("inject") || attr.HasTag("impl") {
|
||
// 只收集使用到的 import
|
||
bc.imports[fileParser.Imports[attr.TypeAlias]] = attr.TypeAlias
|
||
}
|
||
}
|
||
|
||
if goType.Doc.HasAnnotation("@Bean") {
|
||
bc.structList = append(bc.structList, goType)
|
||
}
|
||
}
|
||
}
|
||
|
||
genBean(dir, bc)
|
||
}
|
||
}
|
||
|
||
type beanCache struct {
|
||
name string
|
||
// path => alias
|
||
imports map[string]string
|
||
structList []parser.GoType
|
||
}
|
||
|
||
func newBeanCache() beanCache {
|
||
return beanCache{
|
||
imports: map[string]string{
|
||
"github.com/go-home-admin/home/bootstrap/providers": "github.com/go-home-admin/home/bootstrap/providers",
|
||
},
|
||
structList: make([]parser.GoType, 0),
|
||
}
|
||
}
|
||
|
||
func genBean(dir string, bc beanCache) {
|
||
if len(bc.structList) == 0 {
|
||
return
|
||
}
|
||
context := make([]string, 0)
|
||
context = append(context, "package "+bc.name)
|
||
|
||
// import
|
||
importAlias := parser.GenImportAlias(strings.ReplaceAll(dir, getRootPath(), ""), bc.name, bc.imports)
|
||
if len(importAlias) != 0 {
|
||
context = append(context, "\nimport ("+getImportStr(bc, importAlias)+"\n)")
|
||
}
|
||
|
||
// Single
|
||
context = append(context, genSingle(bc))
|
||
// Provider
|
||
context = append(context, genProvider(bc, importAlias))
|
||
str := "// gen for home toolset"
|
||
for _, s2 := range context {
|
||
str = str + "\n" + s2
|
||
}
|
||
|
||
err := os.WriteFile(dir+"/z_inject_gen.go", []byte(str+"\n"), 0766)
|
||
if err != nil {
|
||
log.Fatal(err)
|
||
}
|
||
}
|
||
|
||
func genSingle(bc beanCache) string {
|
||
str := ""
|
||
allProviderStr := "\n\treturn []interface{}{"
|
||
for _, goType := range bc.structList {
|
||
if goType.Doc.HasAnnotation("@Bean") {
|
||
str = str + "\nvar " + genSingleName(goType.Name) + " *" + goType.Name
|
||
allProviderStr += "\n\t\t" + genInitializeNewStr(goType.Name) + "(),"
|
||
}
|
||
}
|
||
// 返回全部的提供商
|
||
str += "\n\nfunc GetAllProvider() []interface{} {" + allProviderStr + "\n\t}\n}"
|
||
return str
|
||
}
|
||
|
||
func genSingleName(s string) string {
|
||
return "_" + s + "Single"
|
||
}
|
||
|
||
func genProvider(bc beanCache, m map[string]string) string {
|
||
str := ""
|
||
for _, goType := range bc.structList {
|
||
sVar := genSingleName(goType.Name)
|
||
if goType.Doc.HasAnnotation("@Bean") {
|
||
str = str + "\nfunc " + genInitializeNewStr(goType.Name) + "() *" + goType.Name + " {" +
|
||
"\n\tif " + sVar + " == nil {" + // if _provider == nil {
|
||
"\n\t\t" + sVar + " = " + "&" + goType.Name + "{}" // provider := provider{}
|
||
|
||
for _, attrName := range goType.AttrsSort {
|
||
attr := goType.Attrs[attrName]
|
||
pointer := ""
|
||
if !attr.IsPointer() {
|
||
pointer = "*"
|
||
}
|
||
|
||
for tagName := range attr.Tag {
|
||
if tagName == "inject" {
|
||
str = str + "\n\t\t" +
|
||
sVar + "." + attrName + " = " + pointer + getInitializeNewFunName(attr, m)
|
||
} else if tagName == "impl" {
|
||
// 使用接口 接收注入
|
||
tmp := getInitializeNewFunName(attr, m)
|
||
tmp = strings.ReplaceAll(tmp, "*", "")
|
||
|
||
str = str + "\n\t\t" +
|
||
sVar + "." + attrName + " = " + tmp
|
||
}
|
||
}
|
||
}
|
||
|
||
constraint := m["github.com/go-home-admin/home/bootstrap/providers"]
|
||
if constraint != "" {
|
||
constraint += "."
|
||
}
|
||
str = str +
|
||
"\n\t\t" + constraint + "AfterProvider(" + sVar + ", \"" + goType.Doc.GetAlias() + "\")" +
|
||
"\n\t}" +
|
||
"\n\treturn " + sVar +
|
||
"\n}"
|
||
}
|
||
}
|
||
|
||
return str
|
||
}
|
||
|
||
func getInitializeNewFunName(k parser.GoTypeAttr, m map[string]string) string {
|
||
alias := ""
|
||
name := k.TypeName
|
||
|
||
if !k.InPackage {
|
||
a := m[k.TypeImport]
|
||
if a == "" {
|
||
panic("识别到不明确的import, 最后一个目录和package名称不一致时，需要手动， 例如\nredis \"github.com/go-redis/redis/v8\"")
|
||
} else {
|
||
alias = a + "."
|
||
}
|
||
arr := strings.Split(k.TypeName, ".")
|
||
name = arr[len(arr)-1]
|
||
} else if name[0:1] == "*" {
|
||
name = name[1:]
|
||
}
|
||
tag := k.Tag["inject"]
|
||
if tag == "" {
|
||
tag = k.Tag["impl"]
|
||
}
|
||
|
||
if tag == "" {
|
||
return alias + genInitializeNewStr(name) + "()"
|
||
} else {
|
||
beanAlias := tag.Get(0)
|
||
beanValue := tag.Get(1)
|
||
|
||
providers := m["github.com/go-home-admin/home/bootstrap/providers"]
|
||
if providers != "" {
|
||
providers += "."
|
||
}
|
||
|
||
got := providers + "GetBean(\"" + beanAlias + "\").(" + providers + "Bean)"
|
||
if strings.Index(beanValue, "@") != -1 {
|
||
// 嵌套时, inject: "@config("app.name")"
|
||
// 只支持字符串嵌套注入
|
||
|
||
startTemp := strings.Index(beanValue, "(")
|
||
beanValueNextName := beanValue[1:startTemp]
|
||
if beanValue[len(beanValue)-1:] != ")" {
|
||
beanValue = beanValue + ", " + tag.Get(2)
|
||
}
|
||
beanValueNextVal := strings.Trim(beanValue[startTemp+1:], ")")
|
||
got = got + ".GetBean(*" + providers + "GetBean(\"" + beanValueNextName + "\").(" + providers + "Bean).GetBean(\"" + beanValueNextVal + "\").(*string))"
|
||
} else if tag.Count() <= 2 {
|
||
// 如果没有默认值
|
||
if beanValue == "" {
|
||
gotType := "*" + alias + name
|
||
|
||
// 如果`inject:""`检查是否实现Bean接口, 没有实现就不调用
|
||
gotFuncStr := `func() {gotType} {
|
||
var temp = {providers}GetBean("{beanAlias}")
|
||
if bean, ok := temp.({providers}Bean); ok {
|
||
return bean.GetBean("").({gotType})
|
||
}
|
||
return temp.({gotType})
|
||
}()`
|
||
got = strings.ReplaceAll(gotFuncStr, "{providers}", providers)
|
||
got = strings.Replace(got, "{beanAlias}", beanAlias, 1)
|
||
got = strings.ReplaceAll(got, "{gotType}", gotType)
|
||
return got
|
||
} else {
|
||
// 必然是实现了Bean接口
|
||
got = got + ".GetBean(\"" + beanValue + "\")"
|
||
}
|
||
} else if tag.Count() == 3 {
|
||
// 多个参数||有默认值的注入
|
||
beanValue = beanValue + ", " + tag.Get(2)
|
||
got = got + ".GetBean(`" + beanValue + "`)"
|
||
}
|
||
// 如果被注入的是接口时, 这里的*是多余的, 还不支持注入到 type interface 的变量
|
||
// 如果有需要, 自行复制z_inject_gen.go内代码, 把开头*和类型*删除，就兼容了
|
||
return got + ".(*" + alias + name + ")"
|
||
}
|
||
}
|
||
|
||
// 控制对完函数名称
|
||
func genInitializeNewStr(name string) string {
|
||
if name[0:1] == "*" {
|
||
name = name[1:]
|
||
}
|
||
|
||
return strings.Replace(newName, "{name}", name, 1)
|
||
}
|
||
|
||
// 生成 import => alias
|
||
func genImportAlias(m map[string]string) map[string]string {
|
||
aliasMapImport := make(map[string]string)
|
||
importMapAlias := make(map[string]string)
|
||
for iname, imp := range m {
|
||
if iname != imp {
|
||
if aliasMapImport[iname] != "" && aliasMapImport[iname] != imp {
|
||
aliasMapImport[iname+"_2"] = imp
|
||
} else {
|
||
aliasMapImport[iname] = imp
|
||
}
|
||
} else {
|
||
temp := strings.Split(imp, "/")
|
||
key := temp[len(temp)-1]
|
||
|
||
if _, ok := aliasMapImport[key]; ok {
|
||
for i := 1; i < 1000; i++ {
|
||
newKey := key + strconv.Itoa(i)
|
||
if _, ok2 := aliasMapImport[newKey]; !ok2 {
|
||
key = newKey
|
||
break
|
||
}
|
||
}
|
||
}
|
||
aliasMapImport[key] = imp
|
||
}
|
||
}
|
||
for s, s2 := range aliasMapImport {
|
||
importMapAlias[s2] = s
|
||
}
|
||
|
||
return importMapAlias
|
||
}
|
||
|
||
// cm = import => alias
|
||
func getImportStr(bc beanCache, m map[string]string) string {
|
||
has := map[string]bool{
|
||
"github.com/go-home-admin/home/bootstrap/providers": true,
|
||
}
|
||
|
||
for _, goType := range bc.structList {
|
||
if goType.Doc.HasAnnotation("@Bean") {
|
||
for _, attr := range goType.Attrs {
|
||
if !attr.InPackage {
|
||
has[attr.TypeImport] = true
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// 删除未使用的import
|
||
nm := make(map[string]string)
|
||
for s, vv := range m {
|
||
if _, ok := has[s]; ok {
|
||
nm[s] = vv
|
||
}
|
||
}
|
||
|
||
sk := sortMap(nm)
|
||
got := ""
|
||
for _, k := range sk {
|
||
if k != "" {
|
||
got += "\n\t" + nm[k] + " \"" + k + "\""
|
||
}
|
||
}
|
||
|
||
return got
|
||
}
|
||
|
||
func sortMap(m map[string]string) []string {
|
||
var keys []string
|
||
for k := range m {
|
||
keys = append(keys, k)
|
||
}
|
||
sort.Strings(keys)
|
||
return keys
|
||
}
|