package main

import (
	"reflect"
	"strings"
	"testing"
)

var gEvalTests = []struct {
	inp   string
	toks  []string
	exprs []expr
}{
	{
		"",
		[]string{},
		nil,
	},

	{
		"# comments start with '#'",
		[]string{},
		nil,
	},

	{
		"echo hello",
		[]string{"echo", "hello", "\n"},
		[]expr{&callExpr{"echo", []string{"hello"}, 1}},
	},

	{
		"echo hello world",
		[]string{"echo", "hello", "world", "\n"},
		[]expr{&callExpr{"echo", []string{"hello", "world"}, 1}},
	},

	{
		"echo 'hello world'",
		[]string{"echo", "hello world", "\n"},
		[]expr{&callExpr{"echo", []string{"hello world"}, 1}},
	},

	{
		`echo "hello world"`,
		[]string{"echo", "hello world", "\n"},
		[]expr{&callExpr{"echo", []string{"hello world"}, 1}},
	},

	{
		`echo "hello\"world"`,
		[]string{"echo", `hello"world`, "\n"},
		[]expr{&callExpr{"echo", []string{`hello"world`}, 1}},
	},

	{
		`echo "hello\tworld"`,
		[]string{"echo", "hello\tworld", "\n"},
		[]expr{&callExpr{"echo", []string{"hello\tworld"}, 1}},
	},

	{
		`echo "hello\nworld"`,
		[]string{"echo", "hello\nworld", "\n"},
		[]expr{&callExpr{"echo", []string{"hello\nworld"}, 1}},
	},

	{
		`echo "hello\zworld"`,
		[]string{"echo", `hello\zworld`, "\n"},
		[]expr{&callExpr{"echo", []string{`hello\zworld`}, 1}},
	},

	{
		`echo "hello\0world"`,
		[]string{"echo", "hello\000world", "\n"},
		[]expr{&callExpr{"echo", []string{"hello\000world"}, 1}},
	},

	{
		`echo "hello\101world"`,
		[]string{"echo", "hello\101world", "\n"},
		[]expr{&callExpr{"echo", []string{"hello\101world"}, 1}},
	},

	{
		`echo hello\ world`,
		[]string{"echo", "hello world", "\n"},
		[]expr{&callExpr{"echo", []string{"hello world"}, 1}},
	},

	{
		"echo hello\\\tworld",
		[]string{"echo", "hello\tworld", "\n"},
		[]expr{&callExpr{"echo", []string{"hello\tworld"}, 1}},
	},

	{
		"echo hello\\\nworld",
		[]string{"echo", "hello\nworld", "\n"},
		[]expr{&callExpr{"echo", []string{"hello\nworld"}, 1}},
	},

	{
		`echo hello\\world`,
		[]string{"echo", `hello\world`, "\n"},
		[]expr{&callExpr{"echo", []string{`hello\world`}, 1}},
	},

	{
		`echo hello\zworld`,
		[]string{"echo", "hellozworld", "\n"},
		[]expr{&callExpr{"echo", []string{"hellozworld"}, 1}},
	},

	{
		"set hidden # trailing comments are allowed",
		[]string{"set", "hidden", "\n"},
		[]expr{&setExpr{"hidden", ""}},
	},

	{
		"set hidden; set preview",
		[]string{"set", "hidden", ";", "set", "preview", "\n"},
		[]expr{&setExpr{"hidden", ""}, &setExpr{"preview", ""}},
	},

	{
		"set hidden\nset preview",
		[]string{"set", "hidden", "\n", "set", "preview", "\n"},
		[]expr{&setExpr{"hidden", ""}, &setExpr{"preview", ""}},
	},

	{
		`set ifs ""`,
		[]string{"set", "ifs", "", "\n"},
		[]expr{&setExpr{"ifs", ""}},
	},

	{
		`set ifs "\n"`,
		[]string{"set", "ifs", "\n", "\n"},
		[]expr{&setExpr{"ifs", "\n"}},
	},

	{
		"set ratios 1:2:3",
		[]string{"set", "ratios", "1:2:3", "\n"},
		[]expr{&setExpr{"ratios", "1:2:3"}},
	},

	{
		"set ratios 1:2:3;",
		[]string{"set", "ratios", "1:2:3", ";"},
		[]expr{&setExpr{"ratios", "1:2:3"}},
	},

	{
		":set ratios 1:2:3",
		[]string{":", "set", "ratios", "1:2:3", "\n", "\n"},
		[]expr{&listExpr{[]expr{&setExpr{"ratios", "1:2:3"}}, 1}},
	},

	{
		":set ratios 1:2:3\nset hidden",
		[]string{":", "set", "ratios", "1:2:3", "\n", "\n", "set", "hidden", "\n"},
		[]expr{&listExpr{[]expr{&setExpr{"ratios", "1:2:3"}}, 1}, &setExpr{"hidden", ""}},
	},

	{
		":set ratios 1:2:3;",
		[]string{":", "set", "ratios", "1:2:3", ";", "\n"},
		[]expr{&listExpr{[]expr{&setExpr{"ratios", "1:2:3"}}, 1}},
	},

	{
		":set ratios 1:2:3;\nset hidden",
		[]string{":", "set", "ratios", "1:2:3", ";", "\n", "set", "hidden", "\n"},
		[]expr{&listExpr{[]expr{&setExpr{"ratios", "1:2:3"}}, 1}, &setExpr{"hidden", ""}},
	},

	{
		"map gh cd ~",
		[]string{"map", "gh", "cd", "~", "\n"},
		[]expr{&mapExpr{"gh", &callExpr{"cd", []string{"~"}, 1}}},
	},

	{
		"map gh cd ~;",
		[]string{"map", "gh", "cd", "~", ";"},
		[]expr{&mapExpr{"gh", &callExpr{"cd", []string{"~"}, 1}}},
	},

	{
		"map gh :cd ~",
		[]string{"map", "gh", ":", "cd", "~", "\n", "\n"},
		[]expr{&mapExpr{"gh", &listExpr{[]expr{&callExpr{"cd", []string{"~"}, 1}}, 1}}},
	},

	{
		"map gh :cd ~;",
		[]string{"map", "gh", ":", "cd", "~", ";", "\n"},
		[]expr{&mapExpr{"gh", &listExpr{[]expr{&callExpr{"cd", []string{"~"}, 1}}, 1}}},
	},

	{
		"cmap <c-g> cmd-escape",
		[]string{"cmap", "<c-g>", "cmd-escape", "\n"},
		[]expr{&cmapExpr{"<c-g>", "cmd-escape"}},
	},

	{
		"cmd usage $du -h . | less",
		[]string{"cmd", "usage", "$", "du -h . | less", "\n"},
		[]expr{&cmdExpr{"usage", &execExpr{"$", "du -h . | less"}}},
	},

	{
		"cmd 世界 $echo 世界",
		[]string{"cmd", "世界", "$", "echo 世界", "\n"},
		[]expr{&cmdExpr{"世界", &execExpr{"$", "echo 世界"}}},
	},

	{
		"map u usage",
		[]string{"map", "u", "usage", "\n"},
		[]expr{&mapExpr{"u", &callExpr{"usage", nil, 1}}},
	},

	{
		"map u usage;",
		[]string{"map", "u", "usage", ";"},
		[]expr{&mapExpr{"u", &callExpr{"usage", nil, 1}}},
	},

	{
		"map u :usage",
		[]string{"map", "u", ":", "usage", "\n", "\n"},
		[]expr{&mapExpr{"u", &listExpr{[]expr{&callExpr{"usage", nil, 1}}, 1}}},
	},

	{
		"map u :usage;",
		[]string{"map", "u", ":", "usage", ";", "\n"},
		[]expr{&mapExpr{"u", &listExpr{[]expr{&callExpr{"usage", nil, 1}}, 1}}},
	},

	{
		"map r push :rename<space>",
		[]string{"map", "r", "push", ":rename<space>", "\n"},
		[]expr{&mapExpr{"r", &callExpr{"push", []string{":rename<space>"}, 1}}},
	},

	{
		"map r push :rename<space>;",
		[]string{"map", "r", "push", ":rename<space>;", "\n"},
		[]expr{&mapExpr{"r", &callExpr{"push", []string{":rename<space>;"}, 1}}},
	},

	{
		"map r push :rename<space> # trailing comments are allowed after a space",
		[]string{"map", "r", "push", ":rename<space>", "\n"},
		[]expr{&mapExpr{"r", &callExpr{"push", []string{":rename<space>"}, 1}}},
	},

	{
		"map r :push :rename<space>",
		[]string{"map", "r", ":", "push", ":rename<space>", "\n", "\n"},
		[]expr{&mapExpr{"r", &listExpr{[]expr{&callExpr{"push", []string{":rename<space>"}, 1}}, 1}}},
	},

	{
		"map r :push :rename<space> ; set hidden",
		[]string{"map", "r", ":", "push", ":rename<space>", ";", "set", "hidden", "\n", "\n"},
		[]expr{&mapExpr{"r", &listExpr{[]expr{&callExpr{"push", []string{":rename<space>"}, 1}, &setExpr{"hidden", ""}}, 1}}},
	},

	{
		"map u $du -h . | less",
		[]string{"map", "u", "$", "du -h . | less", "\n"},
		[]expr{&mapExpr{"u", &execExpr{"$", "du -h . | less"}}},
	},

	{
		"cmd usage $du -h $1 | less",
		[]string{"cmd", "usage", "$", "du -h $1 | less", "\n"},
		[]expr{&cmdExpr{"usage", &execExpr{"$", "du -h $1 | less"}}},
	},

	{
		"map u usage /",
		[]string{"map", "u", "usage", "/", "\n"},
		[]expr{&mapExpr{"u", &callExpr{"usage", []string{"/"}, 1}}},
	},

	{
		"map ss :set sortby size; set info size",
		[]string{"map", "ss", ":", "set", "sortby", "size", ";", "set", "info", "size", "\n", "\n"},
		[]expr{&mapExpr{"ss", &listExpr{[]expr{&setExpr{"sortby", "size"}, &setExpr{"info", "size"}}, 1}}},
	},

	{
		"map ss :set sortby size; set info size;",
		[]string{"map", "ss", ":", "set", "sortby", "size", ";", "set", "info", "size", ";", "\n"},
		[]expr{&mapExpr{"ss", &listExpr{[]expr{&setExpr{"sortby", "size"}, &setExpr{"info", "size"}}, 1}}},
	},

	{
		`cmd gohome :{{
			cd ~
			set hidden
		}}`,
		[]string{
			"cmd", "gohome", ":", "{{",
			"cd", "~", "\n",
			"set", "hidden", "\n",
			"}}", "\n",
		},
		[]expr{&cmdExpr{
			"gohome", &listExpr{[]expr{
				&callExpr{"cd", []string{"~"}, 1},
				&setExpr{"hidden", ""},
			}, 1},
		}},
	},

	{
		`map gh :{{
			cd ~
			set hidden
		}}`,
		[]string{
			"map", "gh", ":", "{{",
			"cd", "~", "\n",
			"set", "hidden", "\n",
			"}}", "\n",
		},
		[]expr{&mapExpr{
			"gh", &listExpr{[]expr{
				&callExpr{"cd", []string{"~"}, 1},
				&setExpr{"hidden", ""},
			}, 1},
		}},
	},

	{
		`map c ${{
			mkdir foo
			cp $fs foo
			tar -czvf foo.tar.gz foo
			rm -rf foo
		}}`,
		[]string{"map", "c", "$", "{{", `
			mkdir foo
			cp $fs foo
			tar -czvf foo.tar.gz foo
			rm -rf foo
		`, "}}", "\n"},
		[]expr{&mapExpr{"c", &execExpr{"$", `
			mkdir foo
			cp $fs foo
			tar -czvf foo.tar.gz foo
			rm -rf foo
		`}}},
	},

	{
		`cmd compress ${{
			mkdir $1
			cp $fs $1
			tar -czvf $1.tar.gz $1
			rm -rf $1
		}}`,
		[]string{"cmd", "compress", "$", "{{", `
			mkdir $1
			cp $fs $1
			tar -czvf $1.tar.gz $1
			rm -rf $1
		`, "}}", "\n"},
		[]expr{&cmdExpr{"compress", &execExpr{"$", `
			mkdir $1
			cp $fs $1
			tar -czvf $1.tar.gz $1
			rm -rf $1
		`}}},
	},
}

func TestScan(t *testing.T) {
	for _, test := range gEvalTests {
		s := newScanner(strings.NewReader(test.inp))

		for _, tok := range test.toks {
			if s.scan(); s.tok != tok {
				t.Errorf("at input '%s' expected '%s' but scanned '%s'", test.inp, tok, s.tok)
			}
		}

		if s.scan() {
			t.Errorf("at input '%s' unexpected '%s'", test.inp, s.tok)
		}
	}
}

func TestParse(t *testing.T) {
	for _, test := range gEvalTests {
		p := newParser(strings.NewReader(test.inp))

		for _, expr := range test.exprs {
			if p.parse(); !reflect.DeepEqual(p.expr, expr) {
				t.Errorf("at input '%s' expected '%s' but parsed '%s'", test.inp, expr, p.expr)
			}
		}

		if p.parse(); p.expr != nil {
			t.Errorf("at input '%s' unexpected '%s'", test.inp, p.expr)
		}
	}
}

func TestSplitKeys(t *testing.T) {
	inps := []struct {
		s    string
		keys []string
	}{
		{"", nil},
		{"j", []string{"j"}},
		{"jk", []string{"j", "k"}},
		{"1j", []string{"1", "j"}},
		{"42j", []string{"4", "2", "j"}},
		{"<space>", []string{"<space>"}},
		{"j<space>", []string{"j", "<space>"}},
		{"j<space>k", []string{"j", "<space>", "k"}},
		{"1j<space>k", []string{"1", "j", "<space>", "k"}},
		{"1j<space>1k", []string{"1", "j", "<space>", "1", "k"}},
		{"<>", []string{"<>"}},
		{"<space", []string{"<space"}},
		{"<space<", []string{"<space<"}},
		{"<space<>", []string{"<space<>"}},
		{"><space>", []string{">", "<space>"}},
		{"><space>>", []string{">", "<space>", ">"}},
	}

	for _, inp := range inps {
		if keys := splitKeys(inp.s); !reflect.DeepEqual(keys, inp.keys) {
			t.Errorf("at input '%s' expected '%v' but got '%v'", inp.s, inp.keys, keys)
		}
	}
}