Changeset 0.10.2 (#58)

server/store/interfaces.go

@@ -284,6 +284,7 @@ type GlobalConfigStore interface {
 
 type UsageStore interface {
 	Log(ctx context.Context, entry *models.UsageEntry) error
+	CountRecentByRole(ctx context.Context, userID, role string, since time.Time) (int, error)
 	QueryByUser(ctx context.Context, userID string, opts UsageQueryOptions) ([]models.UsageAggregate, error)
 	QueryByUserPersonal(ctx context.Context, userID string, opts UsageQueryOptions) ([]models.UsageAggregate, error)
 	QueryByTeam(ctx context.Context, teamID string, opts UsageQueryOptions) ([]models.UsageAggregate, error)

server/store/postgres/usage.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"strings"
+	"time"
 
 	"git.gobha.me/xcaliber/chat-switchboard/models"
 	"git.gobha.me/xcaliber/chat-switchboard/store"
@@ -143,6 +144,17 @@ func (s *UsageStore) GetPersonalTotals(ctx context.Context, userID string, opts
 	return &t, err
 }
 
+// CountRecentByRole counts how many usage entries a user has for a given
+// role within the specified duration. Used for rate limiting utility calls.
+func (s *UsageStore) CountRecentByRole(ctx context.Context, userID, role string, since time.Time) (int, error) {
+	var count int
+	err := DB.QueryRowContext(ctx, `
+		SELECT COUNT(*) FROM usage_log
+		WHERE user_id = $1 AND role = $2 AND created_at >= $3
+	`, userID, role, since).Scan(&count)
+	return count, err
+}
+
 // ── Internal ───────────────────────────────
 
 func (s *UsageStore) buildFilters(baseWhere []string, baseArgs []interface{}, opts store.UsageQueryOptions) ([]string, []interface{}) {