fin-check-api/db/category.go

package db

import (
	"errors"

	"gorm.io/gorm"
)

type Category struct {
	gorm.Model
	Name string
	// Parent is used as a infinite sub-category structure
	ParentID uint
	Parent   *Category
	UserID   uint
	User     *User
}

var (
	ERROR_CATEGORY_PARENT_NOT_FOUND = errors.New("ParentID is invalid for user")
	ERROR_CATEGORY_NAME_NOT_UNIQUE  = errors.New("Name for Category have to be unique for user")
)

func (c *Category) BeforeSave(tx *gorm.DB) error {
	var parent Category
	if err := tx.Find(&parent, Category{ParentID: c.ParentID, UserID: c.UserID}).Error; err != nil {
		return err
	}
	if parent.ID == 0 {
		return ERROR_CATEGORY_PARENT_NOT_FOUND
	}
	var dup Category
	if err := tx.Find(&dup, Category{Name: c.Name, UserID: c.UserID}).Error; err != nil {
		return err
	}
	if dup.ID != 0 {
		return ERROR_CATEGORY_NAME_NOT_UNIQUE
	}
	return nil
}