129 lines
3.8 KiB
Go
129 lines
3.8 KiB
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"
|
|
)
|
|
|
|
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"`
|
|
TelegramChatID *string `json:"telegram_chat_id"`
|
|
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.TelegramChatID == 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.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,
|
|
TelegramChatID: body.TelegramChatID,
|
|
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)
|
|
}
|