81 lines
2.5 KiB
Go
81 lines
2.5 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
)
|
|
|
|
// TeamMemberRepo implements GORM repository for TeamMember.
|
|
// Reference: Chatwoot app/models/team_member.rb
|
|
type TeamMemberRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewTeamMemberRepo creates a new TeamMember repository.
|
|
func NewTeamMemberRepo(db *gorm.DB) *TeamMemberRepo {
|
|
return &TeamMemberRepo{db: db}
|
|
}
|
|
|
|
// FindByTeamAndUser retrieves a team member by team and user.
|
|
func (r *TeamMemberRepo) FindByTeamAndUser(ctx context.Context, teamID, userID uint) (*model.TeamMember, error) {
|
|
var tm model.TeamMember
|
|
err := r.db.WithContext(ctx).Where("team_id = ? AND user_id = ?", teamID, userID).First(&tm).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &tm, nil
|
|
}
|
|
|
|
// FindByTeam retrieves all members assigned to a team.
|
|
func (r *TeamMemberRepo) FindByTeam(ctx context.Context, teamID uint) ([]model.TeamMember, error) {
|
|
var tms []model.TeamMember
|
|
err := r.db.WithContext(ctx).Preload("User").
|
|
Where("team_id = ?", teamID).
|
|
Order("id ASC").Find(&tms).Error
|
|
return tms, err
|
|
}
|
|
|
|
// FindByUser retrieves all teams a user is assigned to.
|
|
func (r *TeamMemberRepo) FindByUser(ctx context.Context, userID uint) ([]model.TeamMember, error) {
|
|
var tms []model.TeamMember
|
|
err := r.db.WithContext(ctx).Preload("Team").
|
|
Where("user_id = ?", userID).
|
|
Order("id ASC").Find(&tms).Error
|
|
return tms, err
|
|
}
|
|
|
|
// Create creates a new team member assignment.
|
|
func (r *TeamMemberRepo) Create(ctx context.Context, tm *model.TeamMember) error {
|
|
return r.db.WithContext(ctx).Create(tm).Error
|
|
}
|
|
|
|
// CreateBatch creates multiple team member assignments at once.
|
|
func (r *TeamMemberRepo) CreateBatch(ctx context.Context, tms []model.TeamMember) error {
|
|
return r.db.WithContext(ctx).Create(&tms).Error
|
|
}
|
|
|
|
// Delete removes a team member assignment.
|
|
func (r *TeamMemberRepo) Delete(ctx context.Context, teamID, userID uint) error {
|
|
return r.db.WithContext(ctx).
|
|
Where("team_id = ? AND user_id = ?", teamID, userID).
|
|
Delete(&model.TeamMember{}).Error
|
|
}
|
|
|
|
// DeleteByTeam removes all members from a team.
|
|
func (r *TeamMemberRepo) DeleteByTeam(ctx context.Context, teamID uint) error {
|
|
return r.db.WithContext(ctx).
|
|
Where("team_id = ?", teamID).
|
|
Delete(&model.TeamMember{}).Error
|
|
}
|
|
|
|
// CountByTeam counts members in a team.
|
|
func (r *TeamMemberRepo) CountByTeam(ctx context.Context, teamID uint) (int64, error) {
|
|
var count int64
|
|
err := r.db.WithContext(ctx).Model(&model.TeamMember{}).
|
|
Where("team_id = ?", teamID).Count(&count).Error
|
|
return count, err
|
|
}
|