From 082f285bc7ce156ad0750bb48d46ed65a2e4aedb Mon Sep 17 00:00:00 2001
From: Jesse Plamondon-Willard <Pathoschild@users.noreply.github.com>
Date: Sun, 22 Dec 2019 00:44:13 -0500
Subject: streamline local environments, update technical docs & privacy page

---
 src/SMAPI.Web/Startup.cs | 25 ++++++++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

(limited to 'src/SMAPI.Web/Startup.cs')

diff --git a/src/SMAPI.Web/Startup.cs b/src/SMAPI.Web/Startup.cs
index 07ee0c9e..338cd2d5 100644
--- a/src/SMAPI.Web/Startup.cs
+++ b/src/SMAPI.Web/Startup.cs
@@ -1,5 +1,7 @@
+using System;
 using System.Collections.Generic;
 using Hangfire;
+using Hangfire.MemoryStorage;
 using Hangfire.Mongo;
 using Microsoft.AspNetCore.Builder;
 using Microsoft.AspNetCore.Hosting;
@@ -8,6 +10,7 @@ using Microsoft.AspNetCore.Routing;
 using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Options;
+using Mongo2Go;
 using MongoDB.Bson.Serialization;
 using MongoDB.Driver;
 using Newtonsoft.Json;
@@ -89,10 +92,20 @@ namespace StardewModdingAPI.Web
             }
 
             // init MongoDB
+            services.AddSingleton<MongoDbRunner>(serv => !mongoConfig.IsConfigured()
+                ? MongoDbRunner.Start()
+                : throw new InvalidOperationException("The MongoDB connection is configured, so the local development version should not be used.")
+            );
             services.AddSingleton<IMongoDatabase>(serv =>
             {
+                // get connection string
+                string connectionString = mongoConfig.IsConfigured()
+                    ? mongoConfig.GetConnectionString()
+                    : serv.GetRequiredService<MongoDbRunner>().ConnectionString;
+
+                // get client
                 BsonSerializer.RegisterSerializer(new UtcDateTimeOffsetSerializer());
-                return new MongoClient(mongoConfig.GetConnectionString()).GetDatabase(mongoConfig.Database);
+                return new MongoClient(connectionString).GetDatabase(mongoConfig.Database);
             });
             services.AddSingleton<IModCacheRepository>(serv => new ModCacheRepository(serv.GetRequiredService<IMongoDatabase>()));
             services.AddSingleton<IWikiCacheRepository>(serv => new WikiCacheRepository(serv.GetRequiredService<IMongoDatabase>()));
@@ -104,12 +117,18 @@ namespace StardewModdingAPI.Web
                     config
                         .SetDataCompatibilityLevel(CompatibilityLevel.Version_170)
                         .UseSimpleAssemblyNameTypeSerializer()
-                        .UseRecommendedSerializerSettings()
-                        .UseMongoStorage(mongoConfig.GetConnectionString(), $"{mongoConfig.Database}-hangfire", new MongoStorageOptions
+                        .UseRecommendedSerializerSettings();
+
+                    if (mongoConfig.IsConfigured())
+                    {
+                        config.UseMongoStorage(mongoConfig.GetConnectionString(), $"{mongoConfig.Database}-hangfire", new MongoStorageOptions
                         {
                             MigrationOptions = new MongoMigrationOptions(MongoMigrationStrategy.Drop),
                             CheckConnection = false // error on startup takes down entire process
                         });
+                    }
+                    else
+                        config.UseMemoryStorage();
                 });
 
             // init API clients
-- 
cgit