ZeroBot-Plugin/plugin/antiabuse/db.go

package antiabuse

import (
	"errors"
	"strconv"
	"strings"
	"sync"
	"time"

	sqlite "github.com/FloatTech/sqlite"
)

type antidb struct {
	sync.RWMutex
	sqlite.Sqlite
}

type banWord struct {
	Word string `db:"word"`
}

type banTime struct {
	ID   int64 `db:"id"`
	Time int64 `db:"time"`
}

var (
	nilban = &banWord{}
	nilbt  = &banTime{}
)

func newantidb(path string) (*antidb, error) {
	db := &antidb{Sqlite: sqlite.Sqlite{DBPath: path}}
	err := db.Open(time.Hour * banhour)
	if err != nil {
		return nil, err
	}
	_ = db.Del("__bantime__", "WHERE time<="+strconv.FormatInt(time.Now().Add(-time.Hour*banhour).Unix(), 10))
	return db, db.FindFor("__bantime__", nilbt, "", func() error {
		t := time.Unix(nilbt.Time, 0)
		ttl := time.Until(t.Add(time.Hour * banhour)) // second
		if ttl < time.Minute {
			return nil
		}
		cache.Set(nilbt.ID, struct{}{})
		cache.Touch(nilbt.ID, -time.Since(t))
		return nil
	})
}

func (db *antidb) isInAntiList(uid, gid int64, msg string) bool {
	grp := strconv.FormatInt(gid, 36)
	db.RLock()
	defer db.RUnlock()
	return db.CanFind(grp, "WHERE instr('"+msg+"', word)>0")
}

func (db *antidb) insertWord(gid int64, word string) error {
	grp := strconv.FormatInt(gid, 36)
	db.Lock()
	defer db.Unlock()
	err := db.Create(grp, nilban)
	if err != nil {
		return err
	}
	return db.Insert(grp, &banWord{Word: word})
}

func (db *antidb) deleteWord(gid int64, word string) error {
	grp := strconv.FormatInt(gid, 36)
	db.Lock()
	defer db.Unlock()
	if n, _ := db.Count(grp); n == 0 {
		return errors.New("本群还没有违禁词~")
	}
	return db.Del(grp, "WHERE word='"+word+"'")
}

func (db *antidb) listWords(gid int64) string {
	grp := strconv.FormatInt(gid, 36)
	word := &banWord{}
	sb := strings.Builder{}
	sb.WriteByte('[')
	i := 0
	db.RLock()
	defer db.RUnlock()
	_ = db.FindFor(grp, word, "", func() error {
		if i > 0 {
			sb.WriteString(" | ")
		}
		sb.WriteString(word.Word)
		i++
		return nil
	})
	if sb.Len() <= 4 {
		return "[]"
	}
	sb.WriteByte(']')
	return sb.String()
}