package repository

import (
	"context"

	"gorm.io/gorm"

	"github.com/gochat/gochat/internal/model"
)

// WebhookSubscriptionRepo implements GORM repository for WebhookSubscription.
// Reference: Chatwoot webhook integration + P2B M8 spec
type WebhookSubscriptionRepo struct {
	db *gorm.DB
}

// NewWebhookSubscriptionRepo creates a new WebhookSubscription repository.
func NewWebhookSubscriptionRepo(db *gorm.DB) *WebhookSubscriptionRepo {
	return &WebhookSubscriptionRepo{db: db}
}

// FindByID retrieves a webhook subscription by primary key.
func (r *WebhookSubscriptionRepo) FindByID(ctx context.Context, id uint) (*model.WebhookSubscription, error) {
	var s model.WebhookSubscription
	err := r.db.WithContext(ctx).First(&s, id).Error
	if err != nil {
		return nil, err
	}
	return &s, nil
}

// GetByID is an alias for FindByID for compatibility with service callers.
func (r *WebhookSubscriptionRepo) GetByID(ctx context.Context, id uint) (*model.WebhookSubscription, error) {
	return r.FindByID(ctx, id)
}

// ListByAccount retrieves all webhook subscriptions for an account.
func (r *WebhookSubscriptionRepo) ListByAccount(ctx context.Context, accountID uint) ([]model.WebhookSubscription, error) {
	var subs []model.WebhookSubscription
	err := r.db.WithContext(ctx).Where("account_id = ? AND active = ?", accountID, true).Find(&subs).Error
	return subs, err
}

// ListActiveByAccount retrieves all active webhook subscriptions for an account.
func (r *WebhookSubscriptionRepo) ListActiveByAccount(ctx context.Context, accountID uint) ([]model.WebhookSubscription, error) {
	var subs []model.WebhookSubscription
	err := r.db.WithContext(ctx).
		Where("account_id = ? AND active = ?", accountID, true).
		Find(&subs).Error
	return subs, err
}

// ListByAccountAndEvent retrieves webhook subscriptions that include a specific event type.
func (r *WebhookSubscriptionRepo) ListByAccountAndEvent(ctx context.Context, accountID uint, eventType string) ([]model.WebhookSubscription, error) {
	var subs []model.WebhookSubscription
	// Use JSON containment operator for PostgreSQL: events @> '["event_type"]'
	err := r.db.WithContext(ctx).
		Where("account_id = ? AND active = ?", accountID, true).
		Where("events @> ?", eventType).
		Find(&subs).Error
	return subs, err
}

// Create inserts a new webhook subscription.
func (r *WebhookSubscriptionRepo) Create(ctx context.Context, sub *model.WebhookSubscription) error {
	return r.db.WithContext(ctx).Create(sub).Error
}

// Update updates a webhook subscription.
func (r *WebhookSubscriptionRepo) Update(ctx context.Context, sub *model.WebhookSubscription) error {
	return r.db.WithContext(ctx).Save(sub).Error
}

// Delete soft-deletes a webhook subscription.
func (r *WebhookSubscriptionRepo) Delete(ctx context.Context, id uint) error {
	return r.db.WithContext(ctx).Delete(&model.WebhookSubscription{}, id).Error
}

// --- WebhookDelivery repository operations (embedded for simplicity) ---

// CreateDelivery inserts a webhook delivery record.
func (r *WebhookSubscriptionRepo) CreateDelivery(ctx context.Context, delivery *model.WebhookDelivery) error {
	return r.db.WithContext(ctx).Create(delivery).Error
}

// UpdateDelivery updates a webhook delivery record.
func (r *WebhookSubscriptionRepo) UpdateDelivery(ctx context.Context, delivery *model.WebhookDelivery) error {
	return r.db.WithContext(ctx).Save(delivery).Error
}

// ListDeliveriesBySubscription retrieves recent delivery records for a subscription.
func (r *WebhookSubscriptionRepo) ListDeliveriesBySubscription(ctx context.Context, subscriptionID uint, limit int) ([]model.WebhookDelivery, error) {
	var deliveries []model.WebhookDelivery
	err := r.db.WithContext(ctx).
		Where("subscription_id = ?", subscriptionID).
		Order("created_at DESC").
		Limit(limit).
		Find(&deliveries).Error
	return deliveries, err
}