encapsulate mod DB a bit better for use outside SMAPI (#532)

1 files changed, 11 insertions, 92 deletions

diff --git a/src/StardewModdingAPI.Toolkit/Framework/ModData/ModDatabase.cs b/src/StardewModdingAPI.Toolkit/Framework/ModData/ModDatabase.cs
index c60d2bcb..a12e3c67 100644
--- a/src/StardewModdingAPI.Toolkit/Framework/ModData/ModDatabase.cs
+++ b/src/StardewModdingAPI.Toolkit/Framework/ModData/ModDatabase.cs
@@ -11,7 +11,7 @@ namespace StardewModdingAPI.Toolkit.Framework.ModData
         ** Properties
         *********/
         /// <summary>The underlying mod data records indexed by default display name.</summary>
-        private readonly IDictionary<string, ModDataRecord> Records;
+        private readonly ModDataRecord[] Records;
 
         /// <summary>Get an update URL for an update key (if valid).</summary>
         private readonly Func<string, string> GetUpdateUrl;
@@ -22,63 +22,23 @@ namespace StardewModdingAPI.Toolkit.Framework.ModData
         *********/
         /// <summary>Construct an empty instance.</summary>
         public ModDatabase()
-        : this(new Dictionary<string, ModDataRecord>(), key => null) { }
+        : this(new ModDataRecord[0], key => null) { }
 
         /// <summary>Construct an instance.</summary>
         /// <param name="records">The underlying mod data records indexed by default display name.</param>
         /// <param name="getUpdateUrl">Get an update URL for an update key (if valid).</param>
-        public ModDatabase(IDictionary<string, ModDataRecord> records, Func<string, string> getUpdateUrl)
+        public ModDatabase(IEnumerable<ModDataRecord> records, Func<string, string> getUpdateUrl)
         {
-            this.Records = records;
+            this.Records = records.ToArray();
             this.GetUpdateUrl = getUpdateUrl;
         }
 
-        /// <summary>Get a parsed representation of the <see cref="ModDataRecord.Fields"/> which match a given manifest.</summary>
-        /// <param name="manifest">The manifest to match.</param>
-        public ParsedModDataRecord GetParsed(IManifest manifest)
+        /// <summary>Get a mod data record.</summary>
+        /// <param name="modID">The unique mod ID.</param>
+        public ModDataRecord Get(string modID)
         {
-            // get raw record
-            if (!this.TryGetRaw(manifest?.UniqueID, out string displayName, out ModDataRecord record))
-                return null;
-
-            // parse fields
-            ParsedModDataRecord parsed = new ParsedModDataRecord { DisplayName = displayName, DataRecord = record };
-            foreach (ModDataField field in record.GetFields().Where(field => field.IsMatch(manifest)))
-            {
-                switch (field.Key)
-                {
-                    // update key
-                    case ModDataFieldKey.UpdateKey:
-                        parsed.UpdateKey = field.Value;
-                        break;
-
-                    // alternative URL
-                    case ModDataFieldKey.AlternativeUrl:
-                        parsed.AlternativeUrl = field.Value;
-                        break;
-
-                    // status
-                    case ModDataFieldKey.Status:
-                        parsed.Status = (ModStatus)Enum.Parse(typeof(ModStatus), field.Value, ignoreCase: true);
-                        parsed.StatusUpperVersion = field.UpperVersion;
-                        break;
-
-                    // status reason phrase
-                    case ModDataFieldKey.StatusReasonPhrase:
-                        parsed.StatusReasonPhrase = field.Value;
-                        break;
-                }
-            }
-
-            return parsed;
-        }
-
-        /// <summary>Get the display name for a given mod ID (if available).</summary>
-        /// <param name="id">The unique mod ID.</param>
-        public string GetDisplayNameFor(string id)
-        {
-            return this.TryGetRaw(id, out string displayName, out ModDataRecord _)
-                ? displayName
+            return !string.IsNullOrWhiteSpace(modID)
+                ? this.Records.FirstOrDefault(p => p.HasID(modID))
                 : null;
         }
 
@@ -86,55 +46,14 @@ namespace StardewModdingAPI.Toolkit.Framework.ModData
         /// <param name="id">The unique mod ID.</param>
         public string GetModPageUrlFor(string id)
         {
-            // get raw record
-            if (!this.TryGetRaw(id, out string _, out ModDataRecord record))
-                return null;
-
             // get update key
-            ModDataField updateKeyField = record.GetFields().FirstOrDefault(p => p.Key == ModDataFieldKey.UpdateKey);
+            ModDataRecord record = this.Get(id);
+            ModDataField updateKeyField = record?.Fields.FirstOrDefault(p => p.Key == ModDataFieldKey.UpdateKey);
             if (updateKeyField == null)
                 return null;
 
             // get update URL
             return this.GetUpdateUrl(updateKeyField.Value);
         }
-
-
-        /*********
-        ** Private models
-        *********/
-        /// <summary>Get a raw data record.</summary>
-        /// <param name="id">The mod ID to match.</param>
-        /// <param name="displayName">The mod's default display name.</param>
-        /// <param name="record">The raw mod record.</param>
-        private bool TryGetRaw(string id, out string displayName, out ModDataRecord record)
-        {
-            if (!string.IsNullOrWhiteSpace(id))
-            {
-                foreach (var entry in this.Records)
-                {
-                    displayName = entry.Key;
-                    record = entry.Value;
-
-                    // try main ID
-                    if (record.ID != null && record.ID.Equals(id, StringComparison.InvariantCultureIgnoreCase))
-                        return true;
-
-                    // try former IDs
-                    if (record.FormerIDs != null)
-                    {
-                        foreach (string part in record.FormerIDs.Split('|'))
-                        {
-                            if (part.Trim().Equals(id, StringComparison.InvariantCultureIgnoreCase))
-                                return true;
-                        }
-                    }
-                }
-            }
-
-            displayName = null;
-            record = null;
-            return false;
-        }
     }
 }