From c6762726041906bcc5bcce9f63cf99f6a04180b1 Mon Sep 17 00:00:00 2001
From: John Wong <johnw@wonghome.net>
Date: Sat, 8 Apr 2023 22:23:36 +0800
Subject: [PATCH] login_only

---
 src/invidious/config.cr            | 1 +
 src/invidious/routes/before_all.cr | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/src/invidious/config.cr b/src/invidious/config.cr
index 9fc58409..772bc5a3 100644
--- a/src/invidious/config.cr
+++ b/src/invidious/config.cr
@@ -84,6 +84,7 @@ class Config
 
   # Used to tell Invidious it is behind a proxy, so links to resources should be https://
   property https_only : Bool?
+  property login_only : Bool?
   # HMAC signing key for CSRF tokens and verifying pubsub subscriptions
   property hmac_key : String?
   # Domain to be used for links to resources on the site where an absolute URL is required
diff --git a/src/invidious/routes/before_all.cr b/src/invidious/routes/before_all.cr
index 8e2a253f..24582bd5 100644
--- a/src/invidious/routes/before_all.cr
+++ b/src/invidious/routes/before_all.cr
@@ -148,5 +148,11 @@ module Invidious::Routes::BeforeAll
     end
 
     env.set "current_page", URI.encode_www_form(current_page)
+
+    unregistered_path_whitelist = {"/", "/login", "/licenses", "/privacy"}
+    if !env.get?("user") && !unregistered_path_whitelist.includes?(env.request.path) && CONFIG.login_only
+      env.response.headers["Location"] = "/login"
+      haltf env, status_code: 302
+    end
   end
 end