src/SMAPI.Web/Framework/Caching/Wiki/WikiCacheMongoRepository.cs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

using System;
using System.Collections.Generic;
using System.Linq;
using System.Linq.Expressions;
using MongoDB.Driver;
using StardewModdingAPI.Toolkit.Framework.Clients.Wiki;

namespace StardewModdingAPI.Web.Framework.Caching.Wiki
{
    /// <summary>Manages cached wiki data in MongoDB.</summary>
    internal class WikiCacheMongoRepository : BaseCacheRepository, IWikiCacheRepository
    {
        /*********
        ** Fields
        *********/
        /// <summary>The collection for wiki metadata.</summary>
        private readonly IMongoCollection<CachedWikiMetadata> Metadata;

        /// <summary>The collection for wiki mod data.</summary>
        private readonly IMongoCollection<CachedWikiMod> Mods;


        /*********
        ** Public methods
        *********/
        /// <summary>Construct an instance.</summary>
        /// <param name="database">The authenticated MongoDB database.</param>
        public WikiCacheMongoRepository(IMongoDatabase database)
        {
            // get collections
            this.Metadata = database.GetCollection<CachedWikiMetadata>("wiki-metadata");
            this.Mods = database.GetCollection<CachedWikiMod>("wiki-mods");

            // add indexes if needed
            this.Mods.Indexes.CreateOne(new CreateIndexModel<CachedWikiMod>(Builders<CachedWikiMod>.IndexKeys.Ascending(p => p.ID)));
        }

        /// <summary>Get the cached wiki metadata.</summary>
        /// <param name="metadata">The fetched metadata.</param>
        public bool TryGetWikiMetadata(out CachedWikiMetadata metadata)
        {
            metadata = this.Metadata.Find("{}").FirstOrDefault();
            return metadata != null;
        }

        /// <summary>Get the cached wiki mods.</summary>
        /// <param name="filter">A filter to apply, if any.</param>
        public IEnumerable<CachedWikiMod> GetWikiMods(Expression<Func<CachedWikiMod, bool>> filter = null)
        {
            return filter != null
                ? this.Mods.Find(filter).ToList()
                : this.Mods.Find("{}").ToList();
        }

        /// <summary>Save data fetched from the wiki compatibility list.</summary>
        /// <param name="stableVersion">The current stable Stardew Valley version.</param>
        /// <param name="betaVersion">The current beta Stardew Valley version.</param>
        /// <param name="mods">The mod data.</param>
        /// <param name="cachedMetadata">The stored metadata record.</param>
        /// <param name="cachedMods">The stored mod records.</param>
        public void SaveWikiData(string stableVersion, string betaVersion, IEnumerable<WikiModEntry> mods, out CachedWikiMetadata cachedMetadata, out CachedWikiMod[] cachedMods)
        {
            cachedMetadata = new CachedWikiMetadata(stableVersion, betaVersion);
            cachedMods = mods.Select(mod => new CachedWikiMod(mod)).ToArray();

            this.Mods.DeleteMany("{}");
            this.Mods.InsertMany(cachedMods);

            this.Metadata.DeleteMany("{}");
            this.Metadata.InsertOne(cachedMetadata);
        }
    }
}