mirror of https://github.com/databricks/cli.git
58 lines
1.8 KiB
Go
58 lines
1.8 KiB
Go
package auth
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/databricks/cli/libs/auth"
|
|
"github.com/databricks/cli/libs/cmdio"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func New() *cobra.Command {
|
|
cmd := &cobra.Command{
|
|
Use: "auth",
|
|
Short: "Authentication related commands",
|
|
Long: `Authentication related commands. For more information regarding how
|
|
authentication for the Databricks CLI and SDKs work please refer to the documentation
|
|
linked below.
|
|
|
|
AWS: https://docs.databricks.com/dev-tools/auth/index.html
|
|
Azure: https://learn.microsoft.com/azure/databricks/dev-tools/auth
|
|
GCP: https://docs.gcp.databricks.com/dev-tools/auth/index.html`,
|
|
}
|
|
|
|
var perisistentAuth auth.PersistentAuth
|
|
cmd.PersistentFlags().StringVar(&perisistentAuth.Host, "host", perisistentAuth.Host, "Databricks Host")
|
|
cmd.PersistentFlags().StringVar(&perisistentAuth.AccountID, "account-id", perisistentAuth.AccountID, "Databricks Account ID")
|
|
|
|
cmd.AddCommand(newEnvCommand())
|
|
cmd.AddCommand(newLoginCommand(&perisistentAuth))
|
|
cmd.AddCommand(newProfilesCommand())
|
|
cmd.AddCommand(newTokenCommand(&perisistentAuth))
|
|
cmd.AddCommand(newDescribeCommand())
|
|
return cmd
|
|
}
|
|
|
|
func promptForHost(ctx context.Context) (string, error) {
|
|
if !cmdio.IsInTTY(ctx) {
|
|
return "", fmt.Errorf("the command is being run in a non-interactive environment, please specify a host using --host")
|
|
}
|
|
|
|
prompt := cmdio.Prompt(ctx)
|
|
prompt.Label = "Databricks host (e.g. https://<databricks-instance>.cloud.databricks.com)"
|
|
return prompt.Run()
|
|
}
|
|
|
|
func promptForAccountID(ctx context.Context) (string, error) {
|
|
if !cmdio.IsInTTY(ctx) {
|
|
return "", fmt.Errorf("the command is being run in a non-interactive environment, please specify an account ID using --account-id")
|
|
}
|
|
|
|
prompt := cmdio.Prompt(ctx)
|
|
prompt.Label = "Databricks account ID"
|
|
prompt.Default = ""
|
|
prompt.AllowEdit = true
|
|
return prompt.Run()
|
|
}
|