79 lines
2.7 KiB
Go
79 lines
2.7 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// CaptainAssistantRepo provides data access for CaptainAssistant.
|
|
type CaptainAssistantRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewCaptainAssistantRepo(db *gorm.DB) *CaptainAssistantRepo {
|
|
return &CaptainAssistantRepo{db: db}
|
|
}
|
|
|
|
func (r *CaptainAssistantRepo) DB() *gorm.DB { return r.db }
|
|
|
|
func (r *CaptainAssistantRepo) Create(ctx context.Context, assistant *model.CaptainAssistant) error {
|
|
return r.db.WithContext(ctx).Create(assistant).Error
|
|
}
|
|
|
|
func (r *CaptainAssistantRepo) GetByID(ctx context.Context, id uint) (*model.CaptainAssistant, error) {
|
|
var assistant model.CaptainAssistant
|
|
if err := r.db.WithContext(ctx).First(&assistant, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &assistant, nil
|
|
}
|
|
|
|
func (r *CaptainAssistantRepo) GetByAccountAndID(ctx context.Context, accountID, id uint) (*model.CaptainAssistant, error) {
|
|
var assistant model.CaptainAssistant
|
|
if err := r.db.WithContext(ctx).Where("account_id = ? AND id = ?", accountID, id).First(&assistant).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &assistant, nil
|
|
}
|
|
|
|
func (r *CaptainAssistantRepo) GetAccountFeatureFlags(ctx context.Context, accountID uint) (string, error) {
|
|
var account model.Account
|
|
if err := r.db.WithContext(ctx).Select("feature_flags").First(&account, accountID).Error; err != nil {
|
|
return "", err
|
|
}
|
|
return account.FeatureFlags, nil
|
|
}
|
|
|
|
func (r *CaptainAssistantRepo) Update(ctx context.Context, assistant *model.CaptainAssistant) error {
|
|
return r.db.WithContext(ctx).Save(assistant).Error
|
|
}
|
|
|
|
func (r *CaptainAssistantRepo) Delete(ctx context.Context, id uint) error {
|
|
return r.db.WithContext(ctx).Delete(&model.CaptainAssistant{}, id).Error
|
|
}
|
|
|
|
func (r *CaptainAssistantRepo) ListByAccount(ctx context.Context, accountID uint, offset, limit int) ([]model.CaptainAssistant, int64, error) {
|
|
var assistants []model.CaptainAssistant
|
|
var count int64
|
|
db := r.db.WithContext(ctx).Model(&model.CaptainAssistant{}).Where("account_id = ?", accountID)
|
|
db.Count(&count)
|
|
if err := db.Order("created_at DESC").Offset(offset).Limit(limit).Find(&assistants).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return assistants, count, nil
|
|
}
|
|
|
|
// FindByInboxID retrieves assistants linked to a given inbox via the captain_inboxes join table.
|
|
func (r *CaptainAssistantRepo) FindByInboxID(ctx context.Context, inboxID uint) ([]model.CaptainAssistant, error) {
|
|
var assistants []model.CaptainAssistant
|
|
if err := r.db.WithContext(ctx).
|
|
Joins("JOIN captain_inboxes ON captain_inboxes.captain_assistant_id = captain_assistants.id").
|
|
Where("captain_inboxes.inbox_id = ?", inboxID).
|
|
Find(&assistants).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return assistants, nil
|
|
}
|