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"
|
|
)
|
|
|
|
// CaptainCustomToolRepo provides data access for CaptainCustomTool.
|
|
type CaptainCustomToolRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewCaptainCustomToolRepo(db *gorm.DB) *CaptainCustomToolRepo {
|
|
return &CaptainCustomToolRepo{db: db}
|
|
}
|
|
|
|
func (r *CaptainCustomToolRepo) Create(ctx context.Context, tool *model.CaptainCustomTool) error {
|
|
return r.db.WithContext(ctx).Create(tool).Error
|
|
}
|
|
|
|
func (r *CaptainCustomToolRepo) GetByID(ctx context.Context, id uint) (*model.CaptainCustomTool, error) {
|
|
var tool model.CaptainCustomTool
|
|
if err := r.db.WithContext(ctx).First(&tool, id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &tool, nil
|
|
}
|
|
|
|
func (r *CaptainCustomToolRepo) GetByAccountAndID(ctx context.Context, accountID, id uint) (*model.CaptainCustomTool, error) {
|
|
var tool model.CaptainCustomTool
|
|
if err := r.db.WithContext(ctx).Where("account_id = ? AND id = ?", accountID, id).First(&tool).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &tool, nil
|
|
}
|
|
|
|
func (r *CaptainCustomToolRepo) GetBySlug(ctx context.Context, accountID uint, slug string) (*model.CaptainCustomTool, error) {
|
|
var tool model.CaptainCustomTool
|
|
if err := r.db.WithContext(ctx).Where("account_id = ? AND slug = ?", accountID, slug).First(&tool).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &tool, nil
|
|
}
|
|
|
|
func (r *CaptainCustomToolRepo) Update(ctx context.Context, tool *model.CaptainCustomTool) error {
|
|
return r.db.WithContext(ctx).Save(tool).Error
|
|
}
|
|
|
|
func (r *CaptainCustomToolRepo) Delete(ctx context.Context, id uint) error {
|
|
return r.db.WithContext(ctx).Delete(&model.CaptainCustomTool{}, id).Error
|
|
}
|
|
|
|
func (r *CaptainCustomToolRepo) DeleteByAccount(ctx context.Context, accountID, id uint) error {
|
|
return r.db.WithContext(ctx).Where("account_id = ? AND id = ?", accountID, id).Delete(&model.CaptainCustomTool{}).Error
|
|
}
|
|
|
|
func (r *CaptainCustomToolRepo) ListByAccount(ctx context.Context, accountID uint, offset, limit int) ([]model.CaptainCustomTool, int64, error) {
|
|
var tools []model.CaptainCustomTool
|
|
var count int64
|
|
db := r.db.WithContext(ctx).Model(&model.CaptainCustomTool{}).Where("account_id = ?", accountID)
|
|
db.Count(&count)
|
|
if err := db.Offset(offset).Limit(limit).Find(&tools).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
return tools, count, nil
|
|
}
|
|
|
|
func (r *CaptainCustomToolRepo) CountByAccount(ctx context.Context, accountID uint) (int64, error) {
|
|
var count int64
|
|
if err := r.db.WithContext(ctx).Model(&model.CaptainCustomTool{}).Where("account_id = ?", accountID).Count(&count).Error; err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|