🐛 修改听歌问题

Author: guohuiyuan

README.md

@@ -1026,6 +1026,10 @@ print("run[CQ:image,file="+j["img"]+"]")
   - [x] 酷我点歌[xxx]
 
   - [x] 酷狗点歌[xxx]
+  
+  - [x] qq点歌[xxx]
+  
+  - [x] 咪咕点歌[xxx]
 
 </details>
 <details>

console/console_windows.go

@@ -38,12 +38,18 @@ func setConsoleTitle(title string) (err error) {
 }
 
 func init() {
+	debugMode := os.Getenv("DEBUG_MODE") == "1"
 	stdin := windows.Handle(os.Stdin.Fd())
 
 	var mode uint32
 	err := windows.GetConsoleMode(stdin, &mode)
 	if err != nil {
-		panic(err)
+		if debugMode {
+			logrus.Warnf("调试模式下忽略控制台模式获取失败: %v", err)
+			return // 调试模式下直接返回，跳过后续配置
+		} else {
+			panic(err) // 非调试模式下 panic
+		}
 	}
 
 	mode &^= windows.ENABLE_QUICK_EDIT_MODE // 禁用快速编辑模式

plugin/music/selecter.go

@@ -29,7 +29,8 @@ func init() {
 			"- 网易点歌[xxx]\n" +
 			"- 酷我点歌[xxx]\n" +
 			"- 酷狗点歌[xxx]\n" +
-			"- 咪咕点歌[xxx]",
+			"- 咪咕点歌[xxx]\n" +
+			"- qq点歌[xxx]\n",
 	}).OnRegex(`^(.{0,2})点歌\s?(.{1,25})$`).SetBlock(true).Limit(ctxext.LimitByUser).
 		Handle(func(ctx *zero.Ctx) {
 			// switch 平台
@@ -42,8 +43,10 @@ func init() {
 				ctx.SendChain(kugou(ctx.State["regex_matched"].([]string)[2]))
 			case "网易":
 				ctx.SendChain(cloud163(ctx.State["regex_matched"].([]string)[2]))
-			default: // 默认 QQ音乐
+			case "qq":
 				ctx.SendChain(qqmusic(ctx.State["regex_matched"].([]string)[2]))
+			default: // 默认咪咕点歌
+				ctx.SendChain(migu(ctx.State["regex_matched"].([]string)[2]))
 			}
 		})
 }

plugin/score/model.go

@@ -2,6 +2,7 @@ package score
 
 import (
 	"os"
+	"sync"
 	"time"
 
 	"github.com/jinzhu/gorm"
@@ -11,7 +12,10 @@ import (
 var sdb *scoredb
 
 // scoredb 分数数据库
-type scoredb gorm.DB
+type scoredb struct {
+	db      *gorm.DB
+	scoremu sync.Mutex
+}
 
 // scoretable 分数结构体
 type scoretable struct {
@@ -52,25 +56,31 @@ func initialize(dbpath string) *scoredb {
 		panic(err)
 	}
 	gdb.AutoMigrate(&scoretable{}).AutoMigrate(&signintable{})
-	return (*scoredb)(gdb)
+	return &scoredb{
+		db: gdb,
+	}
 }
 
 // Close ...
 func (sdb *scoredb) Close() error {
-	db := (*gorm.DB)(sdb)
+	db := sdb.db
 	return db.Close()
 }
 
 // GetScoreByUID 取得分数
 func (sdb *scoredb) GetScoreByUID(uid int64) (s scoretable) {
-	db := (*gorm.DB)(sdb)
+	sdb.scoremu.Lock()
+	defer sdb.scoremu.Unlock()
+	db := sdb.db
 	db.Model(&scoretable{}).FirstOrCreate(&s, "uid = ? ", uid)
 	return s
 }
 
 // InsertOrUpdateScoreByUID 插入或更新分数
 func (sdb *scoredb) InsertOrUpdateScoreByUID(uid int64, score int) (err error) {
-	db := (*gorm.DB)(sdb)
+	sdb.scoremu.Lock()
+	defer sdb.scoremu.Unlock()
+	db := sdb.db
 	s := scoretable{
 		UID:   uid,
 		Score: score,
@@ -91,14 +101,18 @@ func (sdb *scoredb) InsertOrUpdateScoreByUID(uid int64, score int) (err error) {
 
 // GetSignInByUID 取得签到次数
 func (sdb *scoredb) GetSignInByUID(uid int64) (si signintable) {
-	db := (*gorm.DB)(sdb)
+	sdb.scoremu.Lock()
+	defer sdb.scoremu.Unlock()
+	db := sdb.db
 	db.Model(&signintable{}).FirstOrCreate(&si, "uid = ? ", uid)
 	return si
 }
 
 // InsertOrUpdateSignInCountByUID 插入或更新签到次数
 func (sdb *scoredb) InsertOrUpdateSignInCountByUID(uid int64, count int) (err error) {
-	db := (*gorm.DB)(sdb)
+	sdb.scoremu.Lock()
+	defer sdb.scoremu.Unlock()
+	db := sdb.db
 	si := signintable{
 		UID:   uid,
 		Count: count,
@@ -118,7 +132,9 @@ func (sdb *scoredb) InsertOrUpdateSignInCountByUID(uid int64, count int) (err er
 }
 
 func (sdb *scoredb) GetScoreRankByTopN(n int) (st []scoretable, err error) {
-	db := (*gorm.DB)(sdb)
+	sdb.scoremu.Lock()
+	defer sdb.scoremu.Unlock()
+	db := sdb.db
 	err = db.Model(&scoretable{}).Order("score desc").Limit(n).Find(&st).Error
 	return
 }