void Write(string outputFileName, string templateFileName, string t, string fullPath, ref string log)
{
try {
string format = Resources.Load <TextAsset>(templateFileName).text;
if (!Directory.Exists(fullPath))
{
Directory.CreateDirectory(fullPath);
}
string absPathToScript = fullPath + outputFileName;
string relPathToScript = SmartEditorUtils.ToRelativePath(absPathToScript);
scriptAbsPathToGuid[absPathToScript] = AssetDatabase.AssetPathToGUID(relPathToScript);
File.WriteAllText(absPathToScript, string.Format(format, t, PrettyTypeName(t), Capitalize(t)));
log += "\n " + outputFileName;
} catch (System.FormatException e) {
Debug.LogError("Bad formatting in template " + templateFileName);
throw e;
}
}
void PostProcessMeta()
{
#region Repair GUIDS
bool refresh = false;
foreach (var a in scriptAbsPathToGuid)
{
// Get latest guid of script from assetdatabase and compare to cached to see if changed
string newGuid = AssetDatabase.AssetPathToGUID(SmartEditorUtils.ToRelativePath(a.Key));
if (newGuid != a.Value)
{
Debug.LogWarningFormat("Repairing changed GUID for script {0}", a.Key);
string metaPath = a.Key + ".meta";
#region Repair meta file
if (File.Exists(metaPath))
{
// Find line with latest guid and replace with cached guid
var lines = File.ReadAllLines(a.Key);
for (int i = 0; i < lines.Length; ++i)
{
string l = lines[i];
if (l.Contains(newGuid))
{
Debug.LogFormat("\n\tRestoring {0} to {1}", newGuid, a.Value);
lines[i] = l.Replace(newGuid, a.Value);
// Save and flag for asset database refresh
File.WriteAllLines(a.Key, lines);
refresh = true;
break;
}
}
}
#endregion
}
}
if (refresh)
{
AssetDatabase.Refresh();
}
#endregion
#region Set icons
refresh = false;
foreach (var a in scriptAbsPathToGuid)
{
string metaPath = a.Key + ".meta";
if (File.Exists(metaPath))
{
var lines = File.ReadAllLines(metaPath);
// Read meta file, look for line with "icon:"
for (int i = 0; i < lines.Length; ++i)
{
string l = lines[i];
// Line must have "icon: " and default icon stuff
if (l.Contains(ICON_MATCH))
{
// Get type of script, then get associated icon from utilities
string pathToScript = AssetDatabase.GUIDToAssetPath(a.Value);
MonoScript script = AssetDatabase.LoadAssetAtPath <MonoScript>(pathToScript);
System.Type tData;
var sot = SmartEditorUtils.GetSmartObjectType(script.GetClass(), out tData);
//TODO Implement icons for component types
if (sot == SmartEditorUtils.SmartObjectType.NONE)
{
break;
}
Texture2D icon = SmartEditorUtils.LoadSmartIcon(sot, false, true);
// Get guid of icon and jam it in there
string pathToIcon = AssetDatabase.GetAssetPath(icon);
string iconGuid = AssetDatabase.AssetPathToGUID(pathToIcon);
bool modified = false;
if (l.Contains(ICON_DEFAULT))
{
// Replace entire default icon
l = l.Replace(ICON_DEFAULT, string.Format("{{fileID: 2800000, guid: {0}, type: 3}}", iconGuid));
modified = true;
}
else if (l.Contains(ICON_CUSTOM_HEAD))
{
// Isolate guid to check if changed
string check = l.Trim().Replace(ICON_MATCH, "").Replace(ICON_CUSTOM_HEAD, "").Replace(ICON_CUSTOM_TAIL, "");
if (check != iconGuid)
{
var sb = new System.Text.StringBuilder();
// Whitespace
for (int j = 0; j < l.Length; ++j)
{
if (l[j] != ' ')
{
break;
}
sb.Append(' ');
}
// Build with new guid
sb.Append(ICON_MATCH);
sb.Append(ICON_CUSTOM_HEAD);
sb.Append(iconGuid);
sb.Append(ICON_CUSTOM_TAIL);
l = sb.ToString();
modified = true;
}
}
if (modified)
{
Debug.Log("Setting icon for script " + a.Key);
// Save and flag for asset database refresh
lines[i] = l;
File.WriteAllLines(metaPath, lines);
refresh = true;
}
break;
}
}
}
else
{
Debug.LogError("Meta file not found: " + metaPath);
}
}
#endregion
if (refresh)
{
AssetDatabase.Refresh();
}
}
void OnGUI()
{
bool repaint = false;
if (_t.Count == 0)
{
_t.Add("");
}
Color gbc = GUI.backgroundColor;
EditorGUILayout.BeginVertical(); { // Used to GetLastRect to control total height of window
EditorGUILayout.Space();
if (!_isRegenerating)
{
EditorGUILayout.LabelField("Data Types", GUILayout.MaxWidth(70));
if (InlineHelpButton())
{
_showTypeTooltip = !_showTypeTooltip;
}
if (_isParsingAssemblies)
{
EditorGUI.ProgressBar(GUILayoutUtility.GetLastRect(), _parseProgress, "Parsing Types");
}
++EditorGUI.indentLevel;
if (_showTypeTooltip)
{
EditorGUILayout.HelpBox("Case-sensitive. Include any namespaces/nesting.\nBasic types (e.g. int, float) and Unity types (e.g. Vector3) don't need namespaces.", MessageType.Info);
}
_toRemove.Clear();
int focusedField = -1;
var btnOptions = new GUILayoutOption[] { GUILayout.Width(20), GUILayout.Height(13) };
bool missingGenericArgs = false;
for (int i = 0; i < _t.Count; ++i)
{
bool typeIsOpenGeneric = false;
string focusName = string.Format(TYPE_FIELD_NAME, i);
EditorGUILayout.BeginHorizontal(); {
GUI.SetNextControlName(focusName);
if (_t[i].Contains("#") || _t[i].Contains("<,") || _t[i].Contains(",,") || _t[i].Contains("<>") || _t[i].Contains(",>"))
{
GUI.backgroundColor = Color.red;
missingGenericArgs = true;
typeIsOpenGeneric = true;
}
_t[i] = WithoutSelectAll(() => EditorGUILayout.TextField(_t[i]));
GUI.backgroundColor = gbc;
GUI.backgroundColor = Color.red;
GUI.enabled = _t.Count > 1;
if (GUILayout.Button("-", btnOptions))
{
_toRemove.Add(_t[i]);
}
GUI.enabled = true;
GUI.backgroundColor = gbc;
} EditorGUILayout.EndHorizontal();
if (typeIsOpenGeneric)
{
EditorGUILayout.HelpBox("Replace all # with type arguments", MessageType.Error);
}
// Get focused field for autocomplete
string focus = GUI.GetNameOfFocusedControl();
if (focus == focusName)
{
if (settingsDirty)
{
OnFocus();
}
if (!string.IsNullOrEmpty(_t[i]))
{
focusedField = i;
}
else
{
_lastTypeName = null;
}
}
}
// Autocomplete
if (!_isParsingAssemblies && focusedField >= 0)
{
// Check if changed
string t = _t[focusedField];
if (t != _lastTypeName)
{
_typeAutocompleteScroll = Vector2.zero;
_lastTypeName = t;
// Start match
_typeMatchIndex = 0;
_typesAutocompleted.Clear();
_matching = true;
}
// Match
if (_matching)
{
_acSw.Start();
bool skip = false;
while (_typeMatchIndex < _typeNames.Count)
{
if (_typeNames[_typeMatchIndex].Contains(t))
{
_typesAutocompleted.Add(_typeNames[_typeMatchIndex]);
}
++_typeMatchIndex;
if (_acSw.Elapsed.TotalSeconds > SmartTypeCreatorSettings.DEFAULT.asyncTypeLoadFrameTime)
{
skip = true;
repaint = true;
break;
}
}
if (!skip)
{
// Finished
_matching = false;
}
_acSw.Stop();
_acSw.Reset();
}
// Autocomplete box
if (_typesAutocompleted.Count > 0)
{
_typeAutocompleteScroll = EditorGUILayout.BeginScrollView(
_typeAutocompleteScroll,
GUILayout.Height(GetTypeAutocompleteHeight()),
GUILayout.ExpandWidth(true)
); {
EditorGUILayout.BeginHorizontal(); {
GUILayout.Space(30);
GUI.backgroundColor = _matching ? Color.yellow : Color.green;
EditorGUILayout.BeginVertical(EditorStyles.helpBox, GUILayout.Height(_typesAutocompleted.Count * 20)); {
// Fake header space
// Don't render occluded elements, according to scroll
int offset = Mathf.Max(0, Mathf.FloorToInt(_typeAutocompleteScroll.y / 20) - 1);
GUILayout.Space(offset * 20);
GUI.backgroundColor = gbc;
for (int i = offset; i < _typesAutocompleted.Count; ++i)
{
EditorGUILayout.BeginHorizontal(); {
if (GUILayout.Button("Select", GUILayout.MaxWidth(75)))
{
_t[focusedField] = _typesAutocompleted[i];
GUI.FocusControl(null);
repaint = true;
}
EditorGUILayout.LabelField(_typesAutocompleted[i]);
} EditorGUILayout.EndHorizontal();
if (i > offset + MAX_AUTOCOMPLETE_LINES + 1)
{
break;
}
}
} EditorGUILayout.EndVertical();
} EditorGUILayout.EndHorizontal();
} EditorGUILayout.EndScrollView();
}
}
else
{
_matching = false;
_lastTypeName = null;
_typesAutocompleted.Clear();
}
if (_toRemove.Count > 0)
{
foreach (string s in _toRemove)
{
_t.Remove(s);
}
_toRemove.Clear();
}
EditorGUILayout.BeginHorizontal(); {
GUI.enabled = false;
EditorGUILayout.TextField("");
GUI.enabled = true;
GUI.backgroundColor = Color.green;
if (GUILayout.Button("+", GUILayout.Width(20), GUILayout.Height(14)))
{
_t.Add("");
}
GUI.backgroundColor = gbc;
} EditorGUILayout.EndHorizontal();
--EditorGUI.indentLevel;
EditorGUILayout.BeginHorizontal(); {
_path = EditorGUILayout.TextField(
new GUIContent("Path", "Folder for generated scripts. If Individual Subfolders ticked, subfolders will be created here."),
_path
);
if (GUILayout.Button("Select...", GUILayout.Width(75)))
{
string p = EditorUtility.OpenFolderPanel("Select folder for generated Smart Types", _path, "");
if (!string.IsNullOrEmpty(p))
{
_path = SmartEditorUtils.ToRelativePath(p);
}
}
} EditorGUILayout.EndHorizontal();
_overwrite = EditorGUILayout.Toggle(new GUIContent("Overwrite", "If false, will ignore types that already exist in the target path."), _overwrite);
_subfolders = EditorGUILayout.Toggle(new GUIContent("Individual Subfolders", "Create subfolders for each type within the target path."), _subfolders);
EditorGUILayout.Space();
EditorGUILayout.BeginHorizontal(EditorStyles.helpBox); {
EditorGUILayout.BeginVertical(GUILayout.Width(150)); {
GUI.backgroundColor = Color.green;
GUI.enabled = !missingGenericArgs && _settingsValid;
if (GUILayout.Button("Create"))
{
PopulateTemplates();
int successes = 0;
scriptAbsPathToGuid.Clear();
for (int i = 0; i < _t.Count; ++i)
{
if (CreateType(_t[i], _overwrite))
{
++successes;
_toRemove.Add(_t[i]);
}
}
if (successes > 0)
{
if (successes == _t.Count)
{
ClosePopup();
}
else
{
// Remove successful types
foreach (string s in _toRemove)
{
_t.Remove(s);
}
}
AssetDatabase.Refresh();
PostProcessMeta();
}
}
GUI.enabled = _settingsValid;
GUI.backgroundColor = Color.cyan;
if (GUILayout.Button("Regenerate..."))
{
PopulateRegen();
}
GUI.backgroundColor = Color.magenta;
if (GUILayout.Button("Regenerate All"))
{
PopulateRegen();
RegenNow();
}
GUI.backgroundColor = gbc;
GUI.enabled = true;
} EditorGUILayout.EndVertical();
DrawSettings();
} EditorGUILayout.EndHorizontal();
}
else
{
// Regen mode
EditorGUILayout.LabelField("Regenerate types:");
EditorGUILayout.Space();
++EditorGUI.indentLevel;
foreach (var a in _regenTypesByPath)
{
EditorGUILayout.LabelField(a.Key);
++EditorGUI.indentLevel;
_regenTypes.Clear();
foreach (string type in a.Value.Keys)
{
_regenTypes.Add(type);
}
foreach (string type in _regenTypes)
{
a.Value[type] = EditorGUILayout.ToggleLeft(type, a.Value[type]);
}
--EditorGUI.indentLevel;
}
--EditorGUI.indentLevel;
EditorGUILayout.Space();
EditorGUILayout.Space();
EditorGUILayout.BeginHorizontal(EditorStyles.helpBox); {
EditorGUILayout.BeginVertical(GUILayout.Width(150)); {
GUI.backgroundColor = Color.green;
if (GUILayout.Button("Regenerate!"))
{
RegenNow();
}
GUI.backgroundColor = Color.red;
if (GUILayout.Button("Cancel"))
{
_regenTypesByPath.Clear();
}
GUI.backgroundColor = gbc;
} EditorGUILayout.EndVertical();
DrawSettings();
} EditorGUILayout.EndHorizontal();
}
} EditorGUILayout.EndVertical();
// GetLastRect for entire vertical group to control total height of window
SetHeight(GUILayoutUtility.GetLastRect().height);
if (repaint)
{
Repaint();
}
}