1
This commit is contained in:
MuHua-123
@@ -0,0 +1,356 @@
|
||||
using UnityEngine;
|
||||
using System.Collections;
|
||||
using System.Collections.Generic;
|
||||
using System.Text.RegularExpressions;
|
||||
using UnityEditor;
|
||||
using System.IO;
|
||||
using System.Reflection;
|
||||
using System.Linq;
|
||||
using System.Xml.Serialization;
|
||||
|
||||
namespace MuHuaEditor {
|
||||
public class ClassReferenceCollection : IReferenceCollection {
|
||||
// guid : types
|
||||
private List<CollectionData> references = null;
|
||||
|
||||
// type : guid
|
||||
private Dictionary<System.Type, List<string>> code2FileDic = new Dictionary<System.Type, List<string>>();
|
||||
private List<TypeDate> fileTypeList = null;
|
||||
public static readonly string xmlPath = "referenceType2File.xml";
|
||||
|
||||
private List<TypeDate> fileTypeXML {
|
||||
get {
|
||||
if (File.Exists(xmlPath) == true) {
|
||||
using (var reader = new StreamReader(xmlPath)) {
|
||||
XmlSerializer serialize = new XmlSerializer(typeof(List<TypeDate>));
|
||||
return (List<TypeDate>)serialize.Deserialize(reader);
|
||||
}
|
||||
}
|
||||
else {
|
||||
return new List<TypeDate>();
|
||||
}
|
||||
}
|
||||
set {
|
||||
using (var writer = new StreamWriter(xmlPath)) {
|
||||
XmlSerializer serialize = new XmlSerializer(typeof(List<TypeDate>));
|
||||
serialize.Serialize(writer, value);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
TypeDate GetTypeData(string guid) {
|
||||
if (fileTypeList.Exists(c => c.guid == guid) == false) {
|
||||
|
||||
var path = AssetDatabase.GUIDToAssetPath(guid);
|
||||
|
||||
fileTypeList.Add(new TypeDate() {
|
||||
guid = guid,
|
||||
fileName = path,
|
||||
timeStamp = File.GetLastWriteTime(path)
|
||||
});
|
||||
}
|
||||
|
||||
return fileTypeList.First(c => c.guid == guid);
|
||||
}
|
||||
|
||||
private bool isSaveEditorCode = false;
|
||||
|
||||
public ClassReferenceCollection(bool saveStiroCode = false) {
|
||||
isSaveEditorCode = saveStiroCode;
|
||||
fileTypeList = fileTypeXML;
|
||||
}
|
||||
|
||||
public void Init(List<CollectionData> refs) {
|
||||
references = refs;
|
||||
}
|
||||
|
||||
public void CollectionFiles() {
|
||||
|
||||
// connect each classes.
|
||||
var firstPassList = new List<string>();
|
||||
if (Directory.Exists("Assets/Plugins"))
|
||||
firstPassList.AddRange(CodeList("Assets/Plugins"));
|
||||
if (Directory.Exists("Assets/Standard Assets"))
|
||||
firstPassList.AddRange(CodeList("Assets/Standard Assets"));
|
||||
|
||||
// Connect the files and class.
|
||||
var codes = CodeList("Assets/").Where(c => firstPassList.Contains(c) == false);
|
||||
|
||||
var allFirstpassTypes = collectionAllFastspassClasses();
|
||||
CollectionCodeFileDictionary(allFirstpassTypes, firstPassList.ToArray());
|
||||
|
||||
var alltypes = CollectionAllClasses();
|
||||
CollectionCodeFileDictionary(alltypes, codes.ToArray());
|
||||
alltypes.AddRange(allFirstpassTypes);
|
||||
|
||||
|
||||
fileTypeXML = fileTypeList;
|
||||
|
||||
foreach (var type in alltypes) {
|
||||
List<string> list = null;
|
||||
if (code2FileDic.ContainsKey(type) == false) {
|
||||
list = new List<string>();
|
||||
code2FileDic.Add(type, list);
|
||||
}
|
||||
else {
|
||||
list = code2FileDic[type];
|
||||
}
|
||||
|
||||
var fullName = type.FullName;
|
||||
var assembly = type.Assembly.FullName;
|
||||
if (fileTypeList.Exists(c => c.assemblly == assembly && c.typeFullName.Contains(fullName))) {
|
||||
var datas = fileTypeList.Where(c => c.assemblly == assembly && c.typeFullName.Contains(fullName));
|
||||
foreach (var data in datas) {
|
||||
list.Add(data.guid);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
float count = 1, max = firstPassList.Count;
|
||||
foreach (var codepath in firstPassList) {
|
||||
EditorUtility.DisplayProgressBar("analytics", codepath, count++ / max);
|
||||
CollectionReferenceClasses(AssetDatabase.AssetPathToGUID(codepath), allFirstpassTypes);
|
||||
}
|
||||
count = 1;
|
||||
max = codes.Count();
|
||||
foreach (var codepath in codes) {
|
||||
EditorUtility.DisplayProgressBar("analytics", codepath, count++ / max);
|
||||
CollectionReferenceClasses(AssetDatabase.AssetPathToGUID(codepath), alltypes);
|
||||
}
|
||||
|
||||
if (isSaveEditorCode) {
|
||||
CollectionCustomEditorClasses(alltypes);
|
||||
}
|
||||
}
|
||||
|
||||
List<string> CodeList(string path) {
|
||||
string[] codes = Directory.GetFiles(path, "*.cs", SearchOption.AllDirectories);
|
||||
|
||||
List<string> needUpdateFileList = new List<string>();
|
||||
|
||||
foreach (var code in codes) {
|
||||
var guid = AssetDatabase.AssetPathToGUID(code);
|
||||
if (fileTypeList.Exists(c => c.guid == guid) == false) {
|
||||
needUpdateFileList.Add(code);
|
||||
continue;
|
||||
}
|
||||
|
||||
var filetype = GetTypeData(guid);
|
||||
|
||||
var timeStamp = filetype.timeStamp;
|
||||
var time = File.GetLastWriteTime(code);
|
||||
if (time != timeStamp) {
|
||||
filetype.timeStamp = time;
|
||||
needUpdateFileList.Add(code);
|
||||
continue;
|
||||
}
|
||||
}
|
||||
return needUpdateFileList;
|
||||
}
|
||||
|
||||
void CollectionCodeFileDictionary(List<System.Type> alltypes, string[] codes) {
|
||||
float count = 1;
|
||||
foreach (var codePath in codes) {
|
||||
EditorUtility.DisplayProgressBar("checking", "search files", count++ / codes.Length);
|
||||
|
||||
// connect file and classes.
|
||||
var code = StripComment(System.IO.File.ReadAllText(codePath));
|
||||
var guid = AssetDatabase.AssetPathToGUID(codePath);
|
||||
|
||||
var typeList = GetTypeData(guid);
|
||||
typeList.typeFullName.Clear();
|
||||
|
||||
foreach (var type in alltypes) {
|
||||
|
||||
if (type.IsNested) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (string.IsNullOrEmpty(type.Namespace) == false) {
|
||||
var namespacepattern = string.Format("namespace\\s*{0}[{{\\s\\n]", type.Namespace);
|
||||
if (Regex.IsMatch(code, namespacepattern) == false) {
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
string typeName = type.IsGenericTypeDefinition ? type.GetGenericTypeDefinition().Name.Split('`')[0] : type.Name;
|
||||
if (type.IsClass) {
|
||||
if (Regex.IsMatch(code, string.Format("class\\s*{0}?[\\s:<{{]", typeName))) {
|
||||
typeList.Add(type);
|
||||
|
||||
var nested = type.GetNestedTypes(BindingFlags.Public | BindingFlags.Instance);
|
||||
|
||||
foreach (var nestedType in nested) {
|
||||
typeList.Add(nestedType);
|
||||
}
|
||||
|
||||
continue;
|
||||
}
|
||||
}
|
||||
else if (type.IsInterface) {
|
||||
|
||||
if (Regex.IsMatch(code, string.Format("interface\\s*{0}[\\s<{{]", typeName))) {
|
||||
typeList.Add(type);
|
||||
continue;
|
||||
}
|
||||
}
|
||||
else if (type.IsEnum) {
|
||||
|
||||
if (Regex.IsMatch(code, string.Format("enum\\s*{0}[\\s{{]", type.Name))) {
|
||||
typeList.Add(type);
|
||||
continue;
|
||||
}
|
||||
}
|
||||
else {
|
||||
if (Regex.IsMatch(code, string.Format("struct\\s*{0}[\\s:<{{]", typeName))) {
|
||||
typeList.Add(type);
|
||||
continue;
|
||||
}
|
||||
if (Regex.IsMatch(code, string.Format("delegate\\s*{0}\\s\\(", typeName))) {
|
||||
typeList.Add(type);
|
||||
continue;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
List<System.Type> CollectionAllClasses() {
|
||||
string path = Application.dataPath;
|
||||
string absolutePath = path.Remove(path.Length - 6);
|
||||
List<System.Type> alltypes = new List<System.Type>();
|
||||
if (File.Exists(absolutePath + "Library/ScriptAssemblies/Assembly-CSharp.dll"))
|
||||
alltypes.AddRange(Assembly.LoadFile(absolutePath + "Library/ScriptAssemblies/Assembly-CSharp.dll").GetTypes());
|
||||
if (isSaveEditorCode && File.Exists(absolutePath + "Library/ScriptAssemblies/Assembly-CSharp-Editor.dll"))
|
||||
alltypes.AddRange(Assembly.LoadFile(absolutePath + "Library/ScriptAssemblies/Assembly-CSharp-Editor.dll").GetTypes());
|
||||
|
||||
return alltypes.ToList();
|
||||
}
|
||||
|
||||
List<System.Type> collectionAllFastspassClasses() {
|
||||
string path = Application.dataPath;
|
||||
string absolutePath = path.Remove(path.Length - 6);
|
||||
List<System.Type> alltypes = new List<System.Type>();
|
||||
if (File.Exists(absolutePath + "Library/ScriptAssemblies/Assembly-CSharp-firstpass.dll"))
|
||||
alltypes.AddRange(Assembly.LoadFile(absolutePath + "Library/ScriptAssemblies/Assembly-CSharp-firstpass.dll").GetTypes());
|
||||
if (isSaveEditorCode && File.Exists(absolutePath + "Library/ScriptAssemblies/Assembly-CSharp-Editor-firstpass.dll"))
|
||||
alltypes.AddRange(Assembly.LoadFile(absolutePath + "Library/ScriptAssemblies/Assembly-CSharp-Editor-firstpass.dll").GetTypes());
|
||||
return alltypes;
|
||||
}
|
||||
|
||||
public static string StripComment(string code) {
|
||||
code = Regex.Replace(code, "//.*[\\n\\r]", "");
|
||||
code = Regex.Replace(code, "/\\*.*[\\n\\r]\\*/", "");
|
||||
return code;
|
||||
}
|
||||
|
||||
void CollectionReferenceClasses(string guid, List<System.Type> types) {
|
||||
var codePath = AssetDatabase.GUIDToAssetPath(guid);
|
||||
if (string.IsNullOrEmpty(codePath) || File.Exists(codePath) == false) {
|
||||
return;
|
||||
}
|
||||
|
||||
var code = StripComment(System.IO.File.ReadAllText(codePath));
|
||||
|
||||
List<string> referenceList = null;
|
||||
CollectionData reference = null;
|
||||
|
||||
if (references.Exists(c => c.fileGuid == guid) == false) {
|
||||
referenceList = new List<string>();
|
||||
reference = new CollectionData() {
|
||||
fileGuid = guid,
|
||||
referenceGids = referenceList,
|
||||
};
|
||||
references.Add(reference);
|
||||
}
|
||||
else {
|
||||
reference = references.Find(c => c.fileGuid == guid);
|
||||
referenceList = reference.referenceGids;
|
||||
}
|
||||
|
||||
referenceList.Clear();
|
||||
|
||||
var timestamp = File.GetLastWriteTime(codePath);
|
||||
reference.timeStamp = timestamp;
|
||||
|
||||
foreach (var type in types) {
|
||||
|
||||
if (code2FileDic.ContainsKey(type) == false || code2FileDic[type].Contains(guid)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
if (string.IsNullOrEmpty(type.Namespace) == false) {
|
||||
var namespacepattern = string.Format("([namespace|using][\\s]{0}[{{\\s\\r\\n\\r;]|{0}\\.)", type.Namespace);
|
||||
if (Regex.IsMatch(code, namespacepattern) == false) {
|
||||
continue;
|
||||
}
|
||||
}
|
||||
|
||||
string match = string.Empty;
|
||||
|
||||
if (type.IsGenericTypeDefinition) {
|
||||
string typeName = type.GetGenericTypeDefinition().Name.Split('`')[0];
|
||||
match = string.Format("[!|&\\]\\[\\.\\s<(]{0}[\\.\\s\\n\\r>,<(){{]", typeName);
|
||||
|
||||
}
|
||||
else {
|
||||
string typeName = type.Name.Split('`')[0].Replace("Attribute", "");
|
||||
match = string.Format("[!|&\\]\\[\\.\\s<(]{0}[\\.\\s\\n\\r>,<(){{\\]]", typeName);
|
||||
|
||||
|
||||
// check Extension Methods
|
||||
|
||||
if (Regex.IsMatch(code, string.Format("this\\s{0}\\s", typeName))) {
|
||||
foreach (var file in code2FileDic[type]) {
|
||||
foreach (var baseReference in references.Where(c => c.fileGuid == file)) {
|
||||
baseReference.referenceGids.Add(guid);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
if (Regex.IsMatch(code, match)) {
|
||||
var typeGuids = code2FileDic[type];
|
||||
foreach (var typeGuid in typeGuids) {
|
||||
|
||||
if (referenceList.Contains(typeGuid) == false) {
|
||||
referenceList.Add(typeGuid);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void CollectionCustomEditorClasses(IEnumerable<System.Type> types) {
|
||||
foreach (var type in types) {
|
||||
|
||||
if (code2FileDic.ContainsKey(type) == false) {
|
||||
continue;
|
||||
}
|
||||
|
||||
var attributes = type.GetCustomAttributes(typeof(CustomEditor), true);
|
||||
foreach (var attribute in attributes) {
|
||||
if (attribute is CustomEditor == false) {
|
||||
continue;
|
||||
}
|
||||
var customEditor = attribute as CustomEditor;
|
||||
var customEditorReferenceTypeField = typeof(CustomEditor).GetField("m_InspectedType", BindingFlags.Instance | BindingFlags.NonPublic);
|
||||
var customEditorReferenceType = (System.Type)customEditorReferenceTypeField.GetValue(customEditor);
|
||||
|
||||
if (code2FileDic.ContainsKey(customEditorReferenceType) == false) {
|
||||
continue;
|
||||
}
|
||||
|
||||
foreach (var filePath in code2FileDic[customEditorReferenceType]) {
|
||||
if (references.Exists(c => c.fileGuid == filePath) == false) {
|
||||
continue;
|
||||
}
|
||||
foreach (var refs in references.Where(c => c.fileGuid == filePath)) {
|
||||
var list = refs.referenceGids;
|
||||
list.AddRange(code2FileDic[type]);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user