83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
const (
|
|
runtimeSettingAllowEncryptedForwarding = "mqtt.allow_encrypted_forwarding"
|
|
runtimeSettingTypeBool = "bool"
|
|
)
|
|
|
|
type runtimeSettingsSnapshot struct {
|
|
AllowEncryptedForwarding bool
|
|
}
|
|
|
|
func (s *store) GetRuntimeSettings() (runtimeSettingsSnapshot, error) {
|
|
allowEncrypted, err := s.GetBoolRuntimeSetting(runtimeSettingAllowEncryptedForwarding, false)
|
|
if err != nil {
|
|
return runtimeSettingsSnapshot{}, err
|
|
}
|
|
return runtimeSettingsSnapshot{AllowEncryptedForwarding: allowEncrypted}, nil
|
|
}
|
|
|
|
func (s *store) GetBoolRuntimeSetting(key string, defaultValue bool) (bool, error) {
|
|
key = strings.TrimSpace(key)
|
|
if key == "" {
|
|
return false, fmt.Errorf("runtime setting key is required")
|
|
}
|
|
|
|
var row runtimeSettingRecord
|
|
err := s.db.Where("key = ?", key).Take(&row).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return defaultValue, nil
|
|
}
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if row.ValueType != "" && row.ValueType != runtimeSettingTypeBool {
|
|
return false, fmt.Errorf("runtime setting %s has type %s, want %s", key, row.ValueType, runtimeSettingTypeBool)
|
|
}
|
|
value, err := strconv.ParseBool(strings.TrimSpace(row.Value))
|
|
if err != nil {
|
|
return false, fmt.Errorf("parse runtime setting %s: %w", key, err)
|
|
}
|
|
return value, nil
|
|
}
|
|
|
|
func (s *store) SetBoolRuntimeSetting(key string, value bool, label string) (*runtimeSettingRecord, error) {
|
|
key = strings.TrimSpace(key)
|
|
if key == "" {
|
|
return nil, fmt.Errorf("runtime setting key is required")
|
|
}
|
|
|
|
row := runtimeSettingRecord{
|
|
Key: key,
|
|
Value: strconv.FormatBool(value),
|
|
ValueType: runtimeSettingTypeBool,
|
|
Label: strings.TrimSpace(label),
|
|
}
|
|
if err := s.db.Clauses(clause.OnConflict{
|
|
Columns: []clause.Column{{Name: "key"}},
|
|
DoUpdates: clause.Assignments(map[string]any{
|
|
"value": row.Value,
|
|
"value_type": row.ValueType,
|
|
"label": row.Label,
|
|
"updated_at": time.Now(),
|
|
}),
|
|
}).Create(&row).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
if err := s.db.Where("key = ?", key).Take(&row).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &row, nil
|
|
}
|