dev

main.go

@@ -5,7 +5,9 @@ import (
 	"io"
 	"os"
 	"path/filepath"
+	"strings"
 
+	"git.apinb.com/bsm-sdk/core/utils"
 	"git.apinb.com/bsm-tools/protoc-gen-ts/internal/plugin"
 	"google.golang.org/protobuf/proto"
 	"google.golang.org/protobuf/types/pluginpb"
@@ -16,6 +18,28 @@ func main() {
 		fmt.Fprintf(os.Stderr, "%s: %v\n", filepath.Base(os.Args[0]), err)
 		os.Exit(1)
 	}
+
+	genConst()
+}
+
+func genConst() {
+	var body map[string][]*plugin.SrvMethod = make(map[string][]*plugin.SrvMethod)
+	for _, item := range plugin.AllSrvMethods {
+		name := strings.ToLower(item.PkgName)
+		body[name] = append(body[name], &item)
+	}
+
+	for name, item := range body {
+		constPath := "./sdk/typescript/" + name + "/const.ts"
+
+		var code string
+		for _, row := range item {
+			code += "/** " + strings.TrimSpace(row.Comment) + " */\n"
+			code += fmt.Sprintf("export const URL_%s_%s = \"/%s.%s/%s\"\n", row.ServiceName, row.MethodName, name, row.ServiceName, row.MethodName)
+			code += "\n"
+		}
+		utils.StringToFile(constPath, code)
+	}
 }
 
 func run() error {
@@ -38,5 +62,6 @@ func run() error {
 	if _, err := os.Stdout.Write(out); err != nil {
 		return err
 	}
+
 	return nil
 }