refactor: 入口移至 cmd/fun,go install 后命令名为 fun
This commit is contained in:
@@ -0,0 +1,225 @@
|
||||
package main
|
||||
|
||||
// TS 渲染器：fun v1.3.7 gen_ts.go 的等价移植，输入从 reflect.Type 换成元数据类型树
|
||||
|
||||
import "strings"
|
||||
|
||||
type renderTs struct{}
|
||||
|
||||
// typeName 等价旧 typeToTemplateType：指针 → " | null" 后缀，切片 → "[]"，
|
||||
// 整数 → number（枚举除外），struct/string → 类型名
|
||||
func (renderTs) typeName(t *MetaType) string {
|
||||
suffix := ""
|
||||
if t.Optional {
|
||||
suffix = " | null"
|
||||
}
|
||||
switch t.Kind {
|
||||
case "enum":
|
||||
return t.Name + suffix
|
||||
case "int":
|
||||
return "number" + suffix
|
||||
case "bool":
|
||||
return "boolean" + suffix
|
||||
case "string", "struct":
|
||||
return t.Name + suffix
|
||||
default: // slice
|
||||
return renderTs{}.typeName(t.Elem) + "[]" + suffix
|
||||
}
|
||||
}
|
||||
|
||||
func (r renderTs) genService(svc MetaService, serviceContext *genServiceType, files fileSet) {
|
||||
var nestedImports []*genImportType
|
||||
|
||||
for _, gm := range svc.Methods {
|
||||
var returnValueText string
|
||||
var dtoText string
|
||||
var argsText string
|
||||
var genericTypeText string
|
||||
|
||||
// () error：无数据返回，客户端类型为 void
|
||||
if gm.Void {
|
||||
genericTypeText = "void"
|
||||
returnValueText = "result<void>"
|
||||
if gm.DTO != nil {
|
||||
v := firstLetterToLower(r.typeName(gm.DTO))
|
||||
if !strings.Contains(v, "[]") && strings.Contains(v, "[") {
|
||||
dtoText += "dto:" + getGenericTypeName(v) + parseGenericTypeParams(v)
|
||||
} else {
|
||||
dtoText += "dto:" + v
|
||||
}
|
||||
argsText += ",dto"
|
||||
nestedImports = append(nestedImports, r.genStruct(gm.DTO, files))
|
||||
}
|
||||
serviceContext.IsIncludeRequest = true
|
||||
serviceContext.GenMethodTypeList = append(serviceContext.GenMethodTypeList, &genMethodType{
|
||||
MethodName: firstLetterToLower(gm.Name),
|
||||
ReturnValueText: returnValueText,
|
||||
DtoText: dtoText,
|
||||
ArgsText: argsText,
|
||||
GenericTypeText: genericTypeText,
|
||||
})
|
||||
continue
|
||||
}
|
||||
|
||||
if gm.IsStream {
|
||||
serviceContext.IsIncludeStream = true
|
||||
if gm.Returns == nil {
|
||||
genericTypeText = "any"
|
||||
returnValueText = "void"
|
||||
} else {
|
||||
t := firstLetterToLower(r.typeName(gm.Returns))
|
||||
if !strings.Contains(t, "[]") && strings.Contains(t, "[") {
|
||||
genericTypeText = getGenericTypeName(t) + parseGenericTypeParams(t)
|
||||
} else {
|
||||
genericTypeText = t
|
||||
}
|
||||
returnValueText = "void"
|
||||
nestedImports = r.genReturnTypes(gm.Returns, nestedImports, files)
|
||||
}
|
||||
} else {
|
||||
serviceContext.IsIncludeRequest = true
|
||||
t := firstLetterToLower(r.typeName(gm.Returns))
|
||||
if !strings.Contains(t, "[]") && strings.Contains(t, "[") {
|
||||
returnValueText = getGenericTypeName(t) + parseGenericTypeParams(t)
|
||||
} else {
|
||||
returnValueText = t
|
||||
}
|
||||
genericTypeText = returnValueText
|
||||
nestedImports = r.genReturnTypes(gm.Returns, nestedImports, files)
|
||||
returnValueText = "result<" + returnValueText + ">"
|
||||
}
|
||||
|
||||
if gm.DTO != nil {
|
||||
v := firstLetterToLower(r.typeName(gm.DTO))
|
||||
if !strings.Contains(v, "[]") && strings.Contains(v, "[") {
|
||||
dtoText += "dto:" + getGenericTypeName(v) + parseGenericTypeParams(v)
|
||||
} else {
|
||||
dtoText += "dto:" + v
|
||||
}
|
||||
argsText += ",dto"
|
||||
nestedImports = append(nestedImports, r.genStruct(gm.DTO, files))
|
||||
}
|
||||
|
||||
serviceContext.GenMethodTypeList = append(serviceContext.GenMethodTypeList, &genMethodType{
|
||||
MethodName: firstLetterToLower(gm.Name),
|
||||
ReturnValueText: returnValueText,
|
||||
DtoText: dtoText,
|
||||
ArgsText: argsText,
|
||||
GenericTypeText: firstLetterToLower(genericTypeText),
|
||||
IsStream: gm.IsStream,
|
||||
})
|
||||
}
|
||||
serviceContext.GenImport = deduplicateServiceImports(nestedImports)
|
||||
|
||||
renderCode(
|
||||
templateTs{}.genServiceTemplate(),
|
||||
firstLetterToLower(svc.Name),
|
||||
serviceContext,
|
||||
"ts",
|
||||
files,
|
||||
)
|
||||
}
|
||||
|
||||
// genReturnTypes 递归生成返回类型涉及的 struct/enum 导入
|
||||
func (r renderTs) genReturnTypes(returnType *MetaType, nestedImports []*genImportType, files fileSet) []*genImportType {
|
||||
if returnType.Kind == "struct" {
|
||||
nestedImports = append(nestedImports, r.genStruct(returnType, files))
|
||||
}
|
||||
if returnType.Kind == "slice" && returnType.Elem.Kind == "struct" {
|
||||
nestedImports = append(nestedImports, r.genStruct(returnType.Elem, files))
|
||||
}
|
||||
if returnType.Kind == "enum" {
|
||||
nestedImports = append(nestedImports, r.getEnum(returnType, files))
|
||||
}
|
||||
return nestedImports
|
||||
}
|
||||
|
||||
func (r renderTs) renderAll(meta *FunMeta, files fileSet) {
|
||||
genContext := genType{GenServiceList: []*genServiceType{}}
|
||||
|
||||
for _, svc := range meta.Services {
|
||||
serviceContext := &genServiceType{
|
||||
ServiceName: firstLetterToLower(svc.Name),
|
||||
GenMethodTypeList: []*genMethodType{},
|
||||
}
|
||||
genContext.GenServiceList = append(genContext.GenServiceList, serviceContext)
|
||||
r.genService(svc, serviceContext, files)
|
||||
}
|
||||
renderCode(templateTs{}.genClientTemplate(), "client", nil, "ts", files)
|
||||
renderCode(templateTs{}.genDefaultServiceTemplate(), "fun", genContext, "ts", files)
|
||||
}
|
||||
|
||||
func (r renderTs) genStruct(t *MetaType, files fileSet) *genImportType {
|
||||
var structTemplate genClassType
|
||||
if strings.Contains(t.Name, "[]") == false && strings.Contains(t.Name, "[") {
|
||||
structTemplate = genClassType{
|
||||
Name: firstLetterToLower(getGenericTypeName(t.Name)) + parseGenericTypeParams(t.Name),
|
||||
}
|
||||
} else {
|
||||
structTemplate = genClassType{
|
||||
Name: firstLetterToLower(t.Name),
|
||||
}
|
||||
}
|
||||
var nestedImports []*genImportType
|
||||
|
||||
for _, field := range t.Fields {
|
||||
fieldType := field.Type
|
||||
jsType := r.typeName(fieldType)
|
||||
name := field.Name
|
||||
if fieldType.Optional {
|
||||
name += "?"
|
||||
}
|
||||
if !strings.Contains(jsType, "[]") && strings.Contains(jsType, "[") {
|
||||
structTemplate.GenClassFieldType = append(structTemplate.GenClassFieldType, &genClassFieldType{
|
||||
Name: firstLetterToLower(name),
|
||||
Type: firstLetterToLower(getGenericTypeName(jsType)) + parseGenericTypeParams(jsType),
|
||||
})
|
||||
} else {
|
||||
structTemplate.GenClassFieldType = append(structTemplate.GenClassFieldType, &genClassFieldType{
|
||||
Name: firstLetterToLower(name),
|
||||
Type: firstLetterToLower(jsType),
|
||||
})
|
||||
}
|
||||
|
||||
if fieldType.Kind == "struct" {
|
||||
nestedImports = append(nestedImports, r.genStruct(fieldType, files))
|
||||
}
|
||||
if fieldType.Kind == "slice" && fieldType.Elem.Kind == "struct" {
|
||||
nestedImports = append(nestedImports, r.genStruct(fieldType.Elem, files))
|
||||
}
|
||||
if fieldType.Kind == "enum" {
|
||||
nestedImports = append(nestedImports, r.getEnum(fieldType, files))
|
||||
}
|
||||
}
|
||||
|
||||
structTemplate.GenImport = deduplicateServiceImports(nestedImports)
|
||||
|
||||
renderCode(
|
||||
templateTs{}.genStructTemplate(),
|
||||
structTemplate.Name,
|
||||
structTemplate,
|
||||
"ts",
|
||||
files,
|
||||
)
|
||||
|
||||
if strings.Contains(t.Name, "[]") == false && strings.Contains(t.Name, "[") {
|
||||
return &genImportType{Name: structTemplate.Name}
|
||||
}
|
||||
return &genImportType{Name: firstLetterToLower(t.Name)}
|
||||
}
|
||||
|
||||
func (r renderTs) getEnum(t *MetaType, files fileSet) *genImportType {
|
||||
var enumTemplate genEnumType
|
||||
enumTemplate.Names = t.Names
|
||||
enumTemplate.DisplayNames = t.DisplayNames
|
||||
enumTemplate.Name = firstLetterToLower(t.Name)
|
||||
|
||||
renderCode(
|
||||
templateTs{}.genEnumTemplate(),
|
||||
firstLetterToLower(t.Name),
|
||||
enumTemplate,
|
||||
"ts",
|
||||
files,
|
||||
)
|
||||
return &genImportType{Name: firstLetterToLower(t.Name)}
|
||||
}
|
||||
Reference in New Issue
Block a user