package spec

import (
	"os"
	"path/filepath"
	"strings"
	"testing"
)

func TestFileGenerator(t *testing.T) {
	// Setup mock files
	tmp := t.TempDir()
	_ = os.MkdirAll(filepath.Join(tmp, "src"), 0755)
	_ = os.WriteFile(filepath.Join(tmp, "main.go"), []byte(""), 0644)
	_ = os.WriteFile(filepath.Join(tmp, "README.md"), []byte(""), 0644)
	_ = os.WriteFile(filepath.Join(tmp, ".hidden"), []byte(""), 0644)
	_ = os.WriteFile(filepath.Join(tmp, "src/utils.go"), []byte(""), 0644)

	oldWd, _ := os.Getwd()
	_ = os.Chdir(tmp)
	defer func() { _ = os.Chdir(oldWd) }()


	t.Run("dirOnly shows only dirs", func(t *testing.T) {
		gen := FileGenerator("/")
		results := gen([]string{"cd", ""}, "cd ", "")
		for _, r := range results {
			if !strings.HasSuffix(r.Cmd, "/") {
				t.Errorf("Expected only directories, got %q", r.Cmd)
			}
		}
	})


	t.Run("Filter extension", func(t *testing.T) {
		gen := FileGenerator(".go")
		results := gen([]string{"ls", ""}, "ls ", "")
		foundMain := false
		for _, r := range results {
			if r.Cmd == "main.go" {
				foundMain = true
			}
			if r.Cmd == "README.md" {
				t.Errorf("Did not expect README.md when filtering for .go")
			}
		}
		if !foundMain {
			t.Errorf("FileGenerator(\".go\") did not suggest main.go")
		}
	})


	t.Run("Nested path", func(t *testing.T) {
		gen := FileGenerator()
		results := gen([]string{"ls", "src/u"}, "ls src/u", "src/u")
		foundUtils := false
		for _, r := range results {
			if r.Cmd == "src/utils.go" {
				foundUtils = true
			}
		}
		if !foundUtils {
			t.Errorf("Did not find src/utils.go for nested path src/u")
		}
	})


	t.Run("Deep scan 1 level", func(t *testing.T) {
		gen := FileGenerator()
		results := gen([]string{"ls", "src/"}, "ls src/", "src/")
		foundUtils := false
		for _, r := range results {
			if r.Cmd == "src/utils.go" {
				foundUtils = true
			}
		}
		if !foundUtils {
			t.Errorf("Deep scan did not find src/utils.go")
		}
	})




	t.Run("Hidden files are skipped", func(t *testing.T) {
		gen := FileGenerator()
		results := gen([]string{"ls", ""}, "ls ", "")
		for _, r := range results {
			if strings.HasPrefix(r.Cmd, ".") {
				t.Errorf("Hidden file %q should be skipped", r.Cmd)
			}
		}
	})
}