package main import ( "os" "os/signal" "github.com/hashicorp/consul/command" "github.com/hashicorp/consul/command/agent" "github.com/mitchellh/cli" ) // Commands is the mapping of all the available Consul commands. var Commands map[string]cli.CommandFactory func init() { ui := &cli.BasicUi{Writer: os.Stdout} Commands = map[string]cli.CommandFactory{ "agent": func() (cli.Command, error) { return &agent.Command{ Revision: GitCommit, Version: Version, VersionPrerelease: VersionPrerelease, Ui: ui, ShutdownCh: make(chan struct{}), }, nil }, "event": func() (cli.Command, error) { return &command.EventCommand{ Ui: ui, }, nil }, "exec": func() (cli.Command, error) { return &command.ExecCommand{ ShutdownCh: makeShutdownCh(), Ui: ui, }, nil }, "force-leave": func() (cli.Command, error) { return &command.ForceLeaveCommand{ Ui: ui, }, nil }, "join": func() (cli.Command, error) { return &command.JoinCommand{ Ui: ui, }, nil }, "keygen": func() (cli.Command, error) { return &command.KeygenCommand{ Ui: ui, }, nil }, "keyring": func() (cli.Command, error) { return &command.KeyringCommand{ Ui: ui, }, nil }, "leave": func() (cli.Command, error) { return &command.LeaveCommand{ Ui: ui, }, nil }, "lock": func() (cli.Command, error) { return &command.LockCommand{ ShutdownCh: makeShutdownCh(), Ui: ui, }, nil }, "maint": func() (cli.Command, error) { return &command.MaintCommand{ Ui: ui, }, nil }, "members": func() (cli.Command, error) { return &command.MembersCommand{ Ui: ui, }, nil }, "monitor": func() (cli.Command, error) { return &command.MonitorCommand{ ShutdownCh: makeShutdownCh(), Ui: ui, }, nil }, "info": func() (cli.Command, error) { return &command.InfoCommand{ Ui: ui, }, nil }, "reload": func() (cli.Command, error) { return &command.ReloadCommand{ Ui: ui, }, nil }, "version": func() (cli.Command, error) { ver := Version rel := VersionPrerelease if GitDescribe != "" { ver = GitDescribe } if GitDescribe == "" && rel == "" { rel = "dev" } return &command.VersionCommand{ Revision: GitCommit, Version: ver, VersionPrerelease: rel, Ui: ui, }, nil }, "watch": func() (cli.Command, error) { return &command.WatchCommand{ ShutdownCh: makeShutdownCh(), Ui: ui, }, nil }, } } // makeShutdownCh returns a channel that can be used for shutdown // notifications for commands. This channel will send a message for every // interrupt received. func makeShutdownCh() <-chan struct{} { resultCh := make(chan struct{}) signalCh := make(chan os.Signal, 4) signal.Notify(signalCh, os.Interrupt) go func() { for { <-signalCh resultCh <- struct{}{} } }() return resultCh }