check how many assets need update before updating

also prompt user if they wish to update, and encourage them to make a
backup.

Author: vkovec

Assets/FbxExporters/Editor/FbxExportSettings.cs

@@ -180,27 +180,33 @@ public override void OnInspectorGUI() {
             }
             EditorGUI.EndDisabledGroup ();
 
-            if (!HideRepairMissingScripts ()) {
-                EditorGUILayout.Space ();
-
-                EditorGUI.indentLevel--;
-                EditorGUILayout.LabelField ("Repair Missing Scripts", EditorStyles.boldLabel);
-                EditorGUI.indentLevel++;
+            EditorGUILayout.Space ();
 
-                EditorGUILayout.Space ();
+            EditorGUI.indentLevel--;
+            EditorGUILayout.LabelField ("Run Component Updater", EditorStyles.boldLabel);
+            EditorGUI.indentLevel++;
 
-                var repairMissingScripts = new GUIContent (
-                                           "Repair Missing Scripts",
-                                           "Repair missing FbxPrefab scripts in text assets");
-                if (GUILayout.Button (repairMissingScripts)) {
-                    bool result = FbxExporters.Editor.RepairMissingScripts.ReplaceGUIDInTextAssets ();
+            EditorGUILayout.Space ();
+
+            var repairMissingScripts = new GUIContent (
+                                       "Run Component Updater",
+                                       "Repair missing FbxPrefab scripts in text assets");
+            if (GUILayout.Button (repairMissingScripts)) {
+                var componentUpdater = new FbxExporters.Editor.RepairMissingScripts ();
+                var filesToRepairCount = componentUpdater.GetAssetsToRepairCount ();
+                if (filesToRepairCount > 0) {
+                    bool result = UnityEditor.EditorUtility.DisplayDialog ("Component Updater",
+                        string.Format("Found {0} text assets with components requiring update.\n\n" +
+                        "If you choose 'Go Ahead', the components in these text serialized assets " +
+                        "will be automatically updated to work with the latest FBX exporter.\n" +
+                            "You should make a backup before proceeding.", filesToRepairCount),
+                        "I Made a Backup. Go Ahead!", "No Thanks");
                     if (result) {
-                        UnityEditor.EditorUtility.DisplayDialog ("Finished Repairing Scripts",
-                            "Repaired missing scripts in text serialized assets", "Ok");
-                    } else {
-                        UnityEditor.EditorUtility.DisplayDialog ("Finished Repairing Scripts",
-                            "Couldn't find any assets needing repair", "Ok");
+                        componentUpdater.ReplaceGUIDInTextAssets ();
                     }
+                } else {
+                    UnityEditor.EditorUtility.DisplayDialog ("Component Updater",
+                        "Couldn't find any text assets requiring update", "Ok");
                 }
             }
 
@@ -214,11 +220,6 @@ public override void OnInspectorGUI() {
             }
         }
 
-        private static bool HideRepairMissingScripts(){
-            var docPath = Application.dataPath + "/FbxExporters/FBX_Exporter_User_Guide_v1.1.0b1.pdf";
-            return File.Exists(docPath)? false : true;
-        }
-
         private static string TryFindDCC(string dccPath, string ext, ExportSettings.DCCType dccType){
             string dccName = "";
             switch (dccType) {

Assets/FbxExporters/Editor/FbxExporterRepairMissingScripts.cs

@@ -15,7 +15,6 @@ public class RepairMissingScripts
 
         private const string m_idFormat = "{{fileID: {0}, guid: {1}, type:";
 
-
         private static string m_forumPackageSearchID;
 
         private static string ForumPackageSearchID {
@@ -38,24 +37,74 @@ private static string CurrentPackageSearchID {
             }
         }
 
-        public static bool ReplaceGUIDInTextAssets ()
+        private string[] m_assetsToRepair;
+
+        public RepairMissingScripts(){
+            m_assetsToRepair = GetAssetsToRepair ();
+        }
+
+        public int GetAssetsToRepairCount(){
+            return (m_assetsToRepair != null) ? m_assetsToRepair.Length : 0;
+        }
+
+        public static string[] GetAssetsToRepair()
         {
             // search project for assets containing old GUID
 
             // ignore if forced binary
             if (UnityEditor.EditorSettings.serializationMode == SerializationMode.ForceBinary) {
-                return false;
+                return new string[]{};
             }
 
             // check all scenes and prefabs
             string[] searchFilePatterns = new string[]{ "*.prefab", "*.unity" };
 
-            bool replacedGUID = false;
+            List<string> assetsToRepair = new List<string> ();
             foreach (string searchPattern in searchFilePatterns) {
                 foreach (string file in Directory.GetFiles(Application.dataPath, searchPattern, SearchOption.AllDirectories)) {
-                    replacedGUID |= ReplaceGUIDInFile (file);
+                    if (AssetNeedsRepair (file)) {
+                        assetsToRepair.Add (file);
+                    }
                 }
             }
+            return assetsToRepair.ToArray ();
+        }
+
+        private static bool AssetNeedsRepair(string filePath)
+        {
+            try{
+                using(var sr = new StreamReader (filePath)){
+                    if(sr.Peek() > -1){
+                        var firstLine = sr.ReadLine();
+                        if(!firstLine.StartsWith("%YAML")){
+                            sr.Close();
+                            return false;
+                        }
+                    }
+
+                    var contents = sr.ReadToEnd();
+                    if(contents.Contains(ForumPackageSearchID)){
+                        sr.Close();
+                        return true;
+                    }
+                }
+            }
+            catch(IOException e){
+                Debug.LogError (string.Format ("Failed to check file for component update: {0} (error={1})", filePath, e));
+            }
+            return false;
+        }
+
+        public bool ReplaceGUIDInTextAssets ()
+        {
+            if (m_assetsToRepair == null) {
+                return false;
+            }
+
+            bool replacedGUID = false;
+            foreach (string file in m_assetsToRepair) {
+                replacedGUID |= ReplaceGUIDInFile (file);
+            }
             if (replacedGUID) {
                 AssetDatabase.Refresh ();
             }