package execution

import (
	"io/ioutil"
	"os"
	"path/filepath"
)

const processesDir = "processes"

type StateDir string

func NewStateDir(root, id string) (StateDir, error) {
	path := filepath.Join(root, id)
	if err := os.Mkdir(path, 0700); err != nil {
		return "", err
	}
	if err := os.Mkdir(filepath.Join(path, processesDir), 0700); err != nil {
		os.RemoveAll(path)
		return "", err
	}
	return StateDir(path), nil
}

func (s StateDir) Delete() error {
	return os.RemoveAll(string(s))
}

func (s StateDir) NewProcess() (string, error) {
	return ioutil.TempDir(s.processDir(), "")
}

func (s StateDir) ProcessDir(id string) string {
	return filepath.Join(string(s), id)
}

func (s StateDir) DeleteProcess(id string) error {
	return os.RemoveAll(filepath.Join(s.processDir(), id))
}

func (s StateDir) Processes() ([]string, error) {
	procsDir := s.processDir()
	dirs, err := ioutil.ReadDir(procsDir)
	if err != nil {
		return nil, err
	}

	paths := make([]string, 0)
	for _, d := range dirs {
		if d.IsDir() {
			paths = append(paths, filepath.Join(procsDir, d.Name()))
		}
	}
	return paths, nil
}

func (s StateDir) processDir() string {
	return filepath.Join(string(s), processesDir)
}