package telegram

import (
	"database/sql"
	"embed"
	"encoding/json"
	"log/slog"
	"net/http"

	"prism/service/credentials"
	"prism/service/util"

	"github.com/go-chi/chi/v5"
)

//go:embed templates/*.html
var templates embed.FS

func GetTemplates() embed.FS {
	return templates
}

type authHandler struct {
	db     *sql.DB
	apiKey string
	logger *slog.Logger
}

type telegramAuthRequest struct {
	BotToken string `json:"bot_token"`
	ChatID   string `json:"chat_id"`
}

func (h *authHandler) handleAuth(w http.ResponseWriter, r *http.Request) {
	var req telegramAuthRequest
	if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
		util.JSONError(w, "Invalid request body", http.StatusBadRequest)
		return
	}

	if req.BotToken == "" || req.ChatID == "" {
		util.JSONError(w, "bot_token and chat_id are required", http.StatusBadRequest)
		return
	}

	credStore, err := credentials.NewStore(h.db, h.apiKey)
	if err != nil {
		util.LogAndError(w, h.logger, "Failed to initialize credentials store", http.StatusInternalServerError, err)
		return
	}

	creds := &credentials.TelegramCredentials{
		BotToken: req.BotToken,
		ChatID:   req.ChatID,
	}

	if err := credStore.SaveTelegram(creds); err != nil {
		util.LogAndError(w, h.logger, "Failed to save Telegram credentials", http.StatusInternalServerError, err)
		return
	}

	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(map[string]string{"status": "success"})
}

func (h *authHandler) handleDelete(w http.ResponseWriter, r *http.Request) {
	credStore, err := credentials.NewStore(h.db, h.apiKey)
	if err != nil {
		util.LogAndError(w, h.logger, "Failed to initialize credentials store", http.StatusInternalServerError, err)
		return
	}

	if err := credStore.DeleteIntegration(credentials.IntegrationTelegram); err != nil {
		util.LogAndError(w, h.logger, "Failed to delete integration", http.StatusInternalServerError, err)
		return
	}

	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(map[string]string{"status": "deleted"})
}

func RegisterRoutes(router *chi.Mux, handlers *Handlers, auth func(http.Handler) http.Handler, db *sql.DB, apiKey string, logger *slog.Logger) {
	if handlers == nil {
		return
	}

	handlers.SetDB(db, apiKey)

	authH := &authHandler{db: db, apiKey: apiKey, logger: logger}

	router.With(auth).Get("/fragment/telegram", handlers.HandleFragment)
	router.With(auth).Post("/api/telegram/auth", authH.handleAuth)
	router.With(auth).Delete("/api/telegram/auth", authH.handleDelete)
}