trevstack/server/internal/handlers/user/v1/auth.go

package user

import (
	"context"
	"database/sql"
	"errors"
	"net/http"
	"strconv"
	"time"

	_ "crypto/sha256" // Crypto

	"connectrpc.com/connect"
	"github.com/golang-jwt/jwt/v5"
	userv1 "github.com/spotdemo4/trevstack/server/internal/connect/user/v1"
	"github.com/spotdemo4/trevstack/server/internal/connect/user/v1/userv1connect"
	"github.com/spotdemo4/trevstack/server/internal/interceptors"
	"github.com/spotdemo4/trevstack/server/internal/sqlc"
	"golang.org/x/crypto/bcrypt"
)

type AuthHandler struct {
	db  *sqlc.Queries
	key []byte
}

func (h *AuthHandler) Login(ctx context.Context, req *connect.Request[userv1.LoginRequest]) (*connect.Response[userv1.LoginResponse], error) {
	// Get user
	user, err := h.db.GetUserbyUsername(ctx, req.Msg.Username)
	if err != nil {
		if errors.Is(err, sql.ErrNoRows) {
			return nil, connect.NewError(connect.CodePermissionDenied, err)
		}

		return nil, connect.NewError(connect.CodeInternal, err)
	}

	// Check password
	if err := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(req.Msg.Password)); err != nil {
		return nil, connect.NewError(connect.CodePermissionDenied, errors.New("invalid username or password"))
	}

	// Generate JWT
	t := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.RegisteredClaims{
		Issuer:  "trevstack",
		Subject: strconv.FormatInt(user.ID, 10),
		IssuedAt: &jwt.NumericDate{
			Time: time.Now(),
		},
		ExpiresAt: &jwt.NumericDate{
			Time: time.Now().Add(time.Hour * 24),
		},
	})
	ss, err := t.SignedString(h.key)
	if err != nil {
		return nil, connect.NewError(connect.CodeInternal, err)
	}

	// Create cookie
	cookie := http.Cookie{
		Name:     "token",
		Value:    ss,
		Path:     "/",
		MaxAge:   86400,
		HttpOnly: true,
		Secure:   true,
		SameSite: http.SameSiteStrictMode,
	}

	res := connect.NewResponse(&userv1.LoginResponse{
		Token: ss,
	})
	res.Header().Set("Set-Cookie", cookie.String())
	return res, nil
}

func (h *AuthHandler) SignUp(ctx context.Context, req *connect.Request[userv1.SignUpRequest]) (*connect.Response[userv1.SignUpResponse], error) {
	// Get user
	_, err := h.db.GetUserbyUsername(ctx, req.Msg.Username)
	if err != nil {
		if !errors.Is(err, sql.ErrNoRows) {
			return nil, connect.NewError(connect.CodeInternal, err)
		}
	} else {
		return nil, connect.NewError(connect.CodeAlreadyExists, err)
	}

	// Check if confirmation passwords match
	if req.Msg.Password != req.Msg.ConfirmPassword {
		return nil, connect.NewError(connect.CodeInvalidArgument, errors.New("passwords do not match"))
	}

	// Hash password
	hash, err := bcrypt.GenerateFromPassword([]byte(req.Msg.Password), bcrypt.DefaultCost)
	if err != nil {
		return nil, connect.NewError(connect.CodeInternal, err)
	}

	// Create user
	_, err = h.db.InsertUser(ctx, sqlc.InsertUserParams{
		Username: req.Msg.Username,
		Password: string(hash),
	})
	if err != nil {
		return nil, connect.NewError(connect.CodeInternal, err)
	}

	res := connect.NewResponse(&userv1.SignUpResponse{})
	return res, nil
}

func (h *AuthHandler) Logout(_ context.Context, _ *connect.Request[userv1.LogoutRequest]) (*connect.Response[userv1.LogoutResponse], error) {
	// Clear cookie
	cookie := http.Cookie{
		Name:     "token",
		Value:    "",
		Path:     "/",
		MaxAge:   -1,
		HttpOnly: true,
		Secure:   true,
		SameSite: http.SameSiteStrictMode,
	}

	res := connect.NewResponse(&userv1.LogoutResponse{})
	res.Header().Set("Set-Cookie", cookie.String())
	return res, nil
}

// func (h *AuthHandler) GetPasskeyIDs(_ context.Context, req *connect.Request[userv1.GetPasskeyIDsRequest]) (*connect.Response[userv1.GetPasskeyIDsResponse], error) {
// 	// Get user
// 	user := models.User{}
// 	if err := h.db.Preload("Passkeys").First(&user, "username = ?", req.Msg.Username).Error; err != nil {
// 		return nil, connect.NewError(connect.CodeNotFound, err)
// 	}

// 	// Get IDs
// 	ids := []string{}
// 	for _, passkey := range user.Passkeys {
// 		ids = append(ids, passkey.ID)
// 	}

// 	return connect.NewResponse(&userv1.GetPasskeyIDsResponse{
// 		PasskeyIds: ids,
// 	}), nil
// }

// func (h *AuthHandler) PasskeyLogin(_ context.Context, req *connect.Request[userv1.PasskeyLoginRequest]) (*connect.Response[userv1.PasskeyLoginResponse], error) {
// 	// Get passkey
// 	passkey := models.Passkey{}
// 	if err := h.db.First(&passkey, "id = ?", req.Msg.Id).Error; err != nil {
// 		return nil, connect.NewError(connect.CodeNotFound, err)
// 	}

// 	// create a verifier from a trusted private key
// 	var verifier cose.Verifier
// 	var err error
// 	switch req.Msg.Algorithm {
// 	case -7:
// 		verifier, err = cose.NewVerifier(cose.AlgorithmES256, passkey.PublicKey)

// 	case -257:
// 		verifier, err = cose.NewVerifier(cose.AlgorithmRS256, passkey.PublicKey)

// 	default:
// 		return nil, connect.NewError(connect.CodeInternal, errors.New("decode algorithm not implemented"))
// 	}
// 	if err != nil {
// 		return nil, connect.NewError(connect.CodeInternal, err)
// 	}

// 	// create a sign message from a raw signature payload
// 	var msg cose.Sign1Message
// 	if err = msg.UnmarshalCBOR(req.Msg.Signature); err != nil {
// 		return nil, connect.NewError(connect.CodeInternal, err)
// 	}

// 	// Validate passkey
// 	err = msg.Verify(nil, verifier)
// 	if err != nil {
// 		return nil, connect.NewError(connect.CodeUnauthenticated, err)
// 	}

// 	// Generate JWT
// 	t := jwt.NewWithClaims(jwt.SigningMethodHS256, jwt.RegisteredClaims{
// 		Issuer:  "trevstack",
// 		Subject: strconv.FormatUint(uint64(passkey.UserID), 10),
// 		IssuedAt: &jwt.NumericDate{
// 			Time: time.Now(),
// 		},
// 		ExpiresAt: &jwt.NumericDate{
// 			Time: time.Now().Add(time.Hour * 24),
// 		},
// 	})
// 	ss, err := t.SignedString(h.key)
// 	if err != nil {
// 		return nil, connect.NewError(connect.CodeInternal, err)
// 	}

// 	// Create cookie
// 	cookie := http.Cookie{
// 		Name:     "token",
// 		Value:    ss,
// 		Path:     "/",
// 		MaxAge:   86400,
// 		HttpOnly: true,
// 		Secure:   true,
// 		SameSite: http.SameSiteStrictMode,
// 	}

// 	res := connect.NewResponse(&userv1.PasskeyLoginResponse{
// 		Token: ss,
// 	})
// 	res.Header().Set("Set-Cookie", cookie.String())
// 	return res, nil
// }

func NewAuthHandler(db *sqlc.Queries, key string) (string, http.Handler) {
	interceptors := connect.WithInterceptors(interceptors.NewRateLimitInterceptor(key))

	return userv1connect.NewAuthServiceHandler(
		&AuthHandler{
			db:  db,
			key: []byte(key),
		},
		interceptors,
	)
}