go-cqhttp/global/config.go

package global

import (
	"os"
	"strconv"
	"time"

	"github.com/hjson/hjson-go"
	jsoniter "github.com/json-iterator/go"
	log "github.com/sirupsen/logrus"
)

var json = jsoniter.ConfigCompatibleWithStandardLibrary

type JsonConfig struct {
	Uin               int64  `json:"uin"`
	Password          string `json:"password"`
	EncryptPassword   bool   `json:"encrypt_password"`
	PasswordEncrypted string `json:"password_encrypted"`
	EnableDB          bool   `json:"enable_db"`
	AccessToken       string `json:"access_token"`
	ReLogin           struct {
		Enabled         bool `json:"enabled"`
		ReLoginDelay    int  `json:"relogin_delay"`
		MaxReloginTimes uint `json:"max_relogin_times"`
	} `json:"relogin"`
	RateLimit struct {
		Enabled    bool    `json:"enabled"`
		Frequency  float64 `json:"frequency"`
		BucketSize int     `json:"bucket_size"`
	} `json:"_rate_limit"`
	IgnoreInvalidCQCode bool                          `json:"ignore_invalid_cqcode"`
	ForceFragmented     bool                          `json:"force_fragmented"`
	HeartbeatInterval   time.Duration                 `json:"heartbeat_interval"`
	HttpConfig          *GoCQHttpConfig               `json:"http_config"`
	WSConfig            *GoCQWebsocketConfig          `json:"ws_config"`
	ReverseServers      []*GoCQReverseWebsocketConfig `json:"ws_reverse_servers"`
	PostMessageFormat   string                        `json:"post_message_format"`
	UseSSOAddress       bool                          `json:"use_sso_address"`
	Debug               bool                          `json:"debug"`
	LogLevel            string                        `json:"log_level"`
	WebUi               *GoCqWebUi                    `json:"web_ui"`
}

type CQHttpApiConfig struct {
	Host                         string `json:"host"`
	Port                         uint16 `json:"port"`
	UseHttp                      bool   `json:"use_http"`
	WSHost                       string `json:"ws_host"`
	WSPort                       uint16 `json:"ws_port"`
	UseWS                        bool   `json:"use_ws"`
	WSReverseUrl                 string `json:"ws_reverse_url"`
	WSReverseApiUrl              string `json:"ws_reverse_api_url"`
	WSReverseEventUrl            string `json:"ws_reverse_event_url"`
	WSReverseReconnectInterval   uint16 `json:"ws_reverse_reconnect_interval"`
	WSReverseReconnectOnCode1000 bool   `json:"ws_reverse_reconnect_on_code_1000"`
	UseWsReverse                 bool   `json:"use_ws_reverse"`
	PostUrl                      string `json:"post_url"`
	AccessToken                  string `json:"access_token"`
	Secret                       string `json:"secret"`
	PostMessageFormat            string `json:"post_message_format"`
}

type GoCQHttpConfig struct {
	Enabled  bool              `json:"enabled"`
	Host     string            `json:"host"`
	Port     uint16            `json:"port"`
	Timeout  int32             `json:"timeout"`
	PostUrls map[string]string `json:"post_urls"`
}

type GoCQWebsocketConfig struct {
	Enabled bool   `json:"enabled"`
	Host    string `json:"host"`
	Port    uint16 `json:"port"`
}

type GoCQReverseWebsocketConfig struct {
	Enabled                  bool   `json:"enabled"`
	ReverseUrl               string `json:"reverse_url"`
	ReverseApiUrl            string `json:"reverse_api_url"`
	ReverseEventUrl          string `json:"reverse_event_url"`
	ReverseReconnectInterval uint16 `json:"reverse_reconnect_interval"`
}

type GoCqWebUi struct {
	Enabled   bool   `json:"enabled"`
	Host      string `json:"host"`
	WebUiPort uint64 `json:"web_ui_port"`
	WebInput  bool   `json:"web_input"`
}

func DefaultConfig() *JsonConfig {
	return &JsonConfig{
		EnableDB: true,
		ReLogin: struct {
			Enabled         bool `json:"enabled"`
			ReLoginDelay    int  `json:"relogin_delay"`
			MaxReloginTimes uint `json:"max_relogin_times"`
		}{
			Enabled:         true,
			ReLoginDelay:    3,
			MaxReloginTimes: 0,
		},
		RateLimit: struct {
			Enabled    bool    `json:"enabled"`
			Frequency  float64 `json:"frequency"`
			BucketSize int     `json:"bucket_size"`
		}{
			Enabled:    false,
			Frequency:  1,
			BucketSize: 1,
		},
		PostMessageFormat: "string",
		ForceFragmented:   false,
		HttpConfig: &GoCQHttpConfig{
			Enabled:  true,
			Host:     "0.0.0.0",
			Port:     5700,
			PostUrls: map[string]string{},
		},
		WSConfig: &GoCQWebsocketConfig{
			Enabled: true,
			Host:    "0.0.0.0",
			Port:    6700,
		},
		ReverseServers: []*GoCQReverseWebsocketConfig{
			{
				Enabled:                  false,
				ReverseUrl:               "ws://you_websocket_universal.server",
				ReverseApiUrl:            "ws://you_websocket_api.server",
				ReverseEventUrl:          "ws://you_websocket_event.server",
				ReverseReconnectInterval: 3000,
			},
		},
		WebUi: &GoCqWebUi{
			Enabled:   true,
			Host:      "127.0.0.1",
			WebInput:  false,
			WebUiPort: 9999,
		},
	}
}

func Load(p string) *JsonConfig {
	if !PathExists(p) {
		log.Warnf("尝试加载配置文件 %v 失败: 文件不存在", p)
		return nil
	}
	var dat map[string]interface{}
	var c = JsonConfig{}
	err := hjson.Unmarshal([]byte(ReadAllText(p)), &dat)
	if err == nil {
		b, _ := json.Marshal(dat)
		err = json.Unmarshal(b, &c)
	}
	if err != nil {
		log.Warnf("尝试加载配置文件 %v 时出现错误: %v", p, err)
		log.Infoln("原文件已备份")
		_ = os.Rename(p, p+".backup"+strconv.FormatInt(time.Now().Unix(), 10))
		return nil
	}
	return &c
}

func (c *JsonConfig) Save(p string) error {
	data, err := hjson.MarshalWithOptions(c, hjson.EncoderOptions{
		Eol: "\n",
		BracesSameLine: true,
		IndentBy: "    ",
	})
	if err != nil {
		return err
	}
	WriteAllText(p, string(data))
	return nil
}