Go Agent barebone notification tests

agent/go.mod

@@ -0,0 +1,5 @@
+module github.com/crocofied/CoreControl
+
+go 1.24.3
+
+require github.com/lib/pq v1.10.9

agent/go.sum

@@ -0,0 +1,2 @@
+github.com/lib/pq v1.10.9 h1:YXG7RB+JIjhP29X+OtkiDnYaXQwpS4JEWq7dtCCRUEw=
+github.com/lib/pq v1.10.9/go.mod h1:AlVN5x4E4T544tWzH6hKfbfQvm3HdbOxrmggDNAPY9o=

agent/main.go

@@ -0,0 +1,183 @@
+package main
+
+import (
+	"database/sql"
+	"encoding/json"
+	"fmt"
+	"log"
+	"os"
+	"time"
+
+	_ "github.com/lib/pq"
+)
+
+var db *sql.DB
+
+type NotificationType string
+
+const (
+	Telegram NotificationType = "TELEGRAM"
+	Ntfy     NotificationType = "NTFY"
+	SMTP     NotificationType = "SMTP"
+)
+
+type NotificationTest struct {
+	ID                     int
+	NotificationProviderID int
+	Sent                   bool
+	Success                sql.NullBool
+	ProviderType           string
+	ProviderConfig         json.RawMessage
+}
+
+func main() {
+	var err error
+	db, err = sql.Open("postgres", os.Getenv("DATABASE_URL"))
+	if err != nil {
+		log.Fatal("Failed to connect to database:", err)
+	}
+	defer db.Close()
+
+	if err = db.Ping(); err != nil {
+		log.Fatal("Failed to ping database:", err)
+	}
+
+	ticker := time.NewTicker(10 * time.Second)
+	defer ticker.Stop()
+
+	for range ticker.C {
+		processPendingTests()
+	}
+}
+
+func processPendingTests() {
+	log.Println("Checking for pending notification tests...")
+
+	tests, err := getPendingTests()
+	if err != nil {
+		log.Println("Error fetching tests:", err)
+		return
+	}
+
+	for _, test := range tests {
+		providerType := NotificationType(test.ProviderType)
+		err := SendNotification(providerType, test.ProviderConfig, test)
+
+		if updateErr := updateTestStatus(test.ID, err == nil); updateErr != nil {
+			log.Printf("Error updating test status: %v", updateErr)
+		}
+
+		if err != nil {
+			log.Printf("Failed to send test %d: %v", test.ID, err)
+			continue
+		}
+
+		log.Printf("Successfully sent test %d", test.ID)
+	}
+}
+
+func SendNotification(providerType NotificationType, config json.RawMessage, test NotificationTest) error {
+	switch providerType {
+	case Telegram:
+		return sendTelegramNotification(config, test)
+	case Ntfy:
+		return sendNtfyNotification(config, test)
+	case SMTP:
+		return sendSMTPNotification(config, test)
+	default:
+		return fmt.Errorf("unknown provider type: %s", providerType)
+	}
+}
+
+func sendTelegramNotification(config json.RawMessage, test NotificationTest) error {
+	var cfg struct {
+		BotToken string `json:"botToken"`
+		ChatID   string `json:"chatID"`
+	}
+
+	if err := json.Unmarshal(config, &cfg); err != nil {
+		return fmt.Errorf("invalid Telegram config: %w", err)
+	}
+
+	log.Printf("Sending Telegram test to chat %s", cfg.ChatID)
+	return nil
+}
+
+func sendNtfyNotification(config json.RawMessage, test NotificationTest) error {
+	var cfg struct {
+		Topic  string `json:"topic"`
+		Server string `json:"server"`
+	}
+
+	if err := json.Unmarshal(config, &cfg); err != nil {
+		return fmt.Errorf("invalid NTFY config: %w", err)
+	}
+
+	log.Printf("Sending NTFY test to topic %s on %s", cfg.Topic, cfg.Server)
+	return nil
+}
+
+func sendSMTPNotification(config json.RawMessage, test NotificationTest) error {
+	var cfg struct {
+		Server   string `json:"server"`
+		Port     int    `json:"port"`
+		Username string `json:"username"`
+		To       string `json:"to"`
+	}
+
+	if err := json.Unmarshal(config, &cfg); err != nil {
+		return fmt.Errorf("invalid SMTP config: %w", err)
+	}
+
+	log.Printf("Sending SMTP test to %s via %s:%d", cfg.To, cfg.Server, cfg.Port)
+	return nil
+}
+
+func getPendingTests() ([]NotificationTest, error) {
+	query := `
+        SELECT t.id, 
+               "t"."notificationProviderId",
+               t.sent, 
+               t.success, 
+               p.type, 
+               p.config
+        FROM notification_tests t
+        JOIN notification_providers p 
+          ON "t"."notificationProviderId" = p.id
+        WHERE t.success IS NOT TRUE`
+
+	rows, err := db.Query(query)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var tests []NotificationTest
+	for rows.Next() {
+		var t NotificationTest
+		var pt string
+
+		if err := rows.Scan(
+			&t.ID,
+			&t.NotificationProviderID,
+			&t.Sent,
+			&t.Success,
+			&pt,
+			&t.ProviderConfig,
+		); err != nil {
+			return nil, err
+		}
+
+		t.ProviderType = pt
+		tests = append(tests, t)
+	}
+	return tests, nil
+}
+
+func updateTestStatus(testID int, success bool) error {
+	_, err := db.Exec(
+		"UPDATE notification_tests SET sent = true, success = $1 WHERE id = $2",
+		success, testID,
+	)
+	return err
+}