feat:添加月幕galgame网站图

README.md

@@ -287,6 +287,12 @@ zerobot [-h] [-t token] [-u url] [-n nickname] [-p prefix] [-d|w] [-g 监听地
 - **煎蛋网无聊图** `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin/jandan"`
     - [x] 来份屌图
     - [x] 更新屌图
+- **月幕galgame图** `import _ "github.com/FloatTech/ZeroBot-Plugin/plugin/ymgal"`
+    - [x] 随机galCG
+    - [x] 随机gal表情包
+    - [x] galCG[xxx]
+    - [x] gal表情包[xxx]
+    - [x] 更新gal
 - **TODO...**
 
 ## 使用方法

main.go

@@ -100,6 +100,7 @@ import (
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/vtb_quotation"  // vtb语录
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/wangyiyun"      // 网易云音乐热评
 	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/wordle"         // 猜单词
+	_ "github.com/FloatTech/ZeroBot-Plugin/plugin/ymgal"          // 月幕galgame
 
 	// _ "github.com/FloatTech/ZeroBot-Plugin/plugin/wtf"            // 鬼东西
 	// _ "github.com/FloatTech/ZeroBot-Plugin/plugin/bilibili_push"  // b站推送

plugin/ai_reply/tts.go → plugin/ai_reply/ai_tts.go

@@ -4,6 +4,7 @@ import (
 	"errors"
 	"regexp"
 	"strconv"
+	"sync"
 
 	"github.com/pkumza/numcn"
 	log "github.com/sirupsen/logrus"
@@ -38,18 +39,26 @@ var (
 )
 
 type ttsInstances struct {
+	sync.RWMutex
 	m map[string]tts.TTS
 	l []string
 }
 
 func (t *ttsInstances) List() []string {
-	return t.l
+	t.RLock()
+	cl := make([]string, len(t.l))
+	_ = copy(cl, t.l)
+	t.RUnlock()
+	return cl
 }
 
 func init() {
 	engine := control.Register(ttsServiceName, order.AcquirePrio(), &control.Options{
-		DisableOnDefault: false,
-		Help:             "语音回复(包括拟声鸟和百度)\n- @Bot 任意文本(任意一句话回复)\n- 设置语音模式拟声鸟阿梓 | 设置语音模式拟声鸟药水哥 | 设置语音模式百度女声 | 设置语音模式百度男声| 设置语音模式百度度逍遥 | 设置语音模式百度度丫丫",
+		DisableOnDefault: true,
+		Help: "语音回复(包括拟声鸟和百度)\n" +
+			"- @Bot 任意文本(任意一句话回复)\n" +
+			"- 设置语音模式[拟声鸟阿梓 | 拟声鸟药水哥 | 百度女声 | 百度男声| 百度度逍遥 | 百度度丫丫]\n" +
+			"- 设置默认语音模式[拟声鸟阿梓 | 拟声鸟药水哥 | 百度女声 | 百度男声| 百度度逍遥 | 百度度丫丫]\n",
 	})
 	engine.OnMessage(zero.OnlyToMe).SetBlock(true).Limit(ctxext.LimitByUser).
 		Handle(func(ctx *zero.Ctx) {
@@ -78,7 +87,13 @@ func init() {
 				ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text(err))
 				return
 			}
-			ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("成功"))
+			ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("设置成功，当前模式为", param))
+		})
+	engine.OnRegex(`^设置默认语音模式(.*)$`, ctxext.FirstValueInList(t)).SetBlock(true).
+		Handle(func(ctx *zero.Ctx) {
+			param := ctx.State["regex_matched"].([]string)[1]
+			t.setDefaultSoundMode(param)
+			ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Text("设置成功，默认模式为", param))
 		})
 }
 
@@ -93,12 +108,14 @@ func (t *ttsInstances) setSoundMode(ctx *zero.Ctx, name string) error {
 		gid = -ctx.Event.UserID
 	}
 	var index int64
+	t.RLock()
 	for i, s := range t.l {
 		if s == name {
 			index = int64(i)
 			break
 		}
 	}
+	t.RUnlock()
 	m, ok := control.Lookup(ttsServiceName)
 	if !ok {
 		return errors.New("no such plugin")
@@ -113,10 +130,27 @@ func (t *ttsInstances) getSoundMode(ctx *zero.Ctx) (name string) {
 	}
 	m, ok := control.Lookup(ttsServiceName)
 	if ok {
+		t.RLock()
+		defer t.RUnlock()
 		index := m.GetData(gid)
 		if int(index) < len(t.l) {
 			return t.l[index]
 		}
 	}
 	return "拟声鸟阿梓"
 }
+
+func (t *ttsInstances) setDefaultSoundMode(name string) {
+	var index int
+	t.RLock()
+	for _, s := range t.l {
+		if s == name {
+			break
+		}
+		index++
+	}
+	t.RUnlock()
+	t.Lock()
+	t.l[0], t.l[index] = t.l[index], t.l[0]
+	t.Unlock()
+}

plugin/ymgal/model.go

@@ -0,0 +1,276 @@
+package ymgal
+
+import (
+	"fmt"
+	"github.com/antchfx/htmlquery"
+	_ "github.com/fumiama/sqlite3" // import sql
+	"github.com/jinzhu/gorm"
+	log "github.com/sirupsen/logrus"
+	"math/rand"
+	"net/url"
+	"os"
+	"regexp"
+	"strconv"
+	"sync"
+	"time"
+)
+
+// gdb 得分数据库
+var gdb *ymgaldb
+
+// ymgaldb galgame图片数据库
+type ymgaldb gorm.DB
+
+var mu sync.RWMutex
+
+// ymgal gal图片储存结构体
+type ymgal struct {
+	ID                 int64  `gorm:"column:id" `
+	Title              string `gorm:"column:title" `
+	PictureType        string `gorm:"column:picture_type" `
+	PictureDescription string `gorm:"column:picture_description;type:varchar(1024)" `
+	PictureList        string `gorm:"column:picture_list;type:varchar(20000)" `
+}
+
+// TableName ...
+func (ymgal) TableName() string {
+	return "ymgal"
+}
+
+// initialize 初始化ymgaldb数据库
+func initialize(dbpath string) *ymgaldb {
+	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(&ymgal{})
+	return (*ymgaldb)(gdb)
+}
+
+func (gdb *ymgaldb) insertOrUpdateYmgalByID(id int64, title, pictureType, pictureDescription, pictureList string) (err error) {
+	db := (*gorm.DB)(gdb)
+	y := ymgal{
+		ID:                 id,
+		Title:              title,
+		PictureType:        pictureType,
+		PictureDescription: pictureDescription,
+		PictureList:        pictureList,
+	}
+	if err = db.Debug().Model(&ymgal{}).First(&y, "id = ? ", id).Error; err != nil {
+		if gorm.IsRecordNotFoundError(err) {
+			err = db.Debug().Model(&ymgal{}).Create(&y).Error // newUser not user
+		}
+	} else {
+		err = db.Debug().Model(&ymgal{}).Where("id = ? ", id).Update(map[string]interface{}{
+			"title":               title,
+			"picture_type":        pictureType,
+			"picture_description": pictureDescription,
+			"picture_list":        pictureList,
+		}).Error
+	}
+	return
+}
+
+func (gdb *ymgaldb) getYmgalByID(id string) (y ymgal) {
+	db := (*gorm.DB)(gdb)
+	db.Debug().Model(&ymgal{}).Where("id = ?", id).Take(&y)
+	return
+}
+
+func (gdb *ymgaldb) randomYmgal(pictureType string) (y ymgal) {
+	db := (*gorm.DB)(gdb)
+	var count int
+	s := db.Debug().Model(&ymgal{}).Where("picture_type = ?", pictureType).Count(&count)
+	if count == 0 {
+		return
+	}
+	s.Offset(rand.Intn(count)).Take(&y)
+	return
+}
+
+func (gdb *ymgaldb) getYmgalByKey(pictureType, key string) (y ymgal) {
+	db := (*gorm.DB)(gdb)
+	var count int
+	s := db.Debug().Model(&ymgal{}).Where("picture_type = ? and (picture_description like ? or title like ?) ", pictureType, "%"+key+"%", "%"+key+"%").Count(&count)
+	if count == 0 {
+		return
+	}
+	s.Offset(rand.Intn(count)).Take(&y)
+	return
+}
+
+const (
+	webURL       = "https://www.ymgal.com"
+	cgType       = "Gal CG"
+	emoticonType = "其他"
+	webPicURL    = webURL + "/co/picset/"
+	reNumber     = `\d+`
+)
+
+var (
+	cgURL                = webURL + "/search?type=picset&sort=default&category=" + url.QueryEscape(cgType) + "&page="
+	emoticonURL          = webURL + "/search?type=picset&sort=default&category=" + url.QueryEscape(emoticonType) + "&page="
+	commonPageNumberExpr = "//*[@id='pager-box']/div/a[@class='icon item pager-next']/preceding-sibling::a[1]/text()"
+	cgIDList             []string
+	emoticonIDList       []string
+)
+
+func initPageNumber() (maxCgPageNumber, maxEmoticonPageNumber int) {
+	doc, err := htmlquery.LoadURL(cgURL + "1")
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+	maxCgPageNumber, err = strconv.Atoi(htmlquery.FindOne(doc, commonPageNumberExpr).Data)
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+	doc, err = htmlquery.LoadURL(emoticonURL + "1")
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+	maxEmoticonPageNumber, err = strconv.Atoi(htmlquery.FindOne(doc, commonPageNumberExpr).Data)
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+	return
+}
+
+func getPicID(pageNumber int, pictureType string) {
+	var picURL string
+	if pictureType == cgType {
+		picURL = cgURL + strconv.Itoa(pageNumber)
+	} else if pictureType == emoticonType {
+		picURL = emoticonURL + strconv.Itoa(pageNumber)
+	}
+	doc, err := htmlquery.LoadURL(picURL)
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+	list := htmlquery.Find(doc, "//*[@id='picset-result-list']/ul/div/div[1]/a")
+	for i := 0; i < len(list); i++ {
+		re := regexp.MustCompile(reNumber)
+		picID := re.FindString(list[i].Attr[0].Val)
+		if pictureType == cgType {
+			cgIDList = append(cgIDList, picID)
+		} else if pictureType == emoticonType {
+			emoticonIDList = append(emoticonIDList, picID)
+		}
+	}
+
+}
+
+func updatePic() {
+	maxCgPageNumber, maxEmoticonPageNumber := initPageNumber()
+	for i := 1; i <= maxCgPageNumber; i++ {
+		getPicID(i, cgType)
+		time.Sleep(time.Millisecond * 500)
+	}
+	for i := 1; i <= maxEmoticonPageNumber; i++ {
+		getPicID(i, emoticonType)
+		time.Sleep(time.Millisecond * 500)
+	}
+CGLOOP:
+	for i := len(cgIDList) - 1; i >= 0; i-- {
+		mu.RLock()
+		y := gdb.getYmgalByID(cgIDList[i])
+		mu.RUnlock()
+		if y.PictureList == "" {
+			mu.Lock()
+			storeCgPic(cgIDList[i])
+			mu.Unlock()
+		} else {
+			break CGLOOP
+		}
+		time.Sleep(time.Millisecond * 500)
+	}
+EMOTICONLOOP:
+	for i := len(emoticonIDList) - 1; i >= 0; i-- {
+		mu.RLock()
+		y := gdb.getYmgalByID(emoticonIDList[i])
+		mu.RUnlock()
+		if y.PictureList == "" {
+			mu.Lock()
+			storeEmoticonPic(emoticonIDList[i])
+			mu.Unlock()
+		} else {
+			break EMOTICONLOOP
+		}
+		time.Sleep(time.Millisecond * 500)
+	}
+}
+
+func storeCgPic(picIDStr string) {
+	picID, err := strconv.ParseInt(picIDStr, 10, 64)
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+	pictureType := cgType
+	doc, err := htmlquery.LoadURL(webPicURL + picIDStr)
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+	title := htmlquery.FindOne(doc, "//meta[@name='name']").Attr[1].Val
+	pictureDescription := htmlquery.FindOne(doc, "//meta[@name='description']").Attr[1].Val
+	pictureNumberStr := htmlquery.FindOne(doc, "//div[@class='meta-info']/div[@class='meta-right']/span[2]/text()").Data
+	re := regexp.MustCompile(reNumber)
+	pictureNumber, err := strconv.Atoi(re.FindString(pictureNumberStr))
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+	pictureList := ""
+	for i := 1; i <= pictureNumber; i++ {
+		picURL := htmlquery.FindOne(doc, fmt.Sprintf("//*[@id='main-picset-warp']/div/div[2]/div/div[@class='swiper-wrapper']/div[%d]", i)).Attr[1].Val
+		if i == 1 {
+			pictureList += picURL
+		} else {
+			pictureList += "," + picURL
+		}
+	}
+	err = gdb.insertOrUpdateYmgalByID(picID, title, pictureType, pictureDescription, pictureList)
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+
+}
+
+func storeEmoticonPic(picIDStr string) {
+	picID, err := strconv.ParseInt(picIDStr, 10, 64)
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+	pictureType := emoticonType
+	doc, err := htmlquery.LoadURL(webPicURL + picIDStr)
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+	title := htmlquery.FindOne(doc, "//meta[@name='name']").Attr[1].Val
+	pictureDescription := htmlquery.FindOne(doc, "//meta[@name='description']").Attr[1].Val
+	pictureNumberStr := htmlquery.FindOne(doc, "//div[@class='meta-info']/div[@class='meta-right']/span[2]/text()").Data
+	re := regexp.MustCompile(reNumber)
+	pictureNumber, err := strconv.Atoi(re.FindString(pictureNumberStr))
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+	pictureList := ""
+	for i := 1; i <= pictureNumber; i++ {
+		picURL := htmlquery.FindOne(doc, fmt.Sprintf("//*[@id='main-picset-warp']/div/div[@class='stream-list']/div[%d]/img", i)).Attr[1].Val
+		if i == 1 {
+			pictureList += picURL
+		} else {
+			pictureList += "," + picURL
+		}
+	}
+	err = gdb.insertOrUpdateYmgalByID(picID, title, pictureType, pictureDescription, pictureList)
+	if err != nil {
+		log.Errorln("[ymgal]:", err)
+	}
+}