refactor conf

conf/node.go

@@ -1,15 +1,82 @@
 package conf
 
+import (
+	"github.com/goccy/go-json"
+)
+
 type NodeConfig struct {
-	ApiConfig *ApiConfig `yaml:"ApiConfig"`
-	Options   *Options   `yaml:"Options"`
+	ApiConfig ApiConfig `json:"-"`
+	Options   Options   `json:"-"`
+}
+
+type rawNodeConfig struct {
+	ApiRaw *json.RawMessage `json:"ApiConfig"`
+	OptRaw *json.RawMessage `json:"Options"`
 }
 
 type ApiConfig struct {
-	APIHost      string `yaml:"ApiHost"`
-	NodeID       int    `yaml:"NodeID"`
-	Key          string `yaml:"ApiKey"`
-	NodeType     string `yaml:"NodeType"`
-	Timeout      int    `yaml:"Timeout"`
-	RuleListPath string `yaml:"RuleListPath"`
+	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
 }