package graphql_test

import (
	"context"
	"tercul/internal/domain"
	"time"

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

// mockAnalyticsService is a mock implementation of the AnalyticsService interface.
type mockAnalyticsService struct {
	mock.Mock
}

func (m *mockAnalyticsService) IncrementWorkCounter(ctx context.Context, workID uint, field string, value int) error {
	args := m.Called(ctx, workID, field, value)
	return args.Error(0)
}

func (m *mockAnalyticsService) IncrementTranslationCounter(ctx context.Context, translationID uint, field string, value int) error {
	args := m.Called(ctx, translationID, field, value)
	return args.Error(0)
}

func (m *mockAnalyticsService) UpdateWorkStats(ctx context.Context, workID uint, stats domain.WorkStats) error {
	args := m.Called(ctx, workID, stats)
	return args.Error(0)
}

func (m *mockAnalyticsService) UpdateTranslationStats(ctx context.Context, translationID uint, stats domain.TranslationStats) error {
	args := m.Called(ctx, translationID, stats)
	return args.Error(0)
}

func (m *mockAnalyticsService) GetOrCreateWorkStats(ctx context.Context, workID uint) (*domain.WorkStats, error) {
	args := m.Called(ctx, workID)
	if args.Get(0) == nil {
		return nil, args.Error(1)
	}
	return args.Get(0).(*domain.WorkStats), args.Error(1)
}

func (m *mockAnalyticsService) GetOrCreateTranslationStats(ctx context.Context, translationID uint) (*domain.TranslationStats, error) {
	args := m.Called(ctx, translationID)
	if args.Get(0) == nil {
		return nil, args.Error(1)
	}
	return args.Get(0).(*domain.TranslationStats), args.Error(1)
}

func (m *mockAnalyticsService) GetOrCreateUserEngagement(ctx context.Context, userID uint, date time.Time) (*domain.UserEngagement, error) {
	args := m.Called(ctx, userID, date)
	if args.Get(0) == nil {
		return nil, args.Error(1)
	}
	return args.Get(0).(*domain.UserEngagement), args.Error(1)
}

func (m *mockAnalyticsService) UpdateUserEngagement(ctx context.Context, userEngagement *domain.UserEngagement) error {
	args := m.Called(ctx, userEngagement)
	return args.Error(0)
}

func (m *mockAnalyticsService) UpdateTrendingWorks(ctx context.Context, timePeriod string, trending []*domain.Trending) error {
	args := m.Called(ctx, timePeriod, trending)
	return args.Error(0)
}

func (m *mockAnalyticsService) GetTrendingWorks(ctx context.Context, timePeriod string, limit int) ([]*domain.Work, error) {
	args := m.Called(ctx, timePeriod, limit)
	if args.Get(0) == nil {
		return nil, args.Error(1)
	}
	return args.Get(0).([]*domain.Work), args.Error(1)
}