package cli

import (
	"os"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestPrompt_Good(t *testing.T) {
	SetStdin(strings.NewReader("hello\n"))
	defer SetStdin(nil) // reset

	val, err := Prompt("Name", "")
	assert.NoError(t, err)
	assert.Equal(t, "hello", val)
}

func TestPrompt_Good_Default(t *testing.T) {
	SetStdin(strings.NewReader("\n"))
	defer SetStdin(nil)

	val, err := Prompt("Name", "world")
	assert.NoError(t, err)
	assert.Equal(t, "world", val)
}

func TestSelect_Good(t *testing.T) {
	SetStdin(strings.NewReader("2\n"))
	defer SetStdin(nil)

	val, err := Select("Pick", []string{"a", "b", "c"})
	assert.NoError(t, err)
	assert.Equal(t, "b", val)
}

func TestSelect_Bad_Invalid(t *testing.T) {
	SetStdin(strings.NewReader("5\n"))
	defer SetStdin(nil)

	_, err := Select("Pick", []string{"a", "b"})
	assert.Error(t, err)
}

func TestMultiSelect_Good(t *testing.T) {
	SetStdin(strings.NewReader("1 3\n"))
	defer SetStdin(nil)

	vals, err := MultiSelect("Pick", []string{"a", "b", "c"})
	assert.NoError(t, err)
	assert.Equal(t, []string{"a", "c"}, vals)
}

func TestMultiSelect_Good_CommasAndRanges(t *testing.T) {
	SetStdin(strings.NewReader("1-2,4\n"))
	defer SetStdin(nil)

	vals, err := MultiSelect("Pick", []string{"a", "b", "c", "d"})
	assert.NoError(t, err)
	assert.Equal(t, []string{"a", "b", "d"}, vals)
}

func TestConfirm_Good(t *testing.T) {
	SetStdin(strings.NewReader("y\n"))
	defer SetStdin(nil)

	assert.True(t, Confirm("Proceed?"))
}

func TestConfirm_Bad_EOFUsesDefault(t *testing.T) {
	SetStdin(strings.NewReader(""))
	defer SetStdin(nil)

	assert.False(t, Confirm("Proceed?", Required()))
	assert.True(t, Confirm("Proceed?", DefaultYes(), Required()))
}

func TestQuestion_Good(t *testing.T) {
	SetStdin(strings.NewReader("alice\n"))
	defer SetStdin(nil)

	val := Question("Name:")
	assert.Equal(t, "alice", val)
}

func TestQuestion_Bad_EOFReturnsDefault(t *testing.T) {
	SetStdin(strings.NewReader(""))
	defer SetStdin(nil)

	assert.Equal(t, "anonymous", Question("Name:", WithDefault("anonymous")))
	assert.Equal(t, "", Question("Name:", RequiredInput()))
}

func TestChoose_Good_DefaultIndex(t *testing.T) {
	SetStdin(strings.NewReader("\n"))
	defer SetStdin(nil)

	val := Choose("Pick", []string{"a", "b", "c"}, WithDefaultIndex[string](1))
	assert.Equal(t, "b", val)
}

func TestChoose_Good_Filter(t *testing.T) {
	SetStdin(strings.NewReader("ap\n2\n"))
	defer SetStdin(nil)

	val := Choose("Pick", []string{"apple", "apricot", "banana"}, Filter[string]())
	assert.Equal(t, "apricot", val)
}

func TestChooseMulti_Good_Filter(t *testing.T) {
	SetStdin(strings.NewReader("ap\n1 2\n"))
	defer SetStdin(nil)

	vals := ChooseMulti("Pick", []string{"apple", "apricot", "banana"}, Filter[string]())
	assert.Equal(t, []string{"apple", "apricot"}, vals)
}

func TestChooseMulti_Good_Commas(t *testing.T) {
	SetStdin(strings.NewReader("1,3\n"))
	defer SetStdin(nil)

	vals := ChooseMulti("Pick", []string{"a", "b", "c"})
	assert.Equal(t, []string{"a", "c"}, vals)
}

func TestChooseMulti_Good_CommasAndRanges(t *testing.T) {
	SetStdin(strings.NewReader("1-2,4\n"))
	defer SetStdin(nil)

	vals := ChooseMulti("Pick", []string{"a", "b", "c", "d"})
	assert.Equal(t, []string{"a", "b", "d"}, vals)
}

func TestSetStdin_Good_ResetNil(t *testing.T) {
	original := stdin
	t.Cleanup(func() { stdin = original })

	SetStdin(strings.NewReader("hello\n"))
	assert.NotSame(t, os.Stdin, stdin)

	SetStdin(nil)
	assert.Same(t, os.Stdin, stdin)
}