46 lines
842 B
Go
46 lines
842 B
Go
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
|
|
}
|