mirror of https://github.com/pulumi/pulumi.git
282 lines
8.3 KiB
Go
282 lines
8.3 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
|
|
Example string
|
|
GenerateConfig bool
|
|
DefaultConfig 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", "float32", "float64", "ID", "int", "int16", "int32", "int64", "int8", "string", "URN", "uint",
|
|
"uint16", "uint32", "uint64", "uint8":
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
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) 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
|
|
}
|
|
|
|
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"},
|
|
{Name: "Float32", Type: "float32", Example: "Float32(1.3)", GenerateConfig: true, DefaultConfig: "0"},
|
|
{Name: "Float64", Type: "float64", Example: "Float64(999.9)", GenerateConfig: true, DefaultConfig: "0"},
|
|
{Name: "ID", Type: "ID", inputType: "ID", implements: []string{"String"}, Example: "ID(\"foo\")"},
|
|
{Name: "Input", Type: "interface{}", Example: "String(\"any\")"},
|
|
{Name: "Int", Type: "int", Example: "Int(42)", GenerateConfig: true, DefaultConfig: "0"},
|
|
{Name: "Int16", Type: "int16", Example: "Int16(33)", GenerateConfig: true, DefaultConfig: "0"},
|
|
{Name: "Int32", Type: "int32", Example: "Int32(24)", GenerateConfig: true, DefaultConfig: "0"},
|
|
{Name: "Int64", Type: "int64", Example: "Int64(15)", GenerateConfig: true, DefaultConfig: "0"},
|
|
{Name: "Int8", Type: "int8", Example: "Int8(6)", GenerateConfig: true, DefaultConfig: "0"},
|
|
{Name: "String", Type: "string", Example: "String(\"foo\")"},
|
|
{Name: "URN", Type: "URN", inputType: "URN", implements: []string{"String"}, Example: "URN(\"foo\")"},
|
|
{Name: "Uint", Type: "uint", Example: "Uint(42)", GenerateConfig: true, DefaultConfig: "0"},
|
|
{Name: "Uint16", Type: "uint16", Example: "Uint16(33)", GenerateConfig: true, DefaultConfig: "0"},
|
|
{Name: "Uint32", Type: "uint32", Example: "Uint32(24)", GenerateConfig: true, DefaultConfig: "0"},
|
|
{Name: "Uint64", Type: "uint64", Example: "Uint64(15)", GenerateConfig: true, DefaultConfig: "0"},
|
|
{Name: "Uint8", Type: "uint8", Example: "Uint8(6)", GenerateConfig: true, DefaultConfig: "0"},
|
|
})
|
|
|
|
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),
|
|
})
|
|
}
|
|
builtins = append(builtins, &builtin{
|
|
Name: name + "Array",
|
|
Type: "[]" + name + "Input",
|
|
elementType: "[]" + p.Type,
|
|
Example: fmt.Sprintf("%sArray{%s}", name, p.Example),
|
|
})
|
|
builtins = append(builtins, &builtin{
|
|
Name: name + "Map",
|
|
Type: "map[string]" + name + "Input",
|
|
elementType: "map[string]" + p.Type,
|
|
Example: fmt.Sprintf("%sMap{\"baz\": %s}", name, p.Example),
|
|
})
|
|
builtins = append(builtins, &builtin{
|
|
Name: name + "ArrayMap",
|
|
Type: "map[string]" + name + "ArrayInput",
|
|
elementType: "map[string][]" + p.Type,
|
|
Example: fmt.Sprintf("%sArrayMap{\"baz\": %sArray{%s}}", name, name, p.Example),
|
|
})
|
|
builtins = append(builtins, &builtin{
|
|
Name: name + "MapArray",
|
|
Type: "[]" + name + "MapInput",
|
|
elementType: "[]map[string]" + p.Type,
|
|
Example: fmt.Sprintf("%sMapArray{%sMap{\"baz\": %s}}", name, name, p.Example),
|
|
})
|
|
builtins = append(builtins, &builtin{
|
|
Name: name + "MapMap",
|
|
Type: "map[string]" + name + "MapInput",
|
|
elementType: "map[string]map[string]" + p.Type,
|
|
Example: fmt.Sprintf("%sMapMap{\"baz\": %sMap{\"baz\": %s}}", name, name, p.Example),
|
|
})
|
|
builtins = append(builtins, &builtin{
|
|
Name: name + "ArrayArray",
|
|
Type: "[]" + name + "ArrayInput",
|
|
elementType: "[][]" + p.Type,
|
|
Example: fmt.Sprintf("%sArrayArray{%sArray{%s}}", name, name, p.Example),
|
|
})
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|
|
}
|