diff --git a/login/internal/cmd/seerrobot.go b/login/internal/cmd/seerrobot.go index bdbaaba35..ef1204bcb 100644 --- a/login/internal/cmd/seerrobot.go +++ b/login/internal/cmd/seerrobot.go @@ -1,6 +1,7 @@ package cmd import ( + _ "blazing/modules/config/controller/robot" _ "blazing/modules/player/controller/robot" // _ "github.com/FloatTech/ZeroBot-Plugin/plugin/antiabuse" // 违禁词 diff --git a/modules/config/controller/robot/cdk.go b/modules/config/controller/robot/cdk.go new file mode 100644 index 000000000..71f9ce187 --- /dev/null +++ b/modules/config/controller/robot/cdk.go @@ -0,0 +1,26 @@ +package robot + +import ( + "blazing/modules/config/service" + "strings" + "time" + + zero "github.com/wdvxdr1123/ZeroBot" +) + +func init() { + zero.OnCommand("CDK"). + Handle(func(ctx *zero.Ctx) { + + var cdks []string + for _, v := range service.NewCdkService().All() { + if v.ValidEndTime.After(time.Now()) { + cdks = append(cdks, v.CDKCode) + + } + + } + + ctx.Send(strings.Join(cdks, "\n")) + }) +} diff --git a/modules/config/service/cdk.go b/modules/config/service/cdk.go index 545d8c6cd..9928f1d29 100644 --- a/modules/config/service/cdk.go +++ b/modules/config/service/cdk.go @@ -44,6 +44,13 @@ func (s *CdkService) Get(id string) *model.CDKConfig { return item +} +func (s *CdkService) All() []model.CDKConfig { + var item []model.CDKConfig + dbm_notenable(s.Model).WhereLT("exchange_remain_count", 0).Scan(&item) + + return item + } func (s *CdkService) Set(id string) bool {