redirect on /admin, push tweaks

.gitignore

@@ -3,3 +3,5 @@ puerta.db
 local.db
 config.joao.yaml
 test.db
+puerta.db-shm
+puerta.db-wal

internal/auth/middleware.go

@@ -82,6 +82,17 @@ func RequireAuthOrRedirect(handler httprouter.Handle, target string) httprouter.
 	})
 }
 
+func RequireAdminOrRedirect(handler httprouter.Handle, target string) httprouter.Handle {
+	return withUser(func(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
+		if req.Context().Value(constants.ContextUser) == nil {
+			http.Redirect(w, req, target, http.StatusTemporaryRedirect)
+			return
+		}
+
+		handler(w, req, ps)
+	})
+}
+
 func RegisterSecondFactor() httprouter.Handle {
 	return RequireAuth(func(w http.ResponseWriter, req *http.Request, ps httprouter.Params) {
 		u := user.FromContext(req)

internal/server/server.go

@@ -256,7 +256,7 @@ func Initialize(config *Config) (http.Handler, error) {
 		w.WriteHeader(200)
 		w.Write(buf)
 	})
-	router.GET("/admin", auth.RequireAdmin(renderTemplate(bytes.ReplaceAll(adminTemplate, []byte("$PUSH_KEY$"), []byte(config.WebPush.Key.Public)))))
+	router.GET("/admin", auth.RequireAdminOrRedirect(renderTemplate(bytes.ReplaceAll(adminTemplate, []byte("$PUSH_KEY$"), []byte(config.WebPush.Key.Public))), "/login?next=/admin"))
 
 	// regular api
 	router.POST("/api/login", auth.LoginHandler)

internal/server/static/admin.js

@@ -296,11 +296,29 @@ window.addEventListener("load", async function() {
     scope: "/"
   })
 
-  const sub = await reg.pushManager.getSubscription()
-  console.log(`registered SW, push sub: ${sub}`, reg)
+  try {
+    sub = await reg.pushManager.getSubscription()
+    if (sub) {
+      pnb.classList.add("subscribed")
+      pnb.innerHTML = "🔕"
+    } else {
+      pnb.classList.remove("subscribed")
+      pnb.innerHTML = "🔔"
+    }
+  } catch(err) {
+    console.error("Could not get pushmanager subscription", err)
+  }
 
   const pnb = document.querySelector("#push-notifications")
 
+  pnb.addEventListener('click', async evt =>{
+    let sub
+    try {
+      sub = await reg.pushManager.getSubscription()
+    } catch(err) {
+      console.error("Could not get pushmanager subscription", err)
+    }
+
     if (sub) {
       pnb.classList.add("subscribed")
       pnb.innerHTML = "🔕"
@@ -309,7 +327,6 @@ window.addEventListener("load", async function() {
       pnb.innerHTML = "🔔"
     }
 
-  pnb.addEventListener('click', async evt =>{
     if (!pnb.classList.contains("subscribed")) {
       if (await createPushSubscription()) {
         pnb.classList.add("subscribed")

internal/server/static/login.js

@@ -34,7 +34,17 @@ function submit(evt){
   clearStatus()
 
   Login().then(() => {
-    window.location = "/";
+    let next = "/"
+    try {
+      const follow = window.location.search.replace("?next=", "")
+      if (follow != "") {
+        next = follow
+      }
+    } catch (err) {
+      console.error(`Could not find next path to follow: ${err}`)
+    }
+
+    window.location = next;
   }).catch((err) => {
     form.classList.add("failed")
     document.querySelector('.error').innerText = err