data.go

package control

import (
	"encoding/json"
	"errors"
	"math/bits"

	log "github.com/sirupsen/logrus"
	"github.com/wdvxdr1123/ZeroBot/utils/helper"
)

var (
	// ErrEmptyExtra ...
	ErrEmptyExtra = errors.New("empty extra")
	// ErrUnregisteredExtra ...
	ErrUnregisteredExtra = errors.New("unregistered extra")
)

// GetData 获取某个群的 62 位配置信息
func (m *Control[CTX]) GetData(gid int64) int64 {
	var c GroupConfig
	var err error
	m.Manager.RLock()
	err = m.Manager.D.Find(m.Service, &c, "WHERE gid = ?", gid)
	m.Manager.RUnlock()
	if err == nil && gid == c.GroupID {
		log.Debugf("[control] plugin %s of grp %d : 0x%x", m.Service, c.GroupID, c.Disable>>1)
		return (c.Disable >> 1) & 0x3fffffff_ffffffff
	}
	return 0
}

// SetData 为某个群设置中间 62 位配置数据 (除高低位)
func (m *Control[CTX]) SetData(groupID int64, data int64) error {
	var c GroupConfig
	m.Manager.RLock()
	err := m.Manager.D.Find(m.Service, &c, "WHERE gid = ?", groupID)
	m.Manager.RUnlock()
	if err != nil {
		c.GroupID = groupID
		if m.Options.DisableOnDefault {
			c.Disable = 1
		}
	}
	x := bits.RotateLeft64(uint64(c.Disable), 1)
	x &= 0x03
	x |= uint64(data) << 2
	c.Disable = int64(bits.RotateLeft64(x, -1))
	log.Debugf("[control] set plugin %s of grp %d : 0x%x", m.Service, c.GroupID, data)
	m.Manager.Lock()
	err = m.Manager.D.Insert(m.Service, &c)
	m.Manager.Unlock()
	if err != nil {
		log.Errorf("[control] %v", err)
	}
	return err
}

// GetExtra 取得额外数据, 一个插件一个
func (m *Control[CTX]) GetExtra(obj any) error {
	if m.Options.Extra == 0 {
		return ErrUnregisteredExtra
	}
	return m.Manager.getExtra(int64(m.Options.Extra), obj)
}

// getExtra 取得额外数据
func (manager *Manager[CTX]) getExtra(gid int64, obj any) error {
	if !manager.CanResponse(gid) {
		return errors.New("there is no extra data for a silent group")
	}
	manager.RLock()
	ext, ok := respCache[gid]
	manager.RUnlock()
	if ok {
		if ext == "-" {
			return ErrEmptyExtra
		}
		return json.Unmarshal(helper.StringToBytes(ext), obj)
	}
	var rsp ResponseGroup
	manager.RLock()
	err := manager.D.Find("__resp", &rsp, "where gid = ?", gid)
	manager.RUnlock()
	if err != nil || rsp.Extra == "-" {
		manager.Lock()
		respCache[gid] = "-"
		manager.Unlock()
		return ErrEmptyExtra
	}
	manager.Lock()
	respCache[gid] = rsp.Extra
	manager.Unlock()
	return json.Unmarshal(helper.StringToBytes(rsp.Extra), obj)
}

// SetExtra 设置额外数据, 一个插件一个
func (m *Control[CTX]) SetExtra(obj any) error {
	if m.Options.Extra == 0 {
		return ErrUnregisteredExtra
	}
	_ = m.Manager.Response(int64(m.Options.Extra))
	return m.Manager.setExtra(int64(m.Options.Extra), obj)
}

// setExtra 设置额外数据
func (manager *Manager[CTX]) setExtra(gid int64, obj any) error {
	if !manager.CanResponse(gid) {
		return errors.New("there is no extra data for a silent group")
	}
	data, err := json.Marshal(obj)
	if err != nil {
		return err
	}
	manager.Lock()
	defer manager.Unlock()
	respCache[gid] = helper.BytesToString(data)
	return manager.D.Insert("__resp", &ResponseGroup{GroupID: gid, Extra: helper.BytesToString(data)})
}