Chris Lu
3 years ago
committed by
GitHub
GitHub
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 121 additions and 0 deletions
@ -0,0 +1,109 @@ |
|||
package command |
|||
|
|||
import ( |
|||
"fmt" |
|||
flag "github.com/chrislusf/seaweedfs/weed/util/fla9" |
|||
"github.com/posener/complete" |
|||
completeinstall "github.com/posener/complete/cmd/install" |
|||
"runtime" |
|||
) |
|||
|
|||
func AutocompleteMain(commands []*Command) bool { |
|||
subCommands := make(map[string]complete.Command) |
|||
helpSubCommands := make(map[string]complete.Command) |
|||
for _, cmd := range commands { |
|||
flags := make(map[string]complete.Predictor) |
|||
cmd.Flag.VisitAll(func(flag *flag.Flag) { |
|||
flags["-"+flag.Name] = complete.PredictAnything |
|||
}) |
|||
|
|||
subCommands[cmd.Name()] = complete.Command{ |
|||
Flags: flags, |
|||
} |
|||
helpSubCommands[cmd.Name()] = complete.Command{} |
|||
} |
|||
subCommands["help"] = complete.Command{Sub: helpSubCommands} |
|||
|
|||
globalFlags := make(map[string]complete.Predictor) |
|||
flag.VisitAll(func(flag *flag.Flag) { |
|||
globalFlags["-"+flag.Name] = complete.PredictAnything |
|||
}) |
|||
|
|||
weedCmd := complete.Command{ |
|||
Sub: subCommands, |
|||
Flags: globalFlags, |
|||
GlobalFlags: complete.Flags{"-h": complete.PredictNothing}, |
|||
} |
|||
cmp := complete.New("weed", weedCmd) |
|||
|
|||
return cmp.Complete() |
|||
} |
|||
|
|||
func installAutoCompletion() bool { |
|||
if runtime.GOOS == "windows" { |
|||
fmt.Printf("windows is not supported") |
|||
return false |
|||
} |
|||
|
|||
err := completeinstall.Install("weed") |
|||
if err != nil { |
|||
fmt.Printf("install failed! %s\n", err) |
|||
return false |
|||
} |
|||
fmt.Printf("autocompletion is enabled. Please restart your shell.\n") |
|||
return true |
|||
} |
|||
|
|||
func uninstallAutoCompletion() bool { |
|||
if runtime.GOOS == "windows" { |
|||
fmt.Printf("windows is not supported") |
|||
return false |
|||
} |
|||
|
|||
err := completeinstall.Uninstall("weed") |
|||
if err != nil { |
|||
fmt.Printf("uninstall failed! %s\n", err) |
|||
return false |
|||
} |
|||
fmt.Printf("autocompletion is disable. Please restart your shell.\n") |
|||
return true |
|||
} |
|||
|
|||
var cmdAutocomplete = &Command{ |
|||
Run: runAutocomplete, |
|||
UsageLine: "autocomplete", |
|||
Short: "install autocomplete", |
|||
Long: `weed autocomplete is installed in the shell. |
|||
|
|||
Supported shells are bash, zsh, and fish. |
|||
Windows is not supported. |
|||
|
|||
`, |
|||
} |
|||
|
|||
func runAutocomplete(cmd *Command, args []string) bool { |
|||
if len(args) != 0 { |
|||
cmd.Usage() |
|||
} |
|||
|
|||
return installAutoCompletion() |
|||
} |
|||
|
|||
var cmdUnautocomplete = &Command{ |
|||
Run: runUnautocomplete, |
|||
UsageLine: "unautocomplete", |
|||
Short: "uninstall autocomplete", |
|||
Long: `weed autocomplete is uninstalled in the shell. |
|||
|
|||
Windows is not supported. |
|||
|
|||
`, |
|||
} |
|||
|
|||
func runUnautocomplete(cmd *Command, args []string) bool { |
|||
if len(args) != 0 { |
|||
cmd.Usage() |
|||
} |
|||
|
|||
return uninstallAutoCompletion() |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue