From 09c427f1cfdba14c7020253703283541eb4ee52d Mon Sep 17 00:00:00 2001
From: Vasek - Tom C <tom.chauveau@epitech.eu>
Date: Fri, 17 Dec 2021 20:25:20 +0100
Subject: [PATCH] Add mkdir task

Signed-off-by: Vasek - Tom C <tom.chauveau@epitech.eu>
---
 plan/task/mkdir.go | 81 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100644 plan/task/mkdir.go

diff --git a/plan/task/mkdir.go b/plan/task/mkdir.go
new file mode 100644
index 00000000..8a03d70c
--- /dev/null
+++ b/plan/task/mkdir.go
@@ -0,0 +1,81 @@
+package task
+
+import (
+	"context"
+	"io/fs"
+
+	"cuelang.org/go/cue"
+	"github.com/moby/buildkit/client/llb"
+	"go.dagger.io/dagger/compiler"
+	"go.dagger.io/dagger/plancontext"
+	"go.dagger.io/dagger/solver"
+)
+
+func init() {
+	Register("Mkdir", func() Task { return &mkdirTask{} })
+}
+
+type mkdirTask struct {
+}
+
+func (t *mkdirTask) Run(ctx context.Context, pctx *plancontext.Context, s solver.Solver, v *compiler.Value) (*compiler.Value, error) {
+	path, err := v.Lookup("path").String()
+	if err != nil {
+		return nil, err
+	}
+
+	// Mode (int)
+	mode, err := v.Lookup("mode").Int64()
+	if err != nil {
+		return nil, err
+	}
+
+	// Retrieve options
+	mkdirOpts := []llb.MkdirOption{}
+	var opts struct {
+		Parents bool
+	}
+
+	if err := v.Decode(&opts); err != nil {
+		return nil, err
+	}
+
+	if opts.Parents {
+		mkdirOpts = append(mkdirOpts, llb.WithParents(true))
+	}
+
+	// Retrieve input Filesystem
+	input, err := pctx.FS.FromValue(v.Lookup("input"))
+	if err != nil {
+		return nil, err
+	}
+
+	// Retrieve input llb state
+	inputState, err := input.Result().ToState()
+	if err != nil {
+		return nil, err
+	}
+
+	// Add Mkdir operation on input llb state
+	outputState := inputState.File(
+		llb.Mkdir(path, fs.FileMode(mode), mkdirOpts...),
+		withCustomName(v, "Mkdir %s", path),
+	)
+
+	// Compute state
+	result, err := s.Solve(ctx, outputState, pctx.Platform.Get())
+	if err != nil {
+		return nil, err
+	}
+
+	// Retrieve result result filesystem
+	outputFs := pctx.FS.New(result)
+
+	// Init output
+	output := compiler.NewValue()
+
+	if err := output.FillPath(cue.ParsePath("output"), outputFs.MarshalCUE()); err != nil {
+		return nil, err
+	}
+	return output, nil
+}