bfea3dea34
This add autocompletion for all flags and commands in the cli. To use this you'll have to source the completions file generated by `corso completion [bash|zsh|fish|powershell]`. You can do that by doing the following (example for bash): ``` bash corso completion bash > /tmp/corso_completions source /tmp/corso_completions ```
77 lines
2.1 KiB
Go
77 lines
2.1 KiB
Go
package flags
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
const (
|
|
UserFN = "user"
|
|
MailBoxFN = "mailbox"
|
|
AzureClientTenantFN = "azure-tenant-id"
|
|
AzureClientIDFN = "azure-client-id"
|
|
AzureClientSecretFN = "azure-client-secret"
|
|
)
|
|
|
|
var (
|
|
UserFV []string
|
|
AzureClientTenantFV string
|
|
AzureClientIDFV string
|
|
AzureClientSecretFV string
|
|
)
|
|
|
|
// AddUserFlag adds the --user flag.
|
|
func AddUserFlag(
|
|
cmd *cobra.Command,
|
|
completionFunc func(*cobra.Command, []string, string) ([]string, cobra.ShellCompDirective),
|
|
) {
|
|
cmd.Flags().StringSliceVar(
|
|
&UserFV,
|
|
UserFN, nil,
|
|
"Backup a specific user's data; accepts '"+Wildcard+"' to select all users.")
|
|
cobra.CheckErr(cmd.MarkFlagRequired(UserFN))
|
|
|
|
cobra.CheckErr(cmd.RegisterFlagCompletionFunc(UserFN, completionFunc))
|
|
}
|
|
|
|
// AddMailBoxFlag adds the --user and --mailbox flag.
|
|
func AddMailBoxFlag(
|
|
cmd *cobra.Command,
|
|
completionFunc func(*cobra.Command, []string, string) ([]string, cobra.ShellCompDirective),
|
|
) {
|
|
flags := cmd.Flags()
|
|
|
|
flags.StringSliceVar(
|
|
&UserFV,
|
|
UserFN, nil,
|
|
"Backup a specific user's data; accepts '"+Wildcard+"' to select all users.")
|
|
|
|
cobra.CheckErr(flags.MarkDeprecated(UserFN, fmt.Sprintf("use --%s instead", MailBoxFN)))
|
|
|
|
cobra.CheckErr(cmd.RegisterFlagCompletionFunc(UserFN,
|
|
func(
|
|
cmd *cobra.Command,
|
|
args []string,
|
|
toComplete string,
|
|
) ([]string, cobra.ShellCompDirective) {
|
|
message := fmt.Sprintf("This flag is deprecated, Use --%s instead", MailBoxFN)
|
|
return cobra.AppendActiveHelp(nil, message), cobra.ShellCompDirectiveNoFileComp
|
|
}))
|
|
|
|
flags.StringSliceVar(
|
|
&UserFV,
|
|
MailBoxFN, nil,
|
|
"Backup a specific mailbox's data; accepts '"+Wildcard+"' to select all mailbox.")
|
|
|
|
cobra.CheckErr(cmd.RegisterFlagCompletionFunc(MailBoxFN, completionFunc))
|
|
}
|
|
|
|
// AddAzureCredsFlags adds M365 cred flags
|
|
func AddAzureCredsFlags(cmd *cobra.Command) {
|
|
fs := cmd.Flags()
|
|
fs.StringVar(&AzureClientTenantFV, AzureClientTenantFN, "", "Azure tenant ID")
|
|
fs.StringVar(&AzureClientIDFV, AzureClientIDFN, "", "Azure app client ID")
|
|
fs.StringVar(&AzureClientSecretFV, AzureClientSecretFN, "", "Azure app client secret")
|
|
}
|