commands.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913
  1. package docker
  2. import (
  3. "bytes"
  4. "encoding/json"
  5. "errors"
  6. "fmt"
  7. "github.com/dotcloud/docker/auth"
  8. "github.com/dotcloud/docker/rcli"
  9. "io"
  10. "log"
  11. "net/http"
  12. "net/url"
  13. "runtime"
  14. "strconv"
  15. "strings"
  16. "sync"
  17. "text/tabwriter"
  18. "time"
  19. )
  20. const VERSION = "0.1.0"
  21. func (srv *Server) Name() string {
  22. return "docker"
  23. }
  24. // FIXME: Stop violating DRY by repeating usage here and in Subcmd declarations
  25. func (srv *Server) Help() string {
  26. help := "Usage: docker COMMAND [arg...]\n\nA self-sufficient runtime for linux containers.\n\nCommands:\n"
  27. for _, cmd := range [][]interface{}{
  28. {"attach", "Attach to a running container"},
  29. {"commit", "Create a new image from a container's changes"},
  30. {"diff", "Inspect changes on a container's filesystem"},
  31. {"export", "Stream the contents of a container as a tar archive"},
  32. {"history", "Show the history of an image"},
  33. {"images", "List images"},
  34. {"import", "Create a new filesystem image from the contents of a tarball"},
  35. {"info", "Display system-wide information"},
  36. {"inspect", "Return low-level information on a container"},
  37. {"kill", "Kill a running container"},
  38. {"login", "Register or Login to the docker registry server"},
  39. {"logs", "Fetch the logs of a container"},
  40. {"port", "Lookup the public-facing port which is NAT-ed to PRIVATE_PORT"},
  41. {"ps", "List containers"},
  42. {"pull", "Pull an image or a repository from the docker registry server"},
  43. {"push", "Push an image or a repository to the docker registry server"},
  44. {"restart", "Restart a running container"},
  45. {"rm", "Remove a container"},
  46. {"rmi", "Remove an image"},
  47. {"run", "Run a command in a new container"},
  48. {"start", "Start a stopped container"},
  49. {"stop", "Stop a running container"},
  50. {"tag", "Tag an image into a repository"},
  51. {"version", "Show the docker version information"},
  52. {"wait", "Block until a container stops, then print its exit code"},
  53. } {
  54. help += fmt.Sprintf(" %-10.10s%s\n", cmd[0], cmd[1])
  55. }
  56. return help
  57. }
  58. // 'docker login': login / register a user to registry service.
  59. func (srv *Server) CmdLogin(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  60. cmd := rcli.Subcmd(stdout, "login", "", "Register or Login to the docker registry server")
  61. if err := cmd.Parse(args); err != nil {
  62. return nil
  63. }
  64. var username string
  65. var password string
  66. var email string
  67. fmt.Fprint(stdout, "Username (", srv.runtime.authConfig.Username, "): ")
  68. fmt.Fscanf(stdin, "%s", &username)
  69. if username == "" {
  70. username = srv.runtime.authConfig.Username
  71. }
  72. if username != srv.runtime.authConfig.Username {
  73. fmt.Fprint(stdout, "Password: ")
  74. fmt.Fscanf(stdin, "%s", &password)
  75. if password == "" {
  76. return errors.New("Error : Password Required\n")
  77. }
  78. fmt.Fprint(stdout, "Email (", srv.runtime.authConfig.Email, "): ")
  79. fmt.Fscanf(stdin, "%s", &email)
  80. if email == "" {
  81. email = srv.runtime.authConfig.Email
  82. }
  83. } else {
  84. password = srv.runtime.authConfig.Password
  85. email = srv.runtime.authConfig.Email
  86. }
  87. newAuthConfig := auth.NewAuthConfig(username, password, email, srv.runtime.root)
  88. status, err := auth.Login(newAuthConfig)
  89. if err != nil {
  90. fmt.Fprintf(stdout, "Error : %s\n", err)
  91. } else {
  92. srv.runtime.authConfig = newAuthConfig
  93. }
  94. if status != "" {
  95. fmt.Fprintf(stdout, status)
  96. }
  97. return nil
  98. }
  99. // 'docker wait': block until a container stops
  100. func (srv *Server) CmdWait(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  101. cmd := rcli.Subcmd(stdout, "wait", "[OPTIONS] NAME", "Block until a container stops, then print its exit code.")
  102. if err := cmd.Parse(args); err != nil {
  103. return nil
  104. }
  105. if cmd.NArg() < 1 {
  106. cmd.Usage()
  107. return nil
  108. }
  109. for _, name := range cmd.Args() {
  110. if container := srv.runtime.Get(name); container != nil {
  111. fmt.Fprintln(stdout, container.Wait())
  112. } else {
  113. return errors.New("No such container: " + name)
  114. }
  115. }
  116. return nil
  117. }
  118. // 'docker version': show version information
  119. func (srv *Server) CmdVersion(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  120. fmt.Fprintf(stdout, "Version:%s\n", VERSION)
  121. return nil
  122. }
  123. // 'docker info': display system-wide information.
  124. func (srv *Server) CmdInfo(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  125. images, _ := srv.runtime.graph.All()
  126. var imgcount int
  127. if images == nil {
  128. imgcount = 0
  129. } else {
  130. imgcount = len(images)
  131. }
  132. cmd := rcli.Subcmd(stdout, "info", "", "Display system-wide information.")
  133. if err := cmd.Parse(args); err != nil {
  134. return nil
  135. }
  136. if cmd.NArg() > 0 {
  137. cmd.Usage()
  138. return nil
  139. }
  140. fmt.Fprintf(stdout, "containers: %d\nversion: %s\nimages: %d\n",
  141. len(srv.runtime.List()),
  142. VERSION,
  143. imgcount)
  144. return nil
  145. }
  146. func (srv *Server) CmdStop(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  147. cmd := rcli.Subcmd(stdout, "stop", "[OPTIONS] NAME", "Stop a running container")
  148. if err := cmd.Parse(args); err != nil {
  149. return nil
  150. }
  151. if cmd.NArg() < 1 {
  152. cmd.Usage()
  153. return nil
  154. }
  155. for _, name := range cmd.Args() {
  156. if container := srv.runtime.Get(name); container != nil {
  157. if err := container.Stop(); err != nil {
  158. return err
  159. }
  160. fmt.Fprintln(stdout, container.Id)
  161. } else {
  162. return errors.New("No such container: " + name)
  163. }
  164. }
  165. return nil
  166. }
  167. func (srv *Server) CmdRestart(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  168. cmd := rcli.Subcmd(stdout, "restart", "[OPTIONS] NAME", "Restart a running container")
  169. if err := cmd.Parse(args); err != nil {
  170. return nil
  171. }
  172. if cmd.NArg() < 1 {
  173. cmd.Usage()
  174. return nil
  175. }
  176. for _, name := range cmd.Args() {
  177. if container := srv.runtime.Get(name); container != nil {
  178. if err := container.Restart(); err != nil {
  179. return err
  180. }
  181. fmt.Fprintln(stdout, container.Id)
  182. } else {
  183. return errors.New("No such container: " + name)
  184. }
  185. }
  186. return nil
  187. }
  188. func (srv *Server) CmdStart(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  189. cmd := rcli.Subcmd(stdout, "start", "[OPTIONS] NAME", "Start a stopped container")
  190. if err := cmd.Parse(args); err != nil {
  191. return nil
  192. }
  193. if cmd.NArg() < 1 {
  194. cmd.Usage()
  195. return nil
  196. }
  197. for _, name := range cmd.Args() {
  198. if container := srv.runtime.Get(name); container != nil {
  199. if err := container.Start(); err != nil {
  200. return err
  201. }
  202. fmt.Fprintln(stdout, container.Id)
  203. } else {
  204. return errors.New("No such container: " + name)
  205. }
  206. }
  207. return nil
  208. }
  209. func (srv *Server) CmdInspect(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  210. cmd := rcli.Subcmd(stdout, "inspect", "[OPTIONS] CONTAINER", "Return low-level information on a container")
  211. if err := cmd.Parse(args); err != nil {
  212. return nil
  213. }
  214. if cmd.NArg() < 1 {
  215. cmd.Usage()
  216. return nil
  217. }
  218. name := cmd.Arg(0)
  219. var obj interface{}
  220. if container := srv.runtime.Get(name); container != nil {
  221. obj = container
  222. } else if image, err := srv.runtime.repositories.LookupImage(name); err == nil && image != nil {
  223. obj = image
  224. } else {
  225. // No output means the object does not exist
  226. // (easier to script since stdout and stderr are not differentiated atm)
  227. return nil
  228. }
  229. data, err := json.Marshal(obj)
  230. if err != nil {
  231. return err
  232. }
  233. indented := new(bytes.Buffer)
  234. if err = json.Indent(indented, data, "", " "); err != nil {
  235. return err
  236. }
  237. if _, err := io.Copy(stdout, indented); err != nil {
  238. return err
  239. }
  240. stdout.Write([]byte{'\n'})
  241. return nil
  242. }
  243. func (srv *Server) CmdPort(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  244. cmd := rcli.Subcmd(stdout, "port", "[OPTIONS] CONTAINER PRIVATE_PORT", "Lookup the public-facing port which is NAT-ed to PRIVATE_PORT")
  245. if err := cmd.Parse(args); err != nil {
  246. return nil
  247. }
  248. if cmd.NArg() != 2 {
  249. cmd.Usage()
  250. return nil
  251. }
  252. name := cmd.Arg(0)
  253. privatePort := cmd.Arg(1)
  254. if container := srv.runtime.Get(name); container == nil {
  255. return errors.New("No such container: " + name)
  256. } else {
  257. if frontend, exists := container.NetworkSettings.PortMapping[privatePort]; !exists {
  258. return fmt.Errorf("No private port '%s' allocated on %s", privatePort, name)
  259. } else {
  260. fmt.Fprintln(stdout, frontend)
  261. }
  262. }
  263. return nil
  264. }
  265. // 'docker rmi NAME' removes all images with the name NAME
  266. func (srv *Server) CmdRmi(stdin io.ReadCloser, stdout io.Writer, args ...string) (err error) {
  267. cmd := rcli.Subcmd(stdout, "rmimage", "[OPTIONS] IMAGE", "Remove an image")
  268. if err := cmd.Parse(args); err != nil {
  269. return nil
  270. }
  271. if cmd.NArg() < 1 {
  272. cmd.Usage()
  273. return nil
  274. }
  275. for _, name := range cmd.Args() {
  276. if err := srv.runtime.graph.Delete(name); err != nil {
  277. return err
  278. }
  279. }
  280. return nil
  281. }
  282. func (srv *Server) CmdHistory(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  283. cmd := rcli.Subcmd(stdout, "history", "[OPTIONS] IMAGE", "Show the history of an image")
  284. if err := cmd.Parse(args); err != nil {
  285. return nil
  286. }
  287. if cmd.NArg() != 1 {
  288. cmd.Usage()
  289. return nil
  290. }
  291. image, err := srv.runtime.repositories.LookupImage(cmd.Arg(0))
  292. if err != nil {
  293. return err
  294. }
  295. w := tabwriter.NewWriter(stdout, 20, 1, 3, ' ', 0)
  296. defer w.Flush()
  297. fmt.Fprintf(w, "ID\tCREATED\tCREATED BY\n")
  298. return image.WalkHistory(func(img *Image) error {
  299. fmt.Fprintf(w, "%s\t%s\t%s\n",
  300. srv.runtime.repositories.ImageName(img.Id),
  301. HumanDuration(time.Now().Sub(img.Created))+" ago",
  302. strings.Join(img.ContainerConfig.Cmd, " "),
  303. )
  304. return nil
  305. })
  306. }
  307. func (srv *Server) CmdRm(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  308. cmd := rcli.Subcmd(stdout, "rm", "[OPTIONS] CONTAINER", "Remove a container")
  309. if err := cmd.Parse(args); err != nil {
  310. return nil
  311. }
  312. for _, name := range cmd.Args() {
  313. container := srv.runtime.Get(name)
  314. if container == nil {
  315. return errors.New("No such container: " + name)
  316. }
  317. if err := srv.runtime.Destroy(container); err != nil {
  318. fmt.Fprintln(stdout, "Error destroying container "+name+": "+err.Error())
  319. }
  320. }
  321. return nil
  322. }
  323. // 'docker kill NAME' kills a running container
  324. func (srv *Server) CmdKill(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  325. cmd := rcli.Subcmd(stdout, "kill", "[OPTIONS] CONTAINER [CONTAINER...]", "Kill a running container")
  326. if err := cmd.Parse(args); err != nil {
  327. return nil
  328. }
  329. for _, name := range cmd.Args() {
  330. container := srv.runtime.Get(name)
  331. if container == nil {
  332. return errors.New("No such container: " + name)
  333. }
  334. if err := container.Kill(); err != nil {
  335. fmt.Fprintln(stdout, "Error killing container "+name+": "+err.Error())
  336. }
  337. }
  338. return nil
  339. }
  340. func (srv *Server) CmdImport(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  341. cmd := rcli.Subcmd(stdout, "import", "[OPTIONS] URL|- [REPOSITORY [TAG]]", "Create a new filesystem image from the contents of a tarball")
  342. var archive io.Reader
  343. var resp *http.Response
  344. if err := cmd.Parse(args); err != nil {
  345. return nil
  346. }
  347. src := cmd.Arg(0)
  348. if src == "" {
  349. return errors.New("Not enough arguments")
  350. } else if src == "-" {
  351. archive = stdin
  352. } else {
  353. u, err := url.Parse(src)
  354. if err != nil {
  355. return err
  356. }
  357. if u.Scheme == "" {
  358. u.Scheme = "http"
  359. u.Host = src
  360. u.Path = ""
  361. }
  362. fmt.Fprintf(stdout, "Downloading from %s\n", u.String())
  363. // Download with curl (pretty progress bar)
  364. // If curl is not available, fallback to http.Get()
  365. resp, err = Download(u.String(), stdout)
  366. if err != nil {
  367. return err
  368. }
  369. archive = ProgressReader(resp.Body, int(resp.ContentLength), stdout)
  370. }
  371. img, err := srv.runtime.graph.Create(archive, nil, "Imported from "+src)
  372. if err != nil {
  373. return err
  374. }
  375. // Optionally register the image at REPO/TAG
  376. if repository := cmd.Arg(1); repository != "" {
  377. tag := cmd.Arg(2) // Repository will handle an empty tag properly
  378. if err := srv.runtime.repositories.Set(repository, tag, img.Id, true); err != nil {
  379. return err
  380. }
  381. }
  382. fmt.Fprintln(stdout, img.Id)
  383. return nil
  384. }
  385. func (srv *Server) CmdPush(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  386. cmd := rcli.Subcmd(stdout, "push", "NAME", "Push an image or a repository to the registry")
  387. if err := cmd.Parse(args); err != nil {
  388. return nil
  389. }
  390. local := cmd.Arg(0)
  391. if local == "" {
  392. cmd.Usage()
  393. return nil
  394. }
  395. // If the login failed, abort
  396. if srv.runtime.authConfig == nil || srv.runtime.authConfig.Username == "" {
  397. if err := srv.CmdLogin(stdin, stdout, args...); err != nil {
  398. return err
  399. }
  400. if srv.runtime.authConfig == nil || srv.runtime.authConfig.Username == "" {
  401. return fmt.Errorf("Please login prior to push. ('docker login')")
  402. }
  403. }
  404. var remote string
  405. tmp := strings.SplitN(local, "/", 2)
  406. if len(tmp) == 1 {
  407. return fmt.Errorf(
  408. "Impossible to push a \"root\" repository. Please rename your repository in <user>/<repo> (ex: %s/%s)",
  409. srv.runtime.authConfig.Username, local)
  410. } else {
  411. remote = local
  412. }
  413. Debugf("Pushing [%s] to [%s]\n", local, remote)
  414. // Try to get the image
  415. // FIXME: Handle lookup
  416. // FIXME: Also push the tags in case of ./docker push myrepo:mytag
  417. // img, err := srv.runtime.LookupImage(cmd.Arg(0))
  418. img, err := srv.runtime.graph.Get(local)
  419. if err != nil {
  420. Debugf("The push refers to a repository [%s] (len: %d)\n", local, len(srv.runtime.repositories.Repositories[local]))
  421. // If it fails, try to get the repository
  422. if localRepo, exists := srv.runtime.repositories.Repositories[local]; exists {
  423. if err := srv.runtime.graph.PushRepository(stdout, remote, localRepo, srv.runtime.authConfig); err != nil {
  424. return err
  425. }
  426. return nil
  427. } else {
  428. return err
  429. }
  430. return nil
  431. }
  432. err = srv.runtime.graph.PushImage(stdout, img, srv.runtime.authConfig)
  433. if err != nil {
  434. return err
  435. }
  436. return nil
  437. }
  438. func (srv *Server) CmdPull(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  439. cmd := rcli.Subcmd(stdout, "pull", "NAME", "Pull an image or a repository from the registry")
  440. if err := cmd.Parse(args); err != nil {
  441. return nil
  442. }
  443. remote := cmd.Arg(0)
  444. if remote == "" {
  445. cmd.Usage()
  446. return nil
  447. }
  448. if srv.runtime.graph.LookupRemoteImage(remote, srv.runtime.authConfig) {
  449. if err := srv.runtime.graph.PullImage(stdout, remote, srv.runtime.authConfig); err != nil {
  450. return err
  451. }
  452. return nil
  453. }
  454. // FIXME: Allow pull repo:tag
  455. if err := srv.runtime.graph.PullRepository(stdout, remote, "", srv.runtime.repositories, srv.runtime.authConfig); err != nil {
  456. return err
  457. }
  458. return nil
  459. }
  460. func (srv *Server) CmdImages(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  461. cmd := rcli.Subcmd(stdout, "images", "[OPTIONS] [NAME]", "List images")
  462. //limit := cmd.Int("l", 0, "Only show the N most recent versions of each image")
  463. quiet := cmd.Bool("q", false, "only show numeric IDs")
  464. flAll := cmd.Bool("a", false, "show all images")
  465. if err := cmd.Parse(args); err != nil {
  466. return nil
  467. }
  468. if cmd.NArg() > 1 {
  469. cmd.Usage()
  470. return nil
  471. }
  472. var nameFilter string
  473. if cmd.NArg() == 1 {
  474. nameFilter = cmd.Arg(0)
  475. }
  476. w := tabwriter.NewWriter(stdout, 20, 1, 3, ' ', 0)
  477. if !*quiet {
  478. fmt.Fprintf(w, "REPOSITORY\tTAG\tID\tCREATED\tPARENT\n")
  479. }
  480. var allImages map[string]*Image
  481. var err error
  482. if *flAll {
  483. allImages, err = srv.runtime.graph.Map()
  484. } else {
  485. allImages, err = srv.runtime.graph.Heads()
  486. }
  487. if err != nil {
  488. return err
  489. }
  490. for name, repository := range srv.runtime.repositories.Repositories {
  491. if nameFilter != "" && name != nameFilter {
  492. continue
  493. }
  494. for tag, id := range repository {
  495. image, err := srv.runtime.graph.Get(id)
  496. if err != nil {
  497. log.Printf("Warning: couldn't load %s from %s/%s: %s", id, name, tag, err)
  498. continue
  499. }
  500. delete(allImages, id)
  501. if !*quiet {
  502. for idx, field := range []string{
  503. /* REPOSITORY */ name,
  504. /* TAG */ tag,
  505. /* ID */ id,
  506. /* CREATED */ HumanDuration(time.Now().Sub(image.Created)) + " ago",
  507. /* PARENT */ srv.runtime.repositories.ImageName(image.Parent),
  508. } {
  509. if idx == 0 {
  510. w.Write([]byte(field))
  511. } else {
  512. w.Write([]byte("\t" + field))
  513. }
  514. }
  515. w.Write([]byte{'\n'})
  516. } else {
  517. stdout.Write([]byte(image.Id + "\n"))
  518. }
  519. }
  520. }
  521. // Display images which aren't part of a
  522. if nameFilter == "" {
  523. for id, image := range allImages {
  524. if !*quiet {
  525. for idx, field := range []string{
  526. /* REPOSITORY */ "<none>",
  527. /* TAG */ "<none>",
  528. /* ID */ id,
  529. /* CREATED */ HumanDuration(time.Now().Sub(image.Created)) + " ago",
  530. /* PARENT */ srv.runtime.repositories.ImageName(image.Parent),
  531. } {
  532. if idx == 0 {
  533. w.Write([]byte(field))
  534. } else {
  535. w.Write([]byte("\t" + field))
  536. }
  537. }
  538. w.Write([]byte{'\n'})
  539. } else {
  540. stdout.Write([]byte(image.Id + "\n"))
  541. }
  542. }
  543. }
  544. if !*quiet {
  545. w.Flush()
  546. }
  547. return nil
  548. }
  549. func (srv *Server) CmdPs(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  550. cmd := rcli.Subcmd(stdout,
  551. "ps", "[OPTIONS]", "List containers")
  552. quiet := cmd.Bool("q", false, "Only display numeric IDs")
  553. flAll := cmd.Bool("a", false, "Show all containers. Only running containers are shown by default.")
  554. flFull := cmd.Bool("notrunc", false, "Don't truncate output")
  555. if err := cmd.Parse(args); err != nil {
  556. return nil
  557. }
  558. w := tabwriter.NewWriter(stdout, 12, 1, 3, ' ', 0)
  559. if !*quiet {
  560. fmt.Fprintf(w, "ID\tIMAGE\tCOMMAND\tCREATED\tSTATUS\tCOMMENT\n")
  561. }
  562. for _, container := range srv.runtime.List() {
  563. if !container.State.Running && !*flAll {
  564. continue
  565. }
  566. if !*quiet {
  567. command := fmt.Sprintf("%s %s", container.Path, strings.Join(container.Args, " "))
  568. if !*flFull {
  569. command = Trunc(command, 20)
  570. }
  571. for idx, field := range []string{
  572. /* ID */ container.Id,
  573. /* IMAGE */ srv.runtime.repositories.ImageName(container.Image),
  574. /* COMMAND */ command,
  575. /* CREATED */ HumanDuration(time.Now().Sub(container.Created)) + " ago",
  576. /* STATUS */ container.State.String(),
  577. /* COMMENT */ "",
  578. } {
  579. if idx == 0 {
  580. w.Write([]byte(field))
  581. } else {
  582. w.Write([]byte("\t" + field))
  583. }
  584. }
  585. w.Write([]byte{'\n'})
  586. } else {
  587. stdout.Write([]byte(container.Id + "\n"))
  588. }
  589. }
  590. if !*quiet {
  591. w.Flush()
  592. }
  593. return nil
  594. }
  595. func (srv *Server) CmdCommit(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  596. cmd := rcli.Subcmd(stdout,
  597. "commit", "[OPTIONS] CONTAINER [REPOSITORY [TAG]]",
  598. "Create a new image from a container's changes")
  599. flComment := cmd.String("m", "", "Commit message")
  600. if err := cmd.Parse(args); err != nil {
  601. return nil
  602. }
  603. containerName, repository, tag := cmd.Arg(0), cmd.Arg(1), cmd.Arg(2)
  604. if containerName == "" {
  605. cmd.Usage()
  606. return nil
  607. }
  608. img, err := srv.runtime.Commit(containerName, repository, tag, *flComment)
  609. if err != nil {
  610. return err
  611. }
  612. fmt.Fprintln(stdout, img.Id)
  613. return nil
  614. }
  615. func (srv *Server) CmdExport(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  616. cmd := rcli.Subcmd(stdout,
  617. "export", "CONTAINER",
  618. "Export the contents of a filesystem as a tar archive")
  619. if err := cmd.Parse(args); err != nil {
  620. return nil
  621. }
  622. name := cmd.Arg(0)
  623. if container := srv.runtime.Get(name); container != nil {
  624. data, err := container.Export()
  625. if err != nil {
  626. return err
  627. }
  628. // Stream the entire contents of the container (basically a volatile snapshot)
  629. if _, err := io.Copy(stdout, data); err != nil {
  630. return err
  631. }
  632. return nil
  633. }
  634. return errors.New("No such container: " + name)
  635. }
  636. func (srv *Server) CmdDiff(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  637. cmd := rcli.Subcmd(stdout,
  638. "diff", "CONTAINER [OPTIONS]",
  639. "Inspect changes on a container's filesystem")
  640. if err := cmd.Parse(args); err != nil {
  641. return nil
  642. }
  643. if cmd.NArg() < 1 {
  644. return errors.New("Not enough arguments")
  645. }
  646. if container := srv.runtime.Get(cmd.Arg(0)); container == nil {
  647. return errors.New("No such container")
  648. } else {
  649. changes, err := container.Changes()
  650. if err != nil {
  651. return err
  652. }
  653. for _, change := range changes {
  654. fmt.Fprintln(stdout, change.String())
  655. }
  656. }
  657. return nil
  658. }
  659. func (srv *Server) CmdLogs(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  660. cmd := rcli.Subcmd(stdout, "logs", "[OPTIONS] CONTAINER", "Fetch the logs of a container")
  661. if err := cmd.Parse(args); err != nil {
  662. return nil
  663. }
  664. if cmd.NArg() != 1 {
  665. cmd.Usage()
  666. return nil
  667. }
  668. name := cmd.Arg(0)
  669. if container := srv.runtime.Get(name); container != nil {
  670. logStdout, err := container.ReadLog("stdout")
  671. if err != nil {
  672. return err
  673. }
  674. logStderr, err := container.ReadLog("stderr")
  675. if err != nil {
  676. return err
  677. }
  678. // FIXME: Interpolate stdout and stderr instead of concatenating them
  679. // FIXME: Differentiate stdout and stderr in the remote protocol
  680. if _, err := io.Copy(stdout, logStdout); err != nil {
  681. return err
  682. }
  683. if _, err := io.Copy(stdout, logStderr); err != nil {
  684. return err
  685. }
  686. return nil
  687. }
  688. return errors.New("No such container: " + cmd.Arg(0))
  689. }
  690. func (srv *Server) CmdAttach(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  691. cmd := rcli.Subcmd(stdout, "attach", "[OPTIONS]", "Attach to a running container")
  692. flStdin := cmd.Bool("i", false, "Attach to stdin")
  693. flStdout := cmd.Bool("o", true, "Attach to stdout")
  694. flStderr := cmd.Bool("e", true, "Attach to stderr")
  695. if err := cmd.Parse(args); err != nil {
  696. return nil
  697. }
  698. if cmd.NArg() != 1 {
  699. cmd.Usage()
  700. return nil
  701. }
  702. name := cmd.Arg(0)
  703. container := srv.runtime.Get(name)
  704. if container == nil {
  705. return errors.New("No such container: " + name)
  706. }
  707. var wg sync.WaitGroup
  708. if *flStdin {
  709. cStdin, err := container.StdinPipe()
  710. if err != nil {
  711. return err
  712. }
  713. wg.Add(1)
  714. go func() { io.Copy(cStdin, stdin); wg.Add(-1) }()
  715. }
  716. if *flStdout {
  717. cStdout, err := container.StdoutPipe()
  718. if err != nil {
  719. return err
  720. }
  721. wg.Add(1)
  722. go func() { io.Copy(stdout, cStdout); wg.Add(-1) }()
  723. }
  724. if *flStderr {
  725. cStderr, err := container.StderrPipe()
  726. if err != nil {
  727. return err
  728. }
  729. wg.Add(1)
  730. go func() { io.Copy(stdout, cStderr); wg.Add(-1) }()
  731. }
  732. wg.Wait()
  733. return nil
  734. }
  735. // Ports type - Used to parse multiple -p flags
  736. type ports []int
  737. func (p *ports) String() string {
  738. return fmt.Sprint(*p)
  739. }
  740. func (p *ports) Set(value string) error {
  741. port, err := strconv.Atoi(value)
  742. if err != nil {
  743. return fmt.Errorf("Invalid port: %v", value)
  744. }
  745. *p = append(*p, port)
  746. return nil
  747. }
  748. // ListOpts type
  749. type ListOpts []string
  750. func (opts *ListOpts) String() string {
  751. return fmt.Sprint(*opts)
  752. }
  753. func (opts *ListOpts) Set(value string) error {
  754. *opts = append(*opts, value)
  755. return nil
  756. }
  757. func (srv *Server) CmdTag(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  758. cmd := rcli.Subcmd(stdout, "tag", "[OPTIONS] IMAGE REPOSITORY [TAG]", "Tag an image into a repository")
  759. force := cmd.Bool("f", false, "Force")
  760. if err := cmd.Parse(args); err != nil {
  761. return nil
  762. }
  763. if cmd.NArg() < 2 {
  764. cmd.Usage()
  765. return nil
  766. }
  767. return srv.runtime.repositories.Set(cmd.Arg(1), cmd.Arg(2), cmd.Arg(0), *force)
  768. }
  769. func (srv *Server) CmdRun(stdin io.ReadCloser, stdout io.Writer, args ...string) error {
  770. config, err := ParseRun(args, stdout)
  771. if err != nil {
  772. return err
  773. }
  774. if config.Image == "" {
  775. fmt.Fprintln(stdout, "Error: Image not specified")
  776. return fmt.Errorf("Image not specified")
  777. }
  778. if len(config.Cmd) == 0 {
  779. fmt.Fprintln(stdout, "Error: Command not specified")
  780. return fmt.Errorf("Command not specified")
  781. }
  782. // Create new container
  783. container, err := srv.runtime.Create(config)
  784. if err != nil {
  785. // If container not found, try to pull it
  786. if srv.runtime.graph.IsNotExist(err) {
  787. fmt.Fprintf(stdout, "Image %s not found, trying to pull it from registry.\n", config.Image)
  788. if err = srv.CmdPull(stdin, stdout, config.Image); err != nil {
  789. return err
  790. }
  791. if container, err = srv.runtime.Create(config); err != nil {
  792. return err
  793. }
  794. } else {
  795. return err
  796. }
  797. }
  798. if config.OpenStdin {
  799. cmdStdin, err := container.StdinPipe()
  800. if err != nil {
  801. return err
  802. }
  803. if !config.Detach {
  804. Go(func() error {
  805. _, err := io.Copy(cmdStdin, stdin)
  806. cmdStdin.Close()
  807. return err
  808. })
  809. }
  810. }
  811. // Run the container
  812. if !config.Detach {
  813. cmdStderr, err := container.StderrPipe()
  814. if err != nil {
  815. return err
  816. }
  817. cmdStdout, err := container.StdoutPipe()
  818. if err != nil {
  819. return err
  820. }
  821. if err := container.Start(); err != nil {
  822. return err
  823. }
  824. sendingStdout := Go(func() error {
  825. _, err := io.Copy(stdout, cmdStdout)
  826. return err
  827. })
  828. sendingStderr := Go(func() error {
  829. _, err := io.Copy(stdout, cmdStderr)
  830. return err
  831. })
  832. errSendingStdout := <-sendingStdout
  833. errSendingStderr := <-sendingStderr
  834. if errSendingStdout != nil {
  835. return errSendingStdout
  836. }
  837. if errSendingStderr != nil {
  838. return errSendingStderr
  839. }
  840. container.Wait()
  841. } else {
  842. if err := container.Start(); err != nil {
  843. return err
  844. }
  845. fmt.Fprintln(stdout, container.Id)
  846. }
  847. return nil
  848. }
  849. func NewServer() (*Server, error) {
  850. if runtime.GOARCH != "amd64" {
  851. log.Fatalf("The docker runtime currently only supports amd64 (not %s). This will change in the future. Aborting.", runtime.GOARCH)
  852. }
  853. runtime, err := NewRuntime()
  854. if err != nil {
  855. return nil, err
  856. }
  857. srv := &Server{
  858. runtime: runtime,
  859. }
  860. return srv, nil
  861. }
  862. type Server struct {
  863. runtime *Runtime
  864. }