Pull image if without when create service

Signed-off-by: skanehira <[email protected]>

Author: skanehira

compose-ref.go

@@ -201,6 +201,7 @@ func doUp(project string, config *compose.Config) error {
 		if err != nil {
 			return err
 		}
+
 		return createService(cli, project, prjDir, service, networks)
 	})
 
@@ -221,6 +222,11 @@ func doUp(project string, config *compose.Config) error {
 func createService(cli *client.Client, project string, prjDir string, s compose.ServiceConfig, networks map[string]string) error {
 	ctx := context.Background()
 
+	err := internal.PullImageIfWithout(cli, ctx, s.Name)
+	if err != nil {
+		return err
+	}
+
 	var shmSize int64
 	if s.ShmSize != "" {
 		v, err := units.RAMInBytes(s.ShmSize)

go.mod

@@ -5,6 +5,7 @@ require (
 	github.com/Microsoft/go-winio v0.4.14 // indirect
 	github.com/compose-spec/compose-go v0.0.0-20200131085702-0b38cc2d8e6b
 	github.com/containerd/containerd v1.3.2 // indirect
+	github.com/docker/cli v0.0.0-20200303215952-eb310fca4956
 	github.com/docker/distribution v2.7.1+incompatible // indirect
 	github.com/docker/docker v1.4.2-0.20191113042239-ea84732a7725
 	github.com/docker/go-connections v0.4.0

go.sum

@@ -15,6 +15,8 @@ github.com/cpuguy83/go-md2man/v2 v2.0.0-20190314233015-f79a8a8ca69d/go.mod h1:ma
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/docker/cli v0.0.0-20200303215952-eb310fca4956 h1:5/ZRsUbguX7xFNLlbxVQY/yhD3Psy+vylKZrNme5BJs=
+github.com/docker/cli v0.0.0-20200303215952-eb310fca4956/go.mod h1:JLrzqnKDaYBop7H2jaqPtU4hHvMKP+vjCwu2uszcLI8=
 github.com/docker/distribution v2.7.1+incompatible h1:a5mlkVzth6W5A4fOsS3D2EO5BUmsJpcB+cRlLU7cSug=
 github.com/docker/distribution v2.7.1+incompatible/go.mod h1:J2gT2udsDAN96Uj4KfcMRqY0/ypR+oyYUYmja8H+y+w=
 github.com/docker/docker v1.4.2-0.20191113042239-ea84732a7725 h1:9AYKG7Y/KlVWYycokdsCJTxv9WrH2+YrrOxrrSnBiCc=

internal/image.go

@@ -0,0 +1,51 @@
+/*
+   Copyright 2020 The Compose Specification Authors.
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+*/
+
+package internal
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/docker/cli/cli/streams"
+	"github.com/docker/docker/api/types"
+	"github.com/docker/docker/api/types/filters"
+	"github.com/docker/docker/client"
+	"github.com/docker/docker/pkg/jsonmessage"
+)
+
+func PullImageIfWithout(cli *client.Client, ctx context.Context, name string) error {
+	args := filters.NewArgs(filters.Arg("reference", name))
+	images, err := cli.ImageList(ctx, types.ImageListOptions{All: true, Filters: args})
+	if err != nil {
+		return err
+	}
+	if len(images) == 0 {
+		fmt.Println("pulling image: " + name)
+		res, err := cli.ImagePull(ctx, name, types.ImagePullOptions{})
+		if err != nil {
+			return err
+		}
+
+		out := streams.NewOut(os.Stdout)
+		err = jsonmessage.DisplayJSONMessagesToStream(res, out, nil)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}