54 lines
1.6 KiB
Go
54 lines
1.6 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
)
|
|
|
|
// PushTokenRepo implements GORM repository for PushToken.
|
|
// Reference: Chatwoot PushSubscription model + P2B M8 spec
|
|
type PushTokenRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewPushTokenRepo creates a new PushToken repository.
|
|
func NewPushTokenRepo(db *gorm.DB) *PushTokenRepo {
|
|
return &PushTokenRepo{db: db}
|
|
}
|
|
|
|
// FindByID retrieves a push token by primary key.
|
|
func (r *PushTokenRepo) FindByID(ctx context.Context, id uint) (*model.PushToken, error) {
|
|
var t model.PushToken
|
|
err := r.db.WithContext(ctx).First(&t, id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &t, nil
|
|
}
|
|
|
|
// ListByUser retrieves all push tokens for a user.
|
|
func (r *PushTokenRepo) ListByUser(ctx context.Context, userID uint) ([]model.PushToken, error) {
|
|
var tokens []model.PushToken
|
|
err := r.db.WithContext(ctx).Where("user_id = ?", userID).Find(&tokens).Error
|
|
return tokens, err
|
|
}
|
|
|
|
// Create inserts a new push token.
|
|
func (r *PushTokenRepo) Create(ctx context.Context, token *model.PushToken) error {
|
|
return r.db.WithContext(ctx).Create(token).Error
|
|
}
|
|
|
|
// Delete removes a push token by ID.
|
|
func (r *PushTokenRepo) Delete(ctx context.Context, id uint) error {
|
|
return r.db.WithContext(ctx).Delete(&model.PushToken{}, id).Error
|
|
}
|
|
|
|
// DeleteByTokenAndUser removes a push token by its token string and user ID.
|
|
func (r *PushTokenRepo) DeleteByTokenAndUser(ctx context.Context, token string, userID uint) error {
|
|
return r.db.WithContext(ctx).
|
|
Where("token = ? AND user_id = ?", token, userID).
|
|
Delete(&model.PushToken{}).Error
|
|
} |