协议代理枚举类型修复

gate/client_proto/client_proto_gen_test.go

@@ -7,6 +7,7 @@ import (
 )
 
 func TestClientProtoGen(t *testing.T) {
+	// 生成根据proto类名获取对象实例的switch方法
 	dir, err := os.ReadDir("./proto")
 	if err != nil {
 		panic(err)
@@ -22,7 +23,6 @@ func TestClientProtoGen(t *testing.T) {
 		}
 		nameList = append(nameList, split[len(split)-2])
 	}
-
 	fileData := "package client_proto\n"
 	fileData += "\n"
 	fileData += "import (\n"
@@ -38,9 +38,99 @@ func TestClientProtoGen(t *testing.T) {
 	fileData += "\t\treturn nil\n"
 	fileData += "\t}\n"
 	fileData += "}\n"
-
 	err = os.WriteFile("./client_proto_gen.go", []byte(fileData), 0644)
 	if err != nil {
 		panic(err)
 	}
+	// 处理枚举
+	for _, entry := range dir {
+		rawFileData, err := os.ReadFile("./proto/" + entry.Name())
+		if err != nil {
+			panic(err)
+		}
+		rawFileStr := string(rawFileData)
+		rawFileLine := strings.Split(rawFileStr, "\n")
+		newFileStr := ""
+		for i := 0; i < len(rawFileLine); i++ {
+			line := rawFileLine[i]
+			newFileStr += line + "\n"
+			if !strings.Contains(line, "enum") {
+				continue
+			}
+			split := strings.Split(line, " ")
+			if len(split) != 3 || split[0] != "enum" || split[2] != "{" {
+				continue
+			}
+			enumName := split[1]
+			refEnum := FindEnumInDirFile("../../protocol/proto_hk4e", enumName)
+			if refEnum == nil {
+				continue
+			}
+			i++
+			x := 0
+			for {
+				nextLine := rawFileLine[i]
+				if !strings.Contains(nextLine, "}") && x < len(refEnum) {
+					newFileStr += refEnum[x] + "\n"
+					i++
+					x++
+				} else {
+					newFileStr += line + "\n"
+					break
+				}
+			}
+		}
+		err = os.WriteFile("./proto/"+entry.Name(), []byte(newFileStr), 0644)
+		if err != nil {
+			panic(err)
+		}
+	}
+}
+
+func FindEnumInDirFile(path string, name string) (lineList []string) {
+	dir, err := os.ReadDir(path)
+	if err != nil {
+		panic(err)
+	}
+	for _, entry := range dir {
+		if entry.IsDir() {
+			ret := FindEnumInDirFile(path+"/"+entry.Name(), name)
+			if ret != nil {
+				return ret
+			}
+			continue
+		}
+		fileData, err := os.ReadFile(path + "/" + entry.Name())
+		if err != nil {
+			panic(err)
+		}
+		fileStr := string(fileData)
+		fileLine := strings.Split(fileStr, "\n")
+		for i := 0; i < len(fileLine); i++ {
+			line := fileLine[i]
+			if !strings.Contains(line, "enum") {
+				continue
+			}
+			split := strings.Split(line, " ")
+			if len(split) != 3 || split[0] != "enum" || split[2] != "{" {
+				continue
+			}
+			enumName := split[1]
+			if enumName != name {
+				continue
+			}
+			i++
+			lineList := make([]string, 0)
+			for {
+				nextLine := fileLine[i]
+				if !strings.Contains(nextLine, "}") {
+					lineList = append(lineList, nextLine)
+				} else {
+					return lineList
+				}
+				i++
+			}
+		}
+	}
+	return nil
 }