mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
package graph
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
log "github.com/Sirupsen/logrus"
|
|
"github.com/docker/docker/engine"
|
|
"github.com/docker/docker/pkg/archive"
|
|
"github.com/docker/docker/runconfig"
|
|
"github.com/docker/docker/utils"
|
|
)
|
|
|
|
func (s *TagStore) CmdImport(job *engine.Job) engine.Status {
|
|
if n := len(job.Args); n != 2 && n != 3 {
|
|
return job.Errorf("Usage: %s SRC REPO [TAG]", job.Name)
|
|
}
|
|
var (
|
|
src = job.Args[0]
|
|
repo = job.Args[1]
|
|
tag string
|
|
sf = utils.NewStreamFormatter(job.GetenvBool("json"))
|
|
archive archive.ArchiveReader
|
|
resp *http.Response
|
|
stdoutBuffer = bytes.NewBuffer(nil)
|
|
newConfig runconfig.Config
|
|
)
|
|
if len(job.Args) > 2 {
|
|
tag = job.Args[2]
|
|
}
|
|
|
|
if src == "-" {
|
|
archive = job.Stdin
|
|
} else {
|
|
u, err := url.Parse(src)
|
|
if err != nil {
|
|
return job.Error(err)
|
|
}
|
|
if u.Scheme == "" {
|
|
u.Scheme = "http"
|
|
u.Host = src
|
|
u.Path = ""
|
|
}
|
|
job.Stdout.Write(sf.FormatStatus("", "Downloading from %s", u))
|
|
resp, err = utils.Download(u.String())
|
|
if err != nil {
|
|
return job.Error(err)
|
|
}
|
|
progressReader := utils.ProgressReader(resp.Body, int(resp.ContentLength), job.Stdout, sf, true, "", "Importing")
|
|
defer progressReader.Close()
|
|
archive = progressReader
|
|
}
|
|
|
|
buildConfigJob := job.Eng.Job("build_config")
|
|
buildConfigJob.Stdout.Add(stdoutBuffer)
|
|
buildConfigJob.Setenv("changes", job.Getenv("changes"))
|
|
// FIXME this should be remove when we remove deprecated config param
|
|
buildConfigJob.Setenv("config", job.Getenv("config"))
|
|
|
|
if err := buildConfigJob.Run(); err != nil {
|
|
return job.Error(err)
|
|
}
|
|
if err := json.NewDecoder(stdoutBuffer).Decode(&newConfig); err != nil {
|
|
return job.Error(err)
|
|
}
|
|
|
|
img, err := s.graph.Create(archive, "", "", "Imported from "+src, "", nil, &newConfig)
|
|
if err != nil {
|
|
return job.Error(err)
|
|
}
|
|
// Optionally register the image at REPO/TAG
|
|
if repo != "" {
|
|
if err := s.Set(repo, tag, img.ID, true); err != nil {
|
|
return job.Error(err)
|
|
}
|
|
}
|
|
job.Stdout.Write(sf.FormatStatus("", img.ID))
|
|
logID := img.ID
|
|
if tag != "" {
|
|
logID += ":" + tag
|
|
}
|
|
if err = job.Eng.Job("log", "import", logID, "").Run(); err != nil {
|
|
log.Errorf("Error logging event 'import' for %s: %s", logID, err)
|
|
}
|
|
return engine.StatusOK
|
|
}
|