1a4f36e97b
This change adds a `RegisterInputType` function (similar to the existing `RegisterOutputType`) that is used to register an input interface's type and its associated input type, and adds registrations for the built-in input types. This will be used when copying inputs to an args struct for multi-lang components. When a field is typed as the input interface (e.g. `StringMapInput`) and doesn't need to be an `Output`, we can use this to lookup the non-Output type that implements the interface (e.g. `StringMap`) so it can be instantiated. A subsequent change will update the Go SDK codegen to produce input type registrations for a provider's input types.
361 lines
10 KiB
Go
361 lines
10 KiB
Go
// Copyright 2016-2018, 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.
|
|
|
|
// +build ignore
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"strings"
|
|
"text/template"
|
|
"unicode"
|
|
)
|
|
|
|
type builtin struct {
|
|
Name string
|
|
Type string
|
|
inputType string
|
|
implements []string
|
|
Implements []*builtin
|
|
elementType string
|
|
item *builtin
|
|
ItemType string
|
|
Example string
|
|
elemExample string
|
|
GenerateConfig bool
|
|
DefaultConfig string
|
|
RegisterInput bool
|
|
defaultValue string
|
|
}
|
|
|
|
func (b builtin) DefineInputType() bool {
|
|
return b.inputType == "" && b.Type != "AssetOrArchive"
|
|
}
|
|
|
|
func (b builtin) DefinePtrType() bool {
|
|
return strings.HasSuffix(b.Name, "Ptr")
|
|
}
|
|
|
|
func (b builtin) PtrType() string {
|
|
return b.inputType[1:]
|
|
}
|
|
|
|
func (b builtin) DefineInputMethods() bool {
|
|
return b.Type != "AssetOrArchive"
|
|
}
|
|
|
|
func (b builtin) ImplementsPtrType() bool {
|
|
switch b.Type {
|
|
case "bool", "float64", "ID", "int", "string", "URN":
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (b builtin) DefaultValue() string {
|
|
if b.defaultValue != "" {
|
|
return b.defaultValue
|
|
}
|
|
return b.Name + "{}"
|
|
}
|
|
|
|
func (b builtin) DefineElem() bool {
|
|
return b.DefinePtrType()
|
|
}
|
|
|
|
func (b builtin) ElemReturnType() string {
|
|
return strings.TrimSuffix(b.Name, "Ptr")
|
|
}
|
|
|
|
func (b builtin) ElemElementType() string {
|
|
return strings.TrimPrefix(b.Type, "*")
|
|
}
|
|
|
|
func (b builtin) DefineIndex() bool {
|
|
return strings.HasSuffix(b.Name, "Array")
|
|
}
|
|
|
|
func (b builtin) IndexReturnType() string {
|
|
return strings.TrimSuffix(b.Name, "Array")
|
|
}
|
|
|
|
func (b builtin) IndexElementType() string {
|
|
return strings.TrimPrefix(b.elementType, "[]")
|
|
}
|
|
|
|
func (b builtin) DefineMapIndex() bool {
|
|
return strings.HasSuffix(b.Name, "Map")
|
|
}
|
|
|
|
func (b builtin) MapIndexElementType() string {
|
|
return strings.TrimPrefix(b.elementType, "map[string]")
|
|
}
|
|
|
|
func (b builtin) MapIndexReturnType() string {
|
|
return strings.TrimSuffix(b.Name, "Map")
|
|
}
|
|
|
|
func (b builtin) IndexConversion() string {
|
|
toType := b.item.Name
|
|
switch {
|
|
case b.item.inputType != "":
|
|
// No conversion necessary for types which are their own input type.
|
|
return ""
|
|
case toType == "Input":
|
|
// There is no direct conversion into `Input`, so special case to use `ToOutput`
|
|
return "ToOutput"
|
|
case b.item.item == nil:
|
|
// The item type is a primitive - so use a direct conversion into the defined type (e.g. `String(...)`)
|
|
return toType
|
|
default:
|
|
// The item type is itself a container - so use one of the other code-generated `To<Type>` methods.
|
|
return "To" + toType
|
|
}
|
|
}
|
|
|
|
func (b builtin) ElementType() string {
|
|
if b.elementType != "" {
|
|
return b.elementType
|
|
}
|
|
return b.Type
|
|
}
|
|
|
|
func (b builtin) InputType() string {
|
|
if b.inputType != "" {
|
|
return b.inputType
|
|
}
|
|
return b.Name
|
|
}
|
|
|
|
func (b builtin) DefineToFunction() bool {
|
|
if b.item != nil {
|
|
return b.item.DefineToFunction()
|
|
}
|
|
return b.DefineInputMethods()
|
|
}
|
|
|
|
func (b builtin) ItemExample() string {
|
|
return b.item.Example
|
|
}
|
|
|
|
func (b builtin) ElemExample() string {
|
|
if strings.HasSuffix(b.Name, "Map") {
|
|
return fmt.Sprintf("{\"baz\": %s}", b.item.ElemExample())
|
|
}
|
|
if strings.HasSuffix(b.Name, "Array") {
|
|
return fmt.Sprintf("{%s}", b.item.ElemExample())
|
|
}
|
|
if b.elemExample != "" {
|
|
return b.elemExample
|
|
}
|
|
return b.Example
|
|
}
|
|
|
|
var builtins = makeBuiltins([]*builtin{
|
|
{Name: "Archive", Type: "Archive", inputType: "*archive", implements: []string{"AssetOrArchive"}, Example: "NewFileArchive(\"foo.zip\")"},
|
|
{Name: "Asset", Type: "Asset", inputType: "*asset", implements: []string{"AssetOrArchive"}, Example: "NewFileAsset(\"foo.txt\")"},
|
|
{Name: "AssetOrArchive", Type: "AssetOrArchive", Example: "NewFileArchive(\"foo.zip\")"},
|
|
{Name: "Bool", Type: "bool", Example: "Bool(true)", GenerateConfig: true, DefaultConfig: "false", elemExample: "true", RegisterInput: true, defaultValue: "Bool(false)"},
|
|
{Name: "Float64", Type: "float64", Example: "Float64(999.9)", GenerateConfig: true, DefaultConfig: "0", elemExample: "999.9", RegisterInput: true, defaultValue: "Float64(0)"},
|
|
{Name: "ID", Type: "ID", inputType: "ID", implements: []string{"String"}, Example: "ID(\"foo\")", RegisterInput: true, defaultValue: "ID(\"\")"},
|
|
{Name: "Input", Type: "interface{}", Example: "String(\"any\")"},
|
|
{Name: "Int", Type: "int", Example: "Int(42)", GenerateConfig: true, DefaultConfig: "0", elemExample: "42", RegisterInput: true, defaultValue: "Int(0)"},
|
|
{Name: "String", Type: "string", Example: "String(\"foo\")", elemExample: "\"foo\"", RegisterInput: true, defaultValue: "String(\"\")"},
|
|
{Name: "URN", Type: "URN", inputType: "URN", implements: []string{"String"}, Example: "URN(\"foo\")", RegisterInput: true, defaultValue: "URN(\"\")"},
|
|
})
|
|
|
|
func unexported(s string) string {
|
|
runes := []rune(s)
|
|
|
|
allCaps := true
|
|
for _, r := range runes {
|
|
if !unicode.IsUpper(r) {
|
|
allCaps = false
|
|
break
|
|
}
|
|
}
|
|
|
|
if allCaps {
|
|
return strings.ToLower(s)
|
|
}
|
|
return string(append([]rune{unicode.ToLower(runes[0])}, runes[1:]...))
|
|
}
|
|
|
|
var funcs = template.FuncMap{
|
|
"Unexported": unexported,
|
|
}
|
|
|
|
func makeBuiltins(primitives []*builtin) []*builtin {
|
|
// Augment primitives with array and map types.
|
|
var builtins []*builtin
|
|
for _, p := range primitives {
|
|
name := ""
|
|
if p.Name != "Input" {
|
|
builtins = append(builtins, p)
|
|
name = p.Name
|
|
}
|
|
switch name {
|
|
case "Archive", "Asset", "AssetOrArchive", "":
|
|
// do nothing
|
|
default:
|
|
builtins = append(builtins, &builtin{
|
|
Name: name + "Ptr",
|
|
Type: "*" + p.Type,
|
|
inputType: "*" + unexported(p.Type) + "Ptr",
|
|
Example: fmt.Sprintf("%sPtr(%s(%s))", name, p.Type, p.Example),
|
|
RegisterInput: p.RegisterInput,
|
|
defaultValue: p.defaultValue,
|
|
})
|
|
}
|
|
arrType := &builtin{
|
|
Name: name + "Array",
|
|
Type: "[]" + name + "Input",
|
|
ItemType: name + "Input",
|
|
elementType: "[]" + p.Type,
|
|
item: p,
|
|
Example: fmt.Sprintf("%sArray{%s}", name, p.Example),
|
|
RegisterInput: true,
|
|
}
|
|
builtins = append(builtins, arrType)
|
|
mapType := &builtin{
|
|
Name: name + "Map",
|
|
Type: "map[string]" + name + "Input",
|
|
ItemType: name + "Input",
|
|
elementType: "map[string]" + p.Type,
|
|
item: p,
|
|
Example: fmt.Sprintf("%sMap{\"baz\": %s}", name, p.Example),
|
|
RegisterInput: true,
|
|
}
|
|
builtins = append(builtins, mapType)
|
|
builtins = append(builtins, &builtin{
|
|
Name: name + "ArrayMap",
|
|
Type: "map[string]" + name + "ArrayInput",
|
|
ItemType: name + "ArrayInput",
|
|
elementType: "map[string][]" + p.Type,
|
|
item: arrType,
|
|
Example: fmt.Sprintf("%sArrayMap{\"baz\": %sArray{%s}}", name, name, p.Example),
|
|
RegisterInput: true,
|
|
})
|
|
builtins = append(builtins, &builtin{
|
|
Name: name + "MapArray",
|
|
Type: "[]" + name + "MapInput",
|
|
ItemType: name + "MapInput",
|
|
elementType: "[]map[string]" + p.Type,
|
|
item: mapType,
|
|
Example: fmt.Sprintf("%sMapArray{%sMap{\"baz\": %s}}", name, name, p.Example),
|
|
RegisterInput: true,
|
|
})
|
|
builtins = append(builtins, &builtin{
|
|
Name: name + "MapMap",
|
|
Type: "map[string]" + name + "MapInput",
|
|
ItemType: name + "MapInput",
|
|
elementType: "map[string]map[string]" + p.Type,
|
|
item: mapType,
|
|
Example: fmt.Sprintf("%sMapMap{\"baz\": %sMap{\"baz\": %s}}", name, name, p.Example),
|
|
RegisterInput: true,
|
|
})
|
|
arrayArrayType := &builtin{
|
|
Name: name + "ArrayArray",
|
|
Type: "[]" + name + "ArrayInput",
|
|
ItemType: name + "ArrayInput",
|
|
elementType: "[][]" + p.Type,
|
|
item: arrType,
|
|
Example: fmt.Sprintf("%sArrayArray{%sArray{%s}}", name, name, p.Example),
|
|
RegisterInput: true,
|
|
}
|
|
builtins = append(builtins, arrayArrayType)
|
|
|
|
// TODO - consider expanding to all primitives?
|
|
if name == "" {
|
|
builtins = append(builtins, &builtin{
|
|
Name: "ArrayArrayMap",
|
|
Type: "map[string]ArrayArrayInput",
|
|
ItemType: "ArrayArrayInput",
|
|
elementType: "map[string][][]" + p.Type,
|
|
item: arrayArrayType,
|
|
Example: fmt.Sprintf("%sArrayArrayMap{\"baz\": %sArrayArray{Array{%s}}}", name, name, p.Example),
|
|
RegisterInput: true,
|
|
})
|
|
}
|
|
}
|
|
|
|
nameToBuiltin := map[string]*builtin{}
|
|
for _, b := range builtins {
|
|
nameToBuiltin[b.Name] = b
|
|
}
|
|
|
|
for _, b := range builtins {
|
|
for _, i := range b.implements {
|
|
b.Implements = append(b.Implements, nameToBuiltin[i])
|
|
}
|
|
}
|
|
|
|
return builtins
|
|
}
|
|
|
|
func main() {
|
|
templates, err := template.New("templates").Funcs(funcs).ParseGlob("./templates/*")
|
|
if err != nil {
|
|
log.Fatalf("failed to parse templates: %v", err)
|
|
}
|
|
|
|
data := map[string]interface{}{
|
|
"Builtins": builtins,
|
|
}
|
|
for _, t := range templates.Templates() {
|
|
// template with Name = "foo-bar.go.template" will be written to ./foo/bar.go
|
|
// "bar.go.template" ./bar.go
|
|
// "fizz-buzz-bar.go.template" ./fizz/buzz/bar.go
|
|
|
|
pwd, err := os.Getwd()
|
|
if err != nil {
|
|
log.Fatalf("code generation failed: %v", err.Error())
|
|
}
|
|
|
|
parts := strings.Split(t.Name(), "-")
|
|
filename := strings.TrimRight(parts[len(parts)-1], ".template")
|
|
parts[len(parts)-1] = filename
|
|
|
|
paths := append([]string{pwd}, parts...)
|
|
|
|
fullname := filepath.Join(paths...)
|
|
f, err := os.Create(fullname)
|
|
if err != nil {
|
|
log.Fatalf("failed to create %v: %v", fullname, err)
|
|
}
|
|
if err := t.Execute(f, data); err != nil {
|
|
log.Fatalf("failed to execute %v: %v", t.Name(), err)
|
|
}
|
|
f.Close()
|
|
|
|
gofmt := exec.Command("gofmt", "-s", "-w", fullname)
|
|
stderr, err := gofmt.StderrPipe()
|
|
if err != nil {
|
|
log.Fatalf("failed to pipe stderr from gofmt: %v", err)
|
|
}
|
|
go func() {
|
|
io.Copy(os.Stderr, stderr)
|
|
}()
|
|
if err := gofmt.Run(); err != nil {
|
|
log.Fatalf("failed to gofmt %v: %v", fullname, err)
|
|
}
|
|
}
|
|
}
|