281 lines
7.4 KiB
Go
281 lines
7.4 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/binary"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
"unicode/utf8"
|
|
|
|
"meshtastic_mqtt_server/mqtpp"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
const (
|
|
botDefaultTopicPrefix = "msh/2/e"
|
|
botMessageTypeChannel = "channel"
|
|
botMessageTypeDirect = "direct"
|
|
botMessageStatusPending = "pending"
|
|
botMessageStatusPublished = "published"
|
|
botMessageStatusFailed = "failed"
|
|
)
|
|
|
|
var errBotNodeAlreadyExists = errors.New("bot node already exists")
|
|
|
|
type botNodeInput struct {
|
|
NodeNum *int64
|
|
LongName string
|
|
ShortName string
|
|
Enabled bool
|
|
DefaultChannelID string
|
|
TopicPrefix string
|
|
}
|
|
|
|
type botMessageListOptions struct {
|
|
listOptions
|
|
BotID uint64
|
|
MessageType string
|
|
ChannelID string
|
|
}
|
|
|
|
func (s *store) ListBotNodes(opts listOptions) ([]botNodeRecord, error) {
|
|
opts = normalizeListOptions(opts)
|
|
var rows []botNodeRecord
|
|
q := s.db.Model(&botNodeRecord{}).
|
|
Order("updated_at DESC").
|
|
Order("id DESC").
|
|
Limit(opts.Limit).
|
|
Offset(opts.Offset)
|
|
return rows, q.Find(&rows).Error
|
|
}
|
|
|
|
func (s *store) CountBotNodes(opts listOptions) (int64, error) {
|
|
var total int64
|
|
return total, s.db.Model(&botNodeRecord{}).Count(&total).Error
|
|
}
|
|
|
|
func (s *store) GetBotNode(id uint64) (*botNodeRecord, error) {
|
|
var row botNodeRecord
|
|
if err := s.db.Where("id = ?", id).Take(&row).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &row, nil
|
|
}
|
|
|
|
func (s *store) CreateBotNode(input botNodeInput) (*botNodeRecord, error) {
|
|
row, err := s.normalizedBotNodeRecord(input)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := s.ensureBotNodeUnique(0, row.NodeID, row.NodeNum); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := s.ensureBotNodeDoesNotConflictWithNodeInfo(row.NodeNum); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := s.db.Create(row).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return row, nil
|
|
}
|
|
|
|
func (s *store) UpdateBotNode(id uint64, input botNodeInput) (*botNodeRecord, error) {
|
|
if id == 0 {
|
|
return nil, fmt.Errorf("bot node id is required")
|
|
}
|
|
if _, err := s.GetBotNode(id); err != nil {
|
|
return nil, err
|
|
}
|
|
row, err := s.normalizedBotNodeRecord(input)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if err := s.ensureBotNodeUnique(id, row.NodeID, row.NodeNum); err != nil {
|
|
return nil, err
|
|
}
|
|
if err := s.ensureBotNodeDoesNotConflictWithNodeInfo(row.NodeNum); err != nil {
|
|
return nil, err
|
|
}
|
|
updates := map[string]any{
|
|
"node_id": row.NodeID,
|
|
"node_num": row.NodeNum,
|
|
"long_name": row.LongName,
|
|
"short_name": row.ShortName,
|
|
"enabled": row.Enabled,
|
|
"default_channel_id": row.DefaultChannelID,
|
|
"topic_prefix": row.TopicPrefix,
|
|
"updated_at": time.Now(),
|
|
}
|
|
if err := s.db.Model(&botNodeRecord{}).Where("id = ?", id).Updates(updates).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return s.GetBotNode(id)
|
|
}
|
|
|
|
func (s *store) DeleteBotNode(id uint64) error {
|
|
result := s.db.Where("id = ?", id).Delete(&botNodeRecord{})
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
if result.RowsAffected == 0 {
|
|
return gorm.ErrRecordNotFound
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *store) InsertBotMessage(row *botMessageRecord) error {
|
|
return s.db.Create(row).Error
|
|
}
|
|
|
|
func (s *store) UpdateBotMessageStatus(id uint64, status, errText string, publishedAt *time.Time) error {
|
|
updates := map[string]any{"status": status, "error": strings.TrimSpace(errText), "published_at": publishedAt}
|
|
result := s.db.Model(&botMessageRecord{}).Where("id = ?", id).Updates(updates)
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
if result.RowsAffected == 0 {
|
|
return gorm.ErrRecordNotFound
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *store) ListBotMessages(opts botMessageListOptions) ([]botMessageRecord, error) {
|
|
opts.listOptions = normalizeListOptions(opts.listOptions)
|
|
var rows []botMessageRecord
|
|
q := applyBotMessageFilters(s.db.Model(&botMessageRecord{}), opts).
|
|
Order("created_at DESC").
|
|
Order("id DESC").
|
|
Limit(opts.Limit).
|
|
Offset(opts.Offset)
|
|
return rows, q.Find(&rows).Error
|
|
}
|
|
|
|
func (s *store) CountBotMessages(opts botMessageListOptions) (int64, error) {
|
|
var total int64
|
|
q := applyBotMessageFilters(s.db.Model(&botMessageRecord{}), opts)
|
|
return total, q.Count(&total).Error
|
|
}
|
|
|
|
func applyBotMessageFilters(q *gorm.DB, opts botMessageListOptions) *gorm.DB {
|
|
if opts.BotID != 0 {
|
|
q = q.Where("bot_id = ?", opts.BotID)
|
|
}
|
|
if opts.MessageType != "" {
|
|
q = q.Where("message_type = ?", opts.MessageType)
|
|
}
|
|
if opts.ChannelID != "" {
|
|
q = q.Where("channel_id = ?", opts.ChannelID)
|
|
}
|
|
if opts.Since != nil {
|
|
q = q.Where("created_at >= ?", *opts.Since)
|
|
}
|
|
if opts.Until != nil {
|
|
q = q.Where("created_at <= ?", *opts.Until)
|
|
}
|
|
return q
|
|
}
|
|
|
|
func (s *store) normalizedBotNodeRecord(input botNodeInput) (*botNodeRecord, error) {
|
|
longName := strings.TrimSpace(input.LongName)
|
|
shortName := strings.TrimSpace(input.ShortName)
|
|
channelID := strings.TrimSpace(input.DefaultChannelID)
|
|
topicPrefix := strings.Trim(strings.TrimSpace(input.TopicPrefix), "/")
|
|
if topicPrefix == "" {
|
|
topicPrefix = botDefaultTopicPrefix
|
|
}
|
|
if longName == "" {
|
|
return nil, fmt.Errorf("long name is required")
|
|
}
|
|
if !utf8.ValidString(longName) {
|
|
return nil, fmt.Errorf("long name must be valid utf-8")
|
|
}
|
|
if shortName == "" {
|
|
return nil, fmt.Errorf("short name is required")
|
|
}
|
|
if !utf8.ValidString(shortName) {
|
|
return nil, fmt.Errorf("short name must be valid utf-8")
|
|
}
|
|
if channelID == "" {
|
|
return nil, fmt.Errorf("default channel id is required")
|
|
}
|
|
var nodeNum int64
|
|
if input.NodeNum == nil || *input.NodeNum == 0 {
|
|
generated, err := s.generateBotNodeNum()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
nodeNum = generated
|
|
} else {
|
|
nodeNum = *input.NodeNum
|
|
}
|
|
if err := validateBotNodeNum(nodeNum); err != nil {
|
|
return nil, err
|
|
}
|
|
return &botNodeRecord{NodeID: mqtpp.NodeNumToID(uint32(nodeNum)), NodeNum: nodeNum, LongName: longName, ShortName: shortName, Enabled: input.Enabled, DefaultChannelID: channelID, TopicPrefix: topicPrefix}, nil
|
|
}
|
|
|
|
func validateBotNodeNum(nodeNum int64) error {
|
|
if nodeNum <= 0 || nodeNum >= int64(mqtpp.NodeNumBroadcast) {
|
|
return fmt.Errorf("node num must be between 1 and 4294967294")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *store) generateBotNodeNum() (int64, error) {
|
|
for i := 0; i < 32; i++ {
|
|
var buf [4]byte
|
|
if _, err := rand.Read(buf[:]); err != nil {
|
|
return 0, err
|
|
}
|
|
nodeNum := int64(binary.LittleEndian.Uint32(buf[:]) & 0x7fffffff)
|
|
if err := validateBotNodeNum(nodeNum); err != nil {
|
|
continue
|
|
}
|
|
if err := s.ensureBotNodeUnique(0, mqtpp.NodeNumToID(uint32(nodeNum)), nodeNum); err != nil {
|
|
if errors.Is(err, errBotNodeAlreadyExists) {
|
|
continue
|
|
}
|
|
return 0, err
|
|
}
|
|
if err := s.ensureBotNodeDoesNotConflictWithNodeInfo(nodeNum); err != nil {
|
|
if errors.Is(err, errBotNodeAlreadyExists) {
|
|
continue
|
|
}
|
|
return 0, err
|
|
}
|
|
return nodeNum, nil
|
|
}
|
|
return 0, fmt.Errorf("generate bot node num failed")
|
|
}
|
|
|
|
func (s *store) ensureBotNodeUnique(id uint64, nodeID string, nodeNum int64) error {
|
|
var existing botNodeRecord
|
|
q := s.db.Where("node_id = ? OR node_num = ?", nodeID, nodeNum)
|
|
if id != 0 {
|
|
q = q.Where("id <> ?", id)
|
|
}
|
|
err := q.Take(&existing).Error
|
|
if err == nil {
|
|
return errBotNodeAlreadyExists
|
|
}
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
func (s *store) ensureBotNodeDoesNotConflictWithNodeInfo(nodeNum int64) error {
|
|
var existing nodeInfoRecord
|
|
err := s.db.Where("node_num = ?", nodeNum).Take(&existing).Error
|
|
if err == nil {
|
|
return errBotNodeAlreadyExists
|
|
}
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|