mirror of https://github.com/databricks/cli.git
110 lines
2.1 KiB
Go
110 lines
2.1 KiB
Go
package generator
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"text/template"
|
|
|
|
tfjson "github.com/hashicorp/terraform-json"
|
|
)
|
|
|
|
func normalizeName(name string) string {
|
|
return strings.TrimPrefix(name, "databricks_")
|
|
}
|
|
|
|
type collection struct {
|
|
OutputFile string
|
|
Blocks []*namedBlock
|
|
}
|
|
|
|
func (c *collection) Generate(path string) error {
|
|
tmpl := template.Must(template.ParseFiles(fmt.Sprintf("./templates/%s.tmpl", c.OutputFile)))
|
|
f, err := os.Create(filepath.Join(path, c.OutputFile))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer f.Close()
|
|
|
|
return tmpl.Execute(f, c)
|
|
}
|
|
|
|
func Run(ctx context.Context, schema *tfjson.ProviderSchema, path string) error {
|
|
// Generate types for resources.
|
|
var resources []*namedBlock
|
|
for _, k := range sortKeys(schema.ResourceSchemas) {
|
|
v := schema.ResourceSchemas[k]
|
|
b := &namedBlock{
|
|
filePattern: "resource_%s.go",
|
|
typeNamePrefix: "Resource",
|
|
name: k,
|
|
block: v.Block,
|
|
}
|
|
err := b.Generate(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
resources = append(resources, b)
|
|
}
|
|
|
|
// Generate types for data sources.
|
|
var dataSources []*namedBlock
|
|
for _, k := range sortKeys(schema.DataSourceSchemas) {
|
|
v := schema.DataSourceSchemas[k]
|
|
b := &namedBlock{
|
|
filePattern: "data_source_%s.go",
|
|
typeNamePrefix: "DataSource",
|
|
name: k,
|
|
block: v.Block,
|
|
}
|
|
err := b.Generate(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
dataSources = append(dataSources, b)
|
|
}
|
|
|
|
// Generate type for provider configuration.
|
|
{
|
|
b := &namedBlock{
|
|
filePattern: "%s.go",
|
|
typeNamePrefix: "",
|
|
name: "config",
|
|
block: schema.ConfigSchema.Block,
|
|
}
|
|
err := b.Generate(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Generate resources.go
|
|
{
|
|
cr := &collection{
|
|
OutputFile: "resources.go",
|
|
Blocks: resources,
|
|
}
|
|
err := cr.Generate(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Generate data_sources.go
|
|
{
|
|
cr := &collection{
|
|
OutputFile: "data_sources.go",
|
|
Blocks: dataSources,
|
|
}
|
|
err := cr.Generate(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|