package handlers

import (
	"encoding/json"
	"log"
	"net/http"
)

type OtpRequest struct {
	OpenID string `json:"openid"`
	Num    int    `json:"num"`
	Token  *[]OTP `json:"token"`
}
type OTP struct {
	Issuer string `json:"issuer"`
	Remark string `json:"remark"`
	Secret string `json:"secret"`
}

func (h *Handler) UpdateOrCreateOtp(w http.ResponseWriter, r *http.Request) {
	var req OtpRequest
	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
		WriteError(w, "Failed to parse request body", http.StatusBadRequest)
		return
	}

	if req.OpenID == "" {
		WriteError(w, "OpenID is required", http.StatusBadRequest)
		return
	}

	if req.Token == nil || len(*req.Token) == 0 {
		WriteError(w, "Token is required", http.StatusBadRequest)
		return
	}

	log.Printf("Saving OTP for user: %s token count:: %d", req.OpenID, len(*req.Token))

	// 插入或更新 OTP 记录
	query := `
		INSERT INTO otp (openid, token)
		VALUES ($1, $2)
		ON CONFLICT (openid) DO UPDATE SET token = EXCLUDED.token
	`

	_, err := h.DB.Exec(query, req.OpenID, req.Token)
	if err != nil {
		WriteError(w, "Failed to update or create OTP", http.StatusInternalServerError)
		return
	}

	WriteJSON(w, Response{Code: 0, Message: "Success"}, http.StatusOK)
}

func (h *Handler) GetOtp(w http.ResponseWriter, r *http.Request) {
	openid := r.URL.Query().Get("openid")
	if openid == "" {
		WriteError(w, "OpenID is required", http.StatusBadRequest)
		return
	}

	var otp OtpRequest

	err := h.DB.Get(&otp, "SELECT openid, token FROM otp WHERE openid=$1", openid)
	if err != nil {
		WriteError(w, "Failed to get OTP", http.StatusInternalServerError)
		return
	}

	WriteJSON(w, Response{Code: 0, Message: "Success", Data: otp}, http.StatusOK)
}