app/calendar/calendar.go

package calendar

import (
	"time"

	"github.com/rs/xid"
	"github.com/rs/zerolog"
	"gorm.io/gorm"
	"kumoly.io/kumoly/app/auth"
	"kumoly.io/kumoly/app/errors"
)

var l zerolog.Logger

var db *gorm.DB

var Color []string = []string{
	"red", "pink", "purple", "deep-purple", "indigo", "blue", "light-blue", "cyan", "teal", "green",
	"light-green", "lime", "brown", "amber", "orange", "deep-orange", "blue-grey",
	// "yellow",
}

type Calendar struct {
	ID string `gorm:"primaryKey"`

	Name        string `gorm:"index:idx_cal,unique"`
	Description string

	Timezone string

	Events []Event

	Group     auth.Group `json:"-"`
	GroupName string     `gorm:"-" json:"Group"`
	GroupID   uint       `gorm:"index:idx_cal,unique" json:"-"`

	Creator   string
	CreatedAt time.Time
	UpdatedAt time.Time
}

func (c *Calendar) BeforeSave(tx *gorm.DB) (err error) {
	if c.GroupName != "" {
		var grp_id uint
		db.Raw("select id from groups where name = ?", c.GroupName).Scan(&grp_id)
		if grp_id == 0 {
			return errors.ErrorNotFound
		}
		c.GroupID = grp_id
	}
	return
}

func (c *Calendar) BeforeCreate(tx *gorm.DB) (err error) {
	if c.ID == "" {
		c.ID = xid.New().String()
	}
	if c.Timezone == "" {
		c.Timezone = "Asia/Taipei"
	}
	return
}

func (c *Calendar) AfterFind(tx *gorm.DB) (err error) {
	if c.GroupID != 0 {
		var name string
		db.Raw("select name from groups where id = ?", c.GroupID).Scan(&name)
		c.GroupName = name
	}

	return
}