from typing import Optional, List
from sqlalchemy.orm import Session
from uuid import UUID

from models.book import Book
from schemas.book import BookCreate

def create_book(db: Session, book_data: BookCreate) -> Book:
    """
    Creates a new book in the database.

    Args:
        db (Session): The database session.
        book_data (BookCreate): The data for the book to create.

    Returns:
        Book: The newly created book object.
    """
    db_book = Book(**book_data.dict())
    db.add(db_book)
    db.commit()
    db.refresh(db_book)
    return db_book

def get_book_by_id(db: Session, book_id: UUID) -> Optional[Book]:
    """
    Retrieves a book by its ID.

    Args:
        db (Session): The database session.
        book_id (UUID): The ID of the book to retrieve.

    Returns:
        Optional[Book]: The book object if found, otherwise None.
    """
    return db.query(Book).filter(Book.id == book_id).first()

def get_books_by_author(db: Session, author: str, skip: int = 0, limit: int = 100) -> List[Book]:
    """
    Retrieves books by a specific author.

    Args:
        db (Session): The database session.
        author (str): The author name to filter by.
        skip (int, optional): Number of records to skip for pagination. Defaults to 0.
        limit (int, optional): Maximum number of records to return. Defaults to 100.

    Returns:
        List[Book]: A list of books by the specified author.
    """
    return db.query(Book).filter(Book.author == author).offset(skip).limit(limit).all()

def get_books_by_title(db: Session, title: str, skip: int = 0, limit: int = 100) -> List[Book]:
    """
    Retrieves books by a specific title or partial title match.

    Args:
        db (Session): The database session.
        title (str): The title or partial title to search for.
        skip (int, optional): Number of records to skip for pagination. Defaults to 0.
        limit (int, optional): Maximum number of records to return. Defaults to 100.

    Returns:
        List[Book]: A list of books matching the title search.
    """
    return db.query(Book).filter(Book.title.ilike(f"%{title}%")).offset(skip).limit(limit).all()

def get_all_books(db: Session, skip: int = 0, limit: int = 100) -> List[Book]:
    """
    Retrieves all books with pagination.

    Args:
        db (Session): The database session.
        skip (int, optional): Number of records to skip for pagination. Defaults to 0.
        limit (int, optional): Maximum number of records to return. Defaults to 100.

    Returns:
        List[Book]: A list of book objects.
    """
    return db.query(Book).offset(skip).limit(limit).all()