package api

import (
	"github.com/gorilla/mux"
	"gorm.io/gorm"
	"net/http"
	"sonarqube-badge/router/middlewares"
	"sonarqube-badge/router/utils"
	"sonarqube-badge/store"
)

func changePassword(w http.ResponseWriter, r *http.Request) {
	db, _, user, done := utils.VerifyUser(w, r)
	if done {
		return
	}

	password := r.FormValue("password")
	verifyPassword := r.FormValue("verify_password")

	if password != verifyPassword {
		w.WriteHeader(http.StatusBadRequest)
		w.Write([]byte("passwords do not match"))
		return
	}

	user.ChangePassword(db, password)

	refreshCookie(db, user.ID, r, w)
	w.WriteHeader(http.StatusNoContent)
}

func changeEmail(w http.ResponseWriter, r *http.Request) {
	db, _, user, done := utils.VerifyUser(w, r)
	if done {
		return
	}

	newEmail := r.FormValue("email")
	if user.ChangeEmail(db, newEmail) != nil {
		w.WriteHeader(http.StatusBadRequest)
		return
	}

	user.Email = newEmail

	refreshCookie(db, user.ID, r, w)
	w.WriteHeader(http.StatusNoContent)
}

func changeUsername(w http.ResponseWriter, r *http.Request) {
	db, _, user, done := utils.VerifyUser(w, r)
	if done {
		return
	}

	newUsername := r.FormValue("username")
	user.ChangeUsername(db, newUsername)

	refreshCookie(db, user.ID, r, w)
	w.WriteHeader(http.StatusNoContent)
}

func refreshCookie(db *gorm.DB, userId uint, r *http.Request, w http.ResponseWriter) {
	var user store.User
	db.First(&user, userId)
	cookie, err := utils.CreateJWTCookie(&user, r)
	if err != nil {
		w.WriteHeader(http.StatusInternalServerError)
		return
	}
	http.SetCookie(w, cookie)
}

func UserRouter(r *mux.Router) {
	sr := r.PathPrefix("/").Subrouter()
	sr.Use(middlewares.CheckJwtToken)
	sr.HandleFunc("/password", changePassword).Methods("POST")
	sr.HandleFunc("/email", changeEmail).Methods("POST")
	sr.HandleFunc("/username", changeUsername).Methods("POST")
}