Files
gochat/backend/internal/repository/team_member_repo.go
T
rogee aeddedf2a3 Reorganize repo: backend/, deploy/, docs/ layout + AGENTS.md
Restructure the monorepo into clear top-level directories:
- backend/: Go module root (cmd, internal, pkg, configs, migrations,
  docs/swagger, scripts, tests, go.mod, Makefile, .air.toml)
- deploy/: Docker (Dockerfile, docker-compose*), quickstart, fluentd
- docs/: project documentation + reports/ (moved from repo root)
- AGENTS.md: new AI coding-agent guide at repo root

Update all references to the new layout:
- Dockerfile: COPY backend/go.mod, COPY backend/ (context = repo root)
- docker-compose files: context ../.., dockerfile deploy/docker/Dockerfile,
  env_file ../../.env, volume mounts ../../backend:/app
- deploy/quickstart/compose.yaml: dockerfile deploy/docker/Dockerfile
- CI: working-directory: backend for go commands, file deploy/docker/Dockerfile,
  coverage path backend/coverage.out, health_check backend/scripts/
- backend/Makefile: docker target uses -f ../deploy/docker/Dockerfile ../
- README: architecture tree, quickstart, config paths updated

Move root stray scripts (rename_models.*, run_m11_tests.sh, verify_build.sh,
gorm_bool_main.go) to backend/scripts/legacy/. All moves via git mv to
preserve history. Build, vet, SQLite tests, and docker compose config verified.
2026-07-07 14:44:12 +08:00

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
}