package service

import (
	"github.com/gin-gonic/gin"
	"mc-client-updater-server/pkg/dao"
	"mc-client-updater-server/pkg/dao/entity"
	"mc-client-updater-server/pkg/util"
)

type InstanceService struct {
	ctx *gin.Context
}

func NewInstanceService(c *gin.Context) *InstanceService {
	return &InstanceService{ctx: c}
}

func (s *InstanceService) AddInstance(name, updateURL string) (*entity.Instance, error) {
	instEntity := entity.Instance{
		Name:      name,
		UpdateURL: updateURL,
	}
	tx := dao.DB().Create(&instEntity)
	if tx.Error != nil {
		return nil, tx.Error
	}

	tx = dao.DB().Where(&instEntity).Last(&instEntity)
	return &instEntity, tx.Error
}

func (s *InstanceService) GetInstanceByName(name string) (*entity.Instance, error) {
	instEntity := entity.Instance{}
	tx := dao.DB().Where("name=?", name).Last(&instEntity)
	return &instEntity, tx.Error
}

func (s *InstanceService) NewGrantToken(instName string, expireStr string) (*entity.Grant, error) {
	expireAt := util.MustParseSQLTime(expireStr)
	grantEntity := entity.Grant{GrantTo: instName, ExpireAt: expireAt, Token: util.RandStr(32)}
	tx := dao.DB().Create(&grantEntity)
	if tx.Error != nil {
		return nil, tx.Error
	}

	tx = dao.DB().Where(&grantEntity).Last(&grantEntity)
	return &grantEntity, tx.Error
}

func (s *InstanceService) GetGrantByToken(token string) (*entity.Grant, error) {
	grantEntity := entity.Grant{}
	tx := dao.DB().Where("token=?", token).Last(&grantEntity)
	return &grantEntity, tx.Error
}