mirror of https://github.com/pulumi/pulumi.git
109 lines
3.4 KiB
Go
109 lines
3.4 KiB
Go
// Copyright 2016-2020, Pulumi Corporation.
|
|
//
|
|
// Licensed 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 pcl
|
|
|
|
import (
|
|
"github.com/pulumi/pulumi/pkg/v3/codegen/hcl2/model"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/util/contract"
|
|
)
|
|
|
|
const (
|
|
// IntrinsicApply is the name of the apply intrinsic.
|
|
IntrinsicApply = "__apply"
|
|
// IntrinsicConvert is the name of the conversion intrinsic.
|
|
IntrinsicConvert = "__convert"
|
|
)
|
|
|
|
func isOutput(t model.Type) bool {
|
|
switch t := t.(type) {
|
|
case *model.OutputType:
|
|
return true
|
|
case *model.UnionType:
|
|
for _, t := range t.ElementTypes {
|
|
if _, isOutput := t.(*model.OutputType); isOutput {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// NewApplyCall returns a new expression that represents a call to IntrinsicApply.
|
|
func NewApplyCall(args []model.Expression, then *model.AnonymousFunctionExpression) *model.FunctionCallExpression {
|
|
signature := model.StaticFunctionSignature{
|
|
Parameters: make([]model.Parameter, len(args)+1),
|
|
}
|
|
|
|
returnsOutput := false
|
|
exprs := make([]model.Expression, len(args)+1)
|
|
for i, a := range args {
|
|
exprs[i] = a
|
|
if isOutput := isOutput(a.Type()); isOutput {
|
|
returnsOutput = true
|
|
}
|
|
signature.Parameters[i] = model.Parameter{
|
|
Name: then.Signature.Parameters[i].Name,
|
|
Type: a.Type(),
|
|
}
|
|
}
|
|
exprs[len(exprs)-1] = then
|
|
signature.Parameters[len(signature.Parameters)-1] = model.Parameter{
|
|
Name: "then",
|
|
Type: then.Type(),
|
|
}
|
|
|
|
if returnsOutput {
|
|
signature.ReturnType = model.NewOutputType(then.Signature.ReturnType)
|
|
} else {
|
|
signature.ReturnType = model.NewPromiseType(then.Signature.ReturnType)
|
|
}
|
|
|
|
return &model.FunctionCallExpression{
|
|
Name: IntrinsicApply,
|
|
Signature: signature,
|
|
Args: exprs,
|
|
}
|
|
}
|
|
|
|
// ParseApplyCall extracts the apply arguments and the continuation from a call to the apply intrinsic.
|
|
func ParseApplyCall(c *model.FunctionCallExpression) (applyArgs []model.Expression,
|
|
then *model.AnonymousFunctionExpression,
|
|
) {
|
|
contract.Requiref(c.Name == IntrinsicApply, "c", "must be a call to %q, got %q", IntrinsicApply, c.Name)
|
|
return c.Args[:len(c.Args)-1], c.Args[len(c.Args)-1].(*model.AnonymousFunctionExpression)
|
|
}
|
|
|
|
// NewConvertCall returns a new expression that represents a call to IntrinsicConvert.
|
|
func NewConvertCall(from model.Expression, to model.Type) *model.FunctionCallExpression {
|
|
return &model.FunctionCallExpression{
|
|
Name: IntrinsicConvert,
|
|
Signature: model.StaticFunctionSignature{
|
|
Parameters: []model.Parameter{{
|
|
Name: "from",
|
|
Type: from.Type(),
|
|
}},
|
|
ReturnType: to,
|
|
},
|
|
Args: []model.Expression{from},
|
|
}
|
|
}
|
|
|
|
// ParseConvertCall extracts the value being converted and the type it is being converted to from a call to the convert
|
|
// intrinsic.
|
|
func ParseConvertCall(c *model.FunctionCallExpression) (model.Expression, model.Type) {
|
|
contract.Requiref(c.Name == IntrinsicConvert, "c", "must be a call to %q, got %q", IntrinsicConvert, c.Name)
|
|
return c.Args[0], c.Signature.ReturnType
|
|
}
|