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 } var cachedUser *iam.User var cachedIsServicePrincipal *bool func loadHelpers(ctx context.Context) template.FuncMap { 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 cachedUser == nil { var err error cachedUser, err = w.CurrentUser.Me(ctx) if err != nil { return "", err } } result := cachedUser.UserName if result == "" { result = cachedUser.Id } return result, nil }, "is_service_principal": func() (bool, error) { if cachedIsServicePrincipal != nil { return *cachedIsServicePrincipal, nil } if cachedUser == nil { var err error cachedUser, err = w.CurrentUser.Me(ctx) if err != nil { return false, err } } result := auth.IsServicePrincipal(cachedUser.Id) cachedIsServicePrincipal = &result return result, nil }, } }