更新编辑器MenuItem导表与导出协议

更新编辑器MenuItem导表与导出协议
This commit is contained in:
ALEXTANGXIAO
2023-07-29 00:03:54 +08:00
parent 30192d52cf
commit 98dcb80942
29 changed files with 217 additions and 1018 deletions

View File

@@ -0,0 +1,39 @@
using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using UnityEngine;
#if UNITY_EDITOR
using UnityEditor;
#endif
namespace TEngine.Editor
{
public static class GenTools
{
public const string GenNetFirstTips = "ProtoGenTools.GenNetFirstTips";
[MenuItem("TEngine/导出网络Proto|Gen Proto", false, 100)]
public static void ExportProto()
{
var firstTips = EditorPrefs.GetBool(GenNetFirstTips, false);
if (!firstTips && EditorUtility.DisplayDialog("提示", "导出网络Proto依赖于DotNet下Server解决方案首次打开请编译", "不再提示", "继续"))
{
EditorPrefs.SetBool(GenNetFirstTips, true);
return;
}
Application.OpenURL(System.IO.Path.Combine(Application.dataPath,"../DotNet/start_export.bat"));
Debug.Log("proto2cs succeed!");
}
[MenuItem("TEngine/导出Config|Export Config", false, 100)]
public static void ExportConfig()
{
Application.OpenURL(System.IO.Path.Combine(Application.dataPath,"../Luban/gen_code_bin_to_project.bat"));
Debug.Log("proto2cs succeed!");
}
}
}

View File

@@ -1,291 +0,0 @@
using System;
using System.Collections.Generic;
using System.IO;
using System.Text;
using UnityEngine;
#if UNITY_EDITOR
using UnityEditor;
#endif
namespace TEngine.Editor
{
public static class ProtoGenTools
{
#if UNITY_EDITOR
[MenuItem("TEngine/生成Proto|Gen Proto", false, 10)]
#endif
public static void Export()
{
InnerProto2CS.Proto2CS();
Debug.Log("proto2cs succeed!");
}
}
public static class InnerProto2CS
{
private static string ProtoPath = UnityEngine.Application.dataPath + "/..\\Luban\\Proto\\pb_schemas\\";
private static string OutPutPath =
UnityEngine.Application.dataPath + "/..\\Luban\\Proto\\Gen\\";
private static readonly char[] splitChars = { ' ', '\t' };
public static void Proto2CS()
{
Proto2CS("GameProto", "ProtoBase.proto", OutPutPath,10001,false);
}
public static void Proto2CS(string nameSpace, string protoName, string outputPath, int startOpcode,bool useMemoryPool = false)
{
if (!Directory.Exists(outputPath))
{
Directory.CreateDirectory(outputPath);
}
string proto = Path.Combine(ProtoPath, protoName);
string csPath = Path.Combine(outputPath, Path.GetFileNameWithoutExtension(proto) + ".cs");
string s = File.ReadAllText(proto);
StringBuilder sb = new StringBuilder();
sb.Append("//------------------------------------------------------------------------------\n");
sb.Append("// <auto-generated>\n");
sb.Append("// This code was generated by a tool.\n");
sb.Append("// Changes to this file may cause incorrect behavior and will be lost if\n");
sb.Append("// the code is regenerated.\n");
sb.Append("// </auto-generated>\n");
sb.Append("//------------------------------------------------------------------------------\n");
sb.Append("\n");
sb.Append("using System;\n");
sb.Append("using ProtoBuf;\n");
sb.Append("using TEngine;\n");
sb.Append("using System.Collections.Generic;\n");
sb.Append("\n");
sb.Append($"namespace {nameSpace}\n");
sb.Append("{\n");
bool isMsgStart = false;
bool isEnumStart = false;
foreach (string line in s.Split('\n'))
{
string newline = line.Trim();
if (newline == "")
{
continue;
}
if (newline.StartsWith("//ResponseType"))
{
string responseType = line.Split(' ')[1].TrimEnd('\r', '\n');
sb.AppendLine($"\t[ResponseType(nameof({responseType}))]");
continue;
}
if (newline.StartsWith("//"))
{
sb.Append($"\t{newline}\n");
continue;
}
if (newline.StartsWith("message"))
{
string parentClass = "";
isMsgStart = true;
string msgName = newline.Split(splitChars, StringSplitOptions.RemoveEmptyEntries)[1];
string[] ss = newline.Split(new[] { "//" }, StringSplitOptions.RemoveEmptyEntries);
if (ss.Length == 2)
{
parentClass = ss[1].Trim();
}
sb.Append($"\t[Serializable,global::ProtoBuf.ProtoContract(Name = @\"{msgName}\")]\n");
if (useMemoryPool)
{
sb.Append($"\tpublic partial class {msgName}: IMemory");
}
else
{
sb.Append($"\tpublic partial class {msgName}");
}
if (parentClass != "")
{
sb.Append($", {parentClass}\n");
}
else
{
sb.Append("\n");
}
continue;
}
if (isMsgStart)
{
if (newline == "{")
{
sb.Append("\t{\n");
continue;
}
if (newline == "}")
{
isMsgStart = false;
sb.Append("\t}\n\n");
continue;
}
if (newline.Trim().StartsWith("//"))
{
sb.AppendLine(newline);
continue;
}
if (newline.Trim() != "" && newline != "}")
{
if (newline.StartsWith("repeated"))
{
Repeated(sb, nameSpace, newline);
}
else
{
Members(sb, newline, true);
}
}
}
if (newline.StartsWith("enum"))
{
isEnumStart = true;
string enumName = newline.Split(splitChars, StringSplitOptions.RemoveEmptyEntries)[1];
sb.Append($"\t[global::ProtoBuf.ProtoContract()]\n");
sb.Append($"\tpublic enum {enumName}");
sb.Append("\n");
continue;
}
if (isEnumStart)
{
if (newline == "{")
{
sb.Append("\t{\n");
continue;
}
if (newline == "}")
{
isEnumStart = false;
sb.Append("\t}\n\n");
continue;
}
if (newline.Trim().StartsWith("//"))
{
sb.AppendLine(newline);
continue;
}
int index = newline.IndexOf(";");
newline = newline.Remove(index);
sb.Append($"\t\t{newline},\n\n");
}
}
sb.Append("}\n");
using (FileStream txt = new FileStream(csPath, FileMode.Create, FileAccess.ReadWrite))
{
using (StreamWriter sw = new StreamWriter(txt))
{
Debug.Log(sb.ToString());
sw.Write(sb.ToString());
}
}
}
private static void Repeated(StringBuilder sb, string ns, string newline)
{
try
{
int index = newline.IndexOf(";");
newline = newline.Remove(index);
string[] ss = newline.Split(splitChars, StringSplitOptions.RemoveEmptyEntries);
string type = ss[1];
type = ConvertType(type);
string name = ss[2];
int n = int.Parse(ss[4]);
sb.Append($"\t\t[global::ProtoBuf.ProtoMember({n})]\n");
sb.Append($"\t\tpublic List<{type}> {name} = new List<{type}>();\n\n");
}
catch (Exception e)
{
Console.WriteLine($"{newline}\n {e}");
}
}
private static string ConvertType(string type)
{
string typeCs = "";
switch (type)
{
case "int16":
typeCs = "short";
break;
case "int32":
typeCs = "int";
break;
case "bytes":
typeCs = "byte[]";
break;
case "uint32":
typeCs = "uint";
break;
case "long":
typeCs = "long";
break;
case "int64":
typeCs = "long";
break;
case "uint64":
typeCs = "ulong";
break;
case "uint16":
typeCs = "ushort";
break;
default:
typeCs = type;
break;
}
return typeCs;
}
private static void Members(StringBuilder sb, string newline, bool isRequired)
{
try
{
int index = newline.IndexOf(";");
newline = newline.Remove(index);
string[] ss = newline.Split(splitChars, StringSplitOptions.RemoveEmptyEntries);
string type = ss[0];
string name = ss[1];
int n = int.Parse(ss[3]);
string typeCs = ConvertType(type);
sb.Append($"\t\t[global::ProtoBuf.ProtoMember({n})]\n");
if (string.Equals(type,"string"))
{
sb.Append($"\t\t[global::System.ComponentModel.DefaultValue(\"\")]\n");
}
sb.Append($"\t\tpublic {typeCs} {name} {{ get; set; }}\n\n");
}
catch (Exception e)
{
Console.WriteLine($"{newline}\n {e}");
}
}
}
}

View File

@@ -1,448 +0,0 @@
using System;
using UnityEditor;
using UnityEngine;
using System.Collections.Generic;
using System.IO;
using System.Text;
using System.Xml;
using TEngine;
using TEngine.Editor;
using Debug = UnityEngine.Debug;
class GenNetScriptWindow : EditorWindow
{
private static GenNetScriptWindow _window = null;
//绑定通知协议
private StringBuilder _strBindNotify;
//枚举对应协议号
private readonly Dictionary<string, int> _dicName2ID = new Dictionary<string, int>();
//xxx.xml下的协议枚举list
private readonly Dictionary<string, List<string>> _dicPath2Name = new Dictionary<string, List<string>>();
//协议号有没有被勾选
private readonly Dictionary<int, bool> _dicID2Select = new Dictionary<int, bool>();
//记录回包协议号
private readonly Dictionary<int, int> _dicID2ID = new Dictionary<int, int>();
private Vector2 _scrollPos;
private Vector2 _fileListScrollPos;
private string _path = @"G:\github\TEngine\Luban\Proto\pb_schemas\";
private readonly List<string> _filePathList = new List<string>();
private string _curSelectFile = string.Empty;
private string _filterFileName = string.Empty;
private string _filterProName = string.Empty;
[MenuItem("TEngine/协议生成工具|Protobuf Tools")]
static void OpenGenNetScriptWindow()
{
if (!_window)
{
_window = ScriptableObject.CreateInstance<GenNetScriptWindow>();
_window.maxSize = new Vector2(1000, 800);
_window.minSize = _window.maxSize / 2;
_window.LoadLastPath();
}
_window.ShowUtility();
}
void OnGUI()
{
EditorGUILayout.PrefixLabel("protoPath");
_path = EditorGUILayout.TextField(_path);
var r = EditorGUILayout.BeginHorizontal("Button");
if (GUI.Button(r, GUIContent.none))
{
ReadPath();
}
GUILayout.Label("Search");
EditorGUILayout.EndHorizontal();
//加个文件筛选
//EditorGUILayout.BeginHorizontal();
GUILayout.Label("filter file:");
_filterFileName = GUILayout.TextField(_filterFileName);
//EditorGUILayout.EndHorizontal();
//显示文件名部分
if (_filePathList.Count > 0)
{
_fileListScrollPos = EditorGUILayout.BeginScrollView(_fileListScrollPos, GUILayout.Width(r.width), GUILayout.Height(200));
for (int i = 0; i < _filePathList.Count; ++i)
{
var fileName = Path.GetFileNameWithoutExtension(_filePathList[i]);
if (!string.IsNullOrEmpty(_filterFileName) && fileName.IndexOf(_filterFileName, StringComparison.Ordinal) == -1)
{
continue;
}
if (GUILayout.Button(fileName))
{
_curSelectFile = _filePathList[i];
LoadSelectFile();
}
}
EditorGUILayout.EndScrollView();
}
GUILayout.Label("-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------");
if (!string.IsNullOrEmpty(_curSelectFile))
{
//加个协议筛选
GUILayout.Label("filter proto:");
_filterProName = GUILayout.TextField(_filterProName);
_scrollPos = EditorGUILayout.BeginScrollView(_scrollPos, GUILayout.Width(r.width), GUILayout.Height(200));
var fileName2 = Path.GetFileNameWithoutExtension(_curSelectFile);
List<string> list;
if (_dicPath2Name.TryGetValue(fileName2, out list))
{
EditorGUI.indentLevel++;
for (int i = 0; i < list.Count; ++i)
{
var cmdName = list[i];
//筛选一下,忽略大小写
if (!string.IsNullOrEmpty(_filterProName) && cmdName.ToLower().IndexOf(_filterProName.ToLower()) == -1)
continue;
var cmdID = _dicName2ID[cmdName];
EditorGUILayout.BeginHorizontal(GUIStyle.none);
//协议名
EditorGUILayout.LabelField(cmdName);
//toggle
if (!_dicID2Select.ContainsKey(cmdID))
_dicID2Select[cmdID] = false;
_dicID2Select[cmdID] = EditorGUILayout.Toggle(cmdID.ToString(),_dicID2Select[cmdID]);
//回包协议号
if (!_dicID2ID.ContainsKey(cmdID))
{
if (cmdName.EndsWith("REQ"))
_dicID2ID[cmdID] = cmdID + 1;
else
_dicID2ID[cmdID] = 0;
}
_dicID2ID[cmdID] = EditorGUILayout.IntField(_dicID2ID[cmdID]);
EditorGUILayout.EndHorizontal();
}
EditorGUI.indentLevel--;
}
EditorGUILayout.EndScrollView();
}
if (!string.IsNullOrEmpty(_curSelectFile))
{
if (GUILayout.Button("GenSelect"))
{
OnClickGenBtn(false);
}
if (GUILayout.Button("GenAll"))
{
OnClickGenBtn(true);
}
}
if (GUILayout.Button("导出Proto To Csharp|Export Proto To Csharp"))
{
ExportProto();
}
}
#region
private void LoadLastPath()
{
if (PlayerPrefs.HasKey("GenNetScriptWindow.Path"))
{
_path = PlayerPrefs.GetString("GenNetScriptWindow.Path");
}
ReadPath();
}
private void ReadPath()
{
PlayerPrefs.SetString("GenNetScriptWindow.Path", _path);
_filePathList.Clear();
_curSelectFile = String.Empty;
JustLoadFileList(_filePathList, _path);
}
private void JustLoadFileList(List<string> exportList, string folderPath, bool deep = false)
{
if (!LoadFoldChildFileList(exportList, folderPath, deep))
{
EditorUtility.DisplayDialog("folder not exist", "folder not exist:"+_path, "ok");
}
}
private void LoadSelectFile()
{
_dicID2Select.Clear();
_dicID2ID.Clear();
_dicName2ID.Clear();
var xmlFilePath = _curSelectFile;
var fileName = Path.GetFileNameWithoutExtension(xmlFilePath);
var protocolNameList = new List<string>();
//读xml
if (fileName.StartsWith("proto_cs"))
{
Debug.Log("load xml.name:" + fileName);
XmlDocument xmlDoc = new XmlDocument();
xmlDoc.Load(xmlFilePath);
GenProtocolNameList(xmlDoc.ChildNodes, protocolNameList);
_dicPath2Name[fileName] = protocolNameList;
Debug.Log(fileName + " success.");
}
}
private bool LoadFoldChildFileList(List<string> exportList, string folderPath, bool deep = false)
{
if (!Directory.Exists(folderPath))
{
Log.Error("folder not exist: {0}", folderPath);
return false;
}
string[] subFile = Directory.GetFiles(folderPath);
foreach (string fileName in subFile)
{
//有些筛选条件,直接写死这里了。方便
var name = Path.GetFileNameWithoutExtension(fileName);
if (name.StartsWith("proto_cs"))
exportList.Add(fileName);
}
if (deep)
{
string[] subFolders = Directory.GetDirectories(folderPath);
foreach (string folderName in subFolders)
{
LoadFoldChildFileList(exportList, folderName);
}
}
return true;
}
//把macro里的协议号和协议枚举名对应起来
private void GenProtocolNameList(XmlNodeList nodeList, List<string> nameList)
{
foreach (XmlNode node in nodeList)
{
if (node.Attributes == null)
continue;
if (node.Name == "macro")
{
var name = node.Attributes.GetNamedItem("name").Value;
if (name.StartsWith("CS_CMD") || name.StartsWith("CS_NOTIFY"))
{
var id = Convert.ToInt32(node.Attributes.GetNamedItem("value").Value);
_dicName2ID[name] = id;
if (nameList != null)
nameList.Add(name);
}
}
GenProtocolNameList(node.ChildNodes, nameList);
}
}
#endregion
#region
private void OnClickGenBtn(bool genAll)
{
_strBindNotify = new StringBuilder();
bool needRegCmdHandle = false;
StringBuilder sb = new StringBuilder();
List<int> listGenId = new List<int>();
var iter = _dicID2Select.GetEnumerator();
while (iter.MoveNext())
{
if (iter.Current.Value || genAll)
{
int resId = iter.Current.Key;
int resID;
if (_dicID2ID.TryGetValue(resId, out resID))
{
if (resID != 0)
{
var oneStr = GenOneReq(resId, resID);
sb.Append(oneStr);
sb.Append("\n\n");
listGenId.Add(resId);
listGenId.Add(resID);
}
else if (!listGenId.Contains(resId))
{
needRegCmdHandle = true;
var oneStr = GenOneReq(0, resId);
sb.Append(oneStr);
sb.Append("\n\n");
listGenId.Add(resId);
}
}
}
}
if (needRegCmdHandle)
{
sb.Append("public void RegCmdHandle()\n");
sb.Append("{\n");
sb.Append(_strBindNotify);
sb.Append("}\n");
}
TextEditor te = new TextEditor();
te.text = sb.ToString();
te.SelectAll();
te.Copy();
}
private static readonly Dictionary<string, string> TempParamDic = new Dictionary<string, string>();
private static string _reqClassName = string.Empty;
private static string _resClassName = string.Empty;
private static string _reqEnumName = string.Empty;
private static string _resEnumName = string.Empty;
private static string _reqDesc = string.Empty;
private static bool _resResult;
private string GenOneReq(int reqId, int resId)
{
TempParamDic.Clear();
_reqClassName = string.Empty;
_resClassName = string.Empty;
_reqEnumName = string.Empty;
_resEnumName = string.Empty;
_resResult = false;
var xmlFilePath = _curSelectFile;
var fileName = Path.GetFileNameWithoutExtension(xmlFilePath);
//读xml
if (fileName.StartsWith("proto_cs"))
{
Debug.Log("load xml.name:" + fileName);
XmlDocument xmlDoc = new XmlDocument();
xmlDoc.Load(xmlFilePath);
_GenOneReq(xmlDoc.ChildNodes, reqId,resId);
Debug.Log(fileName + " success.");
var sb = new StringBuilder();
sb.Append("#region ");
sb.Append(_reqDesc);
sb.Append(string.Format("\n//{0}\n", _reqDesc));
if (reqId != 0)
{
sb.Append("public void ");
sb.Append(_reqClassName.Substring(2));
sb.Append("(");
foreach (var item in TempParamDic)
{
sb.Append(string.Format("{0} {1}, ", item.Value, item.Key));
}
sb.Remove(sb.Length - 2, 2);//把多余的逗号和空格删了
sb.Append(")\n{\n");
sb.Append(string.Format("\tCSPkg reqPkg = ProtoUtil.BuildCSMsg(netMacros.{0});\n", _reqEnumName));
sb.Append(string.Format("\t{0} reqData = reqPkg.Body.{1};\n", _reqClassName, _reqClassName.Substring(2)));
foreach (var item in TempParamDic)
{
sb.Append(string.Format("\treqData.{0} = {0};\n", item.Key));
}
sb.Append("\n");
sb.Append(string.Format("\tGameClient.Instance.SendCSMsg(reqPkg, netMacros.{0}, {1});\n", _resEnumName,
_resClassName.Substring(2)));
sb.Append("}\n\n");
}
else
{
_strBindNotify.Append(string.Format("\t\tGameClient.Instance.RegCmdHandle(netMacros.{0}, {1});\n", _resEnumName, _resClassName.Substring(2)));
}
//回包
sb.Append(string.Format("private void {0}(CSMsgResult result, CSPkg msg)\n", _resClassName.Substring(2)));
sb.Append("{\n");
sb.Append(string.Format("\tif (DodUtil.CheckHaveError(result, msg, typeof({0})))\n", _resClassName));
sb.Append("\t\treturn;\n\n");
sb.Append(string.Format("\t{0} resData = msg.Body.{1};\n", _resClassName, _resClassName.Substring(2)));
if (_resResult)
{
sb.Append("\tif (resData.Result.Ret != 0)\n");
sb.Append("\t{\n");
sb.Append("\t\tUISys.Mgr.ShowTipMsg(resData.Result);\n");
sb.Append("\t\treturn;\n");
sb.Append("\t}\n");
}
sb.Append("\t//todo\n");
sb.Append("}\n#endregion");
return sb.ToString();
}
return null;
}
private void _GenOneReq(XmlNodeList nodeList, int reqId,int resId)
{
foreach (XmlNode node in nodeList)
{
if (node.Attributes == null)
continue;
if (node.Name.Equals("macro"))
{
var name = node.Attributes.GetNamedItem("name").Value;
if (name.StartsWith("CS_CMD") || name.StartsWith("CS_NOTIFY"))
{
var id = Convert.ToInt32(node.Attributes.GetNamedItem("value").Value);
if (id == reqId)
_reqEnumName = name;
if (id == resId)
_resEnumName = name;
}
}
if (node.Name.Equals("struct"))
{
if (node.Attributes.GetNamedItem("id") == null)
continue;
var enumName = node.Attributes.GetNamedItem("id").Value;
if (enumName.Equals(_reqEnumName))
{
var name = node.Attributes.GetNamedItem("name").Value;
_reqClassName = name;
if (node.Attributes.GetNamedItem("name") != null)
{
_reqDesc = node.Attributes.GetNamedItem("desc").Value;
}
foreach (XmlNode childNode in node.ChildNodes)
{
if (childNode != null && childNode.Name.Equals("entry"))
{
var paramName = childNode.Attributes.GetNamedItem("name").Value;
var paramType = childNode.Attributes.GetNamedItem("type").Value;
TempParamDic.Add(paramName, paramType);
}
}
}
if (enumName.Equals(_resEnumName))
{
var className = node.Attributes.GetNamedItem("name").Value;
_resClassName = className;
}
}
_GenOneReq(node.ChildNodes, reqId, resId);
}
}
#endregion
#region ToCsharp
private void ExportProto()
{
ProtoGenTools.Export();
}
#endregion
}

View File

@@ -1,11 +0,0 @@
fileFormatVersion: 2
guid: 989d4686a599e7047bba6f0cbb1bb5e7
MonoImporter:
externalObjects: {}
serializedVersion: 2
defaultReferences: []
executionOrder: 0
icon: {instanceID: 0}
userData:
assetBundleName:
assetBundleVariant: