package cui

import (
	"strings"
	"bytes"
	"fmt"
	"bufio"
	"os"
	"os/exec"
)

var shapes = map[string]string{
	"wall": "│",
	"ceiling": "─",
	"tl": "┌",
	"tr": "┐",
	"bl": "└",
	"br": "┘",
	"scroll-thumb": "▉",
	"scroll-track": "░",
}

func drawShape(shape string) {
	if val, ok := shapes[shape]; ok {
		fmt.Printf("%s", val)
	} else {
		fmt.Print("x")
	}
}

func moveThenDrawShape(r, c int, s string) {
	MoveCursorTo(r, c)
	drawShape(s)
}

func MoveCursorTo(row, col int) {
	fmt.Printf("\033[%d;%dH", row, col)
}

func moveCursorToward(dir string, amount int) {
	directions := map[string]string{
		"up": "A",
		"down": "B",
		"left": "D",
		"right": "C",
	}

	if val, ok := directions[dir]; ok {
		fmt.Printf("\033[%d%s", amount, val)
	}
}

func Exit() {
	moveCursorToward("down", 500)
	moveCursorToward("right", 500)
	SetLineMode()
	fmt.Print("\n")
	fmt.Print("\033[?25h")
	os.Exit(0)
}

func Clear(dir string) {
	directions := map[string]string{
		"up": "\033[1J",
		"down": "\033[0J",
		"left": "\033[1K",
		"right": "\033[0K",
		"line": "\033[2K",
		"screen": "\033[2J",
	}

	if val, ok := directions[dir]; ok {
		fmt.Print(val)
	}

}

func WrapLines(s []string, length int) []string {
	out := []string{}
	for _, ln := range s {
		if len(ln) <= length {
			out = append(out, ln)
		} else {
			words := strings.Split(ln, " ")
			var subout bytes.Buffer
			for i, wd := range words {
				sublen := subout.Len()
				if sublen + len(wd) + 1 <= length {
					if sublen > 0 {
						subout.WriteString(" ")
					}
					subout.WriteString(wd)	
					if i == len(words) - 1 {
						out = append(out, subout.String())
					}
				} else {
						out = append(out, subout.String())
						subout.Reset()
						subout.WriteString(wd)
						if i == len(words) - 1 {
							out = append(out, subout.String())
							subout.Reset()
						}
				}
			}
		}
	}
	return out
}

func Getch() rune {
	reader := bufio.NewReader(os.Stdin)
	char, _, err := reader.ReadRune()
	if err != nil {
		return '@'
	}
	return char
}

func GetLine() string {
	SetLineMode()
	reader := bufio.NewReader(os.Stdin)
	fmt.Print(": ")
	text, _ := reader.ReadString('\n')
	SetCharMode()
	return text[:len(text)-1]
}

func SetCharMode() {
	cmd := exec.Command("stty", "cbreak", "-echo")
	cmd.Stdin = os.Stdin
	cmd.Stdout = os.Stdout
	cmd.Run()
	fmt.Print("\033[?25l")
}

func SetLineMode() {
	cmd := exec.Command("stty", "-cbreak", "echo")
	cmd.Stdin = os.Stdin
	cmd.Stdout = os.Stdout
	cmd.Run()
}