✨ 插件名称自动注册

2026-02-12 10:10:25 +00:00 · 2023-09-03 13:53:25 +08:00
parent 5485cc3be9
commit 6bff62b91b
108 changed files with 124 additions and 126 deletions
--- a/plugin/sleepmanage/model.go
+++ b/plugin/sleepmanage/model.go
@@ -0,0 +1,112 @@
+package sleepmanage
+
+import (
+	"os"
+	"time"
+
+	"github.com/jinzhu/gorm"
+	log "github.com/sirupsen/logrus"
+)
+
+// sdb 睡眠数据库全局变量
+var sdb *sleepdb
+
+// 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)
+}
+
+// Close 关闭
+func (sdb *sleepdb) Close() error {
+	db := (*gorm.DB)(sdb)
+	return db.Close()
+}
+
+// SleepManage 睡眠信息
+type SleepManage struct {
+	ID        uint      `gorm:"primary_key"`
+	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.Model(&SleepManage{}).Where("group_id = ? and user_id = ?", gid, uid).First(&st).Error; err != nil {
+		// error handling...
+		if gorm.IsRecordNotFoundError(err) {
+			db.Model(&SleepManage{}).Create(&st) // newUser not user
+		}
+	} else {
+		log.Debugln("sleeptime为", st)
+		awakeTime = now.Sub(st.SleepTime)
+		db.Model(&SleepManage{}).Where("group_id = ? and user_id = ?", gid, uid).Update(
+			map[string]any{
+				"sleep_time": now,
+			})
+	}
+	db.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.Model(&SleepManage{}).Where("group_id = ? and user_id = ?", gid, uid).First(&st).Error; err != nil {
+		// error handling...
+		if gorm.IsRecordNotFoundError(err) {
+			db.Model(&SleepManage{}).Create(&st) // newUser not user
+		}
+	} else {
+		log.Debugln("sleeptime为", st)
+		sleepTime = now.Sub(st.SleepTime)
+		db.Model(&SleepManage{}).Where("group_id = ? and user_id = ?", gid, uid).Update(
+			map[string]any{
+				"sleep_time": now,
+			})
+	}
+	db.Model(&SleepManage{}).Where("group_id = ? and sleep_time <= ? and sleep_time >= ?", gid, now, today).Count(&position)
+	return position, sleepTime
+}