feat: support networks (#41)

Introduces the ability to specify networks for containers, ensuring that tasks can be run within specific network environments.

Author: w-h-a

internal/engine/clients/runner/docker/runner.go

@@ -12,6 +12,7 @@ import (
 	"github.com/docker/docker/api/types/filters"
 	"github.com/docker/docker/api/types/image"
 	"github.com/docker/docker/api/types/mount"
+	"github.com/docker/docker/api/types/network"
 	"github.com/docker/docker/api/types/volume"
 	"github.com/docker/docker/client"
 	"github.com/docker/docker/pkg/stdcopy"
@@ -65,7 +66,15 @@ func (r *dockerRunner) Run(ctx context.Context, opts ...runner.RunOption) (strin
 		Mounts:          mounts,
 	}
 
-	rsp, err := r.client.ContainerCreate(ctx, &cc, &hc, nil, nil, "")
+	nc := network.NetworkingConfig{
+		EndpointsConfig: map[string]*network.EndpointSettings{},
+	}
+
+	for _, nw := range options.Networks {
+		nc.EndpointsConfig[nw] = &network.EndpointSettings{NetworkID: nw}
+	}
+
+	rsp, err := r.client.ContainerCreate(ctx, &cc, &hc, &nc, nil, "")
 	if err != nil {
 		// span
 		slog.ErrorContext(ctx, "failed to create container", "image", options.Image, "error", err)

internal/engine/clients/runner/options.go

@@ -30,12 +30,13 @@ func NewOptions(opts ...Option) Options {
 type RunOption func(o *RunOptions)
 
 type RunOptions struct {
-	ID      string
-	Image   string
-	Cmd     []string
-	Env     []string
-	Volumes []string
-	Context context.Context
+	ID       string
+	Image    string
+	Cmd      []string
+	Env      []string
+	Volumes  []string
+	Networks []string
+	Context  context.Context
 }
 
 func RunWithID(id string) RunOption {
@@ -68,6 +69,12 @@ func RunWithVolumes(volumes []string) RunOption {
 	}
 }
 
+func RunWithNetworks(networks []string) RunOption {
+	return func(o *RunOptions) {
+		o.Networks = networks
+	}
+}
+
 func NewRunOptions(opts ...RunOption) RunOptions {
 	options := RunOptions{
 		Context: context.Background(),

internal/engine/services/worker/service.go

@@ -124,6 +124,7 @@ func (s *Service) runTask(ctx context.Context, t *task.Task) error {
 	for _, pre := range t.Pre {
 		pre.ID = strings.ReplaceAll(uuid.NewString(), "-", "")
 		pre.Volumes = t.Volumes
+		pre.Networks = t.Networks
 		result, err := s.run(ctx, pre)
 		finished := time.Now()
 		if err != nil {
@@ -171,6 +172,7 @@ func (s *Service) runTask(ctx context.Context, t *task.Task) error {
 	for _, post := range t.Post {
 		post.ID = strings.ReplaceAll(uuid.NewString(), "-", "")
 		post.Volumes = t.Volumes
+		post.Networks = t.Networks
 		result, err := s.run(ctx, post)
 		finished := time.Now()
 		if err != nil {
@@ -224,6 +226,7 @@ func (s *Service) run(ctx context.Context, t *task.Task) (string, error) {
 		runner.RunWithCmd(t.Cmd),
 		runner.RunWithEnv(t.Env),
 		runner.RunWithVolumes(t.Volumes),
+		runner.RunWithNetworks(t.Networks),
 	}
 
 	return s.runner.Run(ctx, runOpts...)

internal/task/domain.go

@@ -31,6 +31,7 @@ type Task struct {
 	Pre         []*Task    `json:"pre,omitempty"`
 	Post        []*Task    `json:"post,omitempty"`
 	Volumes     []string   `json:"volumes,omitempty"`
+	Networks    []string   `json:"networks,omitempty"`
 	Retry       *Retry     `json:"retry,omitempty"`
 	Timeout     string     `json:"timeout,omitempty"`
 }