package repository

import (
	"context"
	"log/slog"
	"time"

	"github.com/Sosokker/todolist-backend/internal/cache"
	"github.com/Sosokker/todolist-backend/internal/domain"
	db "github.com/Sosokker/todolist-backend/internal/repository/sqlc/generated"
	"github.com/google/uuid"
	"github.com/jackc/pgx/v5/pgxpool"
)

// Common arguments for list methods
type ListParams struct {
	Limit  int
	Offset int
}

type UserRepository interface {
	Create(ctx context.Context, user *domain.User) (*domain.User, error)
	GetByID(ctx context.Context, id uuid.UUID) (*domain.User, error)
	GetByEmail(ctx context.Context, email string) (*domain.User, error)
	GetByGoogleID(ctx context.Context, googleID string) (*domain.User, error)
	Update(ctx context.Context, id uuid.UUID, updateData *domain.User) (*domain.User, error)
	Delete(ctx context.Context, id uuid.UUID) error
}

type TagRepository interface {
	Create(ctx context.Context, tag *domain.Tag) (*domain.Tag, error)
	GetByID(ctx context.Context, id, userID uuid.UUID) (*domain.Tag, error)
	GetByIDs(ctx context.Context, ids []uuid.UUID, userID uuid.UUID) ([]domain.Tag, error)
	ListByUser(ctx context.Context, userID uuid.UUID) ([]domain.Tag, error)
	Update(ctx context.Context, id, userID uuid.UUID, updateData *domain.Tag) (*domain.Tag, error)
	Delete(ctx context.Context, id, userID uuid.UUID) error
}

type ListTodosParams struct {
	UserID         uuid.UUID
	Status         *domain.TodoStatus
	TagID          *uuid.UUID
	DeadlineBefore *time.Time
	DeadlineAfter  *time.Time
	ListParams
}

type TodoRepository interface {
	Create(ctx context.Context, todo *domain.Todo) (*domain.Todo, error)
	GetByID(ctx context.Context, id, userID uuid.UUID) (*domain.Todo, error)
	ListByUser(ctx context.Context, params ListTodosParams) ([]domain.Todo, error)
	Update(ctx context.Context, id, userID uuid.UUID, updateData *domain.Todo) (*domain.Todo, error)
	Delete(ctx context.Context, id, userID uuid.UUID) error
	// Tag associations
	AddTag(ctx context.Context, todoID, tagID uuid.UUID) error
	RemoveTag(ctx context.Context, todoID, tagID uuid.UUID) error
	SetTags(ctx context.Context, todoID uuid.UUID, tagIDs []uuid.UUID) error
	GetTags(ctx context.Context, todoID uuid.UUID) ([]domain.Tag, error)
	// Attachment URL management
	UpdateAttachmentURL(ctx context.Context, todoID, userID uuid.UUID, attachmentURL *string) error
}

type SubtaskRepository interface {
	Create(ctx context.Context, subtask *domain.Subtask) (*domain.Subtask, error)
	GetByID(ctx context.Context, id, userID uuid.UUID) (*domain.Subtask, error)
	ListByTodo(ctx context.Context, todoID, userID uuid.UUID) ([]domain.Subtask, error)
	Update(ctx context.Context, id, userID uuid.UUID, updateData *domain.Subtask) (*domain.Subtask, error)
	Delete(ctx context.Context, id, userID uuid.UUID) error
	GetParentTodoID(ctx context.Context, id uuid.UUID) (uuid.UUID, error)
}

// Transactioner interface allows services to run operations within a DB transaction
type Transactioner interface {
	BeginTx(ctx context.Context) (*db.Queries, error)
}

// RepositoryRegistry bundles all repositories together, often useful for dependency injection
type RepositoryRegistry struct {
	UserRepo    UserRepository
	TagRepo     TagRepository
	TodoRepo    TodoRepository
	SubtaskRepo SubtaskRepository
	*db.Queries
	Pool *pgxpool.Pool
}

// NewRepositoryRegistry creates a new registry, now with caching decorators
func NewRepositoryRegistry(pool *pgxpool.Pool, cache cache.Cache, logger *slog.Logger) *RepositoryRegistry {
	queries := db.New(pool)

	pgxUserRepo := NewPgxUserRepository(queries)
	pgxTagRepo := NewPgxTagRepository(queries)
	pgxTodoRepo := NewPgxTodoRepository(queries, pool)
	pgxSubtaskRepo := NewPgxSubtaskRepository(queries)

	cachingTagRepo := NewCachingTagRepository(pgxTagRepo, cache, logger)

	return &RepositoryRegistry{
		UserRepo:    pgxUserRepo,    // Not cached yet in this example
		TagRepo:     cachingTagRepo, // Use the caching decorator
		TodoRepo:    pgxTodoRepo,    // Not cached yet in this example
		SubtaskRepo: pgxSubtaskRepo, // Not cached yet in this example
		Queries:     queries,
		Pool:        pool,
	}
}