updated stuff

internal/handlers/register.go

@@ -0,0 +1,95 @@
+package handlers
+
+import (
+	"context"
+	"net/http"
+	"time"
+
+	"git.haelnorr.com/h/golib/hws"
+	"git.haelnorr.com/h/oslstats/internal/config"
+	"git.haelnorr.com/h/oslstats/internal/db"
+	"git.haelnorr.com/h/oslstats/internal/session"
+	"git.haelnorr.com/h/oslstats/internal/view/page"
+	"github.com/uptrace/bun"
+)
+
+func Register(
+	server *hws.Server,
+	conn *bun.DB,
+	cfg *config.Config,
+	store *session.Store,
+) http.Handler {
+	return http.HandlerFunc(
+		func(w http.ResponseWriter, r *http.Request) {
+			sessionCookie, err := r.Cookie("registration_session")
+			if err != nil {
+				http.Redirect(w, r, "/login", http.StatusSeeOther)
+				return
+			}
+			details, ok := store.GetRegistrationSession(sessionCookie.Value)
+			if !ok {
+				http.Redirect(w, r, "/login", http.StatusSeeOther)
+				return
+			}
+			ctx, cancel := context.WithTimeout(r.Context(), 10*time.Second)
+			defer cancel()
+			tx, err := conn.BeginTx(ctx, nil)
+			if err != nil {
+				throwInternalServiceError(server, w, r, "Database transaction failed", err)
+				return
+			}
+			defer tx.Rollback()
+			method := r.Method
+			if method == "GET" {
+				unique, err := db.IsUsernameUnique(ctx, tx, details.DiscordUser.Username)
+				if err != nil {
+					throwInternalServiceError(server, w, r, "Database query failed", err)
+					return
+				}
+				tx.Commit()
+				page.Register(details.DiscordUser.Username, unique).Render(r.Context(), w)
+				return
+			}
+			if method == "POST" {
+				// TODO: register the user
+
+				// get the form data
+				//
+				return
+			}
+		},
+	)
+}
+
+func IsUsernameUnique(
+	server *hws.Server,
+	conn *bun.DB,
+	cfg *config.Config,
+	store *session.Store,
+) http.Handler {
+	return http.HandlerFunc(
+		func(w http.ResponseWriter, r *http.Request) {
+			username := r.FormValue("username")
+			// check if its unique
+			ctx, cancel := context.WithTimeout(r.Context(), 10*time.Second)
+			defer cancel()
+			tx, err := conn.BeginTx(ctx, nil)
+			if err != nil {
+				throwInternalServiceError(server, w, r, "Database transaction failed", err)
+				return
+			}
+			defer tx.Rollback()
+			unique, err := db.IsUsernameUnique(ctx, tx, username)
+			if err != nil {
+				throwInternalServiceError(server, w, r, "Database query failed", err)
+				return
+			}
+			tx.Commit()
+			if !unique {
+				w.WriteHeader(http.StatusConflict)
+			} else {
+				w.WriteHeader(http.StatusOK)
+			}
+		},
+	)
+}