docker.go 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160
  1. package main
  2. import (
  3. "fmt"
  4. "os"
  5. "github.com/Sirupsen/logrus"
  6. "github.com/docker/docker/api/client"
  7. "github.com/docker/docker/api/client/container"
  8. "github.com/docker/docker/api/client/image"
  9. "github.com/docker/docker/api/client/network"
  10. "github.com/docker/docker/api/client/node"
  11. "github.com/docker/docker/api/client/plugin"
  12. "github.com/docker/docker/api/client/registry"
  13. "github.com/docker/docker/api/client/service"
  14. "github.com/docker/docker/api/client/stack"
  15. "github.com/docker/docker/api/client/swarm"
  16. "github.com/docker/docker/api/client/system"
  17. "github.com/docker/docker/api/client/volume"
  18. "github.com/docker/docker/cli"
  19. "github.com/docker/docker/cli/cobraadaptor"
  20. cliflags "github.com/docker/docker/cli/flags"
  21. "github.com/docker/docker/cliconfig"
  22. "github.com/docker/docker/dockerversion"
  23. "github.com/docker/docker/pkg/term"
  24. "github.com/docker/docker/utils"
  25. "github.com/spf13/cobra"
  26. "github.com/spf13/pflag"
  27. )
  28. func newDockerCommand(dockerCli *client.DockerCli) *cobra.Command {
  29. opts := cliflags.NewClientOptions()
  30. cmd := &cobra.Command{
  31. Use: "docker [OPTIONS] COMMAND [arg...]",
  32. Short: "A self-sufficient runtime for containers.",
  33. SilenceUsage: true,
  34. SilenceErrors: true,
  35. TraverseChildren: true,
  36. Args: cli.NoArgs,
  37. RunE: func(cmd *cobra.Command, args []string) error {
  38. if opts.Version {
  39. showVersion()
  40. return nil
  41. }
  42. fmt.Fprintf(dockerCli.Err(), "\n"+cmd.UsageString())
  43. return nil
  44. },
  45. PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
  46. dockerPreRun(cmd.Flags(), opts)
  47. return dockerCli.Initialize(opts)
  48. },
  49. }
  50. cobraadaptor.SetupRootCommand(cmd)
  51. flags := cmd.Flags()
  52. flags.BoolVarP(&opts.Version, "version", "v", false, "Print version information and quit")
  53. flags.StringVar(&opts.ConfigDir, "config", cliconfig.ConfigDir(), "Location of client config files")
  54. opts.Common.InstallFlags(flags)
  55. cmd.SetOutput(dockerCli.Out())
  56. cmd.AddCommand(
  57. newDaemonCommand(),
  58. node.NewNodeCommand(dockerCli),
  59. service.NewServiceCommand(dockerCli),
  60. stack.NewStackCommand(dockerCli),
  61. stack.NewTopLevelDeployCommand(dockerCli),
  62. swarm.NewSwarmCommand(dockerCli),
  63. container.NewAttachCommand(dockerCli),
  64. container.NewCommitCommand(dockerCli),
  65. container.NewCopyCommand(dockerCli),
  66. container.NewCreateCommand(dockerCli),
  67. container.NewDiffCommand(dockerCli),
  68. container.NewExecCommand(dockerCli),
  69. container.NewExportCommand(dockerCli),
  70. container.NewKillCommand(dockerCli),
  71. container.NewLogsCommand(dockerCli),
  72. container.NewPauseCommand(dockerCli),
  73. container.NewPortCommand(dockerCli),
  74. container.NewPsCommand(dockerCli),
  75. container.NewRenameCommand(dockerCli),
  76. container.NewRestartCommand(dockerCli),
  77. container.NewRmCommand(dockerCli),
  78. container.NewRunCommand(dockerCli),
  79. container.NewStartCommand(dockerCli),
  80. container.NewStatsCommand(dockerCli),
  81. container.NewStopCommand(dockerCli),
  82. container.NewTopCommand(dockerCli),
  83. container.NewUnpauseCommand(dockerCli),
  84. container.NewUpdateCommand(dockerCli),
  85. container.NewWaitCommand(dockerCli),
  86. image.NewBuildCommand(dockerCli),
  87. image.NewHistoryCommand(dockerCli),
  88. image.NewImagesCommand(dockerCli),
  89. image.NewLoadCommand(dockerCli),
  90. image.NewRemoveCommand(dockerCli),
  91. image.NewSaveCommand(dockerCli),
  92. image.NewPullCommand(dockerCli),
  93. image.NewPushCommand(dockerCli),
  94. image.NewSearchCommand(dockerCli),
  95. image.NewImportCommand(dockerCli),
  96. image.NewTagCommand(dockerCli),
  97. network.NewNetworkCommand(dockerCli),
  98. system.NewEventsCommand(dockerCli),
  99. system.NewInspectCommand(dockerCli),
  100. registry.NewLoginCommand(dockerCli),
  101. registry.NewLogoutCommand(dockerCli),
  102. system.NewVersionCommand(dockerCli),
  103. volume.NewVolumeCommand(dockerCli),
  104. system.NewInfoCommand(dockerCli),
  105. )
  106. plugin.NewPluginCommand(cmd, dockerCli)
  107. return cmd
  108. }
  109. func main() {
  110. // Set terminal emulation based on platform as required.
  111. stdin, stdout, stderr := term.StdStreams()
  112. logrus.SetOutput(stderr)
  113. dockerCli := client.NewDockerCli(stdin, stdout, stderr)
  114. cmd := newDockerCommand(dockerCli)
  115. if err := cmd.Execute(); err != nil {
  116. if sterr, ok := err.(cli.StatusError); ok {
  117. if sterr.Status != "" {
  118. fmt.Fprintln(stderr, sterr.Status)
  119. }
  120. // StatusError should only be used for errors, and all errors should
  121. // have a non-zero exit status, so never exit with 0
  122. if sterr.StatusCode == 0 {
  123. os.Exit(1)
  124. }
  125. os.Exit(sterr.StatusCode)
  126. }
  127. fmt.Fprintln(stderr, err)
  128. os.Exit(1)
  129. }
  130. }
  131. func showVersion() {
  132. if utils.ExperimentalBuild() {
  133. fmt.Printf("Docker version %s, build %s, experimental\n", dockerversion.Version, dockerversion.GitCommit)
  134. } else {
  135. fmt.Printf("Docker version %s, build %s\n", dockerversion.Version, dockerversion.GitCommit)
  136. }
  137. }
  138. func dockerPreRun(flags *pflag.FlagSet, opts *cliflags.ClientOptions) {
  139. opts.Common.SetDefaultOptions(flags)
  140. cliflags.SetDaemonLogLevel(opts.Common.LogLevel)
  141. if opts.ConfigDir != "" {
  142. cliconfig.SetConfigDir(opts.ConfigDir)
  143. }
  144. if opts.Common.Debug {
  145. utils.EnableDebug()
  146. }
  147. }