Northern_Lights/ZeroBot-Plugin
1
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2025-12-19 13:59:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✏️ vtb drop unsafe

Browse Source
This commit is contained in:
fumiama 2021-11-23 13:05:16 +08:00
parent 3d0a161176
commit 17bb844360
2 changed files with 13 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
plugin_vtb_quotation/cron.go
View File

@ -28,7 +28,7 @@ func vtbDaily() {
} }
func vtbData() { func vtbData() {
db := model.Init(dbfile) db := model.Initialize(dbfile)
if db != nil { if db != nil {
for _, v := range db.GetVtbList() { for _, v := range db.GetVtbList() {
db.StoreVtb(v) db.StoreVtb(v)

25
plugin_vtb_quotation/model/model.go
View File

@ -8,7 +8,6 @@ import (
"strconv" "strconv"
"strings" "strings"
"time" "time"
"unsafe"
"github.com/jinzhu/gorm" "github.com/jinzhu/gorm"
_ "github.com/logoove/sqlite" _ "github.com/logoove/sqlite"
@ -18,7 +17,7 @@ import (
type VtbDB gorm.DB type VtbDB gorm.DB
func Init(dbpath string) *VtbDB { func Initialize(dbpath string) *VtbDB {
var err error var err error
if _, err = os.Stat(dbpath); err != nil || os.IsNotExist(err) { if _, err = os.Stat(dbpath); err != nil || os.IsNotExist(err) {
// 生成文件 // 生成文件
@ -33,7 +32,7 @@ func Init(dbpath string) *VtbDB {
panic(err) panic(err)
} }
gdb.AutoMigrate(FirstCategory{}).AutoMigrate(SecondCategory{}).AutoMigrate(ThirdCategory{}) gdb.AutoMigrate(FirstCategory{}).AutoMigrate(SecondCategory{}).AutoMigrate(ThirdCategory{})
return (*VtbDB)(unsafe.Pointer(gdb)) return (*VtbDB)(gdb)
} }
func Open(dbpath string) (*VtbDB, error) { func Open(dbpath string) (*VtbDB, error) {
@ -41,7 +40,7 @@ func Open(dbpath string) (*VtbDB, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} else { } else {
return (*VtbDB)(unsafe.Pointer(db)), nil return (*VtbDB)(db), nil
} }
} }
@ -91,7 +90,7 @@ func (ThirdCategory) TableName() string {
// GetAllFirstCategoryMessage 取出所有vtb // GetAllFirstCategoryMessage 取出所有vtb
func (vdb *VtbDB) GetAllFirstCategoryMessage() string { func (vdb *VtbDB) GetAllFirstCategoryMessage() string {
db := (*gorm.DB)(unsafe.Pointer(vdb)) db := (*gorm.DB)(vdb)
firstStepMessage := "请选择一个vtb并发送序号:\n" firstStepMessage := "请选择一个vtb并发送序号:\n"
var fc FirstCategory var fc FirstCategory
rows, err := db.Model(&FirstCategory{}).Rows() rows, err := db.Model(&FirstCategory{}).Rows()
@ -111,7 +110,7 @@ func (vdb *VtbDB) GetAllFirstCategoryMessage() string {
// GetAllSecondCategoryMessageByFirstIndex 取得同一个vtb所有语录类别 // GetAllSecondCategoryMessageByFirstIndex 取得同一个vtb所有语录类别
func (vdb *VtbDB) GetAllSecondCategoryMessageByFirstIndex(firstIndex int) string { func (vdb *VtbDB) GetAllSecondCategoryMessageByFirstIndex(firstIndex int) string {
db := (*gorm.DB)(unsafe.Pointer(vdb)) db := (*gorm.DB)(vdb)
SecondStepMessage := "请选择一个语录类别并发送序号:\n" SecondStepMessage := "请选择一个语录类别并发送序号:\n"
var sc SecondCategory var sc SecondCategory
var count int var count int
@ -136,7 +135,7 @@ func (vdb *VtbDB) GetAllSecondCategoryMessageByFirstIndex(firstIndex int) string
// GetAllThirdCategoryMessageByFirstIndexAndSecondIndex 取得同一个vtb同个类别的所有语录 // GetAllThirdCategoryMessageByFirstIndexAndSecondIndex 取得同一个vtb同个类别的所有语录
func (vdb *VtbDB) GetAllThirdCategoryMessageByFirstIndexAndSecondIndex(firstIndex, secondIndex int) string { func (vdb *VtbDB) GetAllThirdCategoryMessageByFirstIndexAndSecondIndex(firstIndex, secondIndex int) string {
db := (*gorm.DB)(unsafe.Pointer(vdb)) db := (*gorm.DB)(vdb)
ThirdStepMessage := "请选择一个语录并发送序号:\n" ThirdStepMessage := "请选择一个语录并发送序号:\n"
var fc FirstCategory var fc FirstCategory
db.Model(FirstCategory{}).Where("first_category_index = ?", firstIndex).First(&fc) db.Model(FirstCategory{}).Where("first_category_index = ?", firstIndex).First(&fc)
@ -160,7 +159,7 @@ func (vdb *VtbDB) GetAllThirdCategoryMessageByFirstIndexAndSecondIndex(firstInde
// GetThirdCategory // GetThirdCategory
func (vdb *VtbDB) GetThirdCategory(firstIndex, secondIndex, thirdIndex int) ThirdCategory { func (vdb *VtbDB) GetThirdCategory(firstIndex, secondIndex, thirdIndex int) ThirdCategory {
db := (*gorm.DB)(unsafe.Pointer(vdb)) db := (*gorm.DB)(vdb)
var fc FirstCategory var fc FirstCategory
db.Model(FirstCategory{}).Where("first_category_index = ?", firstIndex).First(&fc) db.Model(FirstCategory{}).Where("first_category_index = ?", firstIndex).First(&fc)
var tc ThirdCategory var tc ThirdCategory
@ -169,7 +168,7 @@ func (vdb *VtbDB) GetThirdCategory(firstIndex, secondIndex, thirdIndex int) Thir
} }
func (vdb *VtbDB) RandomVtb() ThirdCategory { func (vdb *VtbDB) RandomVtb() ThirdCategory {
db := (*gorm.DB)(unsafe.Pointer(vdb)) db := (*gorm.DB)(vdb)
rand.Seed(time.Now().UnixNano()) rand.Seed(time.Now().UnixNano())
var count int var count int
db.Model(&ThirdCategory{}).Count(&count) db.Model(&ThirdCategory{}).Count(&count)
@ -181,7 +180,7 @@ func (vdb *VtbDB) RandomVtb() ThirdCategory {
} }
func (vdb *VtbDB) GetFirstCategoryByFirstUid(firstUid string) FirstCategory { func (vdb *VtbDB) GetFirstCategoryByFirstUid(firstUid string) FirstCategory {
db := (*gorm.DB)(unsafe.Pointer(vdb)) db := (*gorm.DB)(vdb)
var fc FirstCategory var fc FirstCategory
db.Model(FirstCategory{}).Where("first_category_uid = ?", firstUid).Take(&fc) db.Model(FirstCategory{}).Where("first_category_uid = ?", firstUid).Take(&fc)
// logrus.Info(fc) // logrus.Info(fc)
@ -189,14 +188,14 @@ func (vdb *VtbDB) GetFirstCategoryByFirstUid(firstUid string) FirstCategory {
} }
func (vdb *VtbDB) Close() error { func (vdb *VtbDB) Close() error {
db := (*gorm.DB)(unsafe.Pointer(vdb)) db := (*gorm.DB)(vdb)
return db.Close() return db.Close()
} }
const vtbUrl = "https://vtbkeyboard.moe/api/get_vtb_list" const vtbUrl = "https://vtbkeyboard.moe/api/get_vtb_list"
func (vdb *VtbDB) GetVtbList() []string { func (vdb *VtbDB) GetVtbList() []string {
db := (*gorm.DB)(unsafe.Pointer(vdb)) db := (*gorm.DB)(vdb)
client := &http.Client{} client := &http.Client{}
req, err := http.NewRequest("GET", vtbUrl, nil) req, err := http.NewRequest("GET", vtbUrl, nil)
if err != nil { if err != nil {
@ -256,7 +255,7 @@ func (vdb *VtbDB) GetVtbList() []string {
} }
func (vdb *VtbDB) StoreVtb(uid string) { func (vdb *VtbDB) StoreVtb(uid string) {
db := (*gorm.DB)(unsafe.Pointer(vdb)) db := (*gorm.DB)(vdb)
vtbUrl := "https://vtbkeyboard.moe/api/get_vtb_page?uid=" + uid vtbUrl := "https://vtbkeyboard.moe/api/get_vtb_page?uid=" + uid
client := &http.Client{} client := &http.Client{}
req, err := http.NewRequest("GET", vtbUrl, nil) req, err := http.NewRequest("GET", vtbUrl, nil)