koin_ping_0.2.0/backend-go/internal/handlers/notification_config.go

package handlers

import (
	"encoding/json"
	"log"
	"net/http"
	"regexp"
	"strings"

	"github.com/kjannette/koin-ping/backend-go/internal/domain"
	"github.com/kjannette/koin-ping/backend-go/internal/middleware"
	"github.com/kjannette/koin-ping/backend-go/internal/models"
	"github.com/kjannette/koin-ping/backend-go/internal/notifications"
)

var emailRe = regexp.MustCompile(`^[^\s@]+@[^\s@]+\.[^\s@]+$`)

type NotificationConfigHandler struct {
	configs *models.NotificationConfigModel
}

func NewNotificationConfigHandler(configs *models.NotificationConfigModel) *NotificationConfigHandler {
	return &NotificationConfigHandler{configs: configs}
}

func (h *NotificationConfigHandler) GetConfig(w http.ResponseWriter, r *http.Request) {
	userID := middleware.GetUserID(r.Context())
	log.Printf("User %s getting notification config", userID)

	cfg, err := h.configs.GetConfig(r.Context(), userID)
	if err != nil {
		log.Printf("Error getting notification config: %v", err)
		writeError(w, http.StatusInternalServerError, "INTERNAL_ERROR", "Failed to get notification config")
		return
	}

	if cfg == nil {
		writeJSON(w, http.StatusOK, domain.NotificationConfig{
			UserID:              userID,
			NotificationEnabled: true,
		})
		return
	}

	log.Println("Config found")
	writeJSON(w, http.StatusOK, cfg)
}

func (h *NotificationConfigHandler) UpdateConfig(w http.ResponseWriter, r *http.Request) {
	userID := middleware.GetUserID(r.Context())

	var body struct {
		DiscordWebhookURL   *string `json:"discord_webhook_url"`
		SlackWebhookURL     *string `json:"slack_webhook_url"`
		Email               *string `json:"email"`
		NotificationEnabled *bool   `json:"notification_enabled"`
	}
	if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
		log.Printf("Failed to decode notification config request body: %v", err)
		writeError(w, http.StatusBadRequest, "VALIDATION_ERROR", "Invalid request body")
		return
	}

	log.Printf("User %s updating notification config", userID)

	if body.DiscordWebhookURL == nil && body.SlackWebhookURL == nil &&
		body.Email == nil && body.NotificationEnabled == nil {
		writeError(w, http.StatusBadRequest, "VALIDATION_ERROR",
			"At least one configuration field must be provided")
		return
	}

	if body.DiscordWebhookURL != nil && *body.DiscordWebhookURL != "" &&
		!strings.HasPrefix(*body.DiscordWebhookURL, "https://discord.com/api/webhooks/") {
		writeError(w, http.StatusBadRequest, "VALIDATION_ERROR",
			"Invalid Discord webhook URL format")
		return
	}

	if body.SlackWebhookURL != nil && *body.SlackWebhookURL != "" &&
		!strings.HasPrefix(*body.SlackWebhookURL, "https://hooks.slack.com/") {
		writeError(w, http.StatusBadRequest, "VALIDATION_ERROR",
			"Invalid Slack webhook URL format")
		return
	}

	if body.Email != nil && *body.Email != "" && !emailRe.MatchString(*body.Email) {
		writeError(w, http.StatusBadRequest, "VALIDATION_ERROR",
			"Invalid email address format")
		return
	}

	enabled := true
	if body.NotificationEnabled != nil {
		enabled = *body.NotificationEnabled
	}

	cfg := domain.NotificationConfig{
		DiscordWebhookURL:   body.DiscordWebhookURL,
		SlackWebhookURL:     body.SlackWebhookURL,
		Email:               body.Email,
		NotificationEnabled: enabled,
	}

	updated, err := h.configs.UpsertConfig(r.Context(), userID, cfg)
	if err != nil {
		log.Printf("Error updating notification config: %v", err)
		writeError(w, http.StatusInternalServerError, "INTERNAL_ERROR",
			"Failed to update notification configuration")
		return
	}

	log.Println("Notification config updated")
	writeJSON(w, http.StatusOK, updated)
}

func (h *NotificationConfigHandler) DeleteConfig(w http.ResponseWriter, r *http.Request) {
	userID := middleware.GetUserID(r.Context())
	log.Printf("User %s deleting notification config", userID)

	deleted, err := h.configs.Remove(r.Context(), userID)
	if err != nil {
		log.Printf("Error deleting notification config: %v", err)
		writeError(w, http.StatusInternalServerError, "INTERNAL_ERROR",
			"Failed to delete notification configuration")
		return
	}

	if !deleted {
		writeError(w, http.StatusNotFound, "NOT_FOUND", "No notification configuration found")
		return
	}

	log.Println("Notification config deleted")
	w.WriteHeader(http.StatusNoContent)
}

func (h *NotificationConfigHandler) TestWebhook(w http.ResponseWriter, r *http.Request) {
	_ = middleware.GetUserID(r.Context())

	var body struct {
		Type string `json:"type"`
		URL  string `json:"url"`
	}
	if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
		writeError(w, http.StatusBadRequest, "VALIDATION_ERROR", "Invalid request body")
		return
	}

	if body.URL == "" {
		writeError(w, http.StatusBadRequest, "VALIDATION_ERROR", "URL is required")
		return
	}

	var ok bool
	var err error

	switch body.Type {
	case "slack":
		if !strings.HasPrefix(body.URL, "https://hooks.slack.com/") {
			writeError(w, http.StatusBadRequest, "VALIDATION_ERROR", "Invalid Slack webhook URL")
			return
		}
		ok, err = notifications.TestSlackWebhook(body.URL)
	case "discord":
		if !strings.HasPrefix(body.URL, "https://discord.com/api/webhooks/") {
			writeError(w, http.StatusBadRequest, "VALIDATION_ERROR", "Invalid Discord webhook URL")
			return
		}
		ok, err = notifications.TestDiscordWebhook(body.URL)
	default:
		writeError(w, http.StatusBadRequest, "VALIDATION_ERROR", "Type must be 'slack' or 'discord'")
		return
	}

	if err != nil {
		log.Printf("Webhook test failed: %v", err)
		writeJSON(w, http.StatusOK, map[string]interface{}{"success": false, "error": err.Error()})
		return
	}

	writeJSON(w, http.StatusOK, map[string]interface{}{"success": ok})
}