mirror of https://github.com/databricks/cli.git
115 lines
2.5 KiB
Go
115 lines
2.5 KiB
Go
package databrickscfg
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/databricks/databricks-sdk-go/config"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// Profile holds a subset of the keys in a databrickscfg profile.
|
|
// It should only be used for prompting and filtering.
|
|
// Use its name to construct a config.Config.
|
|
type Profile struct {
|
|
Name string
|
|
Host string
|
|
AccountID string
|
|
}
|
|
|
|
func (p Profile) Cloud() string {
|
|
cfg := config.Config{Host: p.Host}
|
|
switch {
|
|
case cfg.IsAws():
|
|
return "AWS"
|
|
case cfg.IsAzure():
|
|
return "Azure"
|
|
case cfg.IsGcp():
|
|
return "GCP"
|
|
default:
|
|
return ""
|
|
}
|
|
}
|
|
|
|
type Profiles []Profile
|
|
|
|
func (p Profiles) Names() []string {
|
|
names := make([]string, len(p))
|
|
for i, v := range p {
|
|
names[i] = v.Name
|
|
}
|
|
return names
|
|
}
|
|
|
|
// SearchCaseInsensitive implements the promptui.Searcher interface.
|
|
// This allows the user to immediately starting typing to narrow down the list.
|
|
func (p Profiles) SearchCaseInsensitive(input string, index int) bool {
|
|
input = strings.ToLower(input)
|
|
name := strings.ToLower(p[index].Name)
|
|
host := strings.ToLower(p[index].Host)
|
|
return strings.Contains(name, input) || strings.Contains(host, input)
|
|
}
|
|
|
|
type ProfileMatchFunction func(Profile) bool
|
|
|
|
func MatchWorkspaceProfiles(p Profile) bool {
|
|
return p.AccountID == ""
|
|
}
|
|
|
|
func MatchAccountProfiles(p Profile) bool {
|
|
return p.Host != "" && p.AccountID != ""
|
|
}
|
|
|
|
func MatchAllProfiles(p Profile) bool {
|
|
return true
|
|
}
|
|
|
|
const DefaultPath = "~/.databrickscfg"
|
|
|
|
func LoadProfiles(path string, fn ProfileMatchFunction) (file string, profiles Profiles, err error) {
|
|
f, err := config.LoadFile(path)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
homedir, err := os.UserHomeDir()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// Replace homedir with ~ if applicable.
|
|
// This is to make the output more readable.
|
|
file = f.Path()
|
|
if strings.HasPrefix(file, homedir) {
|
|
file = "~" + file[len(homedir):]
|
|
}
|
|
|
|
// Iterate over sections and collect matching profiles.
|
|
for _, v := range f.Sections() {
|
|
all := v.KeysHash()
|
|
host, ok := all["host"]
|
|
if !ok {
|
|
// invalid profile
|
|
continue
|
|
}
|
|
profile := Profile{
|
|
Name: v.Name(),
|
|
Host: host,
|
|
AccountID: all["account_id"],
|
|
}
|
|
if fn(profile) {
|
|
profiles = append(profiles, profile)
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func ProfileCompletion(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
|
|
_, profiles, err := LoadProfiles(DefaultPath, MatchAllProfiles)
|
|
if err != nil {
|
|
return nil, cobra.ShellCompDirectiveError
|
|
}
|
|
return profiles.Names(), cobra.ShellCompDirectiveNoFileComp
|
|
}
|