package osutils_test

import (
	"fmt"
	"os"
	"strings"
	"testing"

	"github.com/pocketbase/pocketbase/tools/osutils"
)

func TestYesNoPrompt(t *testing.T) {
	scenarios := []struct {
		stdin    string
		fallback bool
		expected bool
	}{
		{"", false, false},
		{"", true, true},

		// yes
		{"y", false, true},
		{"Y", false, true},
		{"Yes", false, true},
		{"yes", false, true},

		// no
		{"n", true, false},
		{"N", true, false},
		{"No", true, false},
		{"no", true, false},

		// invalid -> no/yes
		{"invalid|no", true, false},
		{"invalid|yes", false, true},
	}

	for _, s := range scenarios {
		t.Run(fmt.Sprintf("%s_%v", s.stdin, s.fallback), func(t *testing.T) {
			stdinread, stdinwrite, err := os.Pipe()
			if err != nil {
				t.Fatal(err)
			}

			parts := strings.Split(s.stdin, "|")
			for _, p := range parts {
				if _, err := stdinwrite.WriteString(p + "\n"); err != nil {
					t.Fatalf("Failed to write test stdin part %q: %v", p, err)
				}
			}

			if err = stdinwrite.Close(); err != nil {
				t.Fatal(err)
			}

			defer func(oldStdin *os.File) { os.Stdin = oldStdin }(os.Stdin)
			os.Stdin = stdinread

			result := osutils.YesNoPrompt("test", s.fallback)

			if result != s.expected {
				t.Fatalf("Expected %v, got %v", s.expected, result)
			}
		})
	}
}

func TestIsProbablyGoRun(t *testing.T) {
	scenarios := []struct {
		arg      string
		envs     map[string]string
		expected bool
	}{
		{"", nil, false},
		{"a/b/c", nil, false},
		{"/a/b/c", nil, false},
		{"/a/b/c", map[string]string{"GOCACHE": "/b/"}, false},
		{"/a/b/c", map[string]string{"GOCACHE": "/a/"}, true},
		{os.TempDir() + "/a/b/c", nil, true},
	}

	originalArgs := os.Args
	defer func() {
		os.Args = originalArgs
	}()

	for _, s := range scenarios {
		t.Run(s.arg, func(t *testing.T) {
			os.Args = []string{s.arg}

			for k, v := range s.envs {
				defer os.Setenv(k, os.Getenv(k))
				os.Setenv(k, v)
			}

			result := osutils.IsProbablyGoRun()

			if result != s.expected {
				t.Fatalf("Expected %v, got %v", s.expected, result)
			}
		})
	}
}