90 lines
2.4 KiB
C#
90 lines
2.4 KiB
C#
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
using UnityEditor;
|
|
using System.IO;
|
|
using ET;
|
|
|
|
namespace ETEditor
|
|
{
|
|
public class EntityWizard : ScriptableWizard
|
|
{
|
|
public bool isServer = false;
|
|
public bool isHotfix = true;
|
|
public bool isUnit = false;
|
|
public string scriptName = "";
|
|
|
|
[Header("指定路劲")]
|
|
[Tooltip("如果为空，使用默认路径")]
|
|
public GlobalPath globalPath;
|
|
|
|
private void OnWizardCreate()
|
|
{
|
|
if (string.IsNullOrWhiteSpace(scriptName))
|
|
{
|
|
return;
|
|
}
|
|
|
|
// 创建代码的文件夹
|
|
GlobalSettingData globalSetting = MongoHelper.FromJson<GlobalSettingData>(EditorPrefs.GetString("GlobalSettingData"));
|
|
string generatePath = globalSetting.EntityPath;
|
|
if (globalPath != GlobalPath.None)
|
|
{
|
|
generatePath = globalSetting.GlobalPaths[(int)globalPath];
|
|
}
|
|
|
|
string folderName = $"Hotfix/{generatePath}";
|
|
if (isHotfix == false || isServer == true)
|
|
{
|
|
folderName = $"Model/{generatePath}";
|
|
}
|
|
|
|
string generateFolder = $"{Application.dataPath}/{folderName}";
|
|
if (isServer == true)
|
|
{
|
|
generateFolder = $"{Application.dataPath.Replace("Unity/Assets", "Server")}/{folderName}";
|
|
}
|
|
|
|
// 写入代码
|
|
AssistentHelper.CheckDirectory(generateFolder);
|
|
using (FileStream fs = new FileStream($"{generateFolder}/{scriptName}.cs", FileMode.OpenOrCreate, FileAccess.ReadWrite, FileShare.ReadWrite))
|
|
{
|
|
using (StreamWriter sw = new StreamWriter(fs, System.Text.Encoding.UTF8))
|
|
{
|
|
// 替换模板代码的内容
|
|
string code = ScriptTemplate.Entity;
|
|
if (isUnit && isServer == false)
|
|
{
|
|
code = ScriptTemplate.UnitEntity;
|
|
}
|
|
|
|
code = code.Replace("类名", scriptName);
|
|
|
|
string currentNamespace = "ET";
|
|
string otherNamespace = "ET";
|
|
if (isHotfix == false || isServer == true)
|
|
{
|
|
string temp = currentNamespace;
|
|
currentNamespace = otherNamespace;
|
|
otherNamespace = temp;
|
|
}
|
|
code = code.Replace("当前命名空间", currentNamespace);
|
|
code = code.Replace("另一个命名空间", otherNamespace);
|
|
|
|
sw.Write(code);
|
|
}
|
|
}
|
|
|
|
EditorPrefs.SetInt("EntityGlobalPath", (int)globalPath);
|
|
AssetDatabase.Refresh();
|
|
}
|
|
|
|
[MenuItem("助手/Wizard/实体 &2", priority = 2)]
|
|
private static void CreateWizard()
|
|
{
|
|
EntityWizard entity = ScriptableWizard.DisplayWizard<EntityWizard>("Entity Script", "创建");
|
|
entity.globalPath = (GlobalPath)EditorPrefs.GetInt("EntityGlobalPath");
|
|
}
|
|
}
|
|
}
|