go-cqhttp/global/config.go

package global

import (
	"encoding/json"
	log "github.com/sirupsen/logrus"
)

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           bool                          `json:"relogin"`
	ReLoginDelay      int                           `json:"relogin_delay"`
	HttpConfig        *GoCQHttpConfig               `json:"http_config"`
	WSConfig          *GoCQWebsocketConfig          `json:"ws_config"`
	ReverseServers    []*GoCQReverseWebsocketConfig `json:"ws_reverse_servers"`
	Debug             bool                          `json:"debug"`
}

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"`
	PostMessageFormat string            `json:"post_message_format"`
}

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"`
}

func DefaultConfig() *JsonConfig {
	return &JsonConfig{
		EnableDB:     true,
		ReLogin:      true,
		ReLoginDelay: 3,
		HttpConfig: &GoCQHttpConfig{
			Enabled:           true,
			Host:              "0.0.0.0",
			Port:              5700,
			PostUrls:          map[string]string{},
			PostMessageFormat: "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,
			},
		},
	}
}

func Load(p string) *JsonConfig {
	if !PathExists(p) {
		log.Warnf("尝试加载配置文件 %v 失败: 文件不存在", p)
		return nil
	}
	c := JsonConfig{}
	err := json.Unmarshal([]byte(ReadAllText(p)), &c)
	if err != nil {
		log.Warnf("尝试加载配置文件 %v 时出现错误: %v", p, err)
		return nil
	}
	return &c
}

func (c *JsonConfig) Save(p string) error {
	data, err := json.MarshalIndent(c, "", "\t")
	if err != nil {
		return err
	}
	WriteAllText(p, string(data))
	return nil
}