From 828912234705a693f2f57cd14b72927c7c345d63 Mon Sep 17 00:00:00 2001
From: Kieran <kieran@harkin.me>
Date: Thu, 27 Apr 2023 09:51:11 +0100
Subject: [PATCH] Maintenance mode flag

---
 VoidCat/Controllers/UploadController.cs | 8 ++++++++
 VoidCat/Model/VoidSettings.cs           | 5 +++++
 2 files changed, 13 insertions(+)

diff --git a/VoidCat/Controllers/UploadController.cs b/VoidCat/Controllers/UploadController.cs
index 252fd76..cd6b5fc 100644
--- a/VoidCat/Controllers/UploadController.cs
+++ b/VoidCat/Controllers/UploadController.cs
@@ -59,6 +59,10 @@ namespace VoidCat.Controllers
         {
             try
             {
+                if (_settings.MaintenanceMode)
+                {
+                    throw new InvalidOperationException("Site is in maintenance mode");
+                }
                 var uid = HttpContext.GetUserId();
                 var mime = Request.Headers.GetHeader("V-Content-Type");
                 var filename = Request.Headers.GetHeader("V-Filename");
@@ -150,6 +154,10 @@ namespace VoidCat.Controllers
         {
             try
             {
+                if (_settings.MaintenanceMode)
+                {
+                    throw new InvalidOperationException("Site is in maintenance mode");
+                }
                 var gid = id.FromBase58Guid();
                 var meta = await _metadata.Get<SecretFileMeta>(gid);
                 if (meta == default) return UploadResult.Error("File not found");
diff --git a/VoidCat/Model/VoidSettings.cs b/VoidCat/Model/VoidSettings.cs
index 5cdd297..d9ea86e 100644
--- a/VoidCat/Model/VoidSettings.cs
+++ b/VoidCat/Model/VoidSettings.cs
@@ -7,6 +7,11 @@ namespace VoidCat.Model
     /// </summary>
     public class VoidSettings
     {
+        /// <summary>
+        /// Maintenance flag
+        /// </summary>
+        public bool MaintenanceMode { get; init; } = false;
+        
         /// <summary>
         /// Base site url, used for redirect urls
         /// </summary>