repository

type APIKeyRepository interface {
	Create(apiKey *domain.APIKey) error
	GetByKeyHash(keyHash string) (*domain.APIKey, error)
	GetByID(id int64) (*domain.APIKey, error)
	ListByUser(userID int64) ([]*domain.APIKey, error)
	UpdateLastUsed(id int64, ip string) error
	Delete(id int64) error
}

type AliasRepository interface {
	Create(alias *domain.Alias) error
	GetByID(id int64) (*domain.Alias, error)
	GetByEmail(email string) (*domain.Alias, error)
	Update(alias *domain.Alias) error
	Delete(id int64) error
	ListAll() ([]*domain.Alias, error)
	ListByDomain(domainID int64) ([]*domain.Alias, error)
	// Pagination methods
	List(offset, limit int) ([]*domain.Alias, error)
	Count() (int64, error)
	CountByDomain(domainID int64) (int64, error)
}

type DKIMRepository interface {
	GetDKIMConfig(domainName string) (*domain.DKIMConfig, error)
}

type DomainRepository interface {
	Create(domain *domain.Domain) error
	GetByID(id int64) (*domain.Domain, error)
	GetByName(name string) (*domain.Domain, error)
	Update(domain *domain.Domain) error
	Delete(id int64) error
	List(offset, limit int) ([]*domain.Domain, error)
}

type GreylistRepository interface {
	Create(ip, sender, recipient string) (*domain.GreylistTriplet, error)
	Get(ip, sender, recipient string) (*domain.GreylistTriplet, error)
	IncrementPass(id int64) error
	DeleteOlderThan(age time.Duration) error
}

type IPBlacklistRepository interface {
	Add(ip, reason string, expiresAt *time.Time) error
	IsBlacklisted(ip string) (bool, error)
	Remove(ip string) error
	RemoveExpired() error
}

type LoginAttemptRepository interface {
	Record(ip, email string, success bool) error
	GetRecentFailures(ip string, duration time.Duration) (int, error)
	GetRecentUserFailures(email string, duration time.Duration) (int, error)
	Cleanup(age time.Duration) error
}

type MailboxRepository interface {
	Create(mailbox *domain.Mailbox) error
	GetByID(id int64) (*domain.Mailbox, error)
	GetByUser(userID int64) ([]*domain.Mailbox, error)
	GetByName(userID int64, name string) (*domain.Mailbox, error)
	Update(mailbox *domain.Mailbox) error
	Delete(id int64) error
}

type MessageRepository interface {
	Create(message *domain.Message) error
	GetByID(id int64) (*domain.Message, error)
	GetByMailbox(mailboxID int64, offset, limit int) ([]*domain.Message, error)
	Update(message *domain.Message) error
	Delete(id int64) error
}

type QuarantineRepository interface {
	Create(message *domain.QuarantineMessage) error
	GetByID(id int64) (*domain.QuarantineMessage, error)
	List(offset, limit int) ([]*domain.QuarantineMessage, error)
	UpdateAction(id int64, action string) error
	Delete(id int64) error
	DeleteOlderThan(age time.Duration) error
}

type QueueRepository interface {
	Enqueue(item *domain.QueueItem) error
	GetPending() ([]*domain.QueueItem, error)
	GetByID(id int64) (*domain.QueueItem, error)
	UpdateStatus(id int64, status string, errorMsg string) error
	UpdateRetry(id int64, retryCount int, nextRetry time.Time) error
	Delete(id int64) error
	// Pagination and filtering methods
	List(offset, limit int) ([]*domain.QueueItem, error)
	ListByStatus(status string, offset, limit int) ([]*domain.QueueItem, error)
	Count() (int64, error)
	CountByStatus(status string) (int64, error)
}

type RateLimitRepository interface {
	Get(key string, limitType string) (*domain.RateLimitEntry, error)
	CreateOrUpdate(entry *domain.RateLimitEntry) error
	Cleanup(windowDuration time.Duration) error
}

type Repositories struct {
	User         UserRepository
	Domain       DomainRepository
	Message      MessageRepository
	Mailbox      MailboxRepository
	Alias        AliasRepository
	Queue        QueueRepository
	LoginAttempt LoginAttemptRepository
	IPBlacklist  IPBlacklistRepository
	Greylist     GreylistRepository
	RateLimit    RateLimitRepository
	Webhook      WebhookRepository
}

type UserRepository interface {
	Create(user *domain.User) error
	GetByID(id int64) (*domain.User, error)
	GetByEmail(email string) (*domain.User, error)
	Update(user *domain.User) error
	UpdateLastLogin(id int64) error
	UpdatePassword(userID int64, passwordHash string) error
	Delete(id int64) error
	List(domainID int64, offset, limit int) ([]*domain.User, error)
	ListAll() ([]*domain.User, error)
	// Pagination methods
	ListPaginated(offset, limit int) ([]*domain.User, error)
	Count() (int64, error)
	CountByDomain(domainID int64) (int64, error)
}

type WebhookRepository interface {
	// Webhook management
	Create(ctx context.Context, webhook *domain.Webhook) error
	GetByID(ctx context.Context, id int64) (*domain.Webhook, error)
	List(ctx context.Context) ([]*domain.Webhook, error)
	ListActive(ctx context.Context) ([]*domain.Webhook, error)
	Update(ctx context.Context, webhook *domain.Webhook) error
	Delete(ctx context.Context, id int64) error

	// Delivery management
	CreateDelivery(ctx context.Context, delivery *domain.WebhookDelivery) error
	GetDeliveryByID(ctx context.Context, id int64) (*domain.WebhookDelivery, error)
	ListDeliveriesByWebhook(ctx context.Context, webhookID int64, limit int) ([]*domain.WebhookDelivery, error)
	ListPendingDeliveries(ctx context.Context, limit int) ([]*domain.WebhookDelivery, error)
	UpdateDelivery(ctx context.Context, delivery *domain.WebhookDelivery) error
	DeleteOldDeliveries(ctx context.Context, olderThan int) error
}