package termui

import (
	"strings"
	"testing"
)

func TestBreakBlocksIntoStrings(t *testing.T) {
	items := breakBlocksIntoStrings("test [blue](fg:blue,mod:bold) and [red](fg:red) and maybe even [foo](bg:red)!")
	if len(items) != 7 {
		t.Fatal("wrong length", len(items))
	}
}

func TestFindStylePositions(t *testing.T) {
	items := findStylePositions("test [blue](fg:blue,mod:bold) and [red](fg:red) and maybe even [foo](bg:red)!")
	if len(items) != 3 {
		t.Fatal("wrong length", len(items))
	}
	if items[0] != 10 {
		t.Fatal("wrong index", items[0])
	}
	if items[1] != 38 {
		t.Fatal("wrong index", items[1])
	}
	if items[2] != 67 {
		t.Fatal("wrong index", items[2])
	}
}

func TestFindStyleBlocks(t *testing.T) {
	items := findStyleBlocks("test [blue](fg:blue,mod:bold) and [red](fg:red) and maybe even [foo](bg:red)!")
	if len(items) != 3 {
		t.Fatal("wrong length", len(items))
	}
	if items[0].Start != 5 && items[0].End != 28 {
		t.Fatal("wrong index", items[0])
	}
	if items[1].Start != 34 && items[1].End != 46 {
		t.Fatal("wrong index", items[1])
	}
	if items[2].Start != 63 && items[2].End != 75 {
		t.Fatal("wrong index", items[2])
	}
}

func TestParseStyles(t *testing.T) {
	cells := ParseStyles("test nothing", NewStyle(ColorWhite))
	cells = ParseStyles("test [blue](fg:blue,bg:white,mod:bold) and [red](fg:red)", NewStyle(ColorWhite))
	if len(cells) != 17 {
		t.Fatal("wrong length", len(cells))
	}
	for i := 0; i < 5; i++ {
		if cells[i].Style.Fg != ColorWhite {
			t.Fatal("wrong fg color", cells[i], i)
		}
		if cells[i].Style.Bg != ColorClear {
			t.Fatal("wrong bg color", cells[i])
		}
		if cells[i].Style.Modifier != ModifierClear {
			t.Fatal("wrong mod", cells[i])
		}
	}
	for i := 5; i < 9; i++ {
		if cells[i].Style.Fg != ColorBlue {
			t.Fatal("wrong fg color", cells[i])
		}
		if cells[i].Style.Bg != ColorWhite {
			t.Fatal("wrong bg color", cells[i])
		}
		if cells[i].Style.Modifier != ModifierBold {
			t.Fatal("wrong mod", cells[i])
		}
	}

	text := textFromCells(cells)
	if text != "test blue and red" {
		t.Fatal("wrong text", text)
	}

	cells = ParseStyles("[blue](fg:blue) [1]", NewStyle(ColorWhite))
	text = textFromCells(cells)
	if text != "blue [1]" {
		t.Fatal("wrong text", text)
	}

	cells = ParseStyles("[0]", NewStyle(ColorWhite))
	text = textFromCells(cells)
	if text != "[0]" {
		t.Fatal("wrong text", text)
	}

	cells = ParseStyles("[", NewStyle(ColorWhite))
	text = textFromCells(cells)
	if text != "[" {
		t.Fatal("wrong text", text)
	}

}

func textFromCells(cells []Cell) string {
	buff := []string{}
	for _, cell := range cells {
		buff = append(buff, string(cell.Rune))
	}
	return strings.Join(buff, "")
}