Michael Chus
20309d1f0e
Renamed module path git.mchus.pro/mchus/quoteforge → git.mchus.pro/mchus/priceforge, renamed package quoteforge → priceforge, moved binary from cmd/qfs to cmd/pfs.
77 lines
1.9 KiB
Go
77 lines
1.9 KiB
Go
package repository
|
|
|
|
import (
|
|
"git.mchus.pro/mchus/priceforge/internal/models"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type CategoryRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewCategoryRepository(db *gorm.DB) *CategoryRepository {
|
|
return &CategoryRepository{db: db}
|
|
}
|
|
|
|
func (r *CategoryRepository) GetAll() ([]models.Category, error) {
|
|
var categories []models.Category
|
|
err := r.db.Order("display_order ASC").Find(&categories).Error
|
|
return categories, err
|
|
}
|
|
|
|
func (r *CategoryRepository) GetByCode(code string) (*models.Category, error) {
|
|
var category models.Category
|
|
err := r.db.Where("code = ?", code).First(&category).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &category, nil
|
|
}
|
|
|
|
func (r *CategoryRepository) GetByID(id uint) (*models.Category, error) {
|
|
var category models.Category
|
|
err := r.db.First(&category, id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &category, nil
|
|
}
|
|
|
|
// CreateIfNotExists creates a new category if it doesn't exist, returns existing one if it does
|
|
func (r *CategoryRepository) CreateIfNotExists(code string) (*models.Category, error) {
|
|
// Try to find existing
|
|
existing, err := r.GetByCode(code)
|
|
if err == nil {
|
|
return existing, nil
|
|
}
|
|
|
|
// Get max display order to put new category at the end
|
|
var maxOrder int
|
|
r.db.Model(&models.Category{}).Select("COALESCE(MAX(display_order), 0)").Scan(&maxOrder)
|
|
|
|
// Create new category
|
|
newCat := &models.Category{
|
|
Code: code,
|
|
Name: code, // Use code as name initially
|
|
NameRu: code,
|
|
DisplayOrder: maxOrder + 1,
|
|
IsRequired: false,
|
|
}
|
|
|
|
if err := r.db.Create(newCat).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return newCat, nil
|
|
}
|
|
|
|
// Create creates a new category
|
|
func (r *CategoryRepository) Create(category *models.Category) error {
|
|
return r.db.Create(category).Error
|
|
}
|
|
|
|
// Update updates an existing category
|
|
func (r *CategoryRepository) Update(category *models.Category) error {
|
|
return r.db.Save(category).Error
|
|
}
|