aiforge/models/medal_category.go

package models

import "code.gitea.io/gitea/modules/timeutil"

type MedalType int

const (
	CustomizeMedal = iota + 1
	SystemMedal
)

type MedalCategory struct {
	ID          int64 `xorm:"pk autoincr"`
	Name        string
	Position    int64
	Type        MedalType
	CreatedUnix timeutil.TimeStamp `xorm:"created"`
	UpdatedUnix timeutil.TimeStamp `xorm:"updated"`
	DeletedAt   timeutil.TimeStamp `xorm:"deleted"`
}

func (m *MedalCategory) ToShow() *MedalCategory4Show {
	return &MedalCategory4Show{
		ID:          m.ID,
		Name:        m.Name,
		Position:    m.Position,
		Type:        m.Type,
		CreatedUnix: m.CreatedUnix,
	}
}

type MedalCategory4Show struct {
	ID          int64 `xorm:"pk autoincr"`
	Name        string
	Position    int64
	Type        MedalType
	CreatedUnix timeutil.TimeStamp `xorm:"created"`
}

func (m MedalCategory4Show) ToDTO() MedalCategory {
	return MedalCategory{
		ID:          m.ID,
		Name:        m.Name,
		Position:    m.Position,
		Type:        m.Type,
		CreatedUnix: m.CreatedUnix,
	}
}

func GetMedalCategoryList() ([]*MedalCategory, error) {
	r := make([]*MedalCategory, 0)
	if err := x.OrderBy("position asc,created_unix desc").Find(&r); err != nil {
		return nil, err
	}
	return r, nil
}

func AddMedalCategory(m MedalCategory) (int64, error) {
	return x.Insert(&m)
}

func UpdateMedalCategoryById(id int64, param MedalCategory) (int64, error) {
	return x.ID(id).Update(&param)
}

func DelMedalCategory(id int64) (int64, error) {
	return x.ID(id).Delete(&MedalCategory{})
}

func GetMedalCategoryById(id int64) (*MedalCategory, error) {
	m := &MedalCategory{}
	has, err := x.ID(id).Get(m)
	if err != nil {
		return nil, err
	} else if !has {
		return nil, &ErrRecordNotExist{}
	}
	return m, nil
}