266 lines
9.0 KiB
C#
266 lines
9.0 KiB
C#
//////////////////////////////////////////////////////////////////////////
|
||
//
|
||
// 文件：Assets\Editor\Proto\ProtoTools.cs
|
||
// 作者：Xoen Xie
|
||
// 时间：2023/06/19
|
||
// 描述：pb tools
|
||
// 说明：
|
||
//
|
||
//////////////////////////////////////////////////////////////////////////
|
||
using UnityEditor;
|
||
using UnityEngine;
|
||
using System.IO;
|
||
using System;
|
||
using System.Collections.Generic;
|
||
|
||
public static class ProtoTools
|
||
{
|
||
#if UNITY_EDITOR_WIN
|
||
static string PROTOC_DIR = Application.dataPath + "/../../../NLDMisc/proto/bin/win64/";
|
||
static string PROTOC = PROTOC_DIR + "protoc.exe";
|
||
#else
|
||
static string PROTOC_DIR = Application.dataPath + "/../../../NLDMisc/proto/bin/mac/";
|
||
static string PROTOC = PROTOC_DIR + "protoc";
|
||
#endif
|
||
|
||
static string PROTO_DIR = Application.dataPath + "/../../../NLDMisc/proto/";
|
||
|
||
|
||
[MenuItem("Tools/*proto/导出pb到csharp", false, (int)NLDMenuID.PBExportCS)]
|
||
public static void PBExportCS()
|
||
{
|
||
string outputDir = Application.dataPath + "/Code/Scripts/Gameplay/Net/Proto/";
|
||
|
||
string [] csProto = new string[] { "client.proto", "message_id.proto", "common.proto"};
|
||
|
||
// VersionControlSystem.Checkout(outputDir);
|
||
|
||
for (int i = 0; i < csProto.Length; i ++)
|
||
{
|
||
EditorUtil.StartProcess(PROTOC,
|
||
$"--csharp_out={outputDir} --proto_path=../../ {csProto[i]}", PROTOC_DIR);
|
||
}
|
||
|
||
Debug.Log("Export to csharp done...");
|
||
|
||
CSCodeGen();
|
||
|
||
Debug.Log("csharp code gen done...");
|
||
}
|
||
|
||
private static Gameplay.Net.EP GetEP(string msgID) {
|
||
if (msgID.Contains("Clt_"))
|
||
return Gameplay.Net.EP.Client;
|
||
else if (msgID.Contains("G_"))
|
||
return Gameplay.Net.EP.Game;
|
||
else if (msgID.Contains("L_"))
|
||
return Gameplay.Net.EP.Login;
|
||
else if (msgID.Contains("Gt_"))
|
||
return Gameplay.Net.EP.Gate;
|
||
else if (msgID.Contains("C_"))
|
||
return Gameplay.Net.EP.Center;
|
||
else if (msgID.Contains("Mg_"))
|
||
return Gameplay.Net.EP.Manager;
|
||
else
|
||
Debug.LogError("ep not handler, msgID=" + msgID);
|
||
return Gameplay.Net.EP.None;
|
||
}
|
||
|
||
private static Gameplay.Net.EP GetDestEP(string msgID)
|
||
{
|
||
if (msgID.Contains("_G_"))
|
||
return Gameplay.Net.EP.Game;
|
||
else if (msgID.Contains("_L_"))
|
||
return Gameplay.Net.EP.Login;
|
||
else if (msgID.Contains("_Gt_"))
|
||
return Gameplay.Net.EP.Gate;
|
||
else if (msgID.Contains("_Mg_"))
|
||
return Gameplay.Net.EP.Manager;
|
||
else
|
||
{
|
||
Debug.Log("ep not handler try again, msgID=" + msgID);
|
||
return GetEP(msgID);
|
||
}
|
||
}
|
||
|
||
[MenuItem("Tools/*proto/导出pb到go", false, (int)NLDMenuID.PBExportGO)]
|
||
public static void PBExportGo()
|
||
{
|
||
string[] protos = new string[] { "message_id.proto", "server/s_message_id.proto" };
|
||
string codeTemplate = Application.dataPath + "/Editor/Tools/Proto/MSGGoCode.txt";
|
||
string fileTemplate = Application.dataPath + "/Editor/Tools/Proto/MSGTemplateGO.txt";
|
||
|
||
string ftContent = File.ReadAllText(fileTemplate);
|
||
string cdContent = File.ReadAllText(codeTemplate);
|
||
|
||
string messagePoolStr = "";
|
||
|
||
List<string>[] ids = new List<string>[(int)Gameplay.Net.EP.Max];
|
||
for (int i = 0; i < (int)Gameplay.Net.EP.Max; i++) {
|
||
ids[i] = new List<string>();
|
||
}
|
||
|
||
for (int i = 0; i < protos.Length; i++)
|
||
{
|
||
string protoName = protos[i];
|
||
|
||
string protoPath = PROTO_DIR + "/" + protoName;
|
||
string[] lines = File.ReadAllLines(protoPath);
|
||
|
||
bool bGenCode = false;
|
||
|
||
for (int J = 0; J < lines.Length; J++)
|
||
{
|
||
if (bGenCode)
|
||
{
|
||
if (lines[J].Contains("MESSAGE_ID_END"))
|
||
{
|
||
break;
|
||
}
|
||
|
||
string rawMsgID = lines[J].Trim();
|
||
|
||
int index = rawMsgID.IndexOf("=");
|
||
if (index > 0)
|
||
{
|
||
string msgID = rawMsgID.Substring(0, index - 1);
|
||
msgID = msgID.Trim();
|
||
Debug.Log("Find Message ID:" + msgID);
|
||
|
||
var ep = GetEP(msgID);
|
||
ids[(int)ep].Add(msgID);
|
||
|
||
messagePoolStr += $"\txsf_net.SetMessage(new(MSG_{msgID}))\n";
|
||
}
|
||
|
||
|
||
}
|
||
else
|
||
{
|
||
if (lines[J].Contains("MESSAGE_ID_START"))
|
||
{
|
||
bGenCode = true;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
for (int i = 0; i < (int)Gameplay.Net.EP.Max; i++)
|
||
{
|
||
var list = ids[i];
|
||
if (list.Count <= 0)
|
||
continue;
|
||
|
||
Gameplay.Net.EP curEP = (Gameplay.Net.EP)i;
|
||
string codes = "";
|
||
for (int J = 0; J < list.Count; ++J)
|
||
{
|
||
string msgID = list[J];
|
||
string goID = $"SMSGID_{msgID}";
|
||
string goEP = curEP.ToString();
|
||
if (curEP == Gameplay.Net.EP.Client)
|
||
{
|
||
goID = $"MSGID_{msgID}";
|
||
goEP = GetDestEP(msgID).ToString();
|
||
}
|
||
|
||
string c = cdContent.Replace("_MSG_NAME_", msgID);
|
||
c = c.Replace("_MSG_ID_NAME_", goID);
|
||
c = c.Replace("_MSG_EP_", goEP);
|
||
codes += c;
|
||
}
|
||
|
||
string codePath = Application.dataPath + $"/../../../NLDServer/message/messages/msg_{curEP.ToString().ToLower()}.go";
|
||
|
||
DateTime currentDateTime = DateTime.Now;
|
||
string date = currentDateTime.ToShortDateString();
|
||
string content = ftContent.Replace("_MSG_DATE_", date);
|
||
content += codes;
|
||
|
||
File.WriteAllText(codePath, content);
|
||
}
|
||
|
||
string codeMessagePool = Application.dataPath + "/../../../NLDServer/message/messages/messages.go";
|
||
EditorUtil.ReplaceContentByTag(codeMessagePool, "MESSAGE_START", "MESSAGE_END", messagePoolStr);
|
||
|
||
Debug.Log("Export to go done...");
|
||
}
|
||
|
||
private static void CSCodeGen()
|
||
{
|
||
string protoPath = PROTO_DIR + "/message_id.proto";
|
||
string codeTemplate = Application.dataPath + "/Editor/Tools/Proto/MSGTemplateCS.txt";
|
||
string TmpContent = File.ReadAllText(codeTemplate);
|
||
|
||
string [] lines = File.ReadAllLines(protoPath);
|
||
|
||
bool bGenCode = false;
|
||
|
||
string messagePoolStr = "";
|
||
|
||
for(int i = 0; i < lines.Length; i ++)
|
||
{
|
||
if(bGenCode)
|
||
{
|
||
if(lines[i].Contains("MESSAGE_ID_END"))
|
||
{
|
||
break;
|
||
}
|
||
|
||
string rawMsgID = lines[i].Trim();
|
||
|
||
int index = rawMsgID.IndexOf("=");
|
||
if(index > 0)
|
||
{
|
||
string msgID = rawMsgID.Substring(0, index-1);
|
||
msgID = msgID.Trim();
|
||
Debug.Log("Find Message ID:" + msgID);
|
||
|
||
string[] idParts = msgID.Split("_");
|
||
string idName = "";
|
||
for(int J = 0; J < idParts.Length; J++)
|
||
{
|
||
idName += EditorUtil.FirstLetterToUpper(idParts[J]);
|
||
}
|
||
|
||
string Desc = "";
|
||
int descIndex = rawMsgID.IndexOf("//");
|
||
if(descIndex > 0)
|
||
{
|
||
Desc = rawMsgID.Substring(descIndex+2);
|
||
}
|
||
|
||
string codePath = Application.dataPath + $"/Code/Scripts/Gameplay/Net/Messages/MSG_{msgID}.cs";
|
||
if(!File.Exists(codePath))
|
||
{
|
||
// VersionControlSystem.Checkout(Application.dataPath + "/Code/Scripts/Gameplay/Net/Messages");
|
||
|
||
string content = TmpContent.Replace("_MSG_NAME_", msgID);
|
||
DateTime currentDateTime = DateTime.Now;
|
||
string date = currentDateTime.ToShortDateString();
|
||
content = content.Replace("_MSG_DATE_", date);
|
||
content = content.Replace("_MSG_DESC_", Desc.Trim());
|
||
|
||
|
||
content = content.Replace("_MSG_ID_NAME_", idName);
|
||
File.WriteAllText(codePath, content);
|
||
}
|
||
|
||
messagePoolStr += $"\t\t\tm_MessagePool[(int)NLDMID.MSGID.{idName}] = new MSG_{msgID}();\n";
|
||
}
|
||
}
|
||
else
|
||
{
|
||
if(lines[i].Contains("MESSAGE_ID_START"))
|
||
{
|
||
bGenCode = true;
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
string codeMessagePool = Application.dataPath + "/Code/Scripts/Gameplay/Net/Client/NetManager.cs";
|
||
// VersionControlSystem.Checkout(codeMessagePool);
|
||
EditorUtil.ReplaceContentByTag(codeMessagePool, "MESSAGE_START", "MESSAGE_END", messagePoolStr);
|
||
}
|
||
} |