import os
import requests
import logging
from pathlib import Path

# Configure logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# Environment variables
IMMICH_API_KEY = os.getenv('IMMICH_API_KEY')
IMMICH_INSTANCE_URL = os.getenv('IMMICH_INSTANCE_URL')
# ALBUM_NAME = os.getenv('IMMICH_ALBUM_NAME')
ALBUM_ID = os.getenv('IMMICH_ALBUM_ID')
DOWNLOAD_PATH = Path(os.getenv('IMMICH_DOWNLOAD_PATH', './downloads'))


# def get_album_id():
#     """Get album ID by name from Immich API"""
#     headers = {'x-api-key': IMMICH_API_KEY}
#     response = requests.get(
#         f"{IMMICH_INSTANCE_URL}/albums",
#         headers=headers,
#         params={'albumName': ALBUM_NAME}
#     )
#     response.raise_for_status()
#
#     for album in response.json():
#         if album['albumName'] == ALBUM_NAME:
#             return album['id']
#     raise ValueError(f"Album '{ALBUM_NAME}' not found")


def get_album_assets(album_id):
    """Retrieve all assets in album with pagination"""
    headers = {'x-api-key': IMMICH_API_KEY}
    assets = []
    page = 1

    while True:
        response = requests.get(
            f"{IMMICH_INSTANCE_URL}/albums/{album_id}",
            headers=headers,
            params={'page': page}
        )
        response.raise_for_status()

        assets.extend(response.json()["assets"])
        total_pages = int(response.headers.get('X-Pagination-Count', 1))
        if page >= total_pages:
            break
        page += 1
        # break

    return assets


def download_assets(assets):
    """Download missing assets to target directory"""
    DOWNLOAD_PATH.mkdir(parents=True, exist_ok=True)
    headers = {'x-api-key': IMMICH_API_KEY}

    for asset in assets:
        filename = f"{asset['originalFileName']}"
        filepath = DOWNLOAD_PATH / filename

        if not filepath.exists():
            logging.info(f"Downloading {filename}")
            response = requests.get(
                f"{IMMICH_INSTANCE_URL}/assets/{asset['id']}/original",
                headers=headers,
                stream=True
            )
            response.raise_for_status()

            with open(filepath, 'wb') as f:
                for chunk in response.iter_content(chunk_size=8192):
                    f.write(chunk)
        else:
            logging.debug(f"Skipping existing file {filename}")


if __name__ == "__main__":
    try:
        assets = get_album_assets(ALBUM_ID)
        download_assets(assets)
        logging.info(f"Sync complete. {len(assets)} assets processed")
    except Exception as e:
        logging.error(f"Sync failed: {str(e)}")
        raise