Added ability to get files only with a specific extension

Author: Cory Leach

Runtime/SaveLoadManager.cs

@@ -13,18 +13,18 @@ public class SaveLoadManager : ScriptableObject
     {
         [Header("Settings"),SerializeField] private string defaultFolder = "SaveData";
         public string DefaultFolder => defaultFolder;
-        
+
         [SerializeField] private string baseFolder = "GameData";
         public string BaseFolder => baseFolder;
-        
+
         [SerializeField] private SerializationMethodType saveMethod = SerializationMethodType.Default;
 
         [Header("Encryption"),SerializeField] protected string key = string.Empty;
         public string Key => key;
-        
+
         [SerializeField] protected string salt = string.Empty;
         public string Salt => salt;
-        
+
         private Dictionary<SerializationMethodType, ISerializationMethod> _methods;
 
         private void OnEnable()
@@ -53,10 +53,10 @@ public static SaveLoadManager Create(string baseFolder, string defaultFolder, Se
             instance.key = key;
             instance.salt = salt;
             instance.saveMethod = saveMethod;
-            
+
             return instance;
         }
-        
+
         /// <summary>
         /// Save an object to disk
         /// </summary>
@@ -77,14 +77,15 @@ public void Save(object obj, string filename, string folder = null)
         /// Gets the list of save files that have been created
         /// </summary>
         /// <param name="folder">sub folder</param>
+        /// <param name="extension">include only files with this extension</param>
         /// <returns>list of file names (excludes the path)</returns>
-        public string[] GetFiles(string folder = null)
+        public string[] GetFiles(string folder = null, string extension = null)
         {
             if (string.IsNullOrEmpty(folder))
             {
                 folder = defaultFolder;
             }
-            return SaveLoadUtility.GetSavedFiles(folder,baseFolder);
+            return SaveLoadUtility.GetSavedFiles(folder,baseFolder, extension);
         }
 
         /// <summary>
@@ -119,7 +120,7 @@ public T Copy<T>(T obj)
             var saveLoadMethod = GetSaveLoadMethod(saveMethod);
             return (T)saveLoadMethod.Copy(obj);
         }
-        
+
         /// <summary>
         /// Load an object from disk
         /// </summary>
@@ -162,7 +163,7 @@ public void DeleteSave(string filename, string folder = null)
             }
             SaveLoadUtility.DeleteSavedFile(filename,folder, baseFolder);
         }
-        
+
         /// <summary>
         /// Save object to file and specify the method of save/load
         /// </summary>
@@ -217,7 +218,7 @@ public void SaveUnityObject(UnityEngine.Object unityObj, string filename, string
             {
                 jsonData = JsonUtility.ToJson(unityObj)
             };
-            
+
             Save(savedObj,filename,folder);
         }
 
@@ -233,12 +234,12 @@ public void SaveUnityObject(UnityEngine.Object unityObj, string filename, string
         public bool LoadUnityObjectOverwrite(UnityEngine.Object objectToOverwrite, string filename, string folder = null)
         {
             var savedObj = Load<JsonSerializedUnityObject>(filename, folder);
-            
+
             if (savedObj == null || string.IsNullOrEmpty(savedObj.jsonData))
             {
                 return false;
             }
-            
+
             JsonUtility.FromJsonOverwrite(savedObj.jsonData,objectToOverwrite);
             return true;
         }
@@ -253,7 +254,7 @@ public void CopyUnityObjectOverwrite(UnityEngine.Object toCopy, UnityEngine.Obje
             var jsonData = JsonUtility.ToJson(toCopy);
             JsonUtility.FromJsonOverwrite(jsonData,toOverwrite);
         }
-        
+
         /// <summary>
         /// JsonSerializedUnityObject
         /// Wrapper for json data created when using Unity's JsonUtility to serialize an object derived from UnityEngine.Object
@@ -280,7 +281,7 @@ public void SetCustomSerializationMethod(ISerializationMethod customSerializatio
             }
             _methods[SerializationMethodType.Custom] = customSerializationMethod;
         }
-        
+
         private ISerializationMethod GetSaveLoadMethod(SerializationMethodType methodType)
         {
             if (_methods == null)
@@ -305,14 +306,14 @@ private ISerializationMethod GetSaveLoadMethod(SerializationMethodType methodTyp
                 case SerializationMethodType.UnityJson:
                     method = new SerializationMethodUnityJson();
                     break;
-                
+
                 case SerializationMethodType.BinaryEncrypted:
                     method = new SerializationMethodBinaryEncrypted(key,salt);
                     break;
                 case SerializationMethodType.UnityJsonEncrypted:
                     method = new SerializationMethodUnityJsonEncrypted(key,salt);
                     break;
-                
+
 #if JSON_DOT_NET
                 case SerializationMethodType.JsonDotNet:
                     method = new SerializationMethodJsonDotNet();
@@ -321,19 +322,17 @@ private ISerializationMethod GetSaveLoadMethod(SerializationMethodType methodTyp
                     method = new SerializationMethodJsonDotNetEncrypted(key,salt);
                     break;
 #endif
-                
+
                 case SerializationMethodType.Custom:
                     throw new MissingComponentException("SerializationMethodType is Custom but no custom ISerializationMethod was found.");
                 default:
                     throw new ArgumentOutOfRangeException(nameof(methodType), methodType, "SaveLoadMethodType not supported");
             }
 
             _methods[methodType] = method;
-            
+
             return method;
         }
-        
-    }    
-}
-
 
+    }
+}

Runtime/SaveLoadUtility.cs

@@ -10,12 +10,12 @@ public static class SaveLoadUtility
         //Default folder name will be used if none is provided.
         private const string DefaultFolderName = "SaveLoad";
         private const string DefaultBaseFolderPath = "GameData";
-        
+
         public static string GetSavePath(string folderName = null, string baseFolderPath = null)
         {
             return GetRuntimeSavePath(folderName, baseFolderPath);
         }
-        
+
         public static string GetRuntimeSavePath(string folderName = null, string baseFolderPath = null)
         {
             if (string.IsNullOrEmpty(folderName))
@@ -34,17 +34,17 @@ public static string GetRuntimeSavePath(string folderName = null, string baseFol
         }
 
         /// <summary>
-        /// 
+        ///
         /// </summary>
         /// <param name="fileName"></param>
         /// <returns></returns>
         private static string GetSaveFileName(string fileName)
         {
             return fileName;
         }
-        
+
         /// <summary>
-        /// 
+        ///
         /// </summary>
         /// <param name="saveObject"></param>
         /// <param name="serializationMethod"></param>
@@ -55,7 +55,7 @@ public static void Save(object saveObject, ISerializationMethod serializationMet
         {
             var savePath = GetSavePath(folderName,baseFolderPath);
             var saveFilename = GetSaveFileName(filename);
-            
+
             //Create directory if it does not exist
             if (!Directory.Exists(savePath))
             {
@@ -95,43 +95,46 @@ public static object Load(System.Type objectType, ISerializationMethod serializa
                 returnObject = serializationMethod.Load(objectType, saveFile);
                 saveFile.Close();
             }
-            
+
             return returnObject;
         }
-        
+
         /// <summary>
         /// Enumerate files in the save directory
         /// </summary>
         /// <param name="folderName">folder containing the save files</param>
         /// <param name="baseFolderPath">base path to the folder</param>
+        /// <param name="extension">include only files with the specified extension</param>
         /// <returns>list of file names</returns>
-        public static IEnumerable<string> EnumerateSavedFiles(string folderName = null, string baseFolderPath = null)
+        public static IEnumerable<string> EnumerateSavedFiles(string folderName = null, string baseFolderPath = null, string extension = null)
         {
             var savePath = GetSavePath(folderName,baseFolderPath);
-            
+
             //If directory does not exist we're done
             if (!Directory.Exists(savePath))
             {
                 yield break;
             }
-            
-            foreach ( var file in Directory.EnumerateFiles(savePath,"*",SearchOption.AllDirectories) )
+
+            var searchPattern = string.IsNullOrEmpty(extension) ? "*" : $"*.{extension}";
+            foreach ( var file in Directory.EnumerateFiles(savePath,searchPattern,SearchOption.AllDirectories) )
             {
                 yield return Path.GetFileName(file);
             }
         }
-        
+
         /// <summary>
         /// Creates an array list of save files in the given folder and path
         /// </summary>
-        /// <param name="folderName"></param>
-        /// <param name="baseFolderPath"></param>
+        /// <param name="folderName">folder containing the save files</param>
+        /// <param name="baseFolderPath">base path to the folder</param>
+        /// <param name="extension">include only files with this extension</param>
         /// <returns>Array of file names</returns>
-        public static string[] GetSavedFiles(string folderName = null, string baseFolderPath = null)
+        public static string[] GetSavedFiles(string folderName = null, string baseFolderPath = null, string extension = null)
         {
-            return EnumerateSavedFiles(folderName, baseFolderPath).ToArray();
+            return EnumerateSavedFiles(folderName, baseFolderPath, extension).ToArray();
         }
-        
+
         /// <summary>
         /// Check if a saved file exists
         /// </summary>
@@ -175,10 +178,8 @@ public static void DeleteDirectory(string path)
             {
                 DeleteDirectory(dir);
             }
-            
+
             Directory.Delete(path,false);
         }
     }
 }
-
-