api/constants.go at main · RoturTW/api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package main

import (
	"log"
	"os"
	"strconv"
)

var DAILY_CLAIMS_FILE_PATH = "./rotur_daily.json"

const USERS_FILE_PATH = "./users.json"

var (
	LOCAL_POSTS_PATH              string
	FOLLOWERS_FILE_PATH           string
	ITEMS_FILE_PATH               string
	KEYS_FILE_PATH                string
	EVENTS_HISTORY_PATH           string
	SYSTEMS_FILE_PATH             string
	GROUPS_FILE_PATH              string
	WEBSOCKET_SERVER_URL          string
	EVENT_SERVER_URL              string
	SUBSCRIPTION_CHECK_INTERVAL   int
	INACTIVITY_TAX_CHECK_INTERVAL int
	BANNED_WORDS_URL              string
	DISCORD_WEBHOOK_URL           string
	KEY_OWNERSHIP_CACHE_TTL       int
	ADMIN_TOKEN                   string

	bannedDomains = []string{
		"pornhub.com", "xvideos.com", "xnxx.com", "redtube.com", "youporn.com",
		"xhamster.com", "tube8.com", "spankbang.com", "brazzers.com", "onlyfans.com",
		"chaturbate.com", "livejasmine.com", "cam4.com",
	}

	rateLimits = map[string]RateLimitConfig{
		"default":  {Count: 100, Period: 60},
		"post":     {Count: 5, Period: 60},
		"reply":    {Count: 10, Period: 60},
		"follow":   {Count: 20, Period: 60},
		"profile":  {Count: 30, Period: 60},
		"search":   {Count: 20, Period: 60},
		"ai":       {Count: 5, Period: 10},
		"register": {Count: 5, Period: 10},
		"global":   {Count: 10, Period: 10}, // Global rate limit: 10 requests per 10 seconds
	}
)

func mustEnv(key string, def string) string {
	val := os.Getenv(key)
	if val == "" {
		if def != "" {
			return def
		}
		log.Printf("[config] WARNING: %s not set", key)
	}
	return val
}

func intEnv(key string, def int) int {
	raw := os.Getenv(key)
	if raw == "" {
		return def
	}
	v, err := strconv.Atoi(raw)
	if err != nil {
		log.Printf("[config] invalid int for %s=%s (using default %d)", key, raw, def)
		return def
	}
	return v
}

func loadConfigFromEnv() {
	LOCAL_POSTS_PATH = mustEnv("LOCAL_POSTS_PATH", "./posts.json")
	FOLLOWERS_FILE_PATH = mustEnv("FOLLOWERS_FILE_PATH", "./clawusers.json")
	ITEMS_FILE_PATH = mustEnv("ITEMS_FILE_PATH", "./items.json")
	GROUPS_FILE_PATH = mustEnv("GROUPS_FILE_PATH", "./rotur/groups")
	KEYS_FILE_PATH = mustEnv("KEYS_FILE_PATH", "./keys.json")
	EVENTS_HISTORY_PATH = mustEnv("EVENTS_HISTORY_PATH", "./events_history.json")
	SYSTEMS_FILE_PATH = mustEnv("SYSTEMS_FILE_PATH", "./systems.json")

	// External services
	WEBSOCKET_SERVER_URL = mustEnv("WEBSOCKET_SERVER_URL", "")
	EVENT_SERVER_URL = mustEnv("EVENT_SERVER_URL", "")
	BANNED_WORDS_URL = mustEnv("BANNED_WORDS_URL", "")
	DISCORD_WEBHOOK_URL = mustEnv("DISCORD_WEBHOOK_URL", "")

	// Numeric settings
	SUBSCRIPTION_CHECK_INTERVAL = intEnv("SUBSCRIPTION_CHECK_INTERVAL", 3600)
	INACTIVITY_TAX_CHECK_INTERVAL = intEnv("INACTIVITY_TAX_CHECK_INTERVAL", 3600)
	KEY_OWNERSHIP_CACHE_TTL = intEnv("KEY_OWNERSHIP_CACHE_TTL", 600)

	// Auth / admin tokens
	ADMIN_TOKEN = mustEnv("ADMIN_TOKEN", "")
}

func init() {
	loadConfigFromEnv()

	groupsData = make(map[string]*GroupData)
}