mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 05:30:07 +08:00
72 lines
1.3 KiB
Go
72 lines
1.3 KiB
Go
// Package data 加载位于 datapath 的小作文
|
|
package data
|
|
|
|
import (
|
|
"crypto/md5"
|
|
"encoding/binary"
|
|
"time"
|
|
|
|
binutils "github.com/FloatTech/floatbox/binary"
|
|
"github.com/FloatTech/floatbox/file"
|
|
sql "github.com/FloatTech/sqlite"
|
|
"github.com/FloatTech/zbputils/control"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var db = sql.Sqlite{}
|
|
|
|
type text struct {
|
|
ID int64 `db:"id"`
|
|
Data string `db:"data"`
|
|
}
|
|
|
|
// LoadText 加载小作文
|
|
func LoadText(dbfile string) error {
|
|
_, err := file.GetLazyData(dbfile, control.Md5File, false)
|
|
db = sql.New(dbfile)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = db.Open(time.Hour)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = db.Create("text", &text{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
c, err := db.Count("text")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
logrus.Printf("[Diana]读取%d条小作文", c)
|
|
return nil
|
|
}
|
|
|
|
// AddText 添加小作文
|
|
func AddText(txt string) error {
|
|
s := md5.Sum(binutils.StringToBytes(txt))
|
|
i := binary.LittleEndian.Uint64(s[:8])
|
|
return db.Insert("text", &text{ID: int64(i), Data: txt})
|
|
}
|
|
|
|
// RandText 随机小作文
|
|
func RandText() string {
|
|
var t text
|
|
err := db.Pick("text", &t)
|
|
if err != nil {
|
|
return err.Error()
|
|
}
|
|
return t.Data
|
|
}
|
|
|
|
// HentaiText 发大病
|
|
func HentaiText() string {
|
|
var t text
|
|
err := db.Find("text", &t, "WHERE id = -3802576048116006195")
|
|
if err != nil {
|
|
return err.Error()
|
|
}
|
|
return t.Data
|
|
}
|