package files

import (
	"reflect"
	"testing"

	"github.com/gtsteffaniak/filebrowser/backend/settings"
	"github.com/stretchr/testify/assert"
)

func BenchmarkSearchAllIndexes(b *testing.B) {
	Initialize(settings.Source{Name: "test", Path: "/srv"})
	idx := GetIndex("test")

	idx.createMockData(50, 3) // 50 dirs, 3 files per dir

	// Generate 100 random search terms
	searchTerms := generateRandomSearchTerms(100)

	b.ResetTimer()
	b.ReportAllocs()
	for i := 0; i < b.N; i++ {
		// Execute the SearchAllIndexes function
		for _, term := range searchTerms {
			idx.Search(term, "/", "test")
		}
	}
}

func TestParseSearch(t *testing.T) {
	tests := []struct {
		input string
		want  SearchOptions
	}{
		{
			input: "my test search",
			want: SearchOptions{
				Conditions: map[string]bool{"exact": false},
				Terms:      []string{"my test search"},
			},
		},
		{
			input: "case:exact my|test|search",
			want: SearchOptions{
				Conditions: map[string]bool{"exact": true},
				Terms:      []string{"my", "test", "search"},
			},
		},
		{
			input: "type:largerThan=100 type:smallerThan=1000 test",
			want: SearchOptions{
				Conditions:  map[string]bool{"exact": false, "larger": true, "smaller": true},
				Terms:       []string{"test"},
				LargerThan:  100,
				SmallerThan: 1000,
			},
		},
		{
			input: "type:audio thisfile",
			want: SearchOptions{
				Conditions: map[string]bool{"exact": false, "audio": true},
				Terms:      []string{"thisfile"},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.input, func(t *testing.T) {
			value := ParseSearch(tt.input)
			if !reflect.DeepEqual(value, tt.want) {
				t.Fatalf("\n got:  %+v\n want: %+v", value, tt.want)
			}
		})
	}
}

func TestSearchWhileIndexing(t *testing.T) {
	Initialize(settings.Source{Name: "test", Path: "/srv"})
	idx := GetIndex("test")

	searchTerms := generateRandomSearchTerms(10)
	for i := 0; i < 5; i++ {
		go idx.createMockData(100, 100) // Creating mock data concurrently
		for _, term := range searchTerms {
			go idx.Search(term, "/", "test") // Search concurrently
		}
	}
}

func TestSearchIndexes(t *testing.T) {
	index := Index{
		Directories: map[string]*FileInfo{
			"/test":      {Files: []ItemInfo{{Name: "audio1.wav", Type: "audio"}}},
			"/test/path": {Files: []ItemInfo{{Name: "file.txt", Type: "text"}}},
			"/new/test": {Files: []ItemInfo{
				{Name: "audio.wav", Type: "audio"},
				{Name: "video.mp4", Type: "video"},
				{Name: "video.MP4", Type: "video"},
			}},
			"/new/test/path": {Files: []ItemInfo{{Name: "archive.zip", Type: "archive"}}},
			"/firstDir": {
				Files: []ItemInfo{
					{Name: "archive.zip", Size: 100, Type: "archive"},
				},
				Folders: []ItemInfo{
					{Name: "thisIsDir", Type: "directory", Size: 2 * 1024 * 1024},
				},
			},
			"/firstDir/thisIsDir": {
				Files: []ItemInfo{
					{Name: "hi.txt", Type: "text"},
				},
				ItemInfo: ItemInfo{
					Size: 2 * 1024 * 1024,
				},
			},
		},
	}

	tests := []struct {
		search         string
		scope          string
		expectedResult []SearchResult
	}{
		{
			search: "audio",
			scope:  "/new/",
			expectedResult: []SearchResult{
				{
					Path: "test/audio.wav",
					Type: "audio",
					Size: 0,
				},
			},
		},
		{
			search: "test",
			scope:  "/",
			expectedResult: []SearchResult{
				{
					Path: "test/",
					Type: "directory",
					Size: 0,
				},
				{
					Path: "new/test/",
					Type: "directory",
					Size: 0,
				},
			},
		},
		{
			search: "archive",
			scope:  "/",
			expectedResult: []SearchResult{
				{
					Path: "firstDir/archive.zip",
					Type: "archive",
					Size: 100,
				},
				{
					Path: "new/test/path/archive.zip",
					Type: "archive",
					Size: 0,
				},
			},
		},
		{
			search: "arch",
			scope:  "/firstDir",
			expectedResult: []SearchResult{
				{
					Path: "archive.zip",
					Type: "archive",
					Size: 100,
				},
			},
		},
		{
			search: "isdir",
			scope:  "/",
			expectedResult: []SearchResult{
				{
					Path: "firstDir/thisIsDir/",
					Type: "directory",
					Size: 2097152,
				},
			},
		},
		{
			search: "IsDir type:largerThan=1",
			scope:  "/",
			expectedResult: []SearchResult{
				{
					Path: "firstDir/thisIsDir/",
					Type: "directory",
					Size: 2097152,
				},
			},
		},
		{
			search: "video",
			scope:  "/",
			expectedResult: []SearchResult{
				{
					Path: "new/test/video.MP4",
					Type: "video",
					Size: 0,
				},
				{
					Path: "new/test/video.mp4",
					Type: "video",
					Size: 0,
				},
			},
		},
	}

	for _, tt := range tests {
		t.Run(tt.search, func(t *testing.T) {
			result := index.Search(tt.search, tt.scope, "")
			assert.ElementsMatch(t, tt.expectedResult, result)
		})
	}
}