storage

type APIKeyFilter struct {
	EnabledOnly bool
	Offset      int
	Limit       int
}

type APIKeyRepository interface {
	Create(ctx context.Context, key *types.APIKey) error
	Get(ctx context.Context, id string) (*types.APIKey, error)
	Update(ctx context.Context, key *types.APIKey) error
	Delete(ctx context.Context, id string) error
	List(ctx context.Context, filter APIKeyFilter) ([]*types.APIKey, error)
	UpdateLastUsed(ctx context.Context, id string) error
}

type AuditFilter struct {
	RequestID *types.SignRequestID
	APIKeyID  *string
	EventType *types.AuditEventType
	ChainType *types.ChainType
	StartTime *time.Time
	EndTime   *time.Time
	// Cursor-based pagination (preferred over Offset)
	// Cursor is the timestamp of the last item from previous page
	Cursor *time.Time
	// CursorID is the ID of the last item (for tie-breaking when timestamps are equal)
	CursorID *types.AuditID
	Limit    int
}

type AuditRepository interface {
	Log(ctx context.Context, record *types.AuditRecord) error
	Query(ctx context.Context, filter AuditFilter) ([]*types.AuditRecord, error)
	Count(ctx context.Context, filter AuditFilter) (int, error)
	GetByRequestID(ctx context.Context, requestID types.SignRequestID) ([]*types.AuditRecord, error)
}

type BudgetRepository interface {
	Create(ctx context.Context, budget *types.RuleBudget) error
	GetByRuleID(ctx context.Context, ruleID types.RuleID, unit string) (*types.RuleBudget, error)
	Delete(ctx context.Context, id string) error
	DeleteByRuleID(ctx context.Context, ruleID types.RuleID) error
	// AtomicSpend atomically increments spent amount and tx count.
	// Returns ErrBudgetExceeded if the spend would exceed limits.
	// Uses SQL-level conditional UPDATE to prevent race conditions.
	AtomicSpend(ctx context.Context, ruleID types.RuleID, unit string, amount string) error
	// ResetBudget resets spent/txCount/alertSent for a new period.
	// Uses conditional WHERE to ensure idempotent reset (only resets if in old period).
	ResetBudget(ctx context.Context, ruleID types.RuleID, unit string, currentPeriodStart time.Time) error
	ListByRuleID(ctx context.Context, ruleID types.RuleID) ([]*types.RuleBudget, error)
	ListByRuleIDs(ctx context.Context, ruleIDs []types.RuleID) ([]*types.RuleBudget, error)
	// MarkAlertSent sets alert_sent=true for the given rule+unit budget.
	// This prevents duplicate alert notifications within the same period.
	MarkAlertSent(ctx context.Context, ruleID types.RuleID, unit string) error
}

type Config struct {
	DSN string `yaml:"dsn"`
}

type GormAPIKeyRepository struct {
	// contains filtered or unexported fields
}

type GormAuditRepository struct {
	// contains filtered or unexported fields
}

type GormBudgetRepository struct {
	// contains filtered or unexported fields
}

type GormRequestRepository struct {
	// contains filtered or unexported fields
}

type GormRuleRepository struct {
	// contains filtered or unexported fields
}

type GormTemplateRepository struct {
	// contains filtered or unexported fields
}

type InMemoryNonceStore struct {
	// contains filtered or unexported fields
}

type MemoryRuleRepository struct {
	// contains filtered or unexported fields
}

type NonceStore interface {
	// CheckAndStore checks if a nonce exists and stores it if not.
	// Returns true if the nonce was stored (new), false if it already exists (replay).
	CheckAndStore(ctx context.Context, apiKeyID, nonce string, ttl time.Duration) (bool, error)
}

type RequestFilter struct {
	APIKeyID      *string
	SignerAddress *string
	ChainType     *types.ChainType
	ChainID       *string
	Status        []types.SignRequestStatus
	// Cursor-based pagination (preferred over Offset)
	// Cursor is the created_at timestamp of the last item from previous page
	Cursor *time.Time
	// CursorID is the ID of the last item (for tie-breaking when timestamps are equal)
	CursorID *types.SignRequestID
	Limit    int
}

type RequestRepository interface {
	Create(ctx context.Context, req *types.SignRequest) error
	Get(ctx context.Context, id types.SignRequestID) (*types.SignRequest, error)
	Update(ctx context.Context, req *types.SignRequest) error
	// CompareAndUpdate atomically updates a request only if its current status
	// matches expectedStatus. Returns ErrStateConflict if the status has changed.
	CompareAndUpdate(ctx context.Context, req *types.SignRequest, expectedStatus types.SignRequestStatus) error
	List(ctx context.Context, filter RequestFilter) ([]*types.SignRequest, error)
	Count(ctx context.Context, filter RequestFilter) (int, error)
	UpdateStatus(ctx context.Context, id types.SignRequestID, status types.SignRequestStatus) error
}

type RuleFilter struct {
	ChainType     *types.ChainType
	ChainID       *string
	APIKeyID      *string
	SignerAddress *string
	Type          *types.RuleType
	Source        *types.RuleSource
	EnabledOnly   bool
	Offset        int
	Limit         int
}

type RuleRepository interface {
	Create(ctx context.Context, rule *types.Rule) error
	Get(ctx context.Context, id types.RuleID) (*types.Rule, error)
	Update(ctx context.Context, rule *types.Rule) error
	Delete(ctx context.Context, id types.RuleID) error
	List(ctx context.Context, filter RuleFilter) ([]*types.Rule, error)
	Count(ctx context.Context, filter RuleFilter) (int, error)
	ListByChainType(ctx context.Context, chainType types.ChainType) ([]*types.Rule, error)
	IncrementMatchCount(ctx context.Context, id types.RuleID) error
}

type TemplateFilter struct {
	Type        *types.RuleType
	Source      *types.RuleSource
	EnabledOnly bool
	Offset      int
	Limit       int
}

type TemplateRepository interface {
	Create(ctx context.Context, tmpl *types.RuleTemplate) error
	Get(ctx context.Context, id string) (*types.RuleTemplate, error)
	GetByName(ctx context.Context, name string) (*types.RuleTemplate, error)
	Update(ctx context.Context, tmpl *types.RuleTemplate) error
	Delete(ctx context.Context, id string) error
	List(ctx context.Context, filter TemplateFilter) ([]*types.RuleTemplate, error)
	Count(ctx context.Context, filter TemplateFilter) (int, error)
}

type Transactional interface {
	RunInTransaction(ctx context.Context, fn func(txRepo RuleRepository) error) error
}