package exec

import (
	"bufio"
	"bytes"
	"errors"
	"fmt"
	"os"
	"os/exec"
	"path/filepath"
	"strings"
	"text/template"
)

func RunCommand(dir, cmd string, args ...string) (string, error) {
	command := exec.Command(cmd, args...)

	var outb, errb bytes.Buffer
	command.Stdout = &outb
	command.Stderr = &errb
	command.Dir = dir
	if err := command.Run(); err != nil {
		return "", errors.New(errb.String())
	}

	return outb.String(), nil
}

func RunTemplatedScript(dir, fileName, scriptTemplate string, funcMap template.FuncMap, args interface{}) (string, error) {
	if _, err := os.Stat(dir); err != nil {
		return "", err
	}
	scriptPath := filepath.Join(dir, fileName)
	f, err := os.Create(scriptPath)
	if err != nil {
		return "", err
	}
	defer f.Close()
	if err := os.Chmod(scriptPath, 0755); err != nil {
		return "", err
	}
	tmpl, err := template.New(scriptTemplate).Funcs(funcMap).Parse(scriptTemplate)
	if err != nil {
		return "", err
	}
	if err := tmpl.Execute(f, args); err != nil {
		return "", err
	}
	output, err := RunCommand(dir, "bash", "./"+fileName)
	if err != nil {
		return "", err
	}
	return output, nil
}

// UserInput will ask for user input with a given title
func UserInput(title string) bool {
	fmt.Println(title)

	reader := bufio.NewReader(os.Stdin)
	fmt.Print("([Y]es/[N]o/[A]bort) Enter: ")

	input, err := reader.ReadString('\n')
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	input = strings.TrimSpace(input)
	input = strings.ToLower(input)

	// Check user input
	if input == "yes" || input == "y" {
		fmt.Printf("Continue...\n___\n\n")
		return true
	}

	if input == "a" || input == "abort" {
		os.Exit(0)
	}

	fmt.Println("Stop.")
	return false
}