package model

import (
	"errors"
	"time"

	"gitee.com/linqwen/momo/app/conf/dao"
	"gitee.com/linqwen/momo/base"
	"gitee.com/linqwen/momo/cache"
	"gitee.com/linqwen/momo/utils"
	"gorm.io/gorm"
)

func init() {
	dao.ConfDb.AutoMigrate(&AccessctlEntity{})
	initAccCache()
}

type AccessctlEntity struct {
	base.SnowflakeIDModel
	Path        string `gorm:"column:path;unique" json:"Path" form:"Path" comment:"Path"`
	Access      int    `gorm:"column:access;default:1" json:"Access" form:"Access" comment:"Access"`
	Ips         string `gorm:"column:ips" json:"Ips" form:"Ips" comment:"Ips"`
	Description string `gorm:"column:description" json:"Description" form:"Description" comment:"Description"`
	Status      int    `gorm:"column:status;type:int" json:"Status"`
}

func (AccessctlEntity) TableName() string { return "conf_accessctl" }

func (s *AccessctlEntity) AfterSave(tx *gorm.DB) (err error) {
	go initAccCache()
	return nil
}

func (s *AccessctlEntity) AfterDelete(tx *gorm.DB) (err error) {
	go initAccCache()
	return nil
}

type accessctl struct {
	Path   string
	Access int
	Ips    string
}

func initAccCache() {
	time.Sleep(100 * time.Millisecond)
	var objs []AccessctlEntity

	var objmap = make(map[string]accessctl)
	if err := dao.ConfDb.Where("status = ?", 1).Find(&objs).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return
		}
		return
	}
	if len(objs) > 0 {
		var accs []accessctl
		utils.SliceToSlice(objs, &accs)
		for _, a := range accs {
			objmap[a.Path] = a
		}
		cache.SetCache("accessctlMap", objmap, time.Now().Add(365*24*time.Hour))
	}
}