"""Backfill episode numbers for migrated show in DB and megaphone.

run this in dev with command
docker compose run --build backfill-episode-numbers-for-show.py --env=dev --api_key=megaphone_token --podcast_id=299
"""

import argparse
import logging
import sys

from flask import g

from podcast import config
from podcast.api import app
from podcast.connectors import mysql
from podcast.logic import megaphone as megaphone_logic
from podcast.models import episode as episode_model
from podcast.models import podcast as podcast_model
from podcast.utils import api_utils


log = logging.getLogger('main')
log.setLevel(logging.DEBUG)
fmt = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
sh = logging.StreamHandler(sys.stdout)
sh.setFormatter(fmt)
log.addHandler(sh)


users = {
    'dev': '366',
    'qa': '1298',
    'prod': '88'
}

request_headers = None


def get_episodes(podcast_id):
    """Return episodes of a podcast.

    Args:
        podcast_id(int): Id of the podcast
    """
    with mysql.pod_db_session(read_only=True) as session:
        rows = session.query(episode_model.Episode).filter(
            episode_model.Episode.podcast_id == podcast_id,
            episode_model.Episode.is_deleted.isnot(True)
        ).all()
        return [row.to_dict() for row in rows]


def update_episode_details_in_megaphone(mp_podcast_id, episode_data, network_id):
    """Update episode number in megaphone.

    Args:
        mp_podcast_id (str): The podcast unique identifier in megaphone.
        episode_data (dict): The episode data from episode table/model.
        network_id (str): The network unique identifier.

    Returns:
        dict: the megaphone api response.

    """
    megaphone_data = {
        'episodeNumber': episode_data['episode_number'],
        'retainAdLocations': True
    }

    megaphone_response = megaphone_logic.call_episode_api(
        megaphone_data,
        mp_podcast_id,
        episode_data['megaphone_id'],
        network_id,
        episode_data['id']
    )
    return megaphone_response


def backfill_episode_numbers(podcast_id):
    """Backfill episode numbers.

    Args:
        podcast_id(int): Id of the podcast.
    """
    episode_number = 1
    podcast = podcast_model.get_podcast_by_id(podcast_id)
    episodes = get_episodes(podcast_id)
    for episode in episodes:
        episode_id = episode['id']
        print(f'Running for episode_id: {episode_id}')

        updated_episode = episode_model.update_episode(episode_id, {'episode_number': episode_number})
        print(f'Database updated episode: {episode_id} with episode number {episode_number}')

        update_episode_details_in_megaphone(podcast['megaphone_id'], updated_episode, podcast['network_id'])
        print(f'Megaphone updated episode: {episode_id} with episode number {episode_number}')

        episode_number += 1


def main():
    """Extract shell arguments and start backfilling episode numbers for podcast in DB and megaphone."""
    global request_headers
    argparser = argparse.ArgumentParser(prog='Backfill episode numbers for show in DB and megaphone.')
    argparser.add_argument('--api_key', required=True, help='Megaphone API Key')
    argparser.add_argument('--env', required=True, help='env', default='qa')
    argparser.add_argument('--podcast_id', required=True, help='podcast_id')

    args = argparser.parse_args()

    config.MEGAPHONE_API_TOKEN = args.api_key
    env = args.env
    podcast_id = int(args.podcast_id)

    def get_user_id():
        """Mock get user id return value."""
        return users[env]

    api_utils.get_user_id = get_user_id

    request_headers = {
        'Grass-Account-Type': 'vendor',
        'Grass-Account-Id': '1',
        'Correlation-Id': '12',
        'Orchard-Identity-UUID': 'podcast-admin',
        'Content-Type': 'application/json'
    }

    log.info(f'Running episode numbers backfill for {env} Environment and podcast id {podcast_id}\n')

    with app.app_context():
        g.log = log
        backfill_episode_numbers(podcast_id)


if __name__ == '__main__':
    main()