migrated out more modules and refactored db system

internal/handler/register.go

@@ -2,6 +2,7 @@ package handler
 
 import (
 	"context"
+	"database/sql"
 	"net/http"
 	"time"
 
@@ -10,23 +11,23 @@ import (
 	"projectreshoot/internal/view/page"
 	"projectreshoot/pkg/config"
 	"projectreshoot/pkg/cookies"
-	"projectreshoot/pkg/db"
+
+	"git.haelnorr.com/h/golib/hlog"
+	"git.haelnorr.com/h/golib/jwt"
 
 	"github.com/pkg/errors"
-	"github.com/rs/zerolog"
 )
 
 func validateRegistration(
-	ctx context.Context,
-	tx *db.SafeWTX,
+	tx *sql.Tx,
 	r *http.Request,
 ) (*models.User, error) {
 	formUsername := r.FormValue("username")
 	formPassword := r.FormValue("password")
 	formConfirmPassword := r.FormValue("confirm-password")
-	unique, err := models.CheckUsernameUnique(ctx, tx, formUsername)
+	unique, err := models.CheckUsernameUnique(tx, formUsername)
 	if err != nil {
-		return nil, errors.Wrap(err, "db.CheckUsernameUnique")
+		return nil, errors.Wrap(err, "models.CheckUsernameUnique")
 	}
 	if !unique {
 		return nil, errors.New("Username is taken")
@@ -37,9 +38,9 @@ func validateRegistration(
 	if len(formPassword) > 72 {
 		return nil, errors.New("Password exceeds maximum length of 72 bytes")
 	}
-	user, err := models.CreateNewUser(ctx, tx, formUsername, formPassword)
+	user, err := models.CreateNewUser(tx, formUsername, formPassword)
 	if err != nil {
-		return nil, errors.Wrap(err, "db.CreateNewUser")
+		return nil, errors.Wrap(err, "models.CreateNewUser")
 	}
 
 	return user, nil
@@ -47,8 +48,9 @@ func validateRegistration(
 
 func RegisterRequest(
 	config *config.Config,
-	logger *zerolog.Logger,
-	conn *db.SafeConn,
+	tokenGen *jwt.TokenGenerator,
+	logger *hlog.Logger,
+	conn *sql.DB,
 ) http.Handler {
 	return http.HandlerFunc(
 		func(w http.ResponseWriter, r *http.Request) {
@@ -56,14 +58,14 @@ func RegisterRequest(
 			defer cancel()
 
 			// Start the transaction
-			tx, err := conn.Begin(ctx)
+			tx, err := conn.BeginTx(ctx, nil)
 			if err != nil {
 				logger.Warn().Err(err).Msg("Failed to set token cookies")
 				w.WriteHeader(http.StatusServiceUnavailable)
 				return
 			}
 			r.ParseForm()
-			user, err := validateRegistration(ctx, tx, r)
+			user, err := validateRegistration(tx, r)
 			if err != nil {
 				tx.Rollback()
 				if err.Error() != "Username is taken" &&
@@ -78,7 +80,7 @@ func RegisterRequest(
 			}
 
 			rememberMe := checkRememberMe(r)
-			err = cookies.SetTokenCookies(w, r, config, user, true, rememberMe)
+			err = cookies.SetTokenCookies(w, r, config, tokenGen, user, true, rememberMe)
 			if err != nil {
 				tx.Rollback()
 				w.WriteHeader(http.StatusInternalServerError)