commit

Author: mbund

admin.go

@@ -3,8 +3,10 @@ package main
 import (
 	"database/sql"
 	"fmt"
+	"io"
 	"log"
 	"net/http"
+	"os"
 	"slices"
 	"strconv"
 	"strings"
@@ -268,6 +270,36 @@ func (r *Router) adminUsers(w http.ResponseWriter, req *http.Request) {
 	}
 }
 
+const databaseFile = "backup.db"
+
+func (r *Router) adminDownloadDatabase(w http.ResponseWriter, req *http.Request) {
+	_, err := r.db.ExecContext(req.Context(), "VACUUM")
+	if err != nil {
+		log.Println("Failed to vacuum database:", err)
+		http.Error(w, "Failed to vacuum database", http.StatusInternalServerError)
+		return
+	}
+	_ = os.Remove(databaseFile)
+	_, err = r.db.ExecContext(req.Context(), "VACUUM main INTO ?", databaseFile)
+	if err != nil {
+		log.Println("Failed to backup database to file:", err)
+		http.Error(w, "Failed to backup database to file", http.StatusInternalServerError)
+		return
+	}
+	f, err := os.Open(databaseFile)
+	if err != nil {
+		log.Println("Failed to open database file:", err)
+		http.Error(w, "Failed to open database file", http.StatusInternalServerError)
+		return
+	}
+	_, err = io.Copy(w, f)
+	if err != nil {
+		log.Println("Failed to write database to writer:", err)
+		http.Error(w, "Failed to write database to writer", http.StatusInternalServerError)
+		return
+	}
+}
+
 func (r *Router) adminUserEdit(w http.ResponseWriter, req *http.Request) {
 	ny, _ := time.LoadLocation("America/New_York")
 

main.go

@@ -588,6 +588,7 @@ func main() {
 	mux.Handle("/discord/callback", router.InjectJwtMiddleware(router.EnforceJwtMiddleware(http.HandlerFunc(router.DiscordCallback))))
 	mux.Handle("/admin", router.InjectJwtMiddleware(router.EnforceJwtMiddleware(router.EnforceAdminMiddleware(http.HandlerFunc(router.admin)))))
 	mux.Handle("/admin/users", router.InjectJwtMiddleware(router.EnforceJwtMiddleware(router.EnforceAdminMiddleware(http.HandlerFunc(router.adminUsers)))))
+	mux.Handle("/admin/auth.db", router.InjectJwtMiddleware(router.EnforceJwtMiddleware(router.EnforceAdminMiddleware(http.HandlerFunc(router.adminDownloadDatabase)))))
 	mux.Handle("/admin/users/{user_id}", router.InjectJwtMiddleware(router.EnforceJwtMiddleware(router.EnforceAdminMiddleware(http.HandlerFunc(router.adminUserEdit)))))
 	mux.Handle("/admin/vote", router.InjectJwtMiddleware(router.EnforceJwtMiddleware(router.EnforceAdminMiddleware(http.HandlerFunc(router.adminVote)))))
 	mux.Handle("GET /admin/vote/new", router.InjectJwtMiddleware(router.EnforceJwtMiddleware(router.EnforceAdminMiddleware(http.HandlerFunc(router.adminVoteEdit)))))

templates/admin.html.tpl

@@ -15,6 +15,31 @@
         {{ if eq .path `/admin/vote` }}bg-gray-100{{ end }}"
       >Vote</a
     >
+    <a
+      href="/admin/auth.db"
+      target="_top"
+      class="rounded-sm px-4 py-2 hover:bg-gray-100 inline-flex justify-between
+        {{ if eq .path `/admin/download` }}bg-gray-100{{ end }}"
+      >
+        <span>Database</span>
+        <svg
+          xmlns="http://www.w3.org/2000/svg"
+          width="24"
+          height="24"
+          viewBox="0 0 24 24"
+          fill="none"
+          stroke="currentColor"
+          stroke-width="2"
+          stroke-linecap="round"
+          stroke-linejoin="round"
+          class="lucide lucide-download-icon lucide-download"
+        >
+          <path d="M12 15V3" />
+          <path d="M21 15v4a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2v-4" />
+          <path d="m7 10 5 5 5-5" />
+        </svg>
+      </a
+    >
   </div>
   <div class="flex-1 min-w-0">{{ block "admin" . }}{{ end }}</div>
 </div>