102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
// Copyright 2016 Marapongo, Inc. All rights reserved.
|
|
|
|
package encoding
|
|
|
|
import (
|
|
"reflect"
|
|
|
|
"github.com/marapongo/mu/pkg/pack/ast"
|
|
"github.com/marapongo/mu/pkg/util/contract"
|
|
"github.com/marapongo/mu/pkg/util/mapper"
|
|
)
|
|
|
|
func decodeModuleMember(m mapper.Mapper, tree mapper.Object) (ast.ModuleMember, error) {
|
|
k, err := mapper.FieldString(tree, reflect.TypeOf((*ast.ModuleMember)(nil)).Elem(), "kind", true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if k != nil {
|
|
kind := ast.NodeKind(*k)
|
|
switch kind {
|
|
case ast.ClassKind:
|
|
return decodeClass(m, tree)
|
|
case ast.ExportKind:
|
|
return decodeExport(m, tree)
|
|
case ast.ModulePropertyKind:
|
|
return decodeModuleProperty(m, tree)
|
|
case ast.ModuleMethodKind:
|
|
return decodeModuleMethod(m, tree)
|
|
default:
|
|
contract.FailMF("Unrecognized ModuleMember kind: %v\n", kind)
|
|
}
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func decodeClass(m mapper.Mapper, tree mapper.Object) (*ast.Class, error) {
|
|
var class ast.Class
|
|
if err := m.Decode(tree, &class); err != nil {
|
|
return nil, err
|
|
}
|
|
return &class, nil
|
|
}
|
|
|
|
func decodeClassMember(m mapper.Mapper, tree mapper.Object) (ast.ClassMember, error) {
|
|
k, err := mapper.FieldString(tree, reflect.TypeOf((*ast.ClassMember)(nil)).Elem(), "kind", true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if k != nil {
|
|
kind := ast.NodeKind(*k)
|
|
switch kind {
|
|
case ast.ClassPropertyKind:
|
|
return decodeClassProperty(m, tree)
|
|
case ast.ClassMethodKind:
|
|
return decodeClassMethod(m, tree)
|
|
default:
|
|
contract.FailMF("Unrecognized ClassMember kind: %v\n", kind)
|
|
}
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func decodeClassProperty(m mapper.Mapper, tree mapper.Object) (*ast.ClassProperty, error) {
|
|
var prop ast.ClassProperty
|
|
if err := m.Decode(tree, &prop); err != nil {
|
|
return nil, err
|
|
}
|
|
return &prop, nil
|
|
}
|
|
|
|
func decodeClassMethod(m mapper.Mapper, tree mapper.Object) (*ast.ClassMethod, error) {
|
|
var meth ast.ClassMethod
|
|
if err := m.Decode(tree, &meth); err != nil {
|
|
return nil, err
|
|
}
|
|
return &meth, nil
|
|
}
|
|
|
|
func decodeExport(m mapper.Mapper, tree mapper.Object) (*ast.Export, error) {
|
|
var export ast.Export
|
|
if err := m.Decode(tree, &export); err != nil {
|
|
return nil, err
|
|
}
|
|
return &export, nil
|
|
}
|
|
|
|
func decodeModuleProperty(m mapper.Mapper, tree mapper.Object) (*ast.ModuleProperty, error) {
|
|
var prop ast.ModuleProperty
|
|
if err := m.Decode(tree, &prop); err != nil {
|
|
return nil, err
|
|
}
|
|
return &prop, nil
|
|
}
|
|
|
|
func decodeModuleMethod(m mapper.Mapper, tree mapper.Object) (*ast.ModuleMethod, error) {
|
|
var meth ast.ModuleMethod
|
|
if err := m.Decode(tree, &meth); err != nil {
|
|
return nil, err
|
|
}
|
|
return &meth, nil
|
|
}
|