rogee
aeddedf2a3
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.
77 lines
2.6 KiB
Go
77 lines
2.6 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) 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
|
|
}
|