|
@@ -50,7 +50,8 @@ func (srv *Server) ContainerExport(name string, out io.Writer) error {
|
|
}
|
|
}
|
|
|
|
|
|
func (srv *Server) ImagesSearch(term string) ([]ApiSearch, error) {
|
|
func (srv *Server) ImagesSearch(term string) ([]ApiSearch, error) {
|
|
- results, err := srv.registry.SearchRepositories(term)
|
|
|
|
|
|
+
|
|
|
|
+ results, err := registry.NewRegistry(srv.runtime.root).SearchRepositories(term)
|
|
if err != nil {
|
|
if err != nil {
|
|
return nil, err
|
|
return nil, err
|
|
}
|
|
}
|
|
@@ -68,7 +69,7 @@ func (srv *Server) ImagesSearch(term string) ([]ApiSearch, error) {
|
|
return outs, nil
|
|
return outs, nil
|
|
}
|
|
}
|
|
|
|
|
|
-func (srv *Server) ImageInsert(name, url, path string, out io.Writer) (string, error) {
|
|
|
|
|
|
+func (srv *Server) ImageInsert(name, url, path string, out io.Writer, sf *utils.StreamFormatter) (string, error) {
|
|
out = utils.NewWriteFlusher(out)
|
|
out = utils.NewWriteFlusher(out)
|
|
img, err := srv.runtime.repositories.LookupImage(name)
|
|
img, err := srv.runtime.repositories.LookupImage(name)
|
|
if err != nil {
|
|
if err != nil {
|
|
@@ -92,7 +93,7 @@ func (srv *Server) ImageInsert(name, url, path string, out io.Writer) (string, e
|
|
return "", err
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
|
|
- if err := c.Inject(utils.ProgressReader(file.Body, int(file.ContentLength), out, "Downloading %v/%v (%v)\r", false), path); err != nil {
|
|
|
|
|
|
+ if err := c.Inject(utils.ProgressReader(file.Body, int(file.ContentLength), out, sf.FormatProgress("Downloading", "%v/%v (%v)"), sf), path); err != nil {
|
|
return "", err
|
|
return "", err
|
|
}
|
|
}
|
|
// FIXME: Handle custom repo, tag comment, author
|
|
// FIXME: Handle custom repo, tag comment, author
|
|
@@ -100,7 +101,7 @@ func (srv *Server) ImageInsert(name, url, path string, out io.Writer) (string, e
|
|
if err != nil {
|
|
if err != nil {
|
|
return "", err
|
|
return "", err
|
|
}
|
|
}
|
|
- fmt.Fprintf(out, "%s\n", img.Id)
|
|
|
|
|
|
+ out.Write(sf.FormatStatus(img.Id))
|
|
return img.ShortId(), nil
|
|
return img.ShortId(), nil
|
|
}
|
|
}
|
|
|
|
|
|
@@ -292,8 +293,8 @@ func (srv *Server) ContainerTag(name, repo, tag string, force bool) error {
|
|
return nil
|
|
return nil
|
|
}
|
|
}
|
|
|
|
|
|
-func (srv *Server) pullImage(out io.Writer, imgId, registry string, token []string, json bool) error {
|
|
|
|
- history, err := srv.registry.GetRemoteHistory(imgId, registry, token)
|
|
|
|
|
|
+func (srv *Server) pullImage(r *registry.Registry, out io.Writer, imgId, endpoint string, token []string, sf *utils.StreamFormatter) error {
|
|
|
|
+ history, err := r.GetRemoteHistory(imgId, endpoint, token)
|
|
if err != nil {
|
|
if err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
@@ -302,8 +303,8 @@ func (srv *Server) pullImage(out io.Writer, imgId, registry string, token []stri
|
|
// FIXME: Launch the getRemoteImage() in goroutines
|
|
// FIXME: Launch the getRemoteImage() in goroutines
|
|
for _, id := range history {
|
|
for _, id := range history {
|
|
if !srv.runtime.graph.Exists(id) {
|
|
if !srv.runtime.graph.Exists(id) {
|
|
- fmt.Fprintf(out, utils.FormatStatus("Pulling %s metadata", json), id)
|
|
|
|
- imgJson, err := srv.registry.GetRemoteImageJson(id, registry, token)
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("Pulling %s metadata", id))
|
|
|
|
+ imgJson, err := r.GetRemoteImageJson(id, endpoint, token)
|
|
if err != nil {
|
|
if err != nil {
|
|
// FIXME: Keep goging in case of error?
|
|
// FIXME: Keep goging in case of error?
|
|
return err
|
|
return err
|
|
@@ -314,12 +315,12 @@ func (srv *Server) pullImage(out io.Writer, imgId, registry string, token []stri
|
|
}
|
|
}
|
|
|
|
|
|
// Get the layer
|
|
// Get the layer
|
|
- fmt.Fprintf(out, utils.FormatStatus("Pulling %s fs layer", json), id)
|
|
|
|
- layer, contentLength, err := srv.registry.GetRemoteImageLayer(img.Id, registry, token)
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("Pulling %s fs layer", id))
|
|
|
|
+ layer, contentLength, err := r.GetRemoteImageLayer(img.Id, endpoint, token)
|
|
if err != nil {
|
|
if err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
- if err := srv.runtime.graph.Register(utils.ProgressReader(layer, contentLength, out, utils.FormatProgress("%v/%v (%v)", json), json), false, img); err != nil {
|
|
|
|
|
|
+ if err := srv.runtime.graph.Register(utils.ProgressReader(layer, contentLength, out, sf.FormatProgress("Downloading", "%v/%v (%v)"), sf), false, img); err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
@@ -327,9 +328,9 @@ func (srv *Server) pullImage(out io.Writer, imgId, registry string, token []stri
|
|
return nil
|
|
return nil
|
|
}
|
|
}
|
|
|
|
|
|
-func (srv *Server) pullRepository(out io.Writer, remote, askedTag string, json bool) error {
|
|
|
|
- fmt.Fprintf(out, utils.FormatStatus("Pulling repository %s from %s", json), remote, auth.IndexServerAddress())
|
|
|
|
- repoData, err := srv.registry.GetRepositoryData(remote)
|
|
|
|
|
|
+func (srv *Server) pullRepository(r *registry.Registry, out io.Writer, remote, askedTag string, sf *utils.StreamFormatter) error {
|
|
|
|
+ out.Write(sf.FormatStatus("Pulling repository %s from %s", remote, auth.IndexServerAddress()))
|
|
|
|
+ repoData, err := r.GetRepositoryData(remote)
|
|
if err != nil {
|
|
if err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
@@ -341,7 +342,7 @@ func (srv *Server) pullRepository(out io.Writer, remote, askedTag string, json b
|
|
}
|
|
}
|
|
|
|
|
|
utils.Debugf("Retrieving the tag list")
|
|
utils.Debugf("Retrieving the tag list")
|
|
- tagsList, err := srv.registry.GetRemoteTags(repoData.Endpoints, remote, repoData.Tokens)
|
|
|
|
|
|
+ tagsList, err := r.GetRemoteTags(repoData.Endpoints, remote, repoData.Tokens)
|
|
if err != nil {
|
|
if err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
@@ -365,11 +366,11 @@ func (srv *Server) pullRepository(out io.Writer, remote, askedTag string, json b
|
|
utils.Debugf("(%s) does not match %s (id: %s), skipping", img.Tag, askedTag, img.Id)
|
|
utils.Debugf("(%s) does not match %s (id: %s), skipping", img.Tag, askedTag, img.Id)
|
|
continue
|
|
continue
|
|
}
|
|
}
|
|
- fmt.Fprintf(out, utils.FormatStatus("Pulling image %s (%s) from %s", json), img.Id, img.Tag, remote)
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("Pulling image %s (%s) from %s", img.Id, img.Tag, remote))
|
|
success := false
|
|
success := false
|
|
for _, ep := range repoData.Endpoints {
|
|
for _, ep := range repoData.Endpoints {
|
|
- if err := srv.pullImage(out, img.Id, "https://"+ep+"/v1", repoData.Tokens, json); err != nil {
|
|
|
|
- fmt.Fprintf(out, utils.FormatStatus("Error while retrieving image for tag: %s (%s); checking next endpoint\n", json), askedTag, err)
|
|
|
|
|
|
+ if err := srv.pullImage(r, out, img.Id, "https://"+ep+"/v1", repoData.Tokens, sf); err != nil {
|
|
|
|
+ out.Write(sf.FormatStatus("Error while retrieving image for tag: %s (%s); checking next endpoint", askedTag, err))
|
|
continue
|
|
continue
|
|
}
|
|
}
|
|
success = true
|
|
success = true
|
|
@@ -394,16 +395,17 @@ func (srv *Server) pullRepository(out io.Writer, remote, askedTag string, json b
|
|
return nil
|
|
return nil
|
|
}
|
|
}
|
|
|
|
|
|
-func (srv *Server) ImagePull(name, tag, registry string, out io.Writer, json bool) error {
|
|
|
|
|
|
+func (srv *Server) ImagePull(name, tag, endpoint string, out io.Writer, sf *utils.StreamFormatter) error {
|
|
|
|
+ r := registry.NewRegistry(srv.runtime.root)
|
|
out = utils.NewWriteFlusher(out)
|
|
out = utils.NewWriteFlusher(out)
|
|
- if registry != "" {
|
|
|
|
- if err := srv.pullImage(out, name, registry, nil, json); err != nil {
|
|
|
|
|
|
+ if endpoint != "" {
|
|
|
|
+ if err := srv.pullImage(r, out, name, endpoint, nil, sf); err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
return nil
|
|
}
|
|
}
|
|
|
|
|
|
- if err := srv.pullRepository(out, name, tag, json); err != nil {
|
|
|
|
|
|
+ if err := srv.pullRepository(r, out, name, tag, sf); err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
|
|
|
|
@@ -476,53 +478,52 @@ func (srv *Server) getImageList(localRepo map[string]string) ([]*registry.ImgDat
|
|
return imgList, nil
|
|
return imgList, nil
|
|
}
|
|
}
|
|
|
|
|
|
-func (srv *Server) pushRepository(out io.Writer, name string, localRepo map[string]string) error {
|
|
|
|
|
|
+func (srv *Server) pushRepository(r *registry.Registry, out io.Writer, name string, localRepo map[string]string, sf *utils.StreamFormatter) error {
|
|
out = utils.NewWriteFlusher(out)
|
|
out = utils.NewWriteFlusher(out)
|
|
- fmt.Fprintf(out, "Processing checksums\n")
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("Processing checksums"))
|
|
imgList, err := srv.getImageList(localRepo)
|
|
imgList, err := srv.getImageList(localRepo)
|
|
if err != nil {
|
|
if err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
- fmt.Fprintf(out, "Sending image list\n")
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("Sending image list"))
|
|
|
|
|
|
- repoData, err := srv.registry.PushImageJsonIndex(name, imgList, false)
|
|
|
|
|
|
+ repoData, err := r.PushImageJsonIndex(name, imgList, false)
|
|
if err != nil {
|
|
if err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
|
|
|
|
- // FIXME: Send only needed images
|
|
|
|
for _, ep := range repoData.Endpoints {
|
|
for _, ep := range repoData.Endpoints {
|
|
- fmt.Fprintf(out, "Pushing repository %s to %s (%d tags)\r\n", name, ep, len(localRepo))
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("Pushing repository %s to %s (%d tags)", name, ep, len(localRepo)))
|
|
// For each image within the repo, push them
|
|
// For each image within the repo, push them
|
|
for _, elem := range imgList {
|
|
for _, elem := range imgList {
|
|
if _, exists := repoData.ImgList[elem.Id]; exists {
|
|
if _, exists := repoData.ImgList[elem.Id]; exists {
|
|
- fmt.Fprintf(out, "Image %s already on registry, skipping\n", name)
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("Image %s already on registry, skipping", name))
|
|
continue
|
|
continue
|
|
}
|
|
}
|
|
- if err := srv.pushImage(out, name, elem.Id, ep, repoData.Tokens); err != nil {
|
|
|
|
|
|
+ if err := srv.pushImage(r, out, name, elem.Id, ep, repoData.Tokens, sf); err != nil {
|
|
// FIXME: Continue on error?
|
|
// FIXME: Continue on error?
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
- fmt.Fprintf(out, "Pushing tags for rev [%s] on {%s}\n", elem.Id, ep+"/users/"+name+"/"+elem.Tag)
|
|
|
|
- if err := srv.registry.PushRegistryTag(name, elem.Id, elem.Tag, ep, repoData.Tokens); err != nil {
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("Pushing tags for rev [%s] on {%s}", elem.Id, ep+"/users/"+name+"/"+elem.Tag))
|
|
|
|
+ if err := r.PushRegistryTag(name, elem.Id, elem.Tag, ep, repoData.Tokens); err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
- if _, err := srv.registry.PushImageJsonIndex(name, imgList, true); err != nil {
|
|
|
|
|
|
+ if _, err := r.PushImageJsonIndex(name, imgList, true); err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
return nil
|
|
}
|
|
}
|
|
|
|
|
|
-func (srv *Server) pushImage(out io.Writer, remote, imgId, ep string, token []string) error {
|
|
|
|
|
|
+func (srv *Server) pushImage(r *registry.Registry, out io.Writer, remote, imgId, ep string, token []string, sf *utils.StreamFormatter) error {
|
|
out = utils.NewWriteFlusher(out)
|
|
out = utils.NewWriteFlusher(out)
|
|
jsonRaw, err := ioutil.ReadFile(path.Join(srv.runtime.graph.Root, imgId, "json"))
|
|
jsonRaw, err := ioutil.ReadFile(path.Join(srv.runtime.graph.Root, imgId, "json"))
|
|
if err != nil {
|
|
if err != nil {
|
|
return fmt.Errorf("Error while retreiving the path for {%s}: %s", imgId, err)
|
|
return fmt.Errorf("Error while retreiving the path for {%s}: %s", imgId, err)
|
|
}
|
|
}
|
|
- fmt.Fprintf(out, "Pushing %s\r\n", imgId)
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("Pushing %s", imgId))
|
|
|
|
|
|
// Make sure we have the image's checksum
|
|
// Make sure we have the image's checksum
|
|
checksum, err := srv.getChecksum(imgId)
|
|
checksum, err := srv.getChecksum(imgId)
|
|
@@ -535,9 +536,9 @@ func (srv *Server) pushImage(out io.Writer, remote, imgId, ep string, token []st
|
|
}
|
|
}
|
|
|
|
|
|
// Send the json
|
|
// Send the json
|
|
- if err := srv.registry.PushImageJsonRegistry(imgData, jsonRaw, ep, token); err != nil {
|
|
|
|
|
|
+ if err := r.PushImageJsonRegistry(imgData, jsonRaw, ep, token); err != nil {
|
|
if err == registry.ErrAlreadyExists {
|
|
if err == registry.ErrAlreadyExists {
|
|
- fmt.Fprintf(out, "Image %s already uploaded ; skipping\n", imgData.Id)
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("Image %s already uploaded ; skipping", imgData.Id))
|
|
return nil
|
|
return nil
|
|
}
|
|
}
|
|
return err
|
|
return err
|
|
@@ -570,20 +571,22 @@ func (srv *Server) pushImage(out io.Writer, remote, imgId, ep string, token []st
|
|
}
|
|
}
|
|
|
|
|
|
// Send the layer
|
|
// Send the layer
|
|
- if err := srv.registry.PushImageLayerRegistry(imgData.Id, utils.ProgressReader(layerData, int(layerData.Size), out, "", false), ep, token); err != nil {
|
|
|
|
|
|
+ if err := r.PushImageLayerRegistry(imgData.Id, utils.ProgressReader(layerData, int(layerData.Size), out, sf.FormatProgress("", "%v/%v (%v)"), sf), ep, token); err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
return nil
|
|
}
|
|
}
|
|
|
|
|
|
-func (srv *Server) ImagePush(name, registry string, out io.Writer) error {
|
|
|
|
|
|
+func (srv *Server) ImagePush(name, endpoint string, out io.Writer, sf *utils.StreamFormatter) error {
|
|
out = utils.NewWriteFlusher(out)
|
|
out = utils.NewWriteFlusher(out)
|
|
img, err := srv.runtime.graph.Get(name)
|
|
img, err := srv.runtime.graph.Get(name)
|
|
|
|
+ r := registry.NewRegistry(srv.runtime.root)
|
|
|
|
+
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Fprintf(out, "The push refers to a repository [%s] (len: %d)\n", name, len(srv.runtime.repositories.Repositories[name]))
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("The push refers to a repository [%s] (len: %d)", name, len(srv.runtime.repositories.Repositories[name])))
|
|
// If it fails, try to get the repository
|
|
// If it fails, try to get the repository
|
|
if localRepo, exists := srv.runtime.repositories.Repositories[name]; exists {
|
|
if localRepo, exists := srv.runtime.repositories.Repositories[name]; exists {
|
|
- if err := srv.pushRepository(out, name, localRepo); err != nil {
|
|
|
|
|
|
+ if err := srv.pushRepository(r, out, name, localRepo, sf); err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
return nil
|
|
@@ -591,14 +594,14 @@ func (srv *Server) ImagePush(name, registry string, out io.Writer) error {
|
|
|
|
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
- fmt.Fprintf(out, "The push refers to an image: [%s]\n", name)
|
|
|
|
- if err := srv.pushImage(out, name, img.Id, registry, nil); err != nil {
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("The push refers to an image: [%s]", name))
|
|
|
|
+ if err := srv.pushImage(r, out, name, img.Id, endpoint, nil, sf); err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
return nil
|
|
}
|
|
}
|
|
|
|
|
|
-func (srv *Server) ImageImport(src, repo, tag string, in io.Reader, out io.Writer) error {
|
|
|
|
|
|
+func (srv *Server) ImageImport(src, repo, tag string, in io.Reader, out io.Writer, sf *utils.StreamFormatter) error {
|
|
var archive io.Reader
|
|
var archive io.Reader
|
|
var resp *http.Response
|
|
var resp *http.Response
|
|
|
|
|
|
@@ -607,21 +610,21 @@ func (srv *Server) ImageImport(src, repo, tag string, in io.Reader, out io.Write
|
|
} else {
|
|
} else {
|
|
u, err := url.Parse(src)
|
|
u, err := url.Parse(src)
|
|
if err != nil {
|
|
if err != nil {
|
|
- fmt.Fprintf(out, "Error: %s\n", err)
|
|
|
|
|
|
+ return err
|
|
}
|
|
}
|
|
if u.Scheme == "" {
|
|
if u.Scheme == "" {
|
|
u.Scheme = "http"
|
|
u.Scheme = "http"
|
|
u.Host = src
|
|
u.Host = src
|
|
u.Path = ""
|
|
u.Path = ""
|
|
}
|
|
}
|
|
- fmt.Fprintf(out, "Downloading from %s\n", u)
|
|
|
|
|
|
+ out.Write(sf.FormatStatus("Downloading from %s", u))
|
|
// Download with curl (pretty progress bar)
|
|
// Download with curl (pretty progress bar)
|
|
// If curl is not available, fallback to http.Get()
|
|
// If curl is not available, fallback to http.Get()
|
|
resp, err = utils.Download(u.String(), out)
|
|
resp, err = utils.Download(u.String(), out)
|
|
if err != nil {
|
|
if err != nil {
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
- archive = utils.ProgressReader(resp.Body, int(resp.ContentLength), out, "Importing %v/%v (%v)\r", false)
|
|
|
|
|
|
+ archive = utils.ProgressReader(resp.Body, int(resp.ContentLength), out, sf.FormatProgress("Importing", "%v/%v (%v)"), sf)
|
|
}
|
|
}
|
|
img, err := srv.runtime.graph.Create(archive, nil, "Imported from "+src, "", nil)
|
|
img, err := srv.runtime.graph.Create(archive, nil, "Imported from "+src, "", nil)
|
|
if err != nil {
|
|
if err != nil {
|
|
@@ -633,7 +636,7 @@ func (srv *Server) ImageImport(src, repo, tag string, in io.Reader, out io.Write
|
|
return err
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
}
|
|
- fmt.Fprintf(out, "%s\n", img.ShortId())
|
|
|
|
|
|
+ out.Write(sf.FormatStatus(img.ShortId()))
|
|
return nil
|
|
return nil
|
|
}
|
|
}
|
|
|
|
|
|
@@ -883,7 +886,6 @@ func (srv *Server) ContainerAttach(name string, logs, stream, stdin, stdout, std
|
|
if container == nil {
|
|
if container == nil {
|
|
return fmt.Errorf("No such container: %s", name)
|
|
return fmt.Errorf("No such container: %s", name)
|
|
}
|
|
}
|
|
-
|
|
|
|
//logs
|
|
//logs
|
|
if logs {
|
|
if logs {
|
|
if stdout {
|
|
if stdout {
|
|
@@ -909,6 +911,9 @@ func (srv *Server) ContainerAttach(name string, logs, stream, stdin, stdout, std
|
|
if container.State.Ghost {
|
|
if container.State.Ghost {
|
|
return fmt.Errorf("Impossible to attach to a ghost container")
|
|
return fmt.Errorf("Impossible to attach to a ghost container")
|
|
}
|
|
}
|
|
|
|
+ if !container.State.Running {
|
|
|
|
+ return fmt.Errorf("Impossible to attach to a stopped container, start it first")
|
|
|
|
+ }
|
|
|
|
|
|
var (
|
|
var (
|
|
cStdin io.ReadCloser
|
|
cStdin io.ReadCloser
|
|
@@ -967,14 +972,12 @@ func NewServer(autoRestart bool) (*Server, error) {
|
|
return nil, err
|
|
return nil, err
|
|
}
|
|
}
|
|
srv := &Server{
|
|
srv := &Server{
|
|
- runtime: runtime,
|
|
|
|
- registry: registry.NewRegistry(runtime.root),
|
|
|
|
|
|
+ runtime: runtime,
|
|
}
|
|
}
|
|
runtime.srv = srv
|
|
runtime.srv = srv
|
|
return srv, nil
|
|
return srv, nil
|
|
}
|
|
}
|
|
|
|
|
|
type Server struct {
|
|
type Server struct {
|
|
- runtime *Runtime
|
|
|
|
- registry *registry.Registry
|
|
|
|
|
|
+ runtime *Runtime
|
|
}
|
|
}
|