147 lines
5.1 KiB
Go
147 lines
5.1 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.
|
|
|
|
// Pulling out some of the repeated strings tokens into constants would harm readability, so we just ignore the
|
|
// goconst linter's warning.
|
|
//
|
|
// nolint: lll, goconst
|
|
package gen
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/golang/glog"
|
|
"github.com/pulumi/pulumi/pkg/v2/codegen"
|
|
"github.com/pulumi/pulumi/pkg/v2/codegen/schema"
|
|
)
|
|
|
|
// DocLanguageHelper is the Go-specific implementation of the DocLanguageHelper.
|
|
type DocLanguageHelper struct {
|
|
packages map[string]*pkgContext
|
|
}
|
|
|
|
var _ codegen.DocLanguageHelper = DocLanguageHelper{}
|
|
|
|
// GetDocLinkForPulumiType returns the doc link for a Pulumi type.
|
|
func (d DocLanguageHelper) GetDocLinkForPulumiType(pkg *schema.Package, typeName string) string {
|
|
moduleVersion := ""
|
|
if pkg.Version != nil {
|
|
if pkg.Version.Major > 1 {
|
|
moduleVersion = fmt.Sprintf("v%d/", pkg.Version.Major)
|
|
}
|
|
}
|
|
return fmt.Sprintf("https://pkg.go.dev/github.com/pulumi/pulumi/sdk/%sgo/pulumi?tab=doc#%s", moduleVersion, typeName)
|
|
}
|
|
|
|
// GetDocLinkForResourceType returns the godoc URL for a type belonging to a resource provider.
|
|
func (d DocLanguageHelper) GetDocLinkForResourceType(pkg *schema.Package, moduleName string, typeName string) string {
|
|
path := fmt.Sprintf("%s/%s", goPackage(pkg.Name), moduleName)
|
|
typeNameParts := strings.Split(typeName, ".")
|
|
typeName = typeNameParts[len(typeNameParts)-1]
|
|
typeName = strings.TrimPrefix(typeName, "*")
|
|
|
|
moduleVersion := ""
|
|
if pkg.Version != nil {
|
|
if pkg.Version.Major > 1 {
|
|
moduleVersion = fmt.Sprintf("v%d/", pkg.Version.Major)
|
|
}
|
|
}
|
|
|
|
return fmt.Sprintf("https://pkg.go.dev/github.com/pulumi/pulumi-%s/sdk/%sgo/%s?tab=doc#%s", pkg.Name, moduleVersion, path, typeName)
|
|
}
|
|
|
|
// GetDocLinkForResourceInputOrOutputType returns the godoc URL for an input or output type.
|
|
func (d DocLanguageHelper) GetDocLinkForResourceInputOrOutputType(pkg *schema.Package, moduleName, typeName string, input bool) string {
|
|
link := d.GetDocLinkForResourceType(pkg, moduleName, typeName)
|
|
if !input {
|
|
return link + "Output"
|
|
}
|
|
return link + "Args"
|
|
}
|
|
|
|
// GetDocLinkForFunctionInputOrOutputType returns the doc link for an input or output type of a Function.
|
|
func (d DocLanguageHelper) GetDocLinkForFunctionInputOrOutputType(pkg *schema.Package, moduleName, typeName string, input bool) string {
|
|
link := d.GetDocLinkForResourceType(pkg, moduleName, typeName)
|
|
if !input {
|
|
return link
|
|
}
|
|
return link + "Args"
|
|
}
|
|
|
|
// GetLanguageTypeString returns the Go-specific type given a Pulumi schema type.
|
|
func (d DocLanguageHelper) GetLanguageTypeString(pkg *schema.Package, moduleName string, t schema.Type, input, optional bool) string {
|
|
modPkg, ok := d.packages[moduleName]
|
|
if !ok {
|
|
glog.Errorf("cannot calculate type string for type %q. could not find a package for module %q", t.String(), moduleName)
|
|
os.Exit(1)
|
|
}
|
|
if _, ok := t.(*schema.EnumType); ok {
|
|
return modPkg.inputType(t, optional)
|
|
}
|
|
return modPkg.plainType(t, optional)
|
|
}
|
|
|
|
// GeneratePackagesMap generates a map of Go packages for resources, functions and types.
|
|
func (d *DocLanguageHelper) GeneratePackagesMap(pkg *schema.Package, tool string, goInfo GoPackageInfo) {
|
|
d.packages = generatePackageContextMap(tool, pkg, goInfo)
|
|
}
|
|
|
|
// GetPropertyName returns the property name specific to Go.
|
|
func (d DocLanguageHelper) GetPropertyName(p *schema.Property) (string, error) {
|
|
return strings.Title(p.Name), nil
|
|
}
|
|
|
|
// GetEnumName returns the enum name specific to Go.
|
|
func (d DocLanguageHelper) GetEnumName(e *schema.Enum, typeName string) (string, error) {
|
|
name := fmt.Sprintf("%v", e.Value)
|
|
if e.Name != "" {
|
|
name = e.Name
|
|
}
|
|
return makeSafeEnumName(name, typeName)
|
|
}
|
|
|
|
func (d DocLanguageHelper) GetFunctionName(modName string, f *schema.Function) string {
|
|
funcName := tokenToName(f.Token)
|
|
pkg, ok := d.packages[modName]
|
|
if !ok {
|
|
return funcName
|
|
}
|
|
|
|
if override, ok := pkg.functionNames[f]; ok {
|
|
funcName = override
|
|
}
|
|
return funcName
|
|
}
|
|
|
|
// GetResourceFunctionResultName returns the name of the result type when a function is used to lookup
|
|
// an existing resource.
|
|
func (d DocLanguageHelper) GetResourceFunctionResultName(modName string, f *schema.Function) string {
|
|
funcName := d.GetFunctionName(modName, f)
|
|
return funcName + "Result"
|
|
}
|
|
|
|
// GetModuleDocLink returns the display name and the link for a module.
|
|
func (d DocLanguageHelper) GetModuleDocLink(pkg *schema.Package, modName string) (string, string) {
|
|
var displayName string
|
|
var link string
|
|
if modName == "" {
|
|
displayName = goPackage(pkg.Name)
|
|
} else {
|
|
displayName = fmt.Sprintf("%s/%s", goPackage(pkg.Name), modName)
|
|
}
|
|
link = d.GetDocLinkForResourceType(pkg, modName, "")
|
|
return displayName, link
|
|
}
|