mirror of https://github.com/databricks/cli.git
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
|
package flags
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ProgressLogFormat string
|
||
|
|
|
||
|
|
var (
|
||
|
|
ModeAppend = ProgressLogFormat("append")
|
||
|
|
ModeInplace = ProgressLogFormat("inplace")
|
||
|
|
ModeJson = ProgressLogFormat("json")
|
||
|
|
ModeDefault = ProgressLogFormat("default")
|
||
|
|
)
|
||
|
|
|
||
|
|
func (p *ProgressLogFormat) String() string {
|
||
|
|
return string(*p)
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewProgressLogFormat() ProgressLogFormat {
|
||
|
|
return ModeDefault
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *ProgressLogFormat) Set(s string) error {
|
||
|
|
lower := strings.ToLower(s)
|
||
|
|
switch lower {
|
||
|
|
case ModeAppend.String():
|
||
|
|
*p = ProgressLogFormat(ModeAppend.String())
|
||
|
|
case ModeInplace.String():
|
||
|
|
*p = ProgressLogFormat(ModeInplace.String())
|
||
|
|
case ModeJson.String():
|
||
|
|
*p = ProgressLogFormat(ModeJson.String())
|
||
|
|
default:
|
||
|
|
valid := []string{
|
||
|
|
ModeAppend.String(),
|
||
|
|
ModeInplace.String(),
|
||
|
|
ModeJson.String(),
|
||
|
|
}
|
||
|
|
return fmt.Errorf("accepted arguments are [%s]", strings.Join(valid, ", "))
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *ProgressLogFormat) Type() string {
|
||
|
|
return "format"
|
||
|
|
}
|
||
|
|
|
||
|
|
// Complete is the Cobra compatible completion function for this flag.
|
||
|
|
func (f *ProgressLogFormat) Complete(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
|
||
|
|
return []string{
|
||
|
|
"append",
|
||
|
|
"inplace",
|
||
|
|
"json",
|
||
|
|
}, cobra.ShellCompDirectiveNoFileComp
|
||
|
|
}
|