package search

import (
	"context"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
	"tercul/internal/app/localization"
	"tercul/internal/domain"
	domainsearch "tercul/internal/domain/search"
)

type mockLocalizationRepository struct {
	mock.Mock
}

func (m *mockLocalizationRepository) GetTranslation(ctx context.Context, key string, language string) (string, error) {
	args := m.Called(ctx, key, language)
	return args.String(0), args.Error(1)
}

func (m *mockLocalizationRepository) GetTranslations(ctx context.Context, keys []string, language string) (map[string]string, error) {
	args := m.Called(ctx, keys, language)
	if args.Get(0) == nil {
		return nil, args.Error(1)
	}
	return args.Get(0).(map[string]string), args.Error(1)
}

func (m *mockLocalizationRepository) GetAuthorBiography(ctx context.Context, authorID uint, language string) (string, error) {
	args := m.Called(ctx, authorID, language)
	return args.String(0), args.Error(1)
}

func (m *mockLocalizationRepository) GetWorkContent(ctx context.Context, workID uint, language string) (string, error) {
	args := m.Called(ctx, workID, language)
	return args.String(0), args.Error(1)
}

type mockWeaviateWrapper struct {
	mock.Mock
}

func (m *mockWeaviateWrapper) Search(ctx context.Context, params domainsearch.SearchParams) (*domainsearch.SearchResults, error) {
	args := m.Called(ctx, params)
	if args.Get(0) == nil {
		return nil, args.Error(1)
	}
	return args.Get(0).(*domainsearch.SearchResults), args.Error(1)
}

func (m *mockWeaviateWrapper) IndexWork(ctx context.Context, work *domain.Work, content string) error {
	args := m.Called(ctx, work, content)
	return args.Error(0)
}

func TestSearchService_Search(t *testing.T) {
	localizationRepo := new(mockLocalizationRepository)
	localizationService := localization.NewService(localizationRepo)
	weaviateWrapper := new(mockWeaviateWrapper)
	service := NewService(weaviateWrapper, localizationService)

	ctx := context.Background()
	testQuery := "test query"
	testFilters := domainsearch.SearchFilters{
		Languages:  []string{"en"},
		Authors:    []string{"1"},
		Tags:       []string{"test-tag"},
		Categories: []string{"test-category"},
	}
	expectedResults := &domainsearch.SearchResults{
		Results: []domainsearch.SearchResultItem{
			{Type: "Work", Entity: domain.Work{Title: "Test Work"}, Score: 0.9},
			{Type: "Author", Entity: domain.Author{Name: "Test Author"}, Score: 0.8},
		},
		TotalResults: 2,
		Limit:        10,
		Offset:       0,
	}

	params := domainsearch.SearchParams{
		Query:   testQuery,
		Filters: testFilters,
		Limit:   10,
		Offset: 0,
	}
	weaviateWrapper.On("Search", ctx, params).Return(expectedResults, nil)

	results, err := service.Search(ctx, params)

	assert.NoError(t, err)
	assert.Equal(t, expectedResults, results)
	weaviateWrapper.AssertExpectations(t)
}

func TestIndexService_IndexWork(t *testing.T) {
	localizationRepo := new(mockLocalizationRepository)
	localizationService := localization.NewService(localizationRepo)
	weaviateWrapper := new(mockWeaviateWrapper)
	service := NewService(weaviateWrapper, localizationService)

	ctx := context.Background()
	testWork := domain.Work{
		TranslatableModel: domain.TranslatableModel{
			BaseModel: domain.BaseModel{ID: 1},
			Language:  "en",
		},
		Title: "Test Work",
	}
	testContent := "This is the test content for the work."

	// Expect a call to get the work's content.
	localizationRepo.On("GetWorkContent", mock.Anything, testWork.ID, testWork.Language).Return(testContent, nil)

	// Expect a call to the Weaviate wrapper with the fetched content.
	weaviateWrapper.On("IndexWork", mock.Anything, &testWork, testContent).Return(nil)

	err := service.IndexWork(ctx, testWork)

	assert.NoError(t, err)
	localizationRepo.AssertExpectations(t)
	weaviateWrapper.AssertExpectations(t)
}