添加vtb语音包 (#74)

* fix:给msg进行url编码，解决青云客接口返回错误 * feat:增加一个vtbkeyboard.moe接口 * fix:id添加映射 * feat:优化逻辑 * feat:优化逻辑 * feat:优化逻辑 * feat:优化逻辑 * feat:增加一个随机vtb * feat:增加一个随机vtb * refactor:更改代码逻辑 * feat:更新数据库 * feat:更新数据库 * feat:更新数据库 * feat:更新数据库 * feat:更新数据库驱动 * feat:增加定时更新数据脚本 * feat:更换驱动 * fix:更新zerobot * fix:修改全局变量 * fix:微调位置 * fix:解决modernc.org/sqlite版本冲突 * fix:解决go.sum冲突,更换cron库
2026-02-12 18:20:27 +00:00 · 2021-11-20 13:34:37 +08:00
parent a537306307
commit e812d2a074
11 changed files with 629 additions and 7 deletions
--- a/plugin_vtb_quotation/firstVtb/firstVtb.go
+++ b/plugin_vtb_quotation/firstVtb/firstVtb.go
@@ -0,0 +1,78 @@
+package firstVtb
+
+import (
+	"github.com/FloatTech/ZeroBot-Plugin/plugin_vtb_quotation/model"
+	"github.com/FloatTech/ZeroBot-Plugin/plugin_vtb_quotation/utils"
+	"github.com/jinzhu/gorm"
+	log "github.com/sirupsen/logrus"
+	"github.com/tidwall/gjson"
+	"io/ioutil"
+	"net/http"
+	"strconv"
+	"strings"
+)
+
+var vtbUrl = "https://vtbkeyboard.moe/api/get_vtb_list"
+
+func GetVtbListStr() string {
+	client := &http.Client{}
+	req, err := http.NewRequest("GET", vtbUrl, nil)
+	if err != nil {
+		log.Println(err)
+	}
+	// 自定义Header
+	req.Header.Set("User-Agent", utils.GetAgent())
+	resp, err := client.Do(req)
+	if err != nil {
+		log.Println(err)
+	}
+
+	defer resp.Body.Close()
+	bytes, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		log.Println(err)
+	}
+	//log.Println(string(bytes))
+	vtbListStr, err := strconv.Unquote(strings.Replace(strconv.Quote(string(bytes)), `\\u`, `\u`, -1))
+	if err != nil {
+		log.Println(err)
+	}
+	log.Println(vtbListStr)
+	return vtbListStr
+}
+func DealVtbListStr(vtbListStr string) []string {
+	uidList := make([]string, 0)
+	count := gjson.Get(vtbListStr, "#").Int()
+	for i := int64(0); i < count; i++ {
+		item := gjson.Get(vtbListStr, strconv.FormatInt(i, 10))
+		log.Println(item)
+		fc := model.FirstCategory{
+			FirstCategoryIndex:       i,
+			FirstCategoryName:        item.Get("name").String(),
+			FirstCategoryDescription: item.Get("description").String(),
+			FirstCategoryIconPath:    item.Get("icon_path").String(),
+			FirstCategoryUid:         item.Get("uid").String(),
+		}
+		log.Println(fc)
+		//model.Db.Model(FirstCategory{}).Where("first_category_uid = ?", fc.FirstCategoryUid).FirstOrCreate(&fc)
+		if err := model.Db.Debug().Model(&model.FirstCategory{}).Where("first_category_uid = ?", fc.FirstCategoryUid).First(&fc).Error; err != nil {
+			// error handling...
+			if gorm.IsRecordNotFoundError(err) {
+				model.Db.Debug().Model(&model.FirstCategory{}).Create(&fc) // newUser not user
+			}
+		} else {
+			model.Db.Debug().Model(&model.FirstCategory{}).Where("first_category_uid = ?", fc.FirstCategoryUid).Update(
+				map[string]interface{}{
+					"first_category_index":       i,
+					"first_category_name":        item.Get("name").String(),
+					"first_category_description": item.Get("description").String(),
+					"first_category_icon_path":   item.Get("icon_path").String(),
+				})
+		}
+		uidList = append(uidList, fc.FirstCategoryUid)
+
+	}
+
+	log.Println(uidList)
+	return uidList
+}