Login redirect to referer on logged-in user

src/invidious/helpers/utils.cr

@@ -259,7 +259,7 @@ def get_referer(env, fallback = "/", unroll = true)
   end
 
   referer = referer.request_target
-  referer = "/" + referer.gsub(/[^\/?@&%=\-_.0-9a-zA-Z]/, "").lstrip("/\\")
+  referer = "/" + referer.gsub(/[^\/?@&%=\-_.:,0-9a-zA-Z]/, "").lstrip("/\\")
 
   if referer == env.request.path
     referer = fallback

src/invidious/routes/login.cr

@@ -6,14 +6,14 @@ module Invidious::Routes::Login
 
     user = env.get? "user"
 
-    return env.redirect "/feed/subscriptions" if user
+    referer = get_referer(env, "/feed/subscriptions")
+
+    return env.redirect referer if user
 
     if !CONFIG.login_enabled
       return error_template(400, "Login has been disabled by administrator.")
     end
 
-    referer = get_referer(env, "/feed/subscriptions")
-
     email = nil
     password = nil
     captcha = nil