drwily/gogobee
1
0
Fork 0
mirror of https://github.com/prosolis/gogobee.git synced 2026-07-16 08:52:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add rate limiting, API response caching, and fix progress bar panic

Browse Source 
- Rate limit !weather (5/day), !concerts (10/day), fix !translate default to 20/day
- Add generic api_cache table with CacheGet/CacheSet helpers
- Cache !weather (1h), !wiki, !define, !tv (24h), !upcoming movies, !anime season/upcoming (24h)
- Fix ProgressBar panic on negative Repeat count when XP exceeds level threshold
- All rate limits configurable via RATELIMIT_* env vars

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
prosolis
2026-03-08 19:07:48 -07:00
parent ddb196aaad
commit 2c7c15b282
8 changed files with 145 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
internal/plugin/concerts.go
View File

@@ -44,17 +44,19 @@ type bandsintownEvent struct {
// ConcertsPlugin provides concert lookups via Bandsintown.
type ConcertsPlugin struct {
Base
apiKey string
httpClient *http.Client
mu sync.Mutex
cooldowns map[string]time.Time // keyed by userID+artist
apiKey string
httpClient *http.Client
rateLimiter *RateLimitsPlugin
mu sync.Mutex
cooldowns map[string]time.Time // keyed by userID+artist
}
// NewConcertsPlugin creates a new ConcertsPlugin.
func NewConcertsPlugin(client *mautrix.Client) *ConcertsPlugin {
func NewConcertsPlugin(client *mautrix.Client, rateLimiter *RateLimitsPlugin) *ConcertsPlugin {
return &ConcertsPlugin{
Base: NewBase(client),
apiKey: os.Getenv("BANDSINTOWN_API_KEY"),
Base: NewBase(client),
apiKey: os.Getenv("BANDSINTOWN_API_KEY"),
rateLimiter: rateLimiter,
httpClient: &http.Client{
Timeout: 10 * time.Second,
},
@@ -113,7 +115,18 @@ func (p *ConcertsPlugin) handleSearch(ctx MessageContext, artist string) error {
return p.SendReply(ctx.RoomID, ctx.EventID, "Concert lookups are not configured (missing API key).")
}
// Rate limit check
// Daily rate limit (configurable, default 10/day to match TS version)
concertLimit := 10
if v := os.Getenv("RATELIMIT_CONCERTS"); v != "" {
if n, err := fmt.Sscanf(v, "%d", &concertLimit); n != 1 || err != nil {
concertLimit = 10
}
}
if p.rateLimiter != nil && !p.rateLimiter.CheckLimit(ctx.Sender, "concerts", concertLimit) {
return p.SendReply(ctx.RoomID, ctx.EventID, "Concert search rate limit reached for today.")
}
// Per-search cooldown
cooldownKey := string(ctx.Sender) + ":" + strings.ToLower(artist)
p.mu.Lock()
if last, ok := p.cooldowns[cooldownKey]; ok && time.Since(last) < 30*time.Second {