package testutil

import (
	"context"

	"tercul/internal/domain"

	"github.com/stretchr/testify/mock"
	"gorm.io/gorm"
)

// MockWorkRepository is a mock implementation of the domain.WorkRepository interface.
type MockWorkRepository struct {
	mock.Mock
}

// Ensure MockWorkRepository implements the interface.
var _ domain.WorkRepository = (*MockWorkRepository)(nil)

// GetByID mocks the GetByID method.
func (m *MockWorkRepository) GetByID(ctx context.Context, id uint) (*domain.Work, error) {
	args := m.Called(ctx, id)
	if args.Get(0) == nil {
		return nil, args.Error(1)
	}
	return args.Get(0).(*domain.Work), args.Error(1)
}

// IsAuthor mocks the IsAuthor method.
func (m *MockWorkRepository) IsAuthor(ctx context.Context, workID uint, authorID uint) (bool, error) {
	args := m.Called(ctx, workID, authorID)
	return args.Bool(0), args.Error(1)
}

// Empty implementations for the rest of the interface to satisfy the compiler.
func (m *MockWorkRepository) Create(ctx context.Context, entity *domain.Work) error {
	args := m.Called(ctx, entity)
	return args.Error(0)
}
func (m *MockWorkRepository) Update(ctx context.Context, entity *domain.Work) error { return nil }
func (m *MockWorkRepository) Delete(ctx context.Context, id uint) error            { return nil }
func (m *MockWorkRepository) List(ctx context.Context, page, pageSize int) (*domain.PaginatedResult[domain.Work], error) {
	return nil, nil
}
func (m *MockWorkRepository) ListAll(ctx context.Context) ([]domain.Work, error)                 { return nil, nil }
func (m *MockWorkRepository) Count(ctx context.Context) (int64, error)                         { return 0, nil }
func (m *MockWorkRepository) Exists(ctx context.Context, id uint) (bool, error)                { return false, nil }
func (m *MockWorkRepository) WithTx(ctx context.Context, fn func(tx *gorm.DB) error) error     { return fn(nil) }
func (m *MockWorkRepository) FindByTitle(ctx context.Context, title string) ([]domain.Work, error) { return nil, nil }
func (m *MockWorkRepository) FindByAuthor(ctx context.Context, authorID uint) ([]domain.Work, error) {
	return nil, nil
}
func (m *MockWorkRepository) FindByCategory(ctx context.Context, categoryID uint) ([]domain.Work, error) {
	return nil, nil
}
func (m *MockWorkRepository) FindByLanguage(ctx context.Context, language string, page, pageSize int) (*domain.PaginatedResult[domain.Work], error) {
	return nil, nil
}
func (m *MockWorkRepository) GetWithTranslations(ctx context.Context, id uint) (*domain.Work, error) {
	return nil, nil
}
func (m *MockWorkRepository) GetWithAssociations(ctx context.Context, id uint) (*domain.Work, error) {
	return nil, nil
}
func (m *MockWorkRepository) GetWithAssociationsInTx(ctx context.Context, tx *gorm.DB, id uint) (*domain.Work, error) {
	return nil, nil
}
func (m *MockWorkRepository) ListWithTranslations(ctx context.Context, page, pageSize int) (*domain.PaginatedResult[domain.Work], error) {
	return nil, nil
}
func (m *MockWorkRepository) FindWithPreload(ctx context.Context, preloads []string, id uint) (*domain.Work, error) {
	return nil, nil
}
func (m *MockWorkRepository) GetAllForSync(ctx context.Context, batchSize, offset int) ([]domain.Work, error) {
	return nil, nil
}
func (m *MockWorkRepository) CreateInTx(ctx context.Context, tx *gorm.DB, entity *domain.Work) error {
	return nil
}
func (m *MockWorkRepository) GetByIDWithOptions(ctx context.Context, id uint, options *domain.QueryOptions) (*domain.Work, error) {
	args := m.Called(ctx, id, options)
	if args.Get(0) == nil {
		return nil, args.Error(1)
	}
	return args.Get(0).(*domain.Work), args.Error(1)
}
func (m *MockWorkRepository) UpdateInTx(ctx context.Context, tx *gorm.DB, entity *domain.Work) error {
	return nil
}
func (m *MockWorkRepository) DeleteInTx(ctx context.Context, tx *gorm.DB, id uint) error { return nil }
func (m *MockWorkRepository) ListWithOptions(ctx context.Context, options *domain.QueryOptions) ([]domain.Work, error) {
	return nil, nil
}
func (m *MockWorkRepository) CountWithOptions(ctx context.Context, options *domain.QueryOptions) (int64, error) {
	return 0, nil
}
func (m *MockWorkRepository) BeginTx(ctx context.Context) (*gorm.DB, error) { return nil, nil }

func (m *MockWorkRepository) ListByCollectionID(ctx context.Context, collectionID uint) ([]domain.Work, error) {
	args := m.Called(ctx, collectionID)
	if args.Get(0) == nil {
		return nil, args.Error(1)
	}
	return args.Get(0).([]domain.Work), args.Error(1)
}