mirror of https://github.com/v2ray/v2ray-core
62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
// +build json
|
|
|
|
package loader
|
|
|
|
import (
|
|
"encoding/json"
|
|
|
|
"github.com/v2ray/v2ray-core/common/log"
|
|
)
|
|
|
|
type JSONConfigLoader struct {
|
|
*BaseConfigLoader
|
|
idKey string
|
|
configKey string
|
|
}
|
|
|
|
func NewJSONConfigLoader(idKey string, configKey string) *JSONConfigLoader {
|
|
return &JSONConfigLoader{
|
|
idKey: idKey,
|
|
configKey: configKey,
|
|
BaseConfigLoader: NewBaseConfigLoader(),
|
|
}
|
|
}
|
|
|
|
func (this *JSONConfigLoader) LoadWithID(raw []byte, id string) (interface{}, error) {
|
|
config, err := this.CreateConfig(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := json.Unmarshal(raw, config); err != nil {
|
|
return nil, err
|
|
}
|
|
return config, nil
|
|
}
|
|
|
|
func (this *JSONConfigLoader) Load(raw []byte) (interface{}, error) {
|
|
var obj map[string]json.RawMessage
|
|
if err := json.Unmarshal(raw, &obj); err != nil {
|
|
return nil, err
|
|
}
|
|
rawID, found := obj[this.idKey]
|
|
if !found {
|
|
log.Error(this.idKey, " not found in JSON content.")
|
|
return nil, ErrConfigIDKeyNotFound
|
|
}
|
|
var id string
|
|
if err := json.Unmarshal(rawID, &id); err != nil {
|
|
return nil, err
|
|
}
|
|
rawConfig := json.RawMessage(raw)
|
|
if len(this.configKey) > 0 {
|
|
configValue, found := obj[this.configKey]
|
|
if !found {
|
|
log.Error(this.configKey, " not found in JSON content.")
|
|
return nil, ErrConfigIDKeyNotFound
|
|
}
|
|
rawConfig = configValue
|
|
}
|
|
return this.LoadWithID([]byte(rawConfig), id)
|
|
}
|