mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 22:00:11 +08:00
131 lines
2.8 KiB
Go
131 lines
2.8 KiB
Go
package bilibili
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/FloatTech/zbputils/binary"
|
|
"github.com/FloatTech/zbputils/web"
|
|
_ "github.com/fumiama/sqlite3" // use sql
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
const (
|
|
bilibiliCookie = "bilbili_cookie"
|
|
)
|
|
|
|
var (
|
|
vtbURLs = [...]string{"https://api.vtbs.moe/v1/short", "https://api.tokyo.vtbs.moe/v1/short", "https://vtbs.musedash.moe/v1/short"}
|
|
vdb *vupdb
|
|
)
|
|
|
|
// vupdb 分数数据库
|
|
type vupdb gorm.DB
|
|
|
|
type vup struct {
|
|
Mid int64 `gorm:"column:mid;primary_key"`
|
|
Uname string `gorm:"column:uname"`
|
|
Roomid int64 `gorm:"column:roomid"`
|
|
}
|
|
|
|
func (vup) TableName() string {
|
|
return "vup"
|
|
}
|
|
|
|
type config struct {
|
|
Key string `gorm:"column:key;primary_key"`
|
|
Value string `gorm:"column:value"`
|
|
}
|
|
|
|
func (config) TableName() string {
|
|
return "config"
|
|
}
|
|
|
|
// initializeVup 初始化vup数据库
|
|
func initializeVup(dbpath string) (*vupdb, error) {
|
|
if _, err := os.Stat(dbpath); err != nil || os.IsNotExist(err) {
|
|
// 生成文件
|
|
f, err := os.Create(dbpath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer f.Close()
|
|
}
|
|
gdb, err := gorm.Open("sqlite3", dbpath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
gdb.AutoMigrate(&vup{}).AutoMigrate(&config{})
|
|
return (*vupdb)(gdb), nil
|
|
}
|
|
|
|
func (vdb *vupdb) insertVupByMid(mid int64, uname string, roomid int64) (err error) {
|
|
db := (*gorm.DB)(vdb)
|
|
v := vup{
|
|
Mid: mid,
|
|
Uname: uname,
|
|
Roomid: roomid,
|
|
}
|
|
if err = db.Model(&vup{}).First(&v, "mid = ? ", mid).Error; err != nil {
|
|
if gorm.IsRecordNotFoundError(err) {
|
|
err = db.Model(&vup{}).Create(&v).Error
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// filterVup 筛选vup
|
|
func (vdb *vupdb) filterVup(ids []int64) (vups []vup, err error) {
|
|
db := (*gorm.DB)(vdb)
|
|
if err = db.Model(&vup{}).Find(&vups, "mid in (?)", ids).Error; err != nil {
|
|
return vups, err
|
|
}
|
|
return
|
|
}
|
|
|
|
func updateVup() error {
|
|
for _, v := range vtbURLs {
|
|
data, err := web.GetData(v)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
gjson.Get(binary.BytesToString(data), "@this").ForEach(func(key, value gjson.Result) bool {
|
|
mid := value.Get("mid").Int()
|
|
uname := value.Get("uname").String()
|
|
roomid := value.Get("roomid").Int()
|
|
err = vdb.insertVupByMid(mid, uname, roomid)
|
|
return err == nil
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (vdb *vupdb) setBilibiliCookie(cookie string) (err error) {
|
|
db := (*gorm.DB)(vdb)
|
|
c := config{
|
|
Key: bilibiliCookie,
|
|
Value: cookie,
|
|
}
|
|
if err = db.Model(&config{}).First(&c, "key = ? ", bilibiliCookie).Error; err != nil {
|
|
// error handling...
|
|
if gorm.IsRecordNotFoundError(err) {
|
|
err = db.Model(&config{}).Create(&c).Error
|
|
}
|
|
} else {
|
|
err = db.Model(&config{}).Where("key = ? ", bilibiliCookie).Update(
|
|
map[string]interface{}{
|
|
"value": cookie,
|
|
}).Error
|
|
}
|
|
return
|
|
}
|
|
|
|
func (vdb *vupdb) getBilibiliCookie() (c config) {
|
|
db := (*gorm.DB)(vdb)
|
|
db.Model(&config{}).First(&c, "key = ?", bilibiliCookie)
|
|
return
|
|
}
|