You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
2.3 KiB
Go
103 lines
2.3 KiB
Go
package dartgen
|
|
|
|
import (
|
|
"os"
|
|
"text/template"
|
|
|
|
"github.com/zeromicro/go-zero/tools/goctl/api/spec"
|
|
)
|
|
|
|
const dataTemplate = `// --{{with .Info}}{{.Title}}{{end}}--
|
|
{{ range .Types}}
|
|
class {{.Name}}{
|
|
{{range .Members}}
|
|
/// {{.Comment}}
|
|
final {{.Type.Name}} {{lowCamelCase .Name}};
|
|
{{end}}
|
|
{{.Name}}({ {{range .Members}}
|
|
this.{{lowCamelCase .Name}},{{end}}
|
|
});
|
|
factory {{.Name}}.fromJson(Map<String,dynamic> m) {
|
|
return {{.Name}}({{range .Members}}
|
|
{{lowCamelCase .Name}}: {{if isDirectType .Type.Name}}m['{{tagGet .Tag "json"}}']{{else if isClassListType .Type.Name}}(m['{{tagGet .Tag "json"}}'] as List<dynamic>).map((i) => {{getCoreType .Type.Name}}.fromJson(i)){{else}}{{.Type.Name}}.fromJson(m['{{tagGet .Tag "json"}}']){{end}},{{end}}
|
|
);
|
|
}
|
|
Map<String,dynamic> toJson() {
|
|
return { {{range .Members}}
|
|
'{{tagGet .Tag "json"}}': {{if isDirectType .Type.Name}}{{lowCamelCase .Name}}{{else if isClassListType .Type.Name}}{{lowCamelCase .Name}}.map((i) => i.toJson()){{else}}{{lowCamelCase .Name}}.toJson(){{end}},{{end}}
|
|
};
|
|
}
|
|
}
|
|
{{end}}
|
|
`
|
|
|
|
func genData(dir string, api *spec.ApiSpec) error {
|
|
err := os.MkdirAll(dir, 0o755)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = genTokens(dir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
file, err := os.OpenFile(dir+api.Service.Name+".dart", os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0o644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
t := template.New("dataTemplate")
|
|
t = t.Funcs(funcMap)
|
|
t, err = t.Parse(dataTemplate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = convertDataType(api)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return t.Execute(file, api)
|
|
}
|
|
|
|
func genTokens(dir string) error {
|
|
path := dir + "tokens.dart"
|
|
if fileExists(path) {
|
|
return nil
|
|
}
|
|
|
|
tokensFile, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0o644)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer tokensFile.Close()
|
|
_, err = tokensFile.WriteString(tokensFileContent)
|
|
return err
|
|
}
|
|
|
|
func convertDataType(api *spec.ApiSpec) error {
|
|
types := api.Types
|
|
if len(types) == 0 {
|
|
return nil
|
|
}
|
|
|
|
for _, ty := range types {
|
|
defineStruct, ok := ty.(spec.DefineStruct)
|
|
if ok {
|
|
for index, member := range defineStruct.Members {
|
|
tp, err := specTypeToDart(member.Type)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defineStruct.Members[index].Type = buildSpecType(member.Type, tp)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|