Implement cooldowns

internal/util/cooldowns.go

@@ -0,0 +1,45 @@
+package util
+
+import (
+	"net"
+	"net/http"
+	"strings"
+	"sync"
+	"time"
+)
+
+var (
+	PostCooldowns   = make(map[string]time.Time)
+	ThreadCooldowns = make(map[string]time.Time)
+)
+
+var cooldownMutex sync.Mutex
+
+const POST_COOLDOWN = 15 * time.Second
+const THREAD_COOLDOWN = 2 * time.Minute
+
+func IsOnCooldown(ip string, m map[string]time.Time, duration time.Duration) bool {
+	cooldownMutex.Lock()
+	defer cooldownMutex.Unlock()
+
+	last, exists := m[ip]
+	if !exists || time.Since(last) >= duration {
+		m[ip] = time.Now()
+		return false
+	}
+	return true
+}
+
+func GetIP(r *http.Request) string {
+	if xff := r.Header.Get("X-Forwarded-For"); xff != "" {
+		parts := strings.Split(xff, ",")
+		return strings.TrimSpace(parts[0])
+	}
+
+	// Fallback to RemoteAddr
+	ip, _, err := net.SplitHostPort(r.RemoteAddr)
+	if err != nil {
+		return r.RemoteAddr
+	}
+	return ip
+}