databricks-cli/libs/template/helpers.go

116 lines
3.1 KiB
Go

package template
import (
"context"
"errors"
"fmt"
"net/url"
"regexp"
"text/template"
"github.com/databricks/cli/cmd/root"
"github.com/databricks/cli/libs/auth"
"github.com/databricks/databricks-sdk-go/service/iam"
)
type ErrFail struct {
msg string
}
func (err ErrFail) Error() string {
return err.msg
}
type pair struct {
k string
v any
}
func loadHelpers(ctx context.Context) template.FuncMap {
var user *iam.User
var is_service_principal *bool
w := root.WorkspaceClient(ctx)
return template.FuncMap{
"fail": func(format string, args ...any) (any, error) {
return nil, ErrFail{fmt.Sprintf(format, args...)}
},
// Alias for https://pkg.go.dev/net/url#Parse. Allows usage of all methods of url.URL
"url": func(rawUrl string) (*url.URL, error) {
return url.Parse(rawUrl)
},
// Alias for https://pkg.go.dev/regexp#Compile. Allows usage of all methods of regexp.Regexp
"regexp": func(expr string) (*regexp.Regexp, error) {
return regexp.Compile(expr)
},
// A key value pair. This is used with the map function to generate maps
// to use inside a template
"pair": func(k string, v any) pair {
return pair{k, v}
},
// map converts a list of pairs to a map object. This is useful to pass multiple
// objects to templates defined in the library directory. Go text template
// syntax for invoking a template only allows specifying a single argument,
// this function can be used to workaround that limitation.
//
// For example: {{template "my_template" (map (pair "foo" $arg1) (pair "bar" $arg2))}}
// $arg1 and $arg2 can be referred from inside "my_template" as ".foo" and ".bar"
"map": func(pairs ...pair) map[string]any {
result := make(map[string]any, 0)
for _, p := range pairs {
result[p.k] = p.v
}
return result
},
// Get smallest node type (follows Terraform's GetSmallestNodeType)
"smallest_node_type": func() (string, error) {
if w.Config.Host == "" {
return "", errors.New("cannot determine target workspace, please first setup a configuration profile using 'databricks auth login'")
}
if w.Config.IsAzure() {
return "Standard_D3_v2", nil
} else if w.Config.IsGcp() {
return "n1-standard-4", nil
}
return "i3.xlarge", nil
},
"workspace_host": func() (string, error) {
if w.Config.Host == "" {
return "", errors.New("cannot determine target workspace, please first setup a configuration profile using 'databricks auth login'")
}
return w.Config.Host, nil
},
"user_name": func() (string, error) {
if user == nil {
var err error
user, err = w.CurrentUser.Me(ctx)
if err != nil {
return "", err
}
}
result := user.UserName
if result == "" {
result = user.Id
}
return result, nil
},
"is_service_principal": func() (bool, error) {
if is_service_principal != nil {
return *is_service_principal, nil
}
if user == nil {
var err error
user, err = w.CurrentUser.Me(ctx)
if err != nil {
return false, err
}
}
result, err := auth.IsServicePrincipal(ctx, w, user.Id)
if err != nil {
return false, err
}
is_service_principal = &result
return result, nil
},
}
}