mirror of https://github.com/databricks/cli.git
27 lines
953 B
Go
27 lines
953 B
Go
|
package root
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
)
|
||
|
|
|
||
|
|
type skipPrompt int
|
||
|
|
|
||
|
|
var skipPromptKey skipPrompt
|
||
|
|
|
||
|
|
// SkipPrompt allows to skip prompt for profile configuration in MustWorkspaceClient.
|
||
|
|
//
|
||
|
|
// When calling MustWorkspaceClient we want to be able to customise if to show prompt or not.
|
||
|
|
// Since we can't change function interface, in the code we only have an access to `cmd` object.
|
||
|
|
// Command struct does not have any state flag which indicates that it's being called in completion mode and
|
||
|
|
// thus the Context object seems to be the only viable option for us to configure prompt behaviour based on
|
||
|
|
// the context it's executed from.
|
||
|
|
func SkipPrompt(ctx context.Context) context.Context {
|
||
|
|
return context.WithValue(ctx, skipPromptKey, true)
|
||
|
|
}
|
||
|
|
|
||
|
|
// shouldSkipPrompt returns whether or not [SkipPrompt] has been set on the specified context.
|
||
|
|
func shouldSkipPrompt(ctx context.Context) bool {
|
||
|
|
skipPrompt, ok := ctx.Value(skipPromptKey).(bool)
|
||
|
|
return ok && skipPrompt
|
||
|
|
}
|