forked from baron/baron-sso
81 lines
2.6 KiB
Go
81 lines
2.6 KiB
Go
package repository
|
|
|
|
import (
|
|
"baron-sso-backend/internal/domain"
|
|
"context"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type TenantRepository interface {
|
|
Create(ctx context.Context, tenant *domain.Tenant) error
|
|
Update(ctx context.Context, tenant *domain.Tenant) error
|
|
FindByID(ctx context.Context, id string) (*domain.Tenant, error)
|
|
FindBySlug(ctx context.Context, slug string) (*domain.Tenant, error)
|
|
FindByName(ctx context.Context, name string) (*domain.Tenant, error)
|
|
FindByDomain(ctx context.Context, domainName string) (*domain.Tenant, error)
|
|
AddDomain(ctx context.Context, tenantID string, domainName string) error
|
|
}
|
|
|
|
type tenantRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewTenantRepository(db *gorm.DB) TenantRepository {
|
|
return &tenantRepository{db: db}
|
|
}
|
|
|
|
func (r *tenantRepository) Create(ctx context.Context, tenant *domain.Tenant) error {
|
|
return r.db.WithContext(ctx).Create(tenant).Error
|
|
}
|
|
|
|
func (r *tenantRepository) Update(ctx context.Context, tenant *domain.Tenant) error {
|
|
return r.db.WithContext(ctx).Save(tenant).Error
|
|
}
|
|
|
|
func (r *tenantRepository) FindByID(ctx context.Context, id string) (*domain.Tenant, error) {
|
|
var tenant domain.Tenant
|
|
if err := r.db.WithContext(ctx).Preload("Domains").First(&tenant, "id = ?", id).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &tenant, nil
|
|
}
|
|
|
|
func (r *tenantRepository) FindBySlug(ctx context.Context, slug string) (*domain.Tenant, error) {
|
|
var tenant domain.Tenant
|
|
if err := r.db.WithContext(ctx).Preload("Domains").Where("slug = ?", slug).First(&tenant).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &tenant, nil
|
|
}
|
|
|
|
func (r *tenantRepository) FindByName(ctx context.Context, name string) (*domain.Tenant, error) {
|
|
var tenant domain.Tenant
|
|
if err := r.db.WithContext(ctx).Preload("Domains").Where("name = ?", name).First(&tenant).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &tenant, nil
|
|
}
|
|
|
|
func (r *tenantRepository) FindByDomain(ctx context.Context, domainName string) (*domain.Tenant, error) {
|
|
var tenantDomain domain.TenantDomain
|
|
if err := r.db.WithContext(ctx).Where("domain = ?", domainName).First(&tenantDomain).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var tenant domain.Tenant
|
|
if err := r.db.WithContext(ctx).Preload("Domains").First(&tenant, "id = ?", tenantDomain.TenantID).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &tenant, nil
|
|
}
|
|
|
|
func (r *tenantRepository) AddDomain(ctx context.Context, tenantID string, domainName string) error {
|
|
td := domain.TenantDomain{
|
|
TenantID: tenantID,
|
|
Domain: domainName,
|
|
Verified: true, // Auto-verify for internal init/admin usage for now
|
|
}
|
|
return r.db.WithContext(ctx).Create(&td).Error
|
|
}
|