🎨 优化目录结构

2026-02-12 02:00:24 +00:00 · 2022-02-25 22:15:14 +08:00
parent 5ccf753af3
commit 0cfb2e4e06
101 changed files with 116 additions and 116 deletions
--- a/plugin/sleep_manage/model/model.go
+++ b/plugin/sleep_manage/model/model.go
@@ -0,0 +1,120 @@
+// Package model 睡眠管理数据库
+package model
+
+import (
+	"os"
+	"time"
+
+	_ "github.com/fumiama/sqlite3" // use sql
+	"github.com/jinzhu/gorm"
+	log "github.com/sirupsen/logrus"
+)
+
+// SleepDB 睡眠数据库
+type SleepDB gorm.DB
+
+// Initialize 初始化
+func Initialize(dbpath string) *SleepDB {
+	var err error
+	if _, err = os.Stat(dbpath); err != nil || os.IsNotExist(err) {
+		// 生成文件
+		f, err := os.Create(dbpath)
+		if err != nil {
+			return nil
+		}
+		defer f.Close()
+	}
+	gdb, err := gorm.Open("sqlite3", dbpath)
+	if err != nil {
+		panic(err)
+	}
+	gdb.AutoMigrate(&SleepManage{})
+	return (*SleepDB)(gdb)
+}
+
+// Open 打开
+func Open(dbpath string) (*SleepDB, error) {
+	db, err := gorm.Open("sqlite3", dbpath)
+	if err != nil {
+		return nil, err
+	}
+	return (*SleepDB)(db), nil
+}
+
+// Close 关闭
+func (sdb *SleepDB) Close() error {
+	db := (*gorm.DB)(sdb)
+	return db.Close()
+}
+
+// SleepManage 睡眠信息
+type SleepManage struct {
+	gorm.Model
+	GroupID   int64     `gorm:"column:group_id"`
+	UserID    int64     `gorm:"column:user_id"`
+	SleepTime time.Time `gorm:"column:sleep_time"`
+}
+
+// TableName 表名
+func (SleepManage) TableName() string {
+	return "sleep_manage"
+}
+
+// Sleep 更新睡眠时间
+func (sdb *SleepDB) Sleep(gid, uid int64) (position int, awakeTime time.Duration) {
+	db := (*gorm.DB)(sdb)
+	now := time.Now()
+	var today time.Time
+	if now.Hour() >= 21 {
+		today = now.Add(-time.Hour*time.Duration(-21+now.Hour()) - time.Minute*time.Duration(now.Minute()) - time.Second*time.Duration(now.Second()))
+	} else if now.Hour() <= 3 {
+		today = now.Add(-time.Hour*time.Duration(3+now.Hour()) - time.Minute*time.Duration(now.Minute()) - time.Second*time.Duration(now.Second()))
+	}
+	st := SleepManage{
+		GroupID:   gid,
+		UserID:    uid,
+		SleepTime: now,
+	}
+	if err := db.Debug().Model(&SleepManage{}).Where("group_id = ? and user_id = ?", gid, uid).First(&st).Error; err != nil {
+		// error handling...
+		if gorm.IsRecordNotFoundError(err) {
+			db.Debug().Model(&SleepManage{}).Create(&st) // newUser not user
+		}
+	} else {
+		log.Println("sleeptime为", st)
+		awakeTime = now.Sub(st.SleepTime)
+		db.Debug().Model(&SleepManage{}).Where("group_id = ? and user_id = ?", gid, uid).Update(
+			map[string]interface{}{
+				"sleep_time": now,
+			})
+	}
+	db.Debug().Model(&SleepManage{}).Where("group_id = ? and sleep_time <= ? and sleep_time >= ?", gid, now, today).Count(&position)
+	return position, awakeTime
+}
+
+// GetUp 更新起床时间
+func (sdb *SleepDB) GetUp(gid, uid int64) (position int, sleepTime time.Duration) {
+	db := (*gorm.DB)(sdb)
+	now := time.Now()
+	today := now.Add(-time.Hour*time.Duration(-6+now.Hour()) - time.Minute*time.Duration(now.Minute()) - time.Second*time.Duration(now.Second()))
+	st := SleepManage{
+		GroupID:   gid,
+		UserID:    uid,
+		SleepTime: now,
+	}
+	if err := db.Debug().Model(&SleepManage{}).Where("group_id = ? and user_id = ?", gid, uid).First(&st).Error; err != nil {
+		// error handling...
+		if gorm.IsRecordNotFoundError(err) {
+			db.Debug().Model(&SleepManage{}).Create(&st) // newUser not user
+		}
+	} else {
+		log.Println("sleeptime为", st)
+		sleepTime = now.Sub(st.SleepTime)
+		db.Debug().Model(&SleepManage{}).Where("group_id = ? and user_id = ?", gid, uid).Update(
+			map[string]interface{}{
+				"sleep_time": now,
+			})
+	}
+	db.Debug().Model(&SleepManage{}).Where("group_id = ? and sleep_time <= ? and sleep_time >= ?", gid, now, today).Count(&position)
+	return position, sleepTime
+}
--- a/plugin/sleep_manage/sleep_manage.go
+++ b/plugin/sleep_manage/sleep_manage.go
@@ -0,0 +1,79 @@
+// Package sleepmanage 睡眠管理
+package sleepmanage
+
+import (
+	"fmt"
+	"time"
+
+	log "github.com/sirupsen/logrus"
+	zero "github.com/wdvxdr1123/ZeroBot"
+	"github.com/wdvxdr1123/ZeroBot/message"
+
+	control "github.com/FloatTech/zbputils/control"
+
+	"github.com/FloatTech/zbputils/control/order"
+
+	"github.com/FloatTech/ZeroBot-Plugin/plugin/sleep_manage/model"
+)
+
+func init() {
+	engine := control.Register("sleepmanage", order.AcquirePrio(), &control.Options{
+		DisableOnDefault:  false,
+		Help:              "sleepmanage\n- 早安\n- 晚安",
+		PrivateDataFolder: "sleep",
+	})
+	dbfile := engine.DataFolder() + "manage.db"
+	engine.OnFullMatch("早安", isMorning, zero.OnlyGroup).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			db, err := model.Open(dbfile)
+			if err != nil {
+				log.Errorln(err)
+				return
+			}
+			position, getUpTime := db.GetUp(ctx.Event.GroupID, ctx.Event.UserID)
+			log.Println(position, getUpTime)
+			hour, minute, second := timeDuration(getUpTime)
+			if (hour == 0 && minute == 0 && second == 0) || hour >= 24 {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(fmt.Sprintf("早安成功！你是今天第%d个起床的", position)))
+			} else {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(fmt.Sprintf("早安成功！你的睡眠时长为%d时%d分%d秒,你是今天第%d个起床的", hour, minute, second, position)))
+			}
+			db.Close()
+		})
+	engine.OnFullMatch("晚安", isEvening, zero.OnlyGroup).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			db, err := model.Open(dbfile)
+			if err != nil {
+				log.Errorln(err)
+				return
+			}
+			position, sleepTime := db.Sleep(ctx.Event.GroupID, ctx.Event.UserID)
+			log.Println(position, sleepTime)
+			hour, minute, second := timeDuration(sleepTime)
+			if (hour == 0 && minute == 0 && second == 0) || hour >= 24 {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(fmt.Sprintf("晚安成功！你是今天第%d个睡觉的", position)))
+			} else {
+				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(fmt.Sprintf("晚安成功！你的清醒时长为%d时%d分%d秒,你是今天第%d个睡觉的", hour, minute, second, position)))
+			}
+			db.Close()
+		})
+}
+
+func timeDuration(time time.Duration) (hour, minute, second int64) {
+	hour = int64(time) / (1000 * 1000 * 1000 * 60 * 60)
+	minute = (int64(time) - hour*(1000*1000*1000*60*60)) / (1000 * 1000 * 1000 * 60)
+	second = (int64(time) - hour*(1000*1000*1000*60*60) - minute*(1000*1000*1000*60)) / (1000 * 1000 * 1000)
+	return hour, minute, second
+}
+
+// 只统计6点到12点的早安
+func isMorning(ctx *zero.Ctx) bool {
+	now := time.Now().Hour()
+	return now >= 6 && now <= 12
+}
+
+// 只统计21点到凌晨3点的晚安
+func isEvening(ctx *zero.Ctx) bool {
+	now := time.Now().Hour()
+	return now >= 21 || now <= 3
+}