package api

import (
	"encoding/json"
	"github.com/gorilla/mux"
	"net/http"
	"sonarqube-badge/router/middlewares"
	"sonarqube-badge/router/utils"
	"sonarqube-badge/security/aes"
	"sonarqube-badge/store"
	"strconv"
)

type ProjectPayload struct {
	Name  string `json:"name"`
	Token string `json:"token"`
}

func setProject(w http.ResponseWriter, r *http.Request) {
	db, cfg, _, isError := utils.VerifyUser(w, r)
	if isError {
		return
	}

	idString := mux.Vars(r)["id"]
	id, err := strconv.Atoi(idString)
	payload := ProjectPayload{}
	err = json.NewDecoder(r.Body).Decode(&payload)
	if err != nil {
		w.WriteHeader(http.StatusBadRequest)
		return
	}

	project := store.Project{}
	db.Where("id = ?", id).First(&project)
	if project.ID == 0 {
		w.WriteHeader(http.StatusNotFound)
	}

	project.ProjectName = payload.Name
	if project.Token != payload.Token {
		project.Token = aes.EncryptAES(cfg.Secret, payload.Token)
	}
	db.Save(&project)
}

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

	project := user.CreateProject(db)

	w.WriteHeader(http.StatusOK)
	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(project)
}

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

	projectIdString := mux.Vars(r)["id"]
	projectId, err := strconv.Atoi(projectIdString)
	if err != nil {
		w.WriteHeader(http.StatusBadRequest)
		return
	}

	project := store.Project{}
	db.Where("id = ? AND owner_id = ?", projectId, user.ID).First(&project)
	if project.ID == 0 {
		w.WriteHeader(http.StatusNotFound)
		return
	}

	db.Delete(&project)
	w.WriteHeader(http.StatusOK)
}

func ProjectRouter(r *mux.Router) {
	sr := r.PathPrefix("").Subrouter()
	sr.Use(middlewares.CheckJwtToken)
	sr.HandleFunc("", createProject).Methods(http.MethodPost)
	sr.HandleFunc("/{id}", setProject).Methods(http.MethodPost)
	sr.HandleFunc("/{id}", deleteProject).Methods(http.MethodDelete)
}