pulumi/pkg/encoding/decode.go
joeduffy db99092334 Implement mapper.Encode "for real"
This change implements `mapper.Encode` "for real" (that is, in a way
that isn't a complete embarrassment).  It uses the obvious reflection
trickery to encode a tagged struct and its values as a JSON-like
in-memory map and collection of keyed values.

During this, I took the opportunity to also clean up a few other things
that had been bugging me.  Namely, the presence of `mapper.Object` was
always error prone, since it isn't a true "typedef" in the sence that
it carries extra RTTI.  Instead of doing that, let's just use the real
`map[string]interface{}` "JSON-map-like" object type.  Even better, we
no longer require resource providers to deal with the mapper
infrastructure.  Instead, the `Check` function can simply return an
array of errors.  It's still best practice to return field-specific errors
to facilitate better diagnostics, but it's no longer required; and I've
added `resource.NewFieldError` to eliminate the need to import mapper.

As of this change, we can also consistently emit RPC structs with `lumi`
tags, rather than `lumi` tags on the way in and `json` on the way out.

This completes pulumi/lumi#183.
2017-06-05 17:49:00 -07:00

77 lines
3.7 KiB
Go

// Licensed to Pulumi Corporation ("Pulumi") under one or more
// contributor license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright ownership.
// Pulumi licenses this file to You under the Apache License, Version 2.0
// (the "License"); you may not use this file except in compliance with
// the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Package encoding can unmarshal LumiPack and LumiIL metadata formats. Because of their complex structure, we cannot
// rely on the standard JSON marshaling and unmarshaling routines. Instead, we will need to do it mostly "by hand".
package encoding
import (
"reflect"
"github.com/pulumi/lumi/pkg/compiler/ast"
"github.com/pulumi/lumi/pkg/pack"
"github.com/pulumi/lumi/pkg/util/mapper"
)
// Decode unmarshals the entire contents of the given byte array into a Package object.
func Decode(m Marshaler, b []byte) (*pack.Package, error) {
// First convert the whole contents of the metadata into a map. Although it would be more efficient to walk the
// token stream, token by token, this allows us to reuse existing YAML packages in addition to JSON ones.
var obj map[string]interface{}
if err := m.Unmarshal(b, &obj); err != nil {
return nil, err
}
// Now decode the top-level Package metadata; this will automatically recurse throughout the whole structure.
md := mapper.New(&mapper.Opts{CustomDecoders: customDecoders()})
var pack pack.Package
if err := md.Decode(obj, &pack); err != nil {
return nil, err
}
return &pack, nil
}
// customDecoders makes a complete map of all known custom AST decoders. In general, any polymorphic node kind that
// appears as a field in another concrete marshalable structure must have an associated custom decoder. If not, the
// Mapper will error out. This is typically an interface type and that method typically switches on the kind property.
// Note that interfaces that are used as "markers" and don't show up in fields are okay and don't require a decoder.
func customDecoders() mapper.Decoders {
return mapper.Decoders{
reflect.TypeOf((*ast.ModuleMember)(nil)).Elem(): moduleMemberDecoder,
reflect.TypeOf((*ast.ClassMember)(nil)).Elem(): classMemberDecoder,
reflect.TypeOf((*ast.Statement)(nil)).Elem(): statementDecoder,
reflect.TypeOf((*ast.Expression)(nil)).Elem(): expressionDecoder,
reflect.TypeOf((*ast.ObjectLiteralProperty)(nil)).Elem(): objectLiteralPropertyDecoder,
}
}
// Each of the custom decoders is a varaible that points to a decoder function; this is done so that the decode*
// functions can remain strongly typed, as the mapper's decoder signature requires a weakly-typed interface{} return.
var moduleMemberDecoder = func(m mapper.Mapper, obj map[string]interface{}) (interface{}, error) {
return decodeModuleMember(m, obj)
}
var classMemberDecoder = func(m mapper.Mapper, obj map[string]interface{}) (interface{}, error) {
return decodeClassMember(m, obj)
}
var statementDecoder = func(m mapper.Mapper, obj map[string]interface{}) (interface{}, error) {
return decodeStatement(m, obj)
}
var expressionDecoder = func(m mapper.Mapper, obj map[string]interface{}) (interface{}, error) {
return decodeExpression(m, obj)
}
var objectLiteralPropertyDecoder = func(m mapper.Mapper, obj map[string]interface{}) (interface{}, error) {
return decodeObjectLiteralProperty(m, obj)
}