Files
gochat/internal/repository/webhook_subscription_repo.go
T
2026-06-04 15:44:48 +08:00

100 lines
3.7 KiB
Go

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
}