From ea5de4c3686ea2b7f8619ac44a9f3d0e58d334f0 Mon Sep 17 00:00:00 2001
From: Akihiro Suda <suda.akihiro@lab.ntt.co.jp>
Date: Fri, 15 Dec 2017 18:28:33 +0900
Subject: [PATCH] example: add build-using-dockerfile

This command mimics `docker build` CLI flags so that Docker users can
more easily get started with BuildKit.

Signed-off-by: Akihiro Suda <suda.akihiro@lab.ntt.co.jp>
---
 examples/build-using-dockerfile/main.go | 157 ++++++++++++++++++++++++
 1 file changed, 157 insertions(+)
 create mode 100644 examples/build-using-dockerfile/main.go

diff --git a/examples/build-using-dockerfile/main.go b/examples/build-using-dockerfile/main.go
new file mode 100644
index 000000000000..2b29bfe096e4
--- /dev/null
+++ b/examples/build-using-dockerfile/main.go
@@ -0,0 +1,157 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"io/ioutil"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+
+	"github.com/containerd/console"
+	"github.com/moby/buildkit/client"
+	"github.com/moby/buildkit/util/appcontext"
+	"github.com/moby/buildkit/util/appdefaults"
+	"github.com/moby/buildkit/util/progress/progressui"
+	"github.com/pkg/errors"
+	"github.com/sirupsen/logrus"
+	"github.com/urfave/cli"
+	"golang.org/x/sync/errgroup"
+)
+
+func main() {
+	app := cli.NewApp()
+	app.Name = "build-using-dockerfile"
+	app.UsageText = `build-using-dockerfile [OPTIONS] PATH | URL | -`
+	app.Description = `
+build using Dockerfile.
+
+This command mimics behavior of "docker build" command so that people can easily get started with BuildKit.
+This command is NOT the replacement of "docker build", and should NOT be used for building production images.
+
+By default, the built image is loaded to Docker.
+`
+	dockerIncompatibleFlags := []cli.Flag{
+		cli.StringFlag{
+			Name:   "buildkit-addr",
+			Usage:  "buildkit daemon address",
+			EnvVar: "BUILDKIT_HOST",
+			Value:  appdefaults.Address,
+		},
+	}
+	app.Flags = append([]cli.Flag{
+		cli.StringSliceFlag{
+			Name:  "build-arg",
+			Usage: "Set build-time variables",
+		},
+		cli.StringFlag{
+			Name:  "file, f",
+			Usage: "Name of the Dockerfile (Default is 'PATH/Dockerfile')",
+		},
+		cli.StringFlag{
+			Name:  "tag, t",
+			Usage: "Name and optionally a tag in the 'name:tag' format",
+		},
+		cli.StringFlag{
+			Name:  "target",
+			Usage: "Set the target build stage to build.",
+		},
+	}, dockerIncompatibleFlags...)
+	app.Action = action
+	if err := app.Run(os.Args); err != nil {
+		fmt.Fprintf(os.Stderr, "error: %v\n", err)
+		os.Exit(1)
+	}
+}
+
+func action(clicontext *cli.Context) error {
+	if tag := clicontext.String("tag"); tag == "" {
+		return errors.New("tag is not specified")
+	}
+	c, err := client.New(clicontext.String("buildkit-addr"), client.WithBlock())
+	if err != nil {
+		return err
+	}
+	tmpTar, err := ioutil.TempFile("", "buldkit-build-using-dockerfile")
+	if err != nil {
+		return err
+	}
+	defer os.Remove(tmpTar.Name())
+	solveOpt, err := newSolveOpt(clicontext, tmpTar.Name())
+	if err != nil {
+		return err
+	}
+	ch := make(chan *client.SolveStatus)
+	eg, ctx := errgroup.WithContext(appcontext.Context())
+	eg.Go(func() error {
+		return c.Solve(ctx, nil, *solveOpt, ch)
+	})
+	eg.Go(func() error {
+		if c, err := console.ConsoleFromFile(os.Stderr); err == nil {
+			// not using shared context to not disrupt display but let is finish reporting errors
+			return progressui.DisplaySolveStatus(context.TODO(), c, ch)
+		}
+		return nil
+	})
+	if err := eg.Wait(); err != nil {
+		return err
+	}
+	logrus.Infof("Loading the image to Docker as %q. This may take a while.", clicontext.String("tag"))
+	if err := loadDockerTar(tmpTar.Name()); err != nil {
+		return err
+	}
+	logrus.Info("Done")
+	return nil
+}
+
+func newSolveOpt(clicontext *cli.Context, tmpTar string) (*client.SolveOpt, error) {
+	buildCtx := clicontext.Args().First()
+	if buildCtx == "" {
+		return nil, errors.New("please specify build context (e.g. \".\" for the current directory)")
+	} else if buildCtx == "-" {
+		return nil, errors.New("stdin not supported yet")
+	}
+
+	file := clicontext.String("file")
+	if file == "" {
+		file = filepath.Join(buildCtx, "Dockerfile")
+	}
+	localDirs := map[string]string{
+		"context":    buildCtx,
+		"dockerfile": filepath.Dir(file),
+	}
+
+	frontendAttrs := map[string]string{
+		"filename": filepath.Base(file),
+	}
+	if target := clicontext.String("target"); target != "" {
+		frontendAttrs["target"] = target
+	}
+
+	for _, buildArg := range clicontext.StringSlice("build-arg") {
+		kv := strings.SplitN(buildArg, "=", 2)
+		if len(kv) != 2 {
+			return nil, errors.Errorf("invalid build-arg value %s", buildArg)
+		}
+		frontendAttrs["build-arg:"+kv[0]] = kv[1]
+	}
+	return &client.SolveOpt{
+		Exporter: "docker", // TODO: use containerd image store when it is integrated to Docker
+		ExporterAttrs: map[string]string{
+			"name":   clicontext.String("tag"),
+			"output": tmpTar,
+		},
+		LocalDirs:     localDirs,
+		Frontend:      "dockerfile.v0", // TODO: use gateway
+		FrontendAttrs: frontendAttrs,
+	}, nil
+}
+
+func loadDockerTar(tar string) error {
+	// no need to use moby/moby/client here
+	cmd := exec.Command("docker", "load", "-i", tar)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+	return cmd.Run()
+}