driver: start implemeting bootstrap for docker

Signed-off-by: Tonis Tiigi <tonistiigi@gmail.com>

driver/docker/driver.go

@@ -2,24 +2,82 @@ package docker
 
 import (
 	"context"
+	"io"
+	"io/ioutil"
 
+	"github.com/docker/docker/api/types"
 	dockertypes "github.com/docker/docker/api/types"
+	"github.com/docker/docker/api/types/container"
+	"github.com/docker/docker/api/types/network"
+	dockerclient "github.com/docker/docker/client"
 	"github.com/moby/buildkit/client"
 	"github.com/pkg/errors"
 	"github.com/tonistiigi/buildx/driver"
 )
 
 type Driver struct {
-	config  driver.InitConfig
+	driver.InitConfig
 	version dockertypes.Version
 }
 
-func (d *Driver) Bootstrap(context.Context, driver.Logger) error {
-	return errors.Errorf("bootstrap not implemented for %T", d)
+func (d *Driver) Bootstrap(ctx context.Context, l driver.Logger) error {
+	_, err := d.DockerAPI.ContainerInspect(ctx, d.Name)
+	if err != nil {
+		if dockerclient.IsErrNotFound(err) {
+			return d.create(ctx, l)
+		}
+		return err
+	}
+	return d.start(ctx, l)
 }
 
-func (d *Driver) Info(context.Context) (driver.Info, error) {
-	return driver.Info{}, errors.Errorf("info not implemented for %T", d)
+func (d *Driver) create(ctx context.Context, l driver.Logger) error {
+	rc, err := d.DockerAPI.ImageCreate(ctx, "moby/buildkit", types.ImageCreateOptions{})
+	if err != nil {
+		return err
+	}
+	_, err = io.Copy(ioutil.Discard, rc)
+	if err != nil {
+		return err
+	}
+	_, err = d.DockerAPI.ContainerCreate(ctx, &container.Config{
+		Image: "moby/buildkit",
+	}, &container.HostConfig{
+		Privileged: true,
+	}, &network.NetworkingConfig{}, d.Name)
+	if err != nil {
+		return err
+	}
+	if err := d.start(ctx, l); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (d *Driver) start(ctx context.Context, l driver.Logger) error {
+	return d.DockerAPI.ContainerStart(ctx, d.Name, types.ContainerStartOptions{})
+}
+
+func (d *Driver) Info(ctx context.Context) (*driver.Info, error) {
+	container, err := d.DockerAPI.ContainerInspect(ctx, d.Name)
+	if err != nil {
+		if dockerclient.IsErrNotFound(err) {
+			return &driver.Info{
+				Status: driver.Terminated,
+			}, nil
+		}
+		return nil, err
+	}
+
+	if container.State.Running {
+		return &driver.Info{
+			Status: driver.Running,
+		}, nil
+	}
+
+	return &driver.Info{
+		Status: driver.Stopped,
+	}, nil
 }
 
 func (d *Driver) Stop(ctx context.Context, force bool) error {
@@ -30,6 +88,6 @@ func (d *Driver) Rm(ctx context.Context, force bool) error {
 	return errors.Errorf("rm not implemented for %T", d)
 }
 
-func (d *Driver) Client() (*client.Client, error) {
+func (d *Driver) Client(ctx context.Context) (*client.Client, error) {
 	return nil, errors.Errorf("client not implemented for %T", d)
 }

driver/docker/factory.go

@@ -36,5 +36,5 @@ func (*factory) New(ctx context.Context, cfg driver.InitConfig) (driver.Driver,
 		return nil, errors.Wrapf(driver.ErrNotConnecting, err.Error())
 	}
 
-	return &Driver{config: cfg, version: v}, nil
+	return &Driver{InitConfig: cfg, version: v}, nil
 }

driver/driver.go

@@ -4,7 +4,6 @@ import (
 	"context"
 
 	"github.com/moby/buildkit/client"
-	specs "github.com/opencontainers/image-spec/specs-go/v1"
 	"github.com/pkg/errors"
 )
 
@@ -24,14 +23,47 @@ const (
 )
 
 type Info struct {
-	Status    Status
-	Platforms []specs.Platform
+	Status Status
 }
 
 type Driver interface {
 	Bootstrap(context.Context, Logger) error
-	Info(context.Context) (Info, error)
+	Info(context.Context) (*Info, error)
 	Stop(ctx context.Context, force bool) error
 	Rm(ctx context.Context, force bool) error
-	Client() (*client.Client, error)
+	Client(ctx context.Context) (*client.Client, error)
+}
+
+func Boot(ctx context.Context, d Driver, status chan *client.SolveStatus) (*client.Client, error) {
+	try := 0
+	for {
+		info, err := d.Info(ctx)
+		if err != nil {
+			return nil, err
+		}
+		try++
+		if info.Status != Running {
+			if try > 2 {
+				return nil, errors.Errorf("failed to bootstrap %T driver in attempts", d)
+			}
+			if err := d.Bootstrap(ctx, func(s *client.SolveStatus) {
+				if status != nil {
+					status <- s
+				}
+			}); err != nil {
+				return nil, err
+			}
+		}
+
+		c, err := d.Client(ctx)
+		if err != nil {
+			if errors.Cause(err) == ErrNotRunning && try <= 2 {
+				continue
+			}
+			return nil, err
+		}
+		return c, nil
+	}
+
+	return nil, errors.Errorf("boot not implemented")
 }

driver/manager.go

@@ -37,7 +37,7 @@ func Register(f Factory) {
 	drivers[f.Name()] = f
 }
 
-func GetDefaultDriver() (Factory, error) {
+func GetDefaultFactory() (Factory, error) {
 	if len(drivers) == 0 {
 		return nil, errors.Errorf("no drivers available")
 	}
@@ -54,3 +54,17 @@ func GetDefaultDriver() (Factory, error) {
 	})
 	return dd[0].f, nil
 }
+
+func GetDriver(ctx context.Context, name string, f Factory, api dockerclient.APIClient) (Driver, error) {
+	if f == nil {
+		var err error
+		f, err = GetDefaultFactory()
+		if err != nil {
+			return nil, err
+		}
+	}
+	return f.New(ctx, InitConfig{
+		Name:      name,
+		DockerAPI: api,
+	})
+}