import json
import os
import logging
from pathlib import Path
import tvdb_v4_official
import aiofiles

THETVDB_API_KEY = os.getenv("THETVDB_API_KEY")
CACHE_DIR = os.getenv("CACHE_DIR")
SAVE_DIR = os.path.join(CACHE_DIR, "metadata")
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# Initialize TVDB client
tvdb = tvdb_v4_official.TVDB(THETVDB_API_KEY)

def get_series_info(series_id):
    """Fetch series information including episodes from TVDB."""
    try:
        series = tvdb.get_series_extended(series_id, meta="episodes")
        logging.info("Series info fetched successfully.")
        return series
    except Exception as e:
        logging.error(f"Error fetching series info: {e}")
        return None

def filter_episode_data(episode):
    """Filter episode data to include only necessary fields."""
    return {
        "id": episode.get("id"),
        "seriesId": episode.get("seriesId"),
        "name": episode.get("name"),
        "aired": episode.get("aired"),
        "runtime": episode.get("runtime"),
        "overview": episode.get("overview"),
        "image": episode.get("image"),
        "imageType": episode.get("imageType"),
        "isMovie": episode.get("isMovie"),
        "number": episode.get("number"),
        "absoluteNumber": episode.get("absoluteNumber"),
        "seasonNumber": episode.get("seasonNumber"),
        "finaleType": episode.get("finaleType"),
        "year": episode.get("year")
    }

async def save_to_json(data, path):
    """Save data to a JSON file asynchronously."""
    try:
        async with aiofiles.open(path, 'w', encoding='utf-8') as f:
            await f.write(json.dumps(data, indent=4, ensure_ascii=False))
        logging.info(f"Data saved to {path}")
    except IOError as e:
        logging.error(f"Error saving data to {path}: {e}")

async def fetch_and_cache_seasons(series_id):
    """Fetch and cache episodes for a given series ID asynchronously."""
    series_info = get_series_info(series_id)
    if not series_info:
        logging.error("Series info could not be fetched.")
        return

    all_seasons = {}
    for season in series_info.get('seasons', []):
        season_id = season['id']
        season_number = season.get('number')
        
        if season_number == 0:
            season_key = "Specials"
        else:
            season_key = f"Season {season_number}"
        
        all_seasons[season_key] = []
        try:
            season_info = tvdb.get_season_extended(season_id)
            for episode in season_info.get('episodes', []):
                filtered_data = filter_episode_data(episode)
                all_seasons[season_key].append(filtered_data)
            logging.info(f"Fetched episodes for {season_key}.")
        except Exception as e:
            logging.error(f"Error fetching season info for {season_key}: {e}")

    # Create folder for the series
    series_folder = Path(SAVE_DIR) / str(series_id)
    series_folder.mkdir(parents=True, exist_ok=True)
    
    # Save episodes for each season in separate JSON files
    for season_key, episodes in sorted(all_seasons.items()):
        episodes_sorted = sorted(episodes, key=lambda e: e.get('number'))
        season_file = series_folder / f"{season_key}.json"
        await save_to_json(episodes_sorted, season_file)

async def main(series_id):
    """Main function to fetch and cache episodes asynchronously."""
    await fetch_and_cache_seasons(series_id)

if __name__ == "__main__":
    import asyncio
    # Replace with your series ID
    SERIES_ID = "315103"
    asyncio.run(main(SERIES_ID))