102 lines
3.3 KiB
Go
102 lines
3.3 KiB
Go
package pogs
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/mitchellh/mapstructure"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// ScopeUnmarshaler can marshal a Scope pog from JSON.
|
|
type ScopeUnmarshaler struct {
|
|
Scope Scope
|
|
}
|
|
|
|
// UnmarshalJSON takes in a JSON string, and attempts to marshal it into a Scope.
|
|
// If successful, the Scope member of this ScopeUnmarshaler is set and nil is returned.
|
|
// If unsuccessful, returns an error.
|
|
func (su *ScopeUnmarshaler) UnmarshalJSON(b []byte) error {
|
|
var scopeJSON map[string]interface{}
|
|
if err := json.Unmarshal(b, &scopeJSON); err != nil {
|
|
return errors.Wrapf(err, "cannot unmarshal %s into scopeJSON", string(b))
|
|
}
|
|
|
|
if group, ok := scopeJSON["group"]; ok {
|
|
if val, ok := group.(string); ok {
|
|
su.Scope = NewGroup(val)
|
|
return nil
|
|
}
|
|
return fmt.Errorf("JSON should have been a Scope, but the 'group' key contained %v", group)
|
|
}
|
|
|
|
if systemName, ok := scopeJSON["system_name"]; ok {
|
|
if val, ok := systemName.(string); ok {
|
|
su.Scope = NewSystemName(val)
|
|
return nil
|
|
}
|
|
return fmt.Errorf("JSON should have been a Scope, but the 'system_name' key contained %v", systemName)
|
|
}
|
|
|
|
return fmt.Errorf("JSON should have been an object with one root key, either 'system_name' or 'group'")
|
|
}
|
|
|
|
// OriginConfigJSONHandler is a wrapper to serialize OriginConfig with type information, and deserialize JSON
|
|
// into an OriginConfig.
|
|
type OriginConfigJSONHandler struct {
|
|
OriginConfig OriginConfig
|
|
}
|
|
|
|
func (ocjh *OriginConfigJSONHandler) MarshalJSON() ([]byte, error) {
|
|
marshaler := make(map[string]OriginConfig, 1)
|
|
marshaler[ocjh.OriginConfig.jsonType()] = ocjh.OriginConfig
|
|
return json.Marshal(marshaler)
|
|
}
|
|
|
|
func (ocjh *OriginConfigJSONHandler) UnmarshalJSON(b []byte) error {
|
|
var originJSON map[string]interface{}
|
|
if err := json.Unmarshal(b, &originJSON); err != nil {
|
|
return errors.Wrapf(err, "cannot unmarshal %s into originJSON", string(b))
|
|
}
|
|
|
|
if originConfig, ok := originJSON[httpType.String()]; ok {
|
|
httpOriginConfig := &HTTPOriginConfig{}
|
|
if err := mapstructure.Decode(originConfig, httpOriginConfig); err != nil {
|
|
return errors.Wrapf(err, "cannot decode %+v into HTTPOriginConfig", originConfig)
|
|
}
|
|
ocjh.OriginConfig = httpOriginConfig
|
|
return nil
|
|
}
|
|
|
|
if originConfig, ok := originJSON[wsType.String()]; ok {
|
|
wsOriginConfig := &WebSocketOriginConfig{}
|
|
if err := mapstructure.Decode(originConfig, wsOriginConfig); err != nil {
|
|
return errors.Wrapf(err, "cannot decode %+v into WebSocketOriginConfig", originConfig)
|
|
}
|
|
ocjh.OriginConfig = wsOriginConfig
|
|
return nil
|
|
}
|
|
|
|
if originConfig, ok := originJSON[helloWorldType.String()]; ok {
|
|
helloWorldOriginConfig := &HelloWorldOriginConfig{}
|
|
if err := mapstructure.Decode(originConfig, helloWorldOriginConfig); err != nil {
|
|
return errors.Wrapf(err, "cannot decode %+v into HelloWorldOriginConfig", originConfig)
|
|
}
|
|
ocjh.OriginConfig = helloWorldOriginConfig
|
|
return nil
|
|
}
|
|
|
|
return fmt.Errorf("cannot unmarshal %s into OriginConfig", string(b))
|
|
}
|
|
|
|
// FallibleConfigMarshaler is a wrapper for FallibleConfig to implement custom marshal logic
|
|
type FallibleConfigMarshaler struct {
|
|
FallibleConfig FallibleConfig
|
|
}
|
|
|
|
func (fcm *FallibleConfigMarshaler) MarshalJSON() ([]byte, error) {
|
|
marshaler := make(map[string]FallibleConfig, 1)
|
|
marshaler[fcm.FallibleConfig.jsonType()] = fcm.FallibleConfig
|
|
return json.Marshal(marshaler)
|
|
}
|