feat: implement storage mount/unmount as well as storage ensure

Author: josegonzalez

tasks/storage_ensure_task.go

@@ -0,0 +1,81 @@
+package tasks
+
+import (
+	"errors"
+	"omakase/subprocess"
+)
+
+type StorageEnsureTask struct {
+	App   string `required:"true" yaml:"app"`
+	Chown string `required:"false" yaml:"chown"`
+	State string `required:"true" yaml:"state" default:"present"`
+}
+
+func (t StorageEnsureTask) DesiredState() string {
+	return t.State
+}
+
+func (t StorageEnsureTask) Execute() TaskOutputState {
+	funcMap := map[string]func(string, string) TaskOutputState{
+		"present": ensureStorage,
+		"absent":  removeStorage,
+	}
+
+	fn := funcMap[t.State]
+	return fn(t.App, t.Chown)
+}
+
+func ensureStorage(app, chown string) TaskOutputState {
+	state := TaskOutputState{
+		Changed: false,
+		State:   "absent",
+	}
+
+	// ensure chown is a valid value
+	chownValues := map[string]bool{
+		"heroku":    true,
+		"herokuish": true,
+		"packeto":   true,
+		"root":      true,
+		"false":     true,
+	}
+	if !chownValues[chown] {
+		state.Error = errors.New("invalid chown value specified")
+		return state
+	}
+
+	// todo: implement a check to see if the folder exists?
+	result, err := subprocess.CallExecCommand(subprocess.ExecCommandInput{
+		Command: "dokku",
+		Args: []string{
+			"--quiet",
+			"storage:ensure-directory",
+			"--chown",
+			chown,
+			app,
+		},
+	})
+	if err != nil {
+		state.Error = err
+		state.Message = result.StderrContents()
+		return state
+	}
+
+	state.Changed = true
+	state.State = "present"
+	return state
+}
+
+func removeStorage(app, chown string) TaskOutputState {
+	state := TaskOutputState{
+		Changed: false,
+		State:   "absent",
+	}
+
+	state.Error = errors.New("the absent state is not supported for storage:ensure")
+	return state
+}
+
+func init() {
+	RegisterTask(&StorageEnsureTask{})
+}

tasks/storage_mount_task.go

@@ -0,0 +1,128 @@
+package tasks
+
+import (
+	"fmt"
+	"omakase/subprocess"
+)
+
+type StorageMountTask struct {
+	App          string `required:"true" yaml:"app"`
+	HostDir      string `required:"true" yaml:"host_dir"`
+	ContainerDir string `required:"true" yaml:"container_dir"`
+	State        string `required:"true" yaml:"state" default:"present"`
+}
+
+func (t StorageMountTask) DesiredState() string {
+	return t.State
+}
+
+func (t StorageMountTask) Execute() TaskOutputState {
+	funcMap := map[string]func(string, string, string) TaskOutputState{
+		"present": mountStorage,
+		"absent":  unmountStorage,
+	}
+
+	fn := funcMap[t.State]
+	return fn(t.App, t.HostDir, t.ContainerDir)
+}
+
+func mountExists(app, hostDir, containerDir string) bool {
+	result, err := subprocess.CallExecCommand(subprocess.ExecCommandInput{
+		Command: "dokku",
+		Args: []string{
+			"--quiet",
+			"storage:list",
+			app,
+			"--format",
+			"json",
+		},
+	})
+	if err != nil {
+		return false
+	}
+
+	var mounts []struct {
+		HostDir      string `json:"host_dir"`
+		ContainerDir string `json:"container_dir"`
+	}
+
+	err = json.Unmarshal(result.StdoutBytes(), &mounts)
+	if err != nil {
+		return false
+	}
+
+	for _, mount := range mounts {
+		if mount.HostDir == hostDir && mount.ContainerDir == containerDir {
+			return true
+		}
+	}
+	return false
+}
+
+func mountStorage(app, hostDir, containerDir string) TaskOutputState {
+	state := TaskOutputState{
+		Changed: false,
+		State:   "present",
+	}
+
+	// check if the mount already exists
+	if mountExists(app, hostDir, containerDir) {
+		state.Changed = false
+		state.State = "present"
+		return state
+	}
+
+	result, err := subprocess.CallExecCommand(subprocess.ExecCommandInput{
+		Command: "dokku",
+		Args: []string{
+			"--quiet",
+			"storage:mount",
+			app,
+			fmt.Sprintf("%s:%s", hostDir, containerDir),
+		},
+	})
+	if err != nil {
+		state.Error = err
+		state.Message = result.StderrContents()
+		return state
+	}
+
+	state.Changed = true
+	state.State = "present"
+	return state
+}
+
+func unmountStorage(app, hostDir, containerDir string) TaskOutputState {
+	state := TaskOutputState{
+		Changed: false,
+		State:   "present",
+	}
+	if !mountExists(app, hostDir, containerDir) {
+		state.Changed = false
+		state.State = "absent"
+		return state
+	}
+
+	result, err := subprocess.CallExecCommand(subprocess.ExecCommandInput{
+		Command: "dokku",
+		Args: []string{
+			"--quiet",
+			"storage:unmount",
+			app,
+			fmt.Sprintf("%s:%s", hostDir, containerDir),
+		},
+	})
+	if err != nil {
+		state.Error = err
+		state.Message = result.StderrContents()
+		return state
+	}
+
+	state.Changed = true
+	state.State = "absent"
+	return state
+}
+
+func init() {
+	RegisterTask(&StorageMountTask{})
+}