newznab-tmux-NNTmux/nntmux/processing/PostProcess.php

<?php

namespace nntmux\processing;

use nntmux\Nfo;
use nntmux\XXX;
use nntmux\NNTP;
use nntmux\Books;
use nntmux\db\DB;
use nntmux\Games;
use nntmux\Movie;
use nntmux\Music;
use nntmux\Console;
use nntmux\Sharing;
use App\Models\Group;
use nntmux\NameFixer;
use App\Models\Release;
use App\Models\Category;
use App\Models\Settings;
use App\Models\ReleaseFile;
use Illuminate\Support\Carbon;
use nntmux\processing\tv\TMDB;
use nntmux\processing\tv\TVDB;
use dariusiii\rarinfo\Par2Info;
use nntmux\processing\tv\TVMaze;
use nntmux\processing\post\AniDB;
use nntmux\processing\tv\TraktTv;
use nntmux\processing\post\ProcessAdditional;

class PostProcess
{
    /**
     * @var \nntmux\db\DB
     */
    public $pdo;

    /**
     * @var \nntmux\NameFixer
     */
    protected $nameFixer;

    /**
     * @var \dariusiii\rarinfo\Par2Info
     */
    protected $_par2Info;

    /**
     * @var
     */
    protected $_srrInfo;

    /**
     * Use alternate NNTP provider when download fails?
     * @var bool
     */
    private $alternateNNTP;

    /**
     * Add par2 info to rar list?
     * @var bool
     */
    private $addpar2;

    /**
     * Should we echo to CLI?
     * @var bool
     */
    private $echooutput;

    /**
     * @var \nntmux\Nfo
     */
    private $Nfo;

    /**
     * Constructor.
     *
     * @param array $options Pass in class instances.
     *
     * @throws \Exception
     */
    public function __construct(array $options = [])
    {
        $defaults = [
            'Echo'         => true,
            'Logger'       => null,
            'Groups'       => null,
            'NameFixer'    => null,
            'Nfo'          => null,
            'ReleaseFiles' => null,
            'Settings'     => null,
        ];
        $options += $defaults;

        // Various.
        $this->echooutput = ($options['Echo'] && NN_ECHOCLI);

        // Class instances.
        $this->pdo = (($options['Settings'] instanceof DB) ? $options['Settings'] : new DB());
        $this->_par2Info = new Par2Info();
        $this->nameFixer = (($options['NameFixer'] instanceof NameFixer) ? $options['NameFixer'] : new NameFixer(['Echo' => $this->echooutput, 'Settings' => $this->pdo, 'Groups' => null]));
        $this->Nfo = (($options['Nfo'] instanceof Nfo) ? $options['Nfo'] : new Nfo());

        // Site settings.
        $this->addpar2 = (int) Settings::settingValue('..addpar2') !== 0;
        $this->alternateNNTP = (int) Settings::settingValue('..alternate_nntp') === 1;
    }

    /**
     * Go through every type of post proc.
     *
     * @param $nntp
     *
     * @return void
     * @throws \Exception
     */
    public function processAll($nntp): void
    {
        $this->processAdditional($nntp);
        $this->processNfos($nntp);
        $this->processSharing($nntp);
        $this->processMovies();
        $this->processMusic();
        $this->processConsoles();
        $this->processGames();
        $this->processAnime();
        $this->processTv();
        $this->processXXX();
        $this->processBooks();
    }

    /**
     * Lookup anidb if enabled.
     *
     *
     * @throws \Exception
     */
    public function processAnime(): void
    {
        if ((int) Settings::settingValue('..lookupanidb') !== 0) {
            (new AniDB(['Echo' => $this->echooutput, 'Settings' => $this->pdo]))->processAnimeReleases();
        }
    }

    /**
     * Process books using amazon.com.
     *
     * @return void
     * @throws \Exception
     */
    public function processBooks(): void
    {
        if ((int) Settings::settingValue('..lookupbooks') !== 0) {
            (new Books(['Echo' => $this->echooutput, 'Settings' => $this->pdo]))->processBookReleases();
        }
    }

    /**
     * @throws \Exception
     */
    public function processConsoles(): void
    {
        if ((int) Settings::settingValue('..lookupgames') !== 0) {
            (new Console(['Settings' => $this->pdo, 'Echo' => $this->echooutput]))->processConsoleReleases();
        }
    }

    /**
     * @throws \Exception
     */
    public function processGames(): void
    {
        if ((int) Settings::settingValue('..lookupgames') !== 0) {
            (new Games(['Echo' => $this->echooutput, 'Settings' => $this->pdo]))->processGamesReleases();
        }
    }

    /**
     * Lookup imdb if enabled.
     *
     * @param string          $groupID             (Optional) ID of a group to work on.
     * @param string          $guidChar            (Optional) First letter of a release GUID to use to get work.
     * @param int|string|null $processMovies       (Optional) 0 Don't process, 1 process all releases,
     *                                             2 process renamed releases only, '' check site setting
     *
     * @return void
     * @throws \Exception
     */
    public function processMovies($groupID = '', $guidChar = '', $processMovies = ''): void
    {
        $processMovies = (is_numeric($processMovies) ? $processMovies : Settings::settingValue('..lookupimdb'));
        if ($processMovies > 0) {
            (new Movie(['Echo' => $this->echooutput, 'Settings' => $this->pdo]))->processMovieReleases($groupID, $guidChar, $processMovies);
        }
    }

    /**
     * @throws \Exception
     */
    public function processMusic(): void
    {
        if ((int) Settings::settingValue('..lookupmusic') !== 0) {
            (new Music(['Echo' => $this->echooutput, 'Settings' => $this->pdo]))->processMusicReleases();
        }
    }

    /**
     * Process nfo files.
     *
     * @param NNTP   $nntp
     * @param string $groupID  (Optional) ID of a group to work on.
     * @param string $guidChar (Optional) First letter of a release GUID to use to get work.
     *
     * @return void
     * @throws \Exception
     */
    public function processNfos(&$nntp, $groupID = '', $guidChar = ''): void
    {
        if ((int) Settings::settingValue('..lookupnfo') === 1) {
            $this->Nfo->processNfoFiles($nntp, $groupID, $guidChar, (int) Settings::settingValue('..lookupimdb'), (int) Settings::settingValue('..lookuptvrage'));
        }
    }

    /**
     * Process comments.
     *
     * @param NNTP $nntp
     * @throws \Exception
     */
    public function processSharing(&$nntp): void
    {
        (new Sharing(['Settings' => $this->pdo, 'NNTP' => $nntp]))->start();
    }

    /**
     * Process all TV related releases which will assign their series/episode/rage data.
     *
     * @param string     $groupID              (Optional) ID of a group to work on.
     * @param string     $guidChar             (Optional) First letter of a release GUID to use to get work.
     * @param string|int|null $processTV       (Optional) 0 Don't process, 1 process all releases,
     *                                         2 process renamed releases only, '' check site setting
     *
     * @return void
     * @throws \Exception
     */
    public function processTv($groupID = '', $guidChar = '', $processTV = ''): void
    {
        $processTV = (is_numeric($processTV) ? $processTV : Settings::settingValue('..lookuptvrage'));
        if ($processTV > 0) {
            (new TVDB(['Echo' => $this->echooutput, 'Settings' => $this->pdo]))->processSite($groupID, $guidChar, $processTV);
            (new TVMaze(['Echo' => $this->echooutput, 'Settings' => $this->pdo]))->processSite($groupID, $guidChar, $processTV);
            (new TMDB(['Echo' => $this->echooutput, 'Settings' => $this->pdo]))->processSite($groupID, $guidChar, $processTV);
            (new TraktTv(['Echo' => $this->echooutput, 'Settings' => $this->pdo]))->processSite($groupID, $guidChar, $processTV);
        }
    }

    /**
     * Lookup xxx if enabled.
     *
     * @throws \Exception
     */
    public function processXXX(): void
    {
        if ((int) Settings::settingValue('..lookupxxx') === 1) {
            (new XXX(['Echo' => $this->echooutput, 'Settings' => $this->pdo]))->processXXXReleases();
        }
    }

    /**
     * Check for passworded releases, RAR/ZIP contents and Sample/Media info.
     *
     * @note Called externally by tmux/bin/update_per_group and update/postprocess.php
     *
     * @param NNTP $nntp Class NNTP
     * @param int|string $groupID (Optional) ID of a group to work on.
     * @param string $guidChar (Optional) First char of release GUID, can be used to select work.
     *
     * @return void
     * @throws \Exception
     */
    public function processAdditional(&$nntp, $groupID = '', $guidChar = ''): void
    {
        (new ProcessAdditional(['Echo' => $this->echooutput, 'NNTP' => $nntp, 'Settings' => $this->pdo, 'Groups' => $this->groups, 'NameFixer' => $this->nameFixer, 'Nfo' => $this->Nfo]))->start($groupID, $guidChar);
    }

    /**
     * Attempt to get a better name from a par2 file and categorize the release.
     *
     * @note Called from NZBContents.php
     *
     * @param string $messageID MessageID from NZB file.
     * @param int    $relID     ID of the release.
     * @param int    $groupID   Group ID of the release.
     * @param NNTP   $nntp      Class NNTP
     * @param int    $show      Only show result or apply iy.
     *
     * @return bool
     * @throws \Exception
     */
    public function parsePAR2($messageID, $relID, $groupID, &$nntp, $show): bool
    {
        if ($messageID === '') {
            return false;
        }

        $query = Release::query()
            ->where(['isrenamed' => 0, 'id' => $relID])
            ->first(['id', 'groups_id', 'categories_id', 'name', 'searchname', 'postdate', 'id as releases_id']);

        if ($query === null) {
            return false;
        }

        // Only get a new name if the category is OTHER.
        $foundName = true;
        if (\in_array((int) $query['categories_id'], Category::OTHERS_GROUP, false)) {
            $foundName = false;
        }

        // Get the PAR2 file.
        $par2 = $nntp->getMessages(Group::getNameByID($groupID), $messageID, $this->alternateNNTP);
        if ($nntp->isError($par2)) {
            return false;
        }

        // Put the PAR2 into Par2Info, check if there's an error.
        $this->_par2Info->setData($par2);
        if ($this->_par2Info->error) {
            return false;
        }

        // Get the file list from Par2Info.
        $files = $this->_par2Info->getFileList();
        if ($files !== false && \count($files) > 0) {
            $filesAdded = 0;

            // Loop through the files.
            foreach ($files as $file) {
                if (! isset($file['name'])) {
                    continue;
                }

                // If we found a name and added 10 files, stop.
                if ($foundName === true && $filesAdded > 10) {
                    break;
                }

                if ($this->addpar2) {
                    // Add to release files.
                    if ($filesAdded < 11 && ReleaseFile::query()->where(['releases_id' => $relID, 'name' => $file['name']])->first() === null) {

                        // Try to add the files to the DB.
                        if (ReleaseFile::addReleaseFiles($relID, $file['name'], $file['hash_16K'], $file['size'], $query['postdate'] !== null ? Carbon::createFromFormat('Y-m-d H:i:s', $query['postdate']) : Carbon::now(), 0)) {
                            $filesAdded++;
                        }
                    }
                } else {
                    $filesAdded++;
                }

                // Try to get a new name.
                if ($foundName === false) {
                    $query['textstring'] = $file['name'];
                    if ($this->nameFixer->checkName($query, 1, 'PAR2, ', 1, $show) === true) {
                        $foundName = true;
                    }
                }
            }

            // If we found some files.
            if ($filesAdded > 0) {

                // Update the file count with the new file count + old file count.
                Release::query()->where('id', $relID)->increment('rarinnerfilecount', $filesAdded);
            }
            if ($foundName === true) {
                return true;
            }
        }

        return false;
    }
}