<?php

namespace nntmux\processing\tv;

use nntmux\ColorCLI;
use nntmux\ReleaseImage;
use Adrenth\Thetvdb\Client;
use Adrenth\Thetvdb\Exception\UnauthorizedException;
use Adrenth\Thetvdb\Exception\CouldNotLoginException;
use Adrenth\Thetvdb\Exception\RequestFailedException;
use Adrenth\Thetvdb\Exception\InvalidArgumentException;
use Adrenth\Thetvdb\Exception\InvalidJsonInResponseException;

/**
 * Class TVDB -- functions used to post process releases against TVDB.
 */
class TVDB extends TV
{
    const TVDB_URL = 'https://api.thetvdb.com';
    const TVDB_API_KEY = '31740C28BAC74DEF';
    const MATCH_PROBABILITY = 75;

    /**
     * @var Client
     */
    public $client;

    /**
     * @var string Authorization token for TVDB v2 API
     */
    public $token;

    /**
     * @string URL for show poster art
     */
    public $posterUrl;

    /**
     * @var string URL for show fanart
     */
    public $fanartUrl;

    /**
     * @bool Do a local lookup only if server is down
     */
    private $local;

    /**
     * @param array $options Class instances / Echo to cli?
     */
    public function __construct(array $options = [])
    {
        parent::__construct($options);
        $this->client = new Client();
        $this->client->setLanguage('en');
        $this->posterUrl = self::TVDB_URL.DS.'graphical/%s-g.jpg';
        $this->fanartUrl = self::TVDB_URL.DS.'_cache/fanart/original/%s-3.jpg';
        $this->local = false;

        // Check if we can get the time for API status
        // If we can't then we set local to true
        try {
            $this->token = $this->client->authentication()->login(self::TVDB_API_KEY);
        } catch (CouldNotLoginException $error) {
            echo ColorCLI::warning('Could not reach TVDB API. Running in local mode only!');
            $this->local = true;
        } catch (UnauthorizedException $error) {
            echo ColorCLI::warning('Bad response from TVDB API. Running in local mode only!');
            $this->local = true;
        }

        if (strlen($this->token) > 0) {
            $this->client->setToken($this->token);
        }
    }

    /**
     * Main processing director function for scrapers
     * Calls work query function and initiates processing.
     *
     * @param      $groupID
     * @param      $guidChar
     * @param      $process
     * @param bool $local
     */
    public function processSite($groupID, $guidChar, $process, $local = false): void
    {
        $res = $this->getTvReleases($groupID, $guidChar, $process, parent::PROCESS_TVDB);

        $tvcount = $res->rowCount();

        if ($this->echooutput && $tvcount > 0) {
            echo ColorCLI::header('Processing TVDB lookup for '.number_format($tvcount).' release(s).');
        }

        if ($res instanceof \Traversable) {
            $this->titleCache = [];

            foreach ($res as $row) {
                $tvdbid = false;

                // Clean the show name for better match probability
                $release = $this->parseInfo($row['searchname']);
                if (is_array($release) && $release['name'] != '') {
                    if (in_array($release['cleanname'], $this->titleCache, false)) {
                        if ($this->echooutput) {
                            echo ColorCLI::headerOver('Title: ').
									ColorCLI::warningOver($release['cleanname']).
									ColorCLI::header(' already failed lookup for this site.  Skipping.');
                        }
                        $this->setVideoNotFound(parent::PROCESS_TVMAZE, $row['id']);
                        continue;
                    }

                    // Find the Video ID if it already exists by checking the title.
                    $videoId = $this->getByTitle($release['cleanname'], parent::TYPE_TV);

                    if ($videoId !== false) {
                        $tvdbid = $this->getSiteByID('tvdb', $videoId);
                    }

                    // Force local lookup only
                    if ($local === true || $this->local === true) {
                        $lookupSetting = false;
                    } else {
                        $lookupSetting = true;
                    }

                    if ($tvdbid === false && $lookupSetting) {

						// If it doesnt exist locally and lookups are allowed lets try to get it.
                        if ($this->echooutput) {
                            echo ColorCLI::primaryOver('Video ID for ').
								ColorCLI::headerOver($release['cleanname']).
								ColorCLI::primary(' not found in local db, checking web.');
                        }

                        // Check if we have a valid country and set it in the array
                        $country = (isset($release['country']) && strlen($release['country']) == 2
							? (string) $release['country']
							: ''
						);

                        // Get the show from TVDB
                        $tvdbShow = $this->getShowInfo((string) $release['cleanname'], $country);

                        if (is_array($tvdbShow)) {
                            $tvdbShow['country'] = $country;
                            $videoId = $this->add($tvdbShow);
                            $tvdbid = (int) $tvdbShow['tvdb'];
                        }
                    } elseif ($this->echooutput && $tvdbid !== false) {
                        echo ColorCLI::primaryOver('Video ID for ').
							ColorCLI::headerOver($release['cleanname']).
							ColorCLI::primary(' found in local db, attempting episode match.');
                    }

                    if (is_numeric($videoId) && $videoId > 0 && is_numeric($tvdbid) && $tvdbid > 0) {
                        // Now that we have valid video and tvdb ids, try to get the poster
                        $this->getPoster($videoId, $tvdbid);

                        $seasonNo = (! empty($release['season']) ? preg_replace('/^S0*/i', '', $release['season']) : '');
                        $episodeNo = (! empty($release['episode']) ? preg_replace('/^E0*/i', '', $release['episode']) : '');

                        if ($episodeNo === 'all') {
                            // Set the video ID and leave episode 0
                            $this->setVideoIdFound($videoId, $row['id'], 0);
                            echo ColorCLI::primary('Found TVDB Match for Full Season!');
                            continue;
                        }

                        // Download all episodes if new show to reduce API/bandwidth usage
                        if ($this->countEpsByVideoID($videoId) === false) {
                            $this->getEpisodeInfo($tvdbid, -1, -1, '', $videoId);
                        }

                        // Check if we have the episode for this video ID
                        $episode = $this->getBySeasonEp($videoId, $seasonNo, $episodeNo, $release['airdate']);

                        if ($episode === false && $lookupSetting) {
                            // Send the request for the episode to TVDB
                            $tvdbEpisode = $this->getEpisodeInfo(
								$tvdbid,
								$seasonNo,
								$episodeNo,
								$release['airdate']
							);

                            if ($tvdbEpisode) {
                                $episode = $this->addEpisode($videoId, $tvdbEpisode);
                            }
                        }

                        if ($episode !== false && is_numeric($episode) && $episode > 0) {
                            // Mark the releases video and episode IDs
                            $this->setVideoIdFound($videoId, $row['id'], $episode);
                            if ($this->echooutput) {
                                echo ColorCLI::primary('Found TVDB Match!');
                            }
                        } else {
                            //Processing failed, set the episode ID to the next processing group
                            $this->setVideoNotFound(parent::PROCESS_TVMAZE, $row['id']);
                        }
                    } else {
                        //Processing failed, set the episode ID to the next processing group
                        $this->setVideoNotFound(parent::PROCESS_TVMAZE, $row['id']);
                        $this->titleCache[] = $release['cleanname'];
                    }
                } else {
                    //Parsing failed, take it out of the queue for examination
                    $this->setVideoNotFound(parent::FAILED_PARSE, $row['id']);
                    $this->titleCache[] = $release['cleanname'];
                }
            }
        }
    }

    /**
     * Placeholder for Videos getBanner.
     *
     * @param $videoID
     * @param $siteId
     *
     * @return bool
     */
    protected function getBanner($videoID, $siteId): bool
    {
        return false;
    }

    /**
     * Calls the API to perform initial show name match to TVDB title
     * Returns a formatted array of show data or false if no match.
     *
     * @param string $cleanName
     *
     * @param string $country
     *
     * @return array|false
     */
    protected function getShowInfo($cleanName, $country = '')
    {
        $return = $response = false;
        $highestMatch = 0;
        try {
            $response = $this->client->search()->seriesByName($cleanName);
        } catch (InvalidArgumentException $error) {
            return false;
        } catch (InvalidJsonInResponseException $error) {
            if (strpos($error->getMessage(), 'Could not decode JSON data') === 0 || strpos($error->getMessage(), 'Incorrect data structure') === 0) {
                return false;
            }
        } catch (RequestFailedException $error) {
            return false;
        } catch (UnauthorizedException $error) {
            if (strpos($error->getMessage(), 'Unauthorized') === 0) {
                return false;
            }
        }

        if ($response === false && $country !== '') {
            try {
                $response = $this->client->search()->seriesByName(rtrim(str_replace($country, '', $cleanName)));
            } catch (InvalidArgumentException $error) {
                return false;
            } catch (InvalidJsonInResponseException $error) {
                if (strpos($error->getMessage(), 'Could not decode JSON data') === 0 || strpos($error->getMessage(), 'Incorrect data structure') === 0) {
                    return false;
                }
            } catch (RequestFailedException $error) {
                return false;
            } catch (UnauthorizedException $error) {
                if (strpos($error->getMessage(), 'Unauthorized') === 0) {
                    return false;
                }
            }
        }

        sleep(1);

        if (is_array($response)) {
            foreach ($response->getData() as $show) {
                if ($this->checkRequiredAttr($show, 'tvdbS')) {
                    // Check for exact title match first and then terminate if found
                    if (strtolower($show->getSeriesName()) === strtolower($cleanName)) {
                        $highest = $show;
                        break;
                    }

                    // Check each show title for similarity and then find the highest similar value
                    $matchPercent = $this->checkMatch(strtolower($show->getSeriesName()), strtolower($cleanName), self::MATCH_PROBABILITY);

                    // If new match has a higher percentage, set as new matched title
                    if ($matchPercent > $highestMatch) {
                        $highestMatch = $matchPercent;
                        $highest = $show;
                    }

                    // Check for show aliases and try match those too
                    if (! empty($show->getAliases())) {
                        foreach ($show->getAliases() as $key => $name) {
                            $matchPercent = $this->checkMatch(strtolower($name), strtolower($cleanName), $matchPercent);
                            if ($matchPercent > $highestMatch) {
                                $highestMatch = $matchPercent;
                                $highest = $show;
                            }
                        }
                    }
                }
            }
            if (! empty($highest)) {
                $return = $this->formatShowInfo($highest);
            }
        }

        return $return;
    }

    /**
     * Retrieves the poster art for the processed show.
     *
     * @param int $videoId -- the local Video ID
     * @param int $showId  -- the TVDB ID
     *
     * @return int
     */
    public function getPoster($videoId, $showId): int
    {
        $ri = new ReleaseImage($this->pdo);

        // Try to get the Poster
        $hascover = $ri->saveImage($videoId, sprintf($this->posterUrl, $showId), $this->imgSavePath, '', '');

        // Couldn't get poster, try fan art instead
        if ($hascover !== 1) {
            $hascover = $ri->saveImage($videoId, sprintf($this->fanartUrl, $showId), $this->imgSavePath, '', '');
        }
        // Mark it retrieved if we saved an image
        if ($hascover === 1) {
            $this->setCoverFound($videoId);
        }

        return $hascover;
    }

    /**
     * Gets the specific episode info for the parsed release after match
     * Returns a formatted array of episode data or false if no match.
     *
     * @param int $tvdbid
     * @param int $season
     * @param int $episode
     * @param string  $airdate
     * @param int $videoId
     *
     * @return array|false
     */
    protected function getEpisodeInfo($tvdbid, $season, $episode, $airdate = '', $videoId = 0)
    {
        $return = $response = false;

        if ($airdate !== '') {
            try {
                $response = $this->client->series()->getEpisodesWithQuery($tvdbid, ['firstAired' => $airdate]);
            } catch (InvalidArgumentException $error) {
                return false;
            } catch (InvalidJsonInResponseException $error) {
                if (strpos($error->getMessage(), 'Could not decode JSON data') === 0 || strpos($error->getMessage(), 'Incorrect data structure') === 0) {
                    return false;
                }
            } catch (RequestFailedException $error) {
                return false;
            } catch (UnauthorizedException $error) {
                if (strpos($error->getMessage(), 'Unauthorized') === 0) {
                    return false;
                }
            }
        } elseif ($videoId > 0) {
            try {
                $response = $this->client->series()->getEpisodes($tvdbid);
            } catch (InvalidArgumentException $error) {
                return false;
            } catch (InvalidJsonInResponseException $error) {
                if (strpos($error->getMessage(), 'Could not decode JSON data') === 0 || strpos($error->getMessage(), 'Incorrect data structure') === 0) {
                    return false;
                }
            } catch (RequestFailedException $error) {
                return false;
            } catch (UnauthorizedException $error) {
                if (strpos($error->getMessage(), 'Unauthorized') === 0) {
                    return false;
                }
            }
        } else {
            try {
                $response = $this->client->series()->getEpisodesWithQuery($tvdbid, ['airedSeason' => $season, 'airedEpisode' => $episode]);
            } catch (InvalidArgumentException $error) {
                return false;
            } catch (InvalidJsonInResponseException $error) {
                if (strpos($error->getMessage(), 'Could not decode JSON data') === 0 || strpos($error->getMessage(), 'Incorrect data structure') === 0) {
                    return false;
                }
            } catch (RequestFailedException $error) {
                return false;
            } catch (UnauthorizedException $error) {
                if (strpos($error->getMessage(), 'Unauthorized') === 0) {
                    return false;
                }
            }
        }

        sleep(1);

        if (is_object($response->getData())) {
            if ($this->checkRequiredAttr($response->getData(), 'tvdbE')) {
                $return = $this->formatEpisodeInfo($response);
            }
        } elseif ($videoId > 0 && is_array($response->getData())) {
            foreach ($response->getData() as $singleEpisode) {
                if ($this->checkRequiredAttr($singleEpisode, 'tvdbE')) {
                    $this->addEpisode($videoId, $this->formatEpisodeInfo($singleEpisode));
                }
            }
        }

        return $return;
    }

    /**
     * Assigns API show response values to a formatted array for insertion
     * Returns the formatted array.
     *
     * @param $show
     *
     * @return array
     */
    protected function formatShowInfo($show): array
    {
        preg_match('/tt(?P<imdbid>\d{6,7})$/i', $show->imdbId, $imdb);

        return [
			'type'      => (int) parent::TYPE_TV,
			'title'     => (string) $show->getSeriesName(),
			'summary'   => (string) $show->getOverview(),
			'started'   => $show->firstAired->format('Y-m-d'),
			'publisher' => (string) $show->getNetwork(),
			'source'    => (int) parent::SOURCE_TVDB,
			'imdb'      => (int) ($imdb['imdbid'] ?? 0),
			'tvdb'      => (int) $show->getid(),
			'trakt'     => 0,
			'tvrage'    => 0,
			'tvmaze'    => 0,
			'tmdb'      => 0,
			'aliases'   => (! empty($show->getAliases()) ? $show->getAliases() : ''),
			'localzone' => "''",
		];
    }

    /**
     * Assigns API episode response values to a formatted array for insertion
     * Returns the formatted array.
     *
     * @param $episode
     *
     * @return array
     */
    protected function formatEpisodeInfo($episode): array
    {
        return [
			'title'       => (string) $episode->name,
			'series'      => (int) $episode->season,
			'episode'     => (int) $episode->number,
			'se_complete' => (string) 'S'.sprintf('%02d', $episode->season).'E'.sprintf('%02d', $episode->number),
			'firstaired'  => $episode->firstAired->format('Y-m-d'),
			'summary'     => (string) $episode->overview,
		];
    }
}