79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
)
|
|
|
|
// TeamRepo implements GORM repository for Team.
|
|
// Reference: Chatwoot app/models/team.rb
|
|
type TeamRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewTeamRepo creates a new Team repository.
|
|
func NewTeamRepo(db *gorm.DB) *TeamRepo {
|
|
return &TeamRepo{db: db}
|
|
}
|
|
|
|
// FindByID retrieves a team by primary key.
|
|
func (r *TeamRepo) FindByID(ctx context.Context, id uint) (*model.Team, error) {
|
|
var team model.Team
|
|
err := r.db.WithContext(ctx).Preload("Members").First(&team, id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &team, nil
|
|
}
|
|
|
|
// FindByIDAndAccount retrieves a team scoped to an account.
|
|
func (r *TeamRepo) FindByIDAndAccount(ctx context.Context, id, accountID uint) (*model.Team, error) {
|
|
var team model.Team
|
|
err := r.db.WithContext(ctx).Preload("Members").
|
|
Where("id = ? AND account_id = ?", id, accountID).First(&team).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &team, nil
|
|
}
|
|
|
|
// ListByAccount retrieves all teams for an account with pagination.
|
|
func (r *TeamRepo) ListByAccount(ctx context.Context, accountID uint, offset, limit int) ([]model.Team, int64, error) {
|
|
var teams []model.Team
|
|
var total int64
|
|
|
|
if err := r.db.WithContext(ctx).Model(&model.Team{}).
|
|
Where("account_id = ?", accountID).Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
err := r.db.WithContext(ctx).Preload("Members").
|
|
Where("account_id = ?", accountID).
|
|
Order("id ASC").Offset(offset).Limit(limit).
|
|
Find(&teams).Error
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return teams, total, nil
|
|
}
|
|
|
|
// Create creates a new team.
|
|
func (r *TeamRepo) Create(ctx context.Context, team *model.Team) error {
|
|
return r.db.WithContext(ctx).Create(team).Error
|
|
}
|
|
|
|
// Update updates a team.
|
|
func (r *TeamRepo) Update(ctx context.Context, team *model.Team) error {
|
|
return r.db.WithContext(ctx).Save(team).Error
|
|
}
|
|
|
|
// Delete soft-deletes a team.
|
|
func (r *TeamRepo) Delete(ctx context.Context, id, accountID uint) error {
|
|
return r.db.WithContext(ctx).
|
|
Where("id = ? AND account_id = ?", id, accountID).
|
|
Delete(&model.Team{}).Error
|
|
}
|