goctl added
This commit is contained in:
kim
46
tools/goctl/api/javagen/gen.go
Normal file
46
tools/goctl/api/javagen/gen.go
Normal file
@@ -0,0 +1,46 @@
|
||||
package javagen
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"strings"
|
||||
|
||||
"zero/core/lang"
|
||||
"zero/tools/goctl/api/parser"
|
||||
"zero/tools/goctl/util"
|
||||
|
||||
"github.com/logrusorgru/aurora"
|
||||
"github.com/urfave/cli"
|
||||
)
|
||||
|
||||
func JavaCommand(c *cli.Context) error {
|
||||
apiFile := c.String("api")
|
||||
dir := c.String("dir")
|
||||
if len(apiFile) == 0 {
|
||||
return errors.New("missing -api")
|
||||
}
|
||||
if len(dir) == 0 {
|
||||
return errors.New("missing -dir")
|
||||
}
|
||||
|
||||
p, err := parser.NewParser(apiFile)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
api, err := p.Parse()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
packetName := api.Service.Name
|
||||
if strings.HasSuffix(packetName, "-api") {
|
||||
packetName = packetName[:len(packetName)-4]
|
||||
}
|
||||
|
||||
lang.Must(util.MkdirIfNotExist(dir))
|
||||
lang.Must(genPacket(dir, packetName, api))
|
||||
lang.Must(genComponents(dir, packetName, api))
|
||||
|
||||
fmt.Println(aurora.Green("Done."))
|
||||
return nil
|
||||
}
|
||||
85
tools/goctl/api/javagen/gencomponents.go
Normal file
85
tools/goctl/api/javagen/gencomponents.go
Normal file
@@ -0,0 +1,85 @@
|
||||
package javagen
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"path"
|
||||
"strings"
|
||||
"text/template"
|
||||
|
||||
"zero/tools/goctl/api/spec"
|
||||
apiutil "zero/tools/goctl/api/util"
|
||||
"zero/tools/goctl/util"
|
||||
)
|
||||
|
||||
const (
|
||||
componentTemplate = `// DO NOT EDIT, generated by goctl
|
||||
package com.xhb.logic.http.packet.{{.packet}}.model;
|
||||
|
||||
import com.xhb.logic.http.DeProguardable;
|
||||
|
||||
{{.componentType}}
|
||||
`
|
||||
)
|
||||
|
||||
func genComponents(dir, packetName string, api *spec.ApiSpec) error {
|
||||
types := apiutil.GetSharedTypes(api)
|
||||
if len(types) == 0 {
|
||||
return nil
|
||||
}
|
||||
for _, ty := range types {
|
||||
if err := createComponent(dir, packetName, ty); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func createComponent(dir, packetName string, ty spec.Type) error {
|
||||
modelFile := util.Title(ty.Name) + ".java"
|
||||
filename := path.Join(dir, modelDir, modelFile)
|
||||
if err := util.RemoveOrQuit(filename); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
fp, created, err := apiutil.MaybeCreateFile(dir, modelDir, modelFile)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if !created {
|
||||
return nil
|
||||
}
|
||||
defer fp.Close()
|
||||
|
||||
tys, err := buildType(ty)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
t := template.Must(template.New("componentType").Parse(componentTemplate))
|
||||
return t.Execute(fp, map[string]string{
|
||||
"componentType": tys,
|
||||
"packet": packetName,
|
||||
})
|
||||
}
|
||||
|
||||
func buildType(ty spec.Type) (string, error) {
|
||||
var builder strings.Builder
|
||||
if err := writeType(&builder, ty); err != nil {
|
||||
return "", apiutil.WrapErr(err, "Type "+ty.Name+" generate error")
|
||||
}
|
||||
return builder.String(), nil
|
||||
}
|
||||
|
||||
func writeType(writer io.Writer, tp spec.Type) error {
|
||||
fmt.Fprintf(writer, "public class %s implements DeProguardable {\n", util.Title(tp.Name))
|
||||
for _, member := range tp.Members {
|
||||
if err := writeProperty(writer, member, 1); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
genGetSet(writer, tp, 1)
|
||||
fmt.Fprintf(writer, "}\n")
|
||||
return nil
|
||||
}
|
||||
277
tools/goctl/api/javagen/genpacket.go
Normal file
277
tools/goctl/api/javagen/genpacket.go
Normal file
@@ -0,0 +1,277 @@
|
||||
package javagen
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"strings"
|
||||
"text/template"
|
||||
|
||||
"zero/core/stringx"
|
||||
"zero/tools/goctl/api/spec"
|
||||
apiutil "zero/tools/goctl/api/util"
|
||||
"zero/tools/goctl/util"
|
||||
)
|
||||
|
||||
const packetTemplate = `package com.xhb.logic.http.packet.{{.packet}};
|
||||
|
||||
import com.google.gson.Gson;
|
||||
import com.xhb.commons.JSON;
|
||||
import com.xhb.commons.JsonParser;
|
||||
import com.xhb.core.network.HttpRequestClient;
|
||||
import com.xhb.core.packet.HttpRequestPacket;
|
||||
import com.xhb.core.response.HttpResponseData;
|
||||
import com.xhb.logic.http.DeProguardable;
|
||||
{{.import}}
|
||||
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
import org.json.JSONObject;
|
||||
|
||||
public class {{.packetName}} extends HttpRequestPacket<{{.packetName}}.{{.packetName}}Response> {
|
||||
|
||||
{{.paramsDeclaration}}
|
||||
|
||||
public {{.packetName}}({{.params}}{{.requestType}} request) {
|
||||
super(request);
|
||||
this.request = request;{{.paramsSet}}
|
||||
}
|
||||
|
||||
@Override
|
||||
public HttpRequestClient.Method requestMethod() {
|
||||
return HttpRequestClient.Method.{{.method}};
|
||||
}
|
||||
|
||||
@Override
|
||||
public String requestUri() {
|
||||
return {{.uri}};
|
||||
}
|
||||
|
||||
@Override
|
||||
public {{.packetName}}Response newInstanceFrom(JSON json) {
|
||||
return new {{.packetName}}Response(json);
|
||||
}
|
||||
|
||||
public static class {{.packetName}}Response extends HttpResponseData {
|
||||
|
||||
private {{.responseType}} responseData;
|
||||
|
||||
{{.packetName}}Response(@NotNull JSON json) {
|
||||
super(json);
|
||||
JSONObject jsonObject = json.asObject();
|
||||
if (JsonParser.hasKey(jsonObject, "data")) {
|
||||
Gson gson = new Gson();
|
||||
JSONObject dataJson = JsonParser.getJSONObject(jsonObject, "data");
|
||||
responseData = gson.fromJson(dataJson.toString(), {{.responseType}}.class);
|
||||
}
|
||||
}
|
||||
|
||||
public {{.responseType}} get{{.responseType}} () {
|
||||
return responseData;
|
||||
}
|
||||
}
|
||||
|
||||
{{.types}}
|
||||
}
|
||||
`
|
||||
|
||||
func genPacket(dir, packetName string, api *spec.ApiSpec) error {
|
||||
for _, route := range api.Service.Routes {
|
||||
if err := createWith(dir, api, route, packetName); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func createWith(dir string, api *spec.ApiSpec, route spec.Route, packetName string) error {
|
||||
packet, ok := apiutil.GetAnnotationValue(route.Annotations, "server", "handler")
|
||||
packet = strings.Replace(packet, "Handler", "Packet", 1)
|
||||
if !ok {
|
||||
return fmt.Errorf("missing packet annotation for %q", route.Path)
|
||||
}
|
||||
|
||||
javaFile := packet + ".java"
|
||||
fp, created, err := apiutil.MaybeCreateFile(dir, "", javaFile)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if !created {
|
||||
return nil
|
||||
}
|
||||
defer fp.Close()
|
||||
|
||||
var builder strings.Builder
|
||||
var first bool
|
||||
tps := apiutil.GetLocalTypes(api, route)
|
||||
|
||||
for _, tp := range tps {
|
||||
if first {
|
||||
first = false
|
||||
} else {
|
||||
fmt.Fprintln(&builder)
|
||||
}
|
||||
if err := genType(&builder, tp); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
types := builder.String()
|
||||
writeIndent(&builder, 1)
|
||||
|
||||
params := paramsForRoute(route)
|
||||
paramsDeclaration := declarationForRoute(route)
|
||||
paramsSet := paramsSet(route)
|
||||
|
||||
t := template.Must(template.New("packetTemplate").Parse(packetTemplate))
|
||||
var tmplBytes bytes.Buffer
|
||||
err = t.Execute(&tmplBytes, map[string]string{
|
||||
"packetName": packet,
|
||||
"method": strings.ToUpper(route.Method),
|
||||
"uri": processUri(route),
|
||||
"types": strings.TrimSpace(types),
|
||||
"responseType": stringx.TakeOne(util.Title(route.ResponseType.Name), "Object"),
|
||||
"params": params,
|
||||
"paramsDeclaration": strings.TrimSpace(paramsDeclaration),
|
||||
"paramsSet": paramsSet,
|
||||
"packet": packetName,
|
||||
"requestType": util.Title(route.RequestType.Name),
|
||||
"import": getImports(api, route, packetName),
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
formatFile(&tmplBytes, fp)
|
||||
return nil
|
||||
}
|
||||
|
||||
func getImports(api *spec.ApiSpec, route spec.Route, packetName string) string {
|
||||
var builder strings.Builder
|
||||
allTypes := apiutil.GetAllTypes(api, route)
|
||||
sharedTypes := apiutil.GetSharedTypes(api)
|
||||
for _, at := range allTypes {
|
||||
for _, item := range sharedTypes {
|
||||
if item.Name == at.Name {
|
||||
fmt.Fprintf(&builder, "import com.xhb.logic.http.packet.%s.model.%s;\n", packetName, item.Name)
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
return builder.String()
|
||||
}
|
||||
|
||||
func formatFile(tmplBytes *bytes.Buffer, file *os.File) {
|
||||
scanner := bufio.NewScanner(tmplBytes)
|
||||
builder := bufio.NewWriter(file)
|
||||
defer builder.Flush()
|
||||
preIsBreakLine := false
|
||||
for scanner.Scan() {
|
||||
text := strings.TrimSpace(scanner.Text())
|
||||
if text == "" && preIsBreakLine {
|
||||
continue
|
||||
}
|
||||
preIsBreakLine = text == ""
|
||||
builder.WriteString(scanner.Text() + "\n")
|
||||
}
|
||||
if err := scanner.Err(); err != nil {
|
||||
println(err)
|
||||
}
|
||||
}
|
||||
|
||||
func paramsSet(route spec.Route) string {
|
||||
path := route.Path
|
||||
cops := strings.Split(path, "/")
|
||||
var builder strings.Builder
|
||||
for _, cop := range cops {
|
||||
if len(cop) == 0 {
|
||||
continue
|
||||
}
|
||||
if strings.HasPrefix(cop, ":") {
|
||||
param := cop[1:]
|
||||
builder.WriteString("\n")
|
||||
builder.WriteString(fmt.Sprintf("\t\tthis.%s = %s;", param, param))
|
||||
}
|
||||
}
|
||||
result := builder.String()
|
||||
return result
|
||||
}
|
||||
|
||||
func paramsForRoute(route spec.Route) string {
|
||||
path := route.Path
|
||||
cops := strings.Split(path, "/")
|
||||
var builder strings.Builder
|
||||
for _, cop := range cops {
|
||||
if len(cop) == 0 {
|
||||
continue
|
||||
}
|
||||
if strings.HasPrefix(cop, ":") {
|
||||
builder.WriteString(fmt.Sprintf("String %s, ", cop[1:]))
|
||||
}
|
||||
}
|
||||
return builder.String()
|
||||
}
|
||||
|
||||
func declarationForRoute(route spec.Route) string {
|
||||
path := route.Path
|
||||
cops := strings.Split(path, "/")
|
||||
var builder strings.Builder
|
||||
writeIndent(&builder, 1)
|
||||
for _, cop := range cops {
|
||||
if len(cop) == 0 {
|
||||
continue
|
||||
}
|
||||
if strings.HasPrefix(cop, ":") {
|
||||
writeIndent(&builder, 1)
|
||||
builder.WriteString(fmt.Sprintf("private String %s;\n", cop[1:]))
|
||||
}
|
||||
}
|
||||
result := strings.TrimSpace(builder.String())
|
||||
if len(result) > 0 {
|
||||
result = "\n" + result
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func processUri(route spec.Route) string {
|
||||
path := route.Path
|
||||
var builder strings.Builder
|
||||
cops := strings.Split(path, "/")
|
||||
for index, cop := range cops {
|
||||
if len(cop) == 0 {
|
||||
continue
|
||||
}
|
||||
if strings.HasPrefix(cop, ":") {
|
||||
builder.WriteString("/\" + " + cop[1:] + " + \"")
|
||||
} else {
|
||||
builder.WriteString("/" + cop)
|
||||
if index == len(cops)-1 {
|
||||
builder.WriteString("\"")
|
||||
}
|
||||
}
|
||||
}
|
||||
result := builder.String()
|
||||
if strings.HasSuffix(result, " + \"") {
|
||||
result = result[:len(result)-4]
|
||||
}
|
||||
if strings.HasPrefix(result, "/") {
|
||||
result = "\"" + result
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func genType(writer io.Writer, tp spec.Type) error {
|
||||
writeIndent(writer, 1)
|
||||
fmt.Fprintf(writer, "static class %s implements DeProguardable {\n", util.Title(tp.Name))
|
||||
for _, member := range tp.Members {
|
||||
if err := writeProperty(writer, member, 2); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
writeBreakline(writer)
|
||||
writeIndent(writer, 1)
|
||||
genGetSet(writer, tp, 2)
|
||||
writeIndent(writer, 1)
|
||||
fmt.Fprintln(writer, "}")
|
||||
return nil
|
||||
}
|
||||
163
tools/goctl/api/javagen/util.go
Normal file
163
tools/goctl/api/javagen/util.go
Normal file
@@ -0,0 +1,163 @@
|
||||
package javagen
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"strings"
|
||||
"text/template"
|
||||
|
||||
"zero/tools/goctl/api/spec"
|
||||
apiutil "zero/tools/goctl/api/util"
|
||||
"zero/tools/goctl/util"
|
||||
)
|
||||
|
||||
const getSetTemplate = `
|
||||
{{.indent}}{{.decorator}}
|
||||
{{.indent}}public {{.returnType}} get{{.property}}() {
|
||||
{{.indent}} return this.{{.propertyValue}};
|
||||
{{.indent}}}
|
||||
|
||||
{{.indent}}public void set{{.property}}({{.type}} {{.propertyValue}}) {
|
||||
{{.indent}} this.{{.propertyValue}} = {{.propertyValue}};
|
||||
{{.indent}}}
|
||||
`
|
||||
|
||||
func writeProperty(writer io.Writer, member spec.Member, indent int) error {
|
||||
writeIndent(writer, indent)
|
||||
ty, err := goTypeToJava(member.Type)
|
||||
ty = strings.Replace(ty, "*", "", 1)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
name, err := member.GetPropertyName()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_, err = fmt.Fprintf(writer, "private %s %s", ty, name)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
writeDefaultValue(writer, member)
|
||||
fmt.Fprint(writer, ";\n")
|
||||
return err
|
||||
}
|
||||
|
||||
func writeDefaultValue(writer io.Writer, member spec.Member) error {
|
||||
switch member.Type {
|
||||
case "string":
|
||||
_, err := fmt.Fprintf(writer, " = \"\"")
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func writeIndent(writer io.Writer, indent int) {
|
||||
for i := 0; i < indent; i++ {
|
||||
fmt.Fprint(writer, "\t")
|
||||
}
|
||||
}
|
||||
|
||||
func indentString(indent int) string {
|
||||
var result = ""
|
||||
for i := 0; i < indent; i++ {
|
||||
result += "\t"
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func writeBreakline(writer io.Writer) {
|
||||
fmt.Fprint(writer, "\n")
|
||||
}
|
||||
|
||||
func isPrimitiveType(tp string) bool {
|
||||
switch tp {
|
||||
case "int", "int32", "int64":
|
||||
return true
|
||||
case "float", "float32", "float64":
|
||||
return true
|
||||
case "bool":
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func goTypeToJava(tp string) (string, error) {
|
||||
if len(tp) == 0 {
|
||||
return "", errors.New("property type empty")
|
||||
}
|
||||
if strings.HasPrefix(tp, "*") {
|
||||
tp = tp[1:]
|
||||
}
|
||||
switch tp {
|
||||
case "string":
|
||||
return "String", nil
|
||||
case "int64":
|
||||
return "long", nil
|
||||
case "int", "int8", "int32":
|
||||
return "int", nil
|
||||
case "float", "float32", "float64":
|
||||
return "double", nil
|
||||
case "bool":
|
||||
return "boolean", nil
|
||||
}
|
||||
if strings.HasPrefix(tp, "[]") {
|
||||
tys, err := apiutil.DecomposeType(tp)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if len(tys) == 0 {
|
||||
return "", fmt.Errorf("%s tp parse error", tp)
|
||||
}
|
||||
return fmt.Sprintf("java.util.ArrayList<%s>", util.Title(tys[0])), nil
|
||||
} else if strings.HasPrefix(tp, "map") {
|
||||
tys, err := apiutil.DecomposeType(tp)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if len(tys) == 2 {
|
||||
return "", fmt.Errorf("%s tp parse error", tp)
|
||||
}
|
||||
return fmt.Sprintf("java.util.HashMap<String, %s>", util.Title(tys[1])), nil
|
||||
}
|
||||
return util.Title(tp), nil
|
||||
}
|
||||
|
||||
func genGetSet(writer io.Writer, tp spec.Type, indent int) error {
|
||||
t := template.Must(template.New("getSetTemplate").Parse(getSetTemplate))
|
||||
for _, member := range tp.Members {
|
||||
var tmplBytes bytes.Buffer
|
||||
|
||||
oty, err := goTypeToJava(member.Type)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
tyString := oty
|
||||
decorator := ""
|
||||
if !isPrimitiveType(member.Type) {
|
||||
if member.IsOptional() {
|
||||
decorator = "@org.jetbrains.annotations.Nullable "
|
||||
} else {
|
||||
decorator = "@org.jetbrains.annotations.NotNull "
|
||||
}
|
||||
tyString = decorator + tyString
|
||||
}
|
||||
|
||||
err = t.Execute(&tmplBytes, map[string]string{
|
||||
"property": util.Title(member.Name),
|
||||
"propertyValue": util.Untitle(member.Name),
|
||||
"type": tyString,
|
||||
"decorator": decorator,
|
||||
"returnType": oty,
|
||||
"indent": indentString(indent),
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
r := tmplBytes.String()
|
||||
r = strings.Replace(r, " boolean get", " boolean is", 1)
|
||||
writer.Write([]byte(r))
|
||||
}
|
||||
return nil
|
||||
}
|
||||
3
tools/goctl/api/javagen/vars.go
Normal file
3
tools/goctl/api/javagen/vars.go
Normal file
@@ -0,0 +1,3 @@
|
||||
package javagen
|
||||
|
||||
const modelDir = "model"
|
||||
Reference in New Issue
Block a user