Skip to content

Commit

Permalink
example: add build-using-dockerfile
Browse files Browse the repository at this point in the history
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>
  • Loading branch information
AkihiroSuda committed Dec 19, 2017
1 parent 2822846 commit ea5de4c
Showing 1 changed file with 157 additions and 0 deletions.
157 changes: 157 additions & 0 deletions examples/build-using-dockerfile/main.go
Original file line number Diff line number Diff line change
@@ -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()
}

0 comments on commit ea5de4c

Please sign in to comment.