V2bX/conf/node.go

package conf

import (
	"github.com/goccy/go-json"
)

type NodeConfig struct {
	ApiConfig ApiConfig `json:"-"`
	Options   Options   `json:"-"`
}

type rawNodeConfig struct {
	ApiRaw *json.RawMessage `json:"ApiConfig"`
	OptRaw *json.RawMessage `json:"Options"`
}

type ApiConfig struct {
	APIHost      string `json:"ApiHost"`
	NodeID       int    `json:"NodeID"`
	Key          string `json:"ApiKey"`
	NodeType     string `json:"NodeType"`
	Timeout      int    `json:"Timeout"`
	RuleListPath string `json:"RuleListPath"`
}

type Options struct {
	Core        string       `json:"Core"`
	ListenIP    string       `json:"ListenIP"`
	SendIP      string       `json:"SendIP"`
	LimitConfig LimitConfig  `json:"LimitConfig"`
	XrayOptions *XrayOptions `json:"XrayOptions"`
	SingOptions *SingOptions `json:"SingOptions"`
	CertConfig  *CertConfig  `json:"CertConfig"`
}

func (n *NodeConfig) UnmarshalJSON(b []byte) (err error) {
	r := rawNodeConfig{}
	err = json.Unmarshal(b, &r)
	if err != nil {
		return err
	}
	if r.ApiRaw != nil {
		err = json.Unmarshal(*r.ApiRaw, &n.ApiConfig)
		if err != nil {
			return
		}
	} else {
		n.ApiConfig = ApiConfig{
			Timeout: 30,
		}
		err = json.Unmarshal(b, &n.ApiConfig)
		if err != nil {
			return
		}
	}

	if r.OptRaw != nil {
		err = json.Unmarshal(*r.OptRaw, &n.Options)
		if err != nil {
			return
		}
	} else {
		n.Options = Options{
			Core:     "xray",
			ListenIP: "0.0.0.0",
			SendIP:   "0.0.0.0",
		}
		err = json.Unmarshal(b, &n.Options)
		if err != nil {
			return
		}
		switch n.Options.Core {
		case "xray":
			n.Options.XrayOptions = NewXrayOptions()
			return json.Unmarshal(b, n.Options.XrayOptions)
		case "sing":
			n.Options.SingOptions = NewSingOptions()
			return json.Unmarshal(b, n.Options.SingOptions)
		}
	}
	return
}