import logging
from pathlib import Path

from PIL import Image, UnidentifiedImageError
from rich.progress import track

from photoalbum.config import Config

logger = logging.getLogger(__name__)


def generate(config: Config, album_path: Path) -> None:
    """
    Main generation function
    """
    skel_files_created, skel_files = maybe_create_skeleton(config, album_path)
    generate_thumbnails(config, album_path)
    # generate_html(config, album_path)

    if skel_files_created:
        print(
            "Some basic files have been created for your album. Edit them as you need:"
        )
        for p in skel_files:
            print(f" - {p}")


def maybe_create_skeleton(config: Config, album_path: Path) -> tuple[bool, list[Path]]:
    """
    Create basic config, template, and static files if they don't already exist in the
    repo
    """
    files_created = False
    skel_files = []

    skel_dir = Path(__file__).parent / "skel"
    logger.debug(f"Skeleton dir: {skel_dir}")

    for parent_path, dirnames, filenames in skel_dir.walk():
        for filename in filenames:
            skel_file_path = parent_path / filename
            rel_path = skel_file_path.relative_to(skel_dir)
            album_file_path = album_path / rel_path

            skel_files.append(album_file_path)

            if not album_file_path.exists():
                album_file_path.parent.mkdir(exist_ok=True)
                album_file_path.write_bytes(skel_file_path.read_bytes())
                logger.debug(f"Created skeleton file {album_file_path}")
                files_created = True

    return files_created, skel_files


def generate_thumbnails(config: Config, path: Path) -> None:
    """
    Find all of the images and generate thumbnails and on-screen versions
    """
    images: list[tuple[Path, str]] = []
    for parent_path, dirnames, filenames in path.walk():
        for filename in filenames:
            try:
                Image.open(parent_path / filename)
            except UnidentifiedImageError:
                continue  # Not an image file

            # If we modify dirnames in-place, walk() will skip anything we remove
            if "slides" in dirnames:
                dirnames.remove("slides")

            images.append((parent_path, filename))

    for parent_path, filename in track(images, description="Making thumbnails..."):
        file_path = parent_path / filename
        orig_img = Image.open(parent_path / filename)

        slides_path = parent_path / "slides"
        slides_path.mkdir(exist_ok=True)

        thumb_img = orig_img.copy()
        thumb_img.thumbnail(config.thumbnail_size)
        thumb_filename = file_path.stem + ".thumb" + file_path.suffix
        thumb_img.save(slides_path / thumb_filename)
        logger.info(f"Generated thumbnail size {parent_path / filename} -> {thumb_filename}")

        screen_img = orig_img.copy()
        screen_img.thumbnail(config.view_size)
        screen_filename = file_path.stem + ".screen" + file_path.suffix
        screen_img.save(slides_path / screen_filename)
        logger.info(f"Generated screen size {parent_path / filename} -> {screen_filename}")


def generate_html(config: Config, path: Path) -> None:
    """
    Recursively generate HTML files for this directory and all children
    """