Files
newznab-tmux/Blacklight/processing/post/ProcessAdditional.php
T
DariusIII 0f530278ba CS fixes
2024-01-03 21:32:00 +01:00

2127 lines
78 KiB
PHP
Executable File

<?php
namespace Blacklight\processing\post;
use App\Models\Category;
use App\Models\Predb;
use App\Models\Release;
use App\Models\ReleaseFile;
use App\Models\Settings;
use App\Models\UsenetGroup;
use Blacklight\Categorize;
use Blacklight\ColorCLI;
use Blacklight\ElasticSearchSiteSearch;
use Blacklight\ManticoreSearch;
use Blacklight\NameFixer;
use Blacklight\Nfo;
use Blacklight\NNTP;
use Blacklight\NZB;
use Blacklight\ReleaseExtra;
use Blacklight\ReleaseImage;
use Blacklight\Releases;
use Blacklight\utility\Utility;
use dariusiii\rarinfo\ArchiveInfo;
use dariusiii\rarinfo\Par2Info;
use FFMpeg\Coordinate\Dimension;
use FFMpeg\Coordinate\TimeCode;
use FFMpeg\FFMpeg;
use FFMpeg\FFProbe;
use FFMpeg\Filters\Video\ResizeFilter;
use FFMpeg\Format\Audio\Vorbis;
use FFMpeg\Format\Video\Ogg;
use Illuminate\Support\Arr;
use Illuminate\Support\Carbon;
use Illuminate\Support\Facades\DB;
use Illuminate\Support\Facades\File;
use Illuminate\Support\Facades\Log;
use Mhor\MediaInfo\MediaInfo;
class ProcessAdditional
{
/**
* How many compressed (rar/zip) files to check.
*
* @var int
*/
public const maxCompressedFilesToCheck = 10;
protected $_releases;
/**
* Count of releases to work on.
*/
protected int $_totalReleases;
protected $_release;
protected NZB $_nzb;
/**
* List of files with sizes/etc contained in the NZB.
*/
protected array $_nzbContents;
protected Par2Info $_par2Info;
protected ArchiveInfo $_archiveInfo;
/**
* @var bool|null|string
*/
protected mixed $_innerFileBlacklist;
protected int $_maxNestedLevels;
/**
* @var string|null
*/
protected mixed $_7zipPath;
/**
* @var null|string
*/
protected mixed $_unrarPath;
protected string $_killString;
protected string|bool $_showCLIReleaseID;
protected int $_queryLimit;
protected int $_segmentsToDownload;
protected int $_maximumRarSegments;
protected int $_maximumRarPasswordChecks;
protected string $_maxSize;
protected string $_minSize;
protected bool $_processThumbnails;
protected string $_audioSavePath;
protected string $_supportFileRegex;
protected bool $_echoCLI;
protected NNTP $_nntp;
protected Categorize $_categorize;
protected NameFixer $_nameFixer;
protected ReleaseExtra $_releaseExtra;
protected ReleaseImage $_releaseImage;
protected Nfo $_nfo;
protected bool $_extractUsingRarInfo;
protected bool $_alternateNNTP;
protected int $_ffMPEGDuration;
protected bool $_addPAR2Files;
protected bool $_processVideo;
protected bool $_processJPGSample;
protected bool $_processAudioSample;
protected bool $_processMediaInfo;
protected bool $_processAudioInfo;
protected bool $_processPasswords;
protected string $_audioFileRegex;
protected string $_ignoreBookRegex;
protected string $_videoFileRegex;
/**
* Have we created a video file for the current release?
*/
protected bool $_foundVideo;
/**
* Have we found MediaInfo data for a Video for the current release?
*/
protected bool $_foundMediaInfo;
/**
* Have we found MediaInfo data for a Audio file for the current release?
*/
protected bool $_foundAudioInfo;
/**
* Have we created a short Audio file sample for the current release?
*/
protected bool $_foundAudioSample;
/**
* Extension of the found audio file (MP3/FLAC/etc).
*/
protected string $_AudioInfoExtension;
/**
* Have we downloaded a JPG file for the current release?
*/
protected bool $_foundJPGSample;
/**
* Have we created a Video JPG image sample for the current release?
*/
protected bool $_foundSample;
/**
* Have we found PAR2 info on this release?
*/
protected bool $_foundPAR2Info;
/**
* Message ID's for found content to download.
*/
protected array $_sampleMessageIDs;
protected $_JPGMessageIDs;
protected $_MediaInfoMessageIDs;
protected $_AudioInfoMessageIDs;
protected $_RARFileMessageIDs;
/**
* Password status of the current release.
*/
protected int $_passwordStatus;
/**
* Does the current release have a password?
*/
protected bool $_releaseHasPassword;
/**
* Does the current release have an NFO file?
*/
protected bool $_releaseHasNoNFO;
/**
* Name of the current release's usenet group.
*/
protected string $_releaseGroupName;
/**
* Number of file information added to DB (from rar/zip/par2 contents).
*/
protected int $_addedFileInfo;
/**
* Number of file information we found from RAR/ZIP.
* (if some of it was already in DB, this count goes up, while the count above does not).
*/
protected int $_totalFileInfo;
/**
* How many compressed (rar/zip) files have we checked.
*/
protected int $_compressedFilesChecked;
/**
* Should we download the last rar?
*/
protected bool $_fetchLastFiles;
/**
* Are we downloading the last rar?
*/
protected bool $_reverse;
protected ManticoreSearch $manticore;
private FFMpeg $ffmpeg;
private FFProbe $ffprobe;
private MediaInfo $mediaInfo;
private ElasticSearchSiteSearch $elasticsearch;
/**
* ProcessAdditional constructor.
*
*
* @throws \Exception
*/
public function __construct()
{
$this->_echoCLI = config('nntmux.echocli');
$this->_nntp = new NNTP();
$this->_nzb = new NZB();
$this->_archiveInfo = new ArchiveInfo();
$this->_categorize = new Categorize();
$this->_nameFixer = new NameFixer();
$this->_releaseExtra = new ReleaseExtra();
$this->_releaseImage = new ReleaseImage();
$this->_par2Info = new Par2Info();
$this->_nfo = new Nfo();
$this->manticore = new ManticoreSearch();
$this->elasticsearch = new ElasticSearchSiteSearch();
$this->ffmpeg = FFMpeg::create(['timeout' => Settings::settingValue('..timeoutseconds')]);
$this->ffprobe = FFProbe::create();
$this->mediaInfo = new MediaInfo();
$this->mediaInfo->setConfig('use_oldxml_mediainfo_output_format', true);
$this->mediaInfo->setConfig('command', Settings::settingValue('apps..mediainfopath'));
$this->_innerFileBlacklist = Settings::settingValue('indexer.ppa.innerfileblacklist') === '' ? false : Settings::settingValue('indexer.ppa.innerfileblacklist');
$this->_maxNestedLevels = (int) Settings::settingValue('..maxnestedlevels') === 0 ? 3 : (int) Settings::settingValue('..maxnestedlevels');
$this->_extractUsingRarInfo = (int) Settings::settingValue('..extractusingrarinfo') !== 0;
$this->_fetchLastFiles = (int) Settings::settingValue('archive.fetch.end') !== 0;
$this->_7zipPath = false;
$this->_unrarPath = false;
// Pass the binary extractors to ArchiveInfo.
$clients = [];
if (! empty(Settings::settingValue('apps..unrarpath'))) {
$this->_unrarPath = Settings::settingValue('apps..unrarpath');
$clients += [ArchiveInfo::TYPE_RAR => $this->_unrarPath];
}
if (! empty(Settings::settingValue('apps..zippath'))) {
$this->_7zipPath = Settings::settingValue('apps..zippath');
$clients += [ArchiveInfo::TYPE_SZIP => $this->_7zipPath];
$clients += [ArchiveInfo::TYPE_ZIP => $this->_7zipPath];
}
$this->_archiveInfo->setExternalClients($clients);
$this->_killString = '"';
if (! empty(Settings::settingValue('apps..timeoutpath')) && (int) Settings::settingValue('..timeoutseconds') > 0) {
$this->_killString = (
'"'.Settings::settingValue('apps..timeoutpath').
'" --foreground --signal=KILL '.
Settings::settingValue('..timeoutseconds').' "'
);
}
$this->_showCLIReleaseID = (PHP_BINARY.' '.__DIR__.'/ProcessAdditional.php ReleaseID: ');
// Maximum amount of releases to fetch per run.
$this->_queryLimit =
(Settings::settingValue('..maxaddprocessed') !== '') ? (int) Settings::settingValue('..maxaddprocessed') : 25;
// Maximum message ID's to download per file type in the NZB (video, jpg, etc).
$this->_segmentsToDownload =
(Settings::settingValue('..segmentstodownload') !== '') ? (int) Settings::settingValue('..segmentstodownload') : 2;
// Maximum message ID's to download for a RAR file.
$this->_maximumRarSegments =
(Settings::settingValue('..maxpartsprocessed') !== '') ? (int) Settings::settingValue('..maxpartsprocessed') : 3;
// Maximum RAR files to check for a password before stopping.
$this->_maximumRarPasswordChecks =
(Settings::settingValue('..passchkattempts') !== '') ? (int) Settings::settingValue('..passchkattempts') : 1;
$this->_maximumRarPasswordChecks = (max($this->_maximumRarPasswordChecks, 1));
// Maximum size of releases in GB.
$this->_maxSize = (Settings::settingValue('..maxsizetopostprocess') !== '') ? (int) Settings::settingValue('..maxsizetopostprocess') : 100;
// Minimum size of releases in MB.
$this->_minSize = (Settings::settingValue('..minsizetopostprocess') !== '') ? (int) Settings::settingValue('..minsizetopostprocess') : 100;
// Use the alternate NNTP provider for downloading Message-ID's ?
$this->_alternateNNTP = (int) Settings::settingValue('..alternate_nntp') === 1;
$this->_ffMPEGDuration = Settings::settingValue('..ffmpeg_duration') !== '' ? (int) Settings::settingValue('..ffmpeg_duration') : 5;
$this->_addPAR2Files = (int) Settings::settingValue('..addpar2') !== 0;
if (! Settings::settingValue('apps..ffmpegpath')) {
$this->_processAudioSample = $this->_processThumbnails = $this->_processVideo = false;
} else {
$this->_processAudioSample = (int) Settings::settingValue('..saveaudiopreview') !== 0;
$this->_processThumbnails = (int) Settings::settingValue('..processthumbnails') !== 0;
$this->_processVideo = (int) Settings::settingValue('..processvideos') !== 0;
}
$this->_processJPGSample = (int) Settings::settingValue('..processjpg') !== 0;
$this->_processMediaInfo = Settings::settingValue('apps..mediainfopath') !== '';
$this->_processAudioInfo = $this->_processMediaInfo;
$this->_processPasswords = ! empty(Settings::settingValue('..checkpasswordedrar')) && ! empty(Settings::settingValue('apps..unrarpath'));
$this->_audioSavePath = storage_path('covers/audiosample/');
$this->_audioFileRegex = '\.(AAC|AIFF|APE|AC3|ASF|DTS|FLAC|MKA|MKS|MP2|MP3|RA|OGG|OGM|W64|WAV|WMA)';
$this->_ignoreBookRegex = '/\b(epub|lit|mobi|pdf|sipdf|html)\b.*\.rar(?!.{20,})/i';
$this->_supportFileRegex = '/\.(vol\d{1,3}\+\d{1,3}|par2|srs|sfv|nzb';
$this->_videoFileRegex = '\.(AVI|F4V|IFO|M1V|M2V|M4V|MKV|MOV|MP4|MPEG|MPG|MPGV|MPV|OGV|QT|RM|RMVB|TS|VOB|WMV)';
}
/**
* Clear out the main temp path when done.
*/
public function __destruct()
{
$this->_clearMainTmpPath();
}
/**
* @throws \Exception
*/
public function start(string $groupID = '', string $guidChar = ''): void
{
$this->_setMainTempPath($guidChar, $groupID);
// Fetch all the releases to work on.
$this->_fetchReleases($groupID, $guidChar);
// Check if we have releases to work on.
if ($this->_totalReleases > 0) {
// Echo start time and process description.
$this->_echoDescription();
$this->_processReleases();
}
}
/**
* @var string Main temp path to work on.
*/
protected string $_mainTmpPath;
/**
* @var string Temp path for current release.
*/
protected string $tmpPath;
/**
* @throws \RuntimeException
* @throws \Exception
*/
protected function _setMainTempPath(&$guidChar, string &$groupID = ''): void
{
// Set up the temporary files folder location.
$this->_mainTmpPath = (string) Settings::settingValue('..tmpunrarpath');
// Check if it ends with a dir separator.
if (! preg_match('/[\/\\\\]$/', $this->_mainTmpPath)) {
$this->_mainTmpPath .= '/';
}
// If we are doing per group, use the groupID has a inner path, so other scripts don't delete the files we are working on.
if ($groupID !== '') {
$this->_mainTmpPath .= ($groupID.'/');
} elseif ($guidChar !== '') {
$this->_mainTmpPath .= ($guidChar.'/');
}
if (! File::isDirectory($this->_mainTmpPath)) {
if (! File::makeDirectory($this->_mainTmpPath, 0777, true, true) && ! File::isDirectory($this->_mainTmpPath)) {
throw new \RuntimeException(sprintf('Directory "%s" was not created', $this->_mainTmpPath));
}
}
if (! File::isDirectory($this->_mainTmpPath)) {
throw new \RuntimeException('Could not create the tmpunrar folder ('.$this->_mainTmpPath.')');
}
$this->_clearMainTmpPath();
$this->tmpPath = $this->_mainTmpPath;
}
/**
* Clear out old folders/files from the main temp folder.
*/
protected function _clearMainTmpPath(): void
{
if ($this->_mainTmpPath !== '') {
$this->_recursivePathDelete(
$this->_mainTmpPath,
// These are folders we don't want to delete.
[
// This is the actual temp folder.
$this->_mainTmpPath,
]
);
}
}
/**
* Get all releases that need to be processed.
*
*
* @void
*/
protected function _fetchReleases(int|string $groupID, string &$guidChar): void
{
$releasesQuery = Release::query()
->where('releases.nzbstatus', '=', 1)
->where('releases.passwordstatus', '=', -1)
->where('releases.haspreview', '=', -1)
->where('categories.disablepreview', '=', 0);
if ($this->_maxSize > 0) {
$releasesQuery->where('releases.size', '<', (int) $this->_maxSize * 1073741824);
}
if ($this->_minSize > 0) {
$releasesQuery->where('releases.size', '>', (int) $this->_minSize * 1048576);
}
if (! empty($groupID)) {
$releasesQuery->where('releases.groups_id', $groupID);
}
if (! empty($guidChar)) {
$releasesQuery->where('releases.leftguid', $guidChar);
}
$releasesQuery->select(['releases.id', 'releases.id as releases_id', 'releases.guid', 'releases.name', 'releases.size', 'releases.groups_id', 'releases.nfostatus', 'releases.fromname', 'releases.completion', 'releases.categories_id', 'releases.searchname', 'releases.predb_id', 'categories.disablepreview'])
->leftJoin('categories', 'categories.id', '=', 'releases.categories_id')
->orderBy('releases.passwordstatus')
->orderByDesc('releases.postdate')
->limit($this->_queryLimit);
$this->_releases = $releasesQuery->get();
$this->_totalReleases = $this->_releases->count();
}
/**
* Output the description and start time.
*
* @void
*/
protected function _echoDescription(): void
{
if ($this->_totalReleases > 1 && $this->_echoCLI) {
$this->_echo(
PHP_EOL.
'Additional post-processing, started at: '.
now()->format('D M d, Y G:i a').
PHP_EOL.
'Downloaded: (xB) = yEnc article, f= Failed ;Processing: z = ZIP file, r = RAR file'.
PHP_EOL.
'Added: s = Sample image, j = JPEG image, A = Audio sample, a = Audio MediaInfo, v = Video sample'.
PHP_EOL.
'Added: m = Video MediaInfo, n = NFO, ^ = File details from inside the RAR/ZIP',
'header'
);
}
}
/**
* Loop through the releases, processing them 1 at a time.
*
* @throws \RuntimeException
* @throws \Exception
*/
protected function _processReleases(): void
{
foreach ($this->_releases as $this->_release) {
$this->_echo(
PHP_EOL.'['.$this->_release->id.']['.
human_filesize($this->_release->size, 1).']',
'primaryOver'
);
cli_set_process_title($this->_showCLIReleaseID.$this->_release->id);
// Create folder to store temporary files.
if (! $this->_createTempFolder()) {
continue;
}
// Get NZB contents.
if (! $this->_getNZBContents()) {
continue;
}
// Reset the current release variables.
$this->_resetReleaseStatus();
// Go through the files in the NZB, get the amount of book files.
$totalBooks = $this->_processNZBContents();
// Check if this NZB is a large collection of books.
$bookFlood = false;
if ($totalBooks > 80 && ($totalBooks * 2) >= \count($this->_nzbContents)) {
$bookFlood = true;
}
if ($this->_processPasswords || $this->_processThumbnails || $this->_processMediaInfo || $this->_processAudioInfo || $this->_processVideo
) {
// Process usenet Message-ID downloads.
$this->_processMessageIDDownloads();
// Process compressed (RAR/ZIP) files inside the NZB.
if (! $bookFlood && $this->_NZBHasCompressedFile) {
// Download the RARs/ZIPs, extract the files inside them and insert the file info into the DB.
$this->_processNZBCompressedFiles();
// Download rar/zip in reverse order, to get the last rar or zip file.
if ($this->_fetchLastFiles) {
$this->_processNZBCompressedFiles(true);
}
if (! $this->_releaseHasPassword) {
// Process the extracted files to get video/audio samples/etc.
$this->_processExtractedFiles();
}
}
}
// Update the release to say we processed it.
$this->_finalizeRelease();
// Delete all files / folders for this release.
$this->_recursivePathDelete($this->tmpPath);
}
if ($this->_echoCLI) {
echo PHP_EOL;
}
}
protected function _recursivePathDelete(string $path, array $ignoredFolders = []): void
{
if (File::isDirectory($path)) {
if (\in_array($path, $ignoredFolders, false)) {
return;
}
foreach (File::allFiles($path) as $file) {
$this->_recursivePathDelete($file, $ignoredFolders);
}
File::deleteDirectory($path);
} elseif (File::isFile($path)) {
File::delete($path);
}
}
/**
* Create a temporary storage folder for the current release.
*
*
*
* @throws \Exception
*/
protected function _createTempFolder(): bool
{
// Per release defaults.
$this->tmpPath = $this->_mainTmpPath.$this->_release->guid.'/';
if (! File::isDirectory($this->tmpPath)) {
if (! File::makeDirectory($this->tmpPath, 0777, true, false) && ! File::isDirectory($this->tmpPath)) {
$this->_echo('Unable to create directory: '.$this->tmpPath, 'warning');
$this->_deleteRelease();
return false;
}
}
return true;
}
/**
* Get list of contents inside a release's NZB file.
*
*
* @throws \Exception
*/
protected function _getNZBContents(): bool
{
$nzbPath = $this->_nzb->NZBPath($this->_release->guid);
if ($nzbPath !== false) {
$nzbContents = Utility::unzipGzipFile($nzbPath);
if (! $nzbContents) {
$this->_echo('NZB is empty or broken for GUID: '.$this->_release->guid, 'warning');
$this->_deleteRelease();
return false;
}
// Get a list of files in the nzb.
$this->_nzbContents = $this->_nzb->nzbFileList($nzbContents, ['no-file-key' => false, 'strip-count' => true]);
if (\count($this->_nzbContents) === 0) {
$this->_echo('NZB is potentially broken for GUID: '.$this->_release->guid, 'warning');
$this->_deleteRelease();
return false;
}
// Sort keys.
ksort($this->_nzbContents, SORT_NATURAL);
return true;
}
$this->_echo('NZB not found for GUID: '.$this->_release->guid, 'warning');
$this->_deleteRelease();
return false;
}
protected function _deleteRelease(): void
{
Release::whereId($this->_release->id)->delete();
}
/**
* Current file we are working on inside a NZB.
*/
protected array $_currentNZBFile;
/**
* Does the current NZB contain a compressed (RAR/ZIP) file?
*/
protected bool $_NZBHasCompressedFile;
/**
* Process the files inside the NZB, find Message-ID's to download.
* If we find files with book extensions, return the amount.
*/
protected function _processNZBContents(): int
{
$totalBookFiles = 0;
foreach ($this->_nzbContents as $this->_currentNZBFile) {
try {
// Check if it's not a nfo, nzb, par2 etc...
if (preg_match($this->_supportFileRegex.'|nfo\b|inf\b|ofn\b)($|[ ")\]-])(?!.{20,})/i', $this->_currentNZBFile['title'])) {
continue;
}
// Check if it's a rar/zip.
if (! $this->_NZBHasCompressedFile &&
preg_match(
'/\.(part\d+|[r|z]\d+|rar|0+|0*10?|zipr\d{2,3}|zipx?)(\s*\.rar)*($|[ ")\]-])|"[a-f0-9]{32}\.[1-9]\d{1,2}".*\(\d+\/\d{2,}\)$/i',
$this->_currentNZBFile['title']
)
) {
$this->_NZBHasCompressedFile = true;
}
// Look for a video sample, make sure it's not an image.
if ($this->_processThumbnails && empty($this->_sampleMessageIDs) && isset($this->_currentNZBFile['segments']) && stripos($this->_currentNZBFile['title'], 'sample') !== false && ! preg_match('/\.jpe?g$/i', $this->_currentNZBFile['title'])
) {
// Get the amount of segments for this file.
$segCount = (\count($this->_currentNZBFile['segments']) - 1);
// If it's more than 1 try to get up to the site specified value of segments.
for ($i = 0; $i < $this->_segmentsToDownload; $i++) {
if ($i > $segCount) {
break;
}
$this->_sampleMessageIDs[] = (string) $this->_currentNZBFile['segments'][$i];
}
}
// Look for a JPG picture, make sure it's not a CD cover.
if ($this->_processJPGSample && empty($this->_JPGMessageIDs) && isset($this->_currentNZBFile['segments']) && ! preg_match('/flac|lossless|mp3|music|inner-sanctum|sound/i', $this->_releaseGroupName) && preg_match('/\.jpe?g[. ")\]]/i', $this->_currentNZBFile['title'])
) {
// Get the amount of segments for this file.
$segCount = (\count($this->_currentNZBFile['segments']) - 1);
// If it's more than 1 try to get up to the site specified value of segments.
for ($i = 0; $i < $this->_segmentsToDownload; $i++) {
if ($i > $segCount) {
break;
}
$this->_JPGMessageIDs[] = (string) $this->_currentNZBFile['segments'][$i];
}
}
// Look for a video file, make sure it's not a sample, for MediaInfo.
if ($this->_processMediaInfo && empty($this->_MediaInfoMessageIDs) && isset($this->_currentNZBFile['segments'][0]) && stripos($this->_currentNZBFile['title'], 'sample') !== false && preg_match('/'.$this->_videoFileRegex.'[. ")\]]/i', $this->_currentNZBFile['title'])
) {
$this->_MediaInfoMessageIDs = (string) $this->_currentNZBFile['segments'][0];
}
// Look for an audio file.
if ($this->_processAudioInfo && empty($this->_AudioInfoMessageIDs) && isset($this->_currentNZBFile['segments']) && preg_match('/'.$this->_audioFileRegex.'[. ")\]]/i', $this->_currentNZBFile['title'], $type)
) {
// Get the extension.
$this->_AudioInfoExtension = $type[1];
$this->_AudioInfoMessageIDs = (string) $this->_currentNZBFile['segments'][0];
}
// Some releases contain many books, increment this to ignore them later.
if (preg_match($this->_ignoreBookRegex, $this->_currentNZBFile['title'])) {
$totalBookFiles++;
}
} catch (\ErrorException $e) {
Log::debug($e->getTraceAsString());
}
}
return $totalBookFiles;
}
/**
* List of message-id's we have tried for rar/zip files.
*/
protected array $_triedCompressedMids = [];
/**
* @throws \Exception
*/
protected function _processNZBCompressedFiles(bool $reverse = false): void
{
$this->_reverse = $reverse;
if ($this->_reverse) {
if (! krsort($this->_nzbContents)) {
return;
}
} else {
$this->_triedCompressedMids = [];
}
$failed = $downloaded = 0;
// Loop through the files, attempt to find if password-ed and files. Starting with what not to process.
foreach ($this->_nzbContents as $nzbFile) {
if ($downloaded >= $this->_maximumRarSegments) {
break;
}
if ($failed >= $this->_maximumRarPasswordChecks) {
break;
}
if ($this->_releaseHasPassword) {
$this->_echo('Skipping processing of rar '.$nzbFile['title'].' it has a password.', 'primaryOver');
break;
}
// Probably not a rar/zip.
if (! preg_match(
'/\.(part\d+|[r|z]\d+|rar|0+|0*10?|zipr\d{2,3}|zipx?)(\s*\.rar)*($|[ ")\]-])|"[a-f0-9]{32}\.[1-9]\d{1,2}".*\(\d+\/\d{2,}\)$/i',
$nzbFile['title']
)
) {
continue;
}
// Get message-id's for the rar file.
$segCount = (\count($nzbFile['segments']) - 1);
$mID = [];
for ($i = 0; $i < $this->_maximumRarSegments; $i++) {
if ($i > $segCount) {
break;
}
$segment = (string) $nzbFile['segments'][$i];
if (! $this->_reverse) {
$this->_triedCompressedMids[] = $segment;
} elseif (\in_array($segment, $this->_triedCompressedMids, false)) {
// We already downloaded this file.
continue 2;
}
$mID[] = $segment;
}
// Nothing to download.
if (empty($mID)) {
continue;
}
// Download the article(s) from usenet.
$fetchedBinary = $this->_nntp->getMessages($this->_releaseGroupName, $mID, $this->_alternateNNTP);
if ($this->_nntp::isError($fetchedBinary)) {
$fetchedBinary = false;
}
if ($fetchedBinary !== false) {
// Echo we downloaded compressed file.
if ($this->_echoCLI) {
$this->_echo('(cB)', 'primaryOver');
}
$downloaded++;
// Process the compressed file.
$decompressed = $this->_processCompressedData($fetchedBinary);
if ($decompressed || $this->_releaseHasPassword) {
break;
}
} else {
$failed++;
if ($this->_echoCLI) {
$this->_echo('f('.$failed.')', 'warningOver');
}
}
}
}
/**
* Check if the data is a ZIP / RAR file, extract files, get file info.
*
*
* @throws \Exception
*/
protected function _processCompressedData(string &$compressedData): bool
{
$this->_compressedFilesChecked++;
// Give the data to archive info so it can check if it's a rar.
if (! $this->_archiveInfo->setData($compressedData, true)) {
if (config('app.debug') === true) {
$this->_debug('Data is probably not RAR or ZIP.');
}
return false;
}
// Check if there's an error.
if ($this->_archiveInfo->error !== '') {
if (config('app.debug') === true) {
$this->_debug('ArchiveInfo Error: '.$this->_archiveInfo->error);
}
return false;
}
try {
// Get a summary of the compressed file.
$dataSummary = $this->_archiveInfo->getSummary(true);
} catch (\Exception $exception) {
// Log the exception and continue to next item
if (config('app.debug') === true) {
Log::warning($exception->getTraceAsString());
}
return false;
}
// Check if the compressed file is encrypted.
if (! empty($this->_archiveInfo->isEncrypted) || (isset($dataSummary['is_encrypted']) && (int) $dataSummary['is_encrypted'] !== 0)) {
if (config('app.debug') === true) {
$this->_debug('ArchiveInfo: Compressed file has a password.');
}
$this->_releaseHasPassword = true;
$this->_passwordStatus = Releases::PASSWD_RAR;
return false;
}
if ($this->_reverse) {
$fileData = $dataSummary['file_list'] ?? [];
if (! empty($fileData)) {
$rarFileName = Arr::pluck($fileData, 'name');
if (preg_match(NameFixer::PREDB_REGEX, $rarFileName[0], $hit)) {
$preCheck = Predb::whereTitle($hit[0])->first();
$this->_release->preid = $preCheck !== null ? $preCheck->value('id') : 0;
(new NameFixer())->updateRelease($this->_release, $preCheck->title ?? ucwords($hit[0], '.'), 'RarInfo FileName Match', true, 'Filenames, ', true, true, $this->_release->preid);
} elseif (! empty($dataSummary['archives']) && ! empty($dataSummary['archives'][$rarFileName[0]]['file_list'])) {
$archiveData = $dataSummary['archives'][$rarFileName[0]]['file_list'];
$archiveFileName = Arr::pluck($archiveData, 'name');
if (preg_match(NameFixer::PREDB_REGEX, $archiveFileName[0], $match2)) {
$preCheck = Predb::whereTitle($match2[0])->first();
$this->_release->preid = $preCheck !== null ? $preCheck->value('id') : 0;
(new NameFixer())->updateRelease($this->_release, $preCheck->title ?? ucwords($match2[0], '.'), 'RarInfo FileName Match', true, 'Filenames, ', true, true, $this->_release->preid);
}
}
}
}
switch ($dataSummary['main_type']) {
case ArchiveInfo::TYPE_RAR:
if ($this->_echoCLI) {
$this->_echo('r', 'primaryOver');
}
if (! $this->_extractUsingRarInfo && $this->_unrarPath !== false) {
$fileName = $this->tmpPath.uniqid('', true).'.rar';
File::put($fileName, $compressedData);
runCmd($this->_killString.$this->_unrarPath.'" e -ai -ep -c- -id -inul -kb -or -p- -r -y "'.$fileName.'" "'.$this->tmpPath.'unrar/"');
File::delete($fileName);
}
break;
case ArchiveInfo::TYPE_SZIP:
case ArchiveInfo::TYPE_ZIP:
if ($this->_echoCLI) {
$this->_echo('7z', 'primaryOver');
}
if (! $this->_extractUsingRarInfo && ! empty($this->_7zipPath)) {
$fileName = $this->tmpPath.uniqid('', true).'.zip';
File::put($fileName, $compressedData);
// Pass the -p flag to the 7zip command to make sure it doesn't get stuck in password prompt
runCmd($this->_killString.$this->_7zipPath.'" x -tzip "'.$fileName.'" -p -bd -y -o"'.$this->tmpPath.'unzip/"');
File::delete($fileName);
}
break;
default:
return false;
}
return $this->_processCompressedFileList();
}
/**
* Get a list of all files in the compressed file, add the file info to the DB.
*
*
* @throws \Exception
*/
protected function _processCompressedFileList(): bool
{
// Get a list of files inside the Compressed file.
$files = $this->_archiveInfo->getArchiveFileList();
if (! \is_array($files) || \count($files) === 0) {
return false;
}
// Loop through the files.
foreach ($files as $file) {
if ($this->_releaseHasPassword) {
break;
}
if (isset($file['name'])) {
if (isset($file['error'])) {
if (config('app.debug') === true) {
$this->_debug("Error: {$file['error']} (in: {$file['source']})");
}
continue;
}
if (isset($file['pass']) && $file['pass'] === true) {
$this->_releaseHasPassword = true;
$this->_passwordStatus = Releases::PASSWD_RAR;
break;
}
if ($this->_innerFileBlacklist !== false && preg_match($this->_innerFileBlacklist, $file['name'])) {
$this->_releaseHasPassword = true;
$this->_passwordStatus = Releases::PASSWD_RAR;
break;
}
$fileName = [];
if (preg_match('/[^\/\\\\]*\.[a-zA-Z0-9]*$/', $file['name'], $fileName)) {
$fileName = $fileName[0];
} else {
$fileName = '';
}
if ($this->_extractUsingRarInfo) {
// Extract files from the rar.
if (isset($file['compressed']) && (int) $file['compressed'] === 0) {
File::put(
$this->tmpPath.random_int(10, 999999).'_'.$fileName,
$this->_archiveInfo->getFileData($file['name'], $file['source'])
);
} // If the files are compressed, use a binary extractor.
else {
$this->_archiveInfo->extractFile($file['name'], $this->tmpPath.random_int(10, 999999).'_'.$fileName);
}
}
}
$this->_addFileInfo($file);
}
if ($this->_addedFileInfo > 0) {
if (config('nntmux.elasticsearch_enabled') === true) {
$this->elasticsearch->updateRelease($this->_release->id);
} else {
$this->manticore->updateRelease($this->_release->id);
}
}
return $this->_totalFileInfo > 0;
}
/**
* @throws \Exception
*/
protected function _addFileInfo(&$file): void
{
// Don't add rar/zip files to the DB.
if (! isset($file['error']) && isset($file['source']) &&
! preg_match($this->_supportFileRegex.'|part\d+|[r|z]\d{1,3}|zipr\d{2,3}|\d{2,3}|zipx|zip|rar)(\s*\.rar)?$/i', $file['name'])
) {
// Cache the amount of files we find in the RAR or ZIP, return this to say we did find RAR or ZIP content.
// This is so we don't download more RAR or ZIP files for no reason.
$this->_totalFileInfo++;
/* Check if we already have the file or not.
* Also make sure we don't add too many files, some releases have 100's of files, like PS3 releases.
*/
if ($this->_addedFileInfo < 11 && ReleaseFile::query()->where(['releases_id' => $this->_release->id, 'name' => $file['name'], 'size' => $file['size']])->first() === null) {
$addReleaseFiles = ReleaseFile::addReleaseFiles($this->_release->id, $file['name'], $file['size'], $file['date'], $file['pass'], '', $file['crc32'] ?? '');
if (! empty($addReleaseFiles)) {
$this->_addedFileInfo++;
if ($this->_echoCLI) {
$this->_echo('^', 'primaryOver');
}
// Check for "codec spam"
if (preg_match('/alt\.binaries\.movies($|\.divx$)/', $this->_releaseGroupName) &&
preg_match('/[\/\\\\]Codec[\/\\\\]Setup\.exe$/i', $file['name'])
) {
if (config('app.debug') === true) {
$this->_debug('Codec spam found, setting release to potentially passworded.');
}
$this->_releaseHasPassword = true;
$this->_passwordStatus = Releases::PASSWD_RAR;
} //Run a PreDB filename check on insert to try and match the release
elseif ($file['name'] !== '' && ! str_starts_with($file['name'], '.')) {
$this->_release['filename'] = $file['name'];
$this->_release['releases_id'] = $this->_release->id;
$this->_nameFixer->matchPreDbFiles($this->_release, 1, 1, true);
}
}
}
}
}
/**
* Go through all the extracted files in the temp folder and process them.
*
* @throws \Exception
*/
protected function _processExtractedFiles(): void
{
$nestedLevels = 0;
// Go through all the files in the temp folder, look for compressed files, extract them and the nested ones.
while ($nestedLevels < $this->_maxNestedLevels) {
// Break out if we checked more than x compressed files.
if ($this->_compressedFilesChecked >= self::maxCompressedFilesToCheck) {
break;
}
$foundCompressedFile = false;
// Get all the compressed files in the temp folder.
$files = $this->_getTempDirectoryContents('/.*\.([rz]\d{2,}|rar|zipx?|0{0,2}1)($|[^a-z0-9])/i');
if ($files !== false) {
foreach ($files as $file) {
// Check if the file exists.
if (File::isFile($file[0])) {
$rarData = @File::get($file[0]);
if ($rarData !== false) {
$this->_processCompressedData($rarData);
$foundCompressedFile = true;
}
File::delete($file[0]);
}
}
}
// If we found no compressed files, break out.
if (! $foundCompressedFile) {
break;
}
$nestedLevels++;
}
$fileType = [];
// Get all the remaining files in the temp dir.
$files = $this->_getTempDirectoryContents();
if ($files !== false) {
foreach ($files as $file) {
$file = $file->getPathname();
// Skip /. and /..
if (preg_match('/[\/\\\\]\.{1,2}$/', $file)) {
continue;
}
if (File::isFile($file)) {
// Process PAR2 files.
if (! $this->_foundPAR2Info && preg_match('/\.par2$/', $file)) {
$this->_siftPAR2Info($file);
} // Process NFO files.
elseif ($this->_releaseHasNoNFO && preg_match('/(\.(nfo|inf|ofn)|info\.txt)$/i', $file)) {
$this->_processNfoFile($file);
} // Process audio files.
elseif ((! $this->_foundAudioInfo || ! $this->_foundAudioSample) && preg_match('/(.*)'.$this->_audioFileRegex.'$/i', $file, $fileType)) {
// Try to get audio sample/audio media info.
File::move($file, $this->tmpPath.'audiofile.'.$fileType[2]);
$this->_getAudioInfo($this->tmpPath.'audiofile.'.$fileType[2], $fileType[2]);
File::delete($this->tmpPath.'audiofile.'.$fileType[2]);
} // Process JPG files.
elseif (! $this->_foundJPGSample && preg_match('/\.jpe?g$/i', $file)) {
$this->_getJPGSample($file);
File::delete($file);
} // Video sample // video clip // video media info.
elseif ((! $this->_foundSample || ! $this->_foundVideo || ! $this->_foundMediaInfo) && preg_match('/(.*)'.$this->_videoFileRegex.'$/i', $file)) {
$this->_processVideoFile($file);
} // Check file's magic info.
else {
$output = Utility::fileInfo($file);
if (! empty($output)) {
switch (true) {
case ! $this->_foundJPGSample && preg_match('/^JPE?G/i', $output):
$this->_getJPGSample($file);
File::delete($file);
break;
case (! $this->_foundMediaInfo || ! $this->_foundSample || ! $this->_foundVideo) && preg_match('/Matroska data|MPEG v4|MPEG sequence, v2|\WAVI\W/i', $output):
$this->_processVideoFile($file);
break;
case (! $this->_foundAudioSample || ! $this->_foundAudioInfo) && preg_match('/^FLAC|layer III|Vorbis audio/i', $output, $fileType):
switch ($fileType[0]) {
case 'FLAC':
$fileType = 'FLAC';
break;
case 'layer III':
$fileType = 'MP3';
break;
case 'Vorbis audio':
$fileType = 'OGG';
break;
}
File::move($file, $this->tmpPath.'audiofile.'.$fileType);
$this->_getAudioInfo($this->tmpPath.'audiofile.'.$fileType, $fileType);
File::delete($this->tmpPath.'audiofile.'.$fileType);
break;
case ! $this->_foundPAR2Info && stripos($output, 'Parity') === 0:
$this->_siftPAR2Info($file);
break;
}
}
}
}
}
}
}
/**
* Download all binaries from usenet and form samples / get media info / etc from them.
*
* @void
*
* @throws \Exception
*/
protected function _processMessageIDDownloads(): void
{
$this->_processSampleMessageIDs();
$this->_processMediaInfoMessageIDs();
$this->_processAudioInfoMessageIDs();
$this->_processJPGMessageIDs();
}
/**
* Download and process binaries for sample videos.
*
* @void
*
* @throws \Exception
*/
protected function _processSampleMessageIDs(): void
{
// Download and process sample image.
if (! $this->_foundSample || ! $this->_foundVideo) {
if (! empty($this->_sampleMessageIDs)) {
// Download it from usenet.
$sampleBinary = $this->_nntp->getMessages($this->_releaseGroupName, $this->_sampleMessageIDs, $this->_alternateNNTP);
if ($this->_nntp::isError($sampleBinary)) {
$sampleBinary = false;
}
if ($sampleBinary !== false) {
if ($this->_echoCLI) {
$this->_echo('(sB)', 'primaryOver');
}
// Check if it's more than 40 bytes.
if (\strlen($sampleBinary) > 40) {
$fileLocation = $this->tmpPath.'sample_'.random_int(0, 99999).'.avi';
// Try to create the file.
File::put($fileLocation, $sampleBinary);
// Try to get a sample picture.
if (! $this->_foundSample) {
$this->_foundSample = $this->_getSample($fileLocation);
}
// Try to get a sample video.
if (! $this->_foundVideo) {
$this->_foundVideo = $this->_getVideo($fileLocation);
}
}
} elseif ($this->_echoCLI) {
$this->_echo('f', 'warningOver');
}
}
}
}
/**
* Download and process binaries for media info from videos.
*
* @void
*
* @throws \Exception
*/
protected function _processMediaInfoMessageIDs(): void
{
// Download and process mediainfo. Also try to get a sample if we didn't get one yet.
if (! $this->_foundMediaInfo || ! $this->_foundSample || ! $this->_foundVideo) {
if (! empty($this->_MediaInfoMessageIDs)) {
// Try to download it from usenet.
$mediaBinary = $this->_nntp->getMessages($this->_releaseGroupName, $this->_MediaInfoMessageIDs, $this->_alternateNNTP);
if ($this->_nntp::isError($mediaBinary)) {
// If error set it to false.
$mediaBinary = false;
}
if ($mediaBinary !== false) {
if ($this->_echoCLI) {
$this->_echo('(mB)', 'primaryOver');
}
// If it's more than 40 bytes...
if (\strlen($mediaBinary) > 40) {
$fileLocation = $this->tmpPath.'media.avi';
// Create a file on the disk with it.
File::put($fileLocation, $mediaBinary);
// Try to get media info.
if (! $this->_foundMediaInfo) {
$this->_foundMediaInfo = $this->_getMediaInfo($fileLocation);
}
// Try to get a sample picture.
if (! $this->_foundSample) {
$this->_foundSample = $this->_getSample($fileLocation);
}
// Try to get a sample video.
if (! $this->_foundVideo) {
$this->_foundVideo = $this->_getVideo($fileLocation);
}
}
} elseif ($this->_echoCLI) {
$this->_echo('f', 'warningOver');
}
}
}
}
/**
* Download and process binaries for media info from songs.
*
* @void
*
* @throws \Exception
*/
protected function _processAudioInfoMessageIDs(): void
{
// Download audio file, use media info to try to get the artist / album.
if (! $this->_foundAudioInfo || ! $this->_foundAudioSample) {
if (! empty($this->_AudioInfoMessageIDs)) {
// Try to download it from usenet.
$audioBinary = $this->_nntp->getMessages($this->_releaseGroupName, $this->_AudioInfoMessageIDs, $this->_alternateNNTP);
if ($this->_nntp::isError($audioBinary)) {
$audioBinary = false;
}
if ($audioBinary !== false) {
if ($this->_echoCLI) {
$this->_echo('(aB)', 'primaryOver');
}
$fileLocation = $this->tmpPath.'audio.'.$this->_AudioInfoExtension;
// Create a file with it.
File::put($fileLocation, $audioBinary);
// Try to get media info / sample of the audio file.
$this->_getAudioInfo($fileLocation, $this->_AudioInfoExtension);
} elseif ($this->_echoCLI) {
$this->_echo('f', 'warningOver');
}
}
}
}
/**
* Download and process binaries for JPG pictures.
*
* @void
*
* @throws \Exception
*/
protected function _processJPGMessageIDs(): void
{
// Download JPG file.
if (! $this->_foundJPGSample && ! empty($this->_JPGMessageIDs)) {
// Try to download it.
$jpgBinary = $this->_nntp->getMessages($this->_releaseGroupName, $this->_JPGMessageIDs, $this->_alternateNNTP);
if ($this->_nntp::isError($jpgBinary)) {
$jpgBinary = false;
}
if ($jpgBinary !== false) {
if ($this->_echoCLI) {
$this->_echo('(jB)', 'primaryOver');
}
// Try to create a file with it.
File::put($this->tmpPath.'samplepicture.jpg', $jpgBinary);
// Check if image exif data is jpeg
if (exif_imagetype($this->tmpPath.'samplepicture.jpg') === IMAGETYPE_JPEG) {
// Try to resize and move it.
$this->_foundJPGSample = ($this->_releaseImage->saveImage($this->_release->guid.'_thumb',
$this->tmpPath.'samplepicture.jpg', $this->_releaseImage->jpgSavePath, 650, 650) === 1);
if ($this->_foundJPGSample) {
// Update the DB to say we got it.
Release::query()->where('id', $this->_release->id)->update(['jpgstatus' => 1]);
if ($this->_echoCLI) {
$this->_echo('j', 'primaryOver');
}
}
File::delete($this->tmpPath.'samplepicture.jpg');
}
} elseif ($this->_echoCLI) {
$this->_echo('f', 'warningOver');
}
}
}
/**
* Update the release to say we processed it.
*/
protected function _finalizeRelease(): void
{
$updateRows = ['haspreview' => 0];
// If samples exist from previous runs, set flags.
if (File::isFile($this->_releaseImage->imgSavePath.$this->_release->guid.'_thumb.jpg')) {
$updateRows = ['haspreview' => 1];
}
if (File::isFile($this->_releaseImage->vidSavePath.$this->_release->guid.'.ogv')) {
$updateRows += ['videostatus' => 1];
}
if (File::isFile($this->_releaseImage->jpgSavePath.$this->_release->guid.'_thumb.jpg')) {
$updateRows += ['jpgstatus' => 1];
}
// Get the amount of files we found inside the RAR/ZIP files.
$releaseFilesCount = ReleaseFile::whereReleasesId($this->_release->id)->count('releases_id');
if ($releaseFilesCount === null) {
$releaseFilesCount = 0;
}
$this->_passwordStatus = max([$this->_passwordStatus]);
// Set the release to no password if password processing is off.
if (! $this->_processPasswords) {
$this->_releaseHasPassword = false;
}
// If we failed to get anything from the RAR/ZIPs, decrement the passwordstatus, if the rar/zip has no password.
if (! $this->_releaseHasPassword && $this->_NZBHasCompressedFile && $releaseFilesCount === 0) {
$release = Release::query()->where('id', $this->_release->id);
$release->decrement('passwordstatus');
$release->update(
$updateRows
);
} // Else update the release with the password status (if the admin enabled the setting).
else {
$updateRows += ['passwordstatus' => $this->_processPasswords ? $this->_passwordStatus : Releases::PASSWD_NONE,
'rarinnerfilecount' => $releaseFilesCount, ];
Release::query()->where('id', $this->_release->id)->update(
$updateRows
);
}
}
/**
* @return bool|string|\Symfony\Component\Finder\SplFileInfo[]
*/
protected function _getTempDirectoryContents(string $pattern = '', string $path = ''): array|bool|string
{
if ($path === '') {
$path = $this->tmpPath;
}
$files = File::allFiles($path);
try {
if ($pattern !== '') {
$allFiles = [];
foreach ($files as $file) {
if (preg_match($pattern, $file->getRelativePathname())) {
$allFiles .= $file;
}
}
return $allFiles;
}
return $files;
} catch (\Throwable $e) {
if (config('app.debug') === true) {
Log::error($e->getTraceAsString());
$this->_debug('ERROR: Could not open temp dir: '.$e->getMessage());
}
return false;
}
}
/**
* @throws \Exception
*/
protected function _getAudioInfo($fileLocation, $fileExtension): bool
{
// Return values.
$retVal = $audVal = false;
// Check if audio sample fetching is on.
if (! $this->_processAudioSample) {
$audVal = true;
}
// Check if media info fetching is on.
if (! $this->_processAudioInfo) {
$retVal = true;
}
$rQuery = Release::query()->where('proc_pp', '=', 0)->where('id', $this->_release->id)->select(['searchname', 'fromname', 'categories_id'])->first();
$musicParent = (string) Category::MUSIC_ROOT;
if ($rQuery === null || ! preg_match(
sprintf(
'/%d\d{3}|%d|%d|%d/',
$musicParent[0],
Category::OTHER_MISC,
Category::MOVIE_OTHER,
Category::TV_OTHER
),
$rQuery->id
)
) {
return false;
}
if (File::isFile($fileLocation)) {
// Check if media info is enabled.
if (! $retVal) {
// Get the media info for the file.
try {
$xmlArray = $this->mediaInfo->getInfo($fileLocation, false);
if ($xmlArray !== null) {
foreach ($xmlArray->getAudios() as $track) {
if ($track->get('album') !== null && $track->get('performer') !== null) {
if ((int) $this->_release->predb_id === 0 && config('nntmux.rename_music_mediainfo')) {
// Make the extension upper case.
$ext = strtoupper($fileExtension);
// Form a new search name.
if (! empty($track->get('recorded_date')) && preg_match('/(?:19|20)\d\d/', $track->get('recorded_date')->getFullname(), $Year)) {
$newName = $track->get('performer')->getFullName().' - '.$track->get('album')->getFullName().' ('.$Year[0].') '.$ext;
} else {
$newName = $track->get('performer')->getFullName().' - '.$track->get('album')->getFullName().' '.$ext;
}
// Get the category or try to determine it.
if ($ext === 'MP3') {
$newCat = Category::MUSIC_MP3;
} elseif ($ext === 'FLAC') {
$newCat = Category::MUSIC_LOSSLESS;
} else {
$newCat = $this->_categorize->determineCategory($rQuery->groups_id, $newName, $rQuery->fromname);
}
$newTitle = escapeString(substr($newName, 0, 255));
// Update the search name.
$release = Release::whereId($this->_release->id);
$release->update(['searchname' => $newTitle, 'categories_id' => $newCat['categories_id'], 'iscategorized' => 1, 'isrenamed' => 1, 'proc_pp' => 1]);
if (config('nntmux.elasticsearch_enabled') === true) {
$this->elasticsearch->updateRelease($this->_release->id);
} else {
$this->manticore->updateRelease($this->_release->id);
}
// Echo the changed name.
if ($this->_echoCLI) {
NameFixer::echoChangedReleaseName(
[
'new_name' => $newTitle,
'old_name' => $rQuery->searchname,
'new_category' => $newCat,
'old_category' => $rQuery->id,
'group' => $rQuery->groups_id,
'releases_id' => $this->_release->id,
'method' => 'ProcessAdditional->_getAudioInfo',
]
);
}
}
// Add the media info.
$this->_releaseExtra->addFromXml($this->_release->id, $xmlArray);
$retVal = true;
$this->_foundAudioInfo = true;
if ($this->_echoCLI) {
$this->_echo('a', 'primaryOver');
}
break;
}
}
}
} catch (\RuntimeException $e) {
Log::debug($e->getMessage());
} catch (\TypeError $e) {
Log::debug($e->getMessage());
}
}
// Check if creating audio samples is enabled.
if (! $audVal) {
// File name to store audio file.
$audioFileName = ($this->_release->guid.'.ogg');
// Create an audio sample.
if ($this->ffprobe->isValid($fileLocation)) {
try {
$audioSample = $this->ffmpeg->open($fileLocation);
$format = new Vorbis();
$audioSample->clip(TimeCode::fromSeconds(30), TimeCode::fromSeconds(30));
$audioSample->save($format, $this->tmpPath.$audioFileName);
} catch (\InvalidArgumentException $e) {
if (config('app.debug') === true) {
Log::error($e->getTraceAsString());
}
//We do nothing, just prevent displaying errors because the file cannot be open(corrupted or incomplete file)
}
}
// Check if the new file was created.
if (File::isFile($this->tmpPath.$audioFileName)) {
// Try to move the temp audio file.
$renamed = File::move($this->tmpPath.$audioFileName, $this->_audioSavePath.$audioFileName);
if (! $renamed) {
// Try to copy it if it fails.
$copied = File::copy($this->tmpPath.$audioFileName, $this->_audioSavePath.$audioFileName);
// Delete the old file.
File::delete($this->tmpPath.$audioFileName);
// If it didn't copy continue.
if (! $copied) {
return false;
}
}
// Try to set the file perms.
@chmod($this->_audioSavePath.$audioFileName, 0764);
// Update DB to said we got a audio sample.
Release::query()->where('id', $this->_release->id)->update(['audiostatus' => 1]);
$audVal = $this->_foundAudioSample = true;
if ($this->_echoCLI) {
$this->_echo('A', 'primaryOver');
}
}
}
}
return $retVal && $audVal;
}
/**
* Try to get JPG picture, resize it and store it on disk.
*/
protected function _getJPGSample(string $fileLocation): void
{
// Try to resize/move the image.
$this->_foundJPGSample = (
$this->_releaseImage->saveImage(
$this->_release->guid.'_thumb',
$fileLocation,
$this->_releaseImage->jpgSavePath,
650,
650
) === 1
);
// If it's successful, tell the DB.
if ($this->_foundJPGSample) {
Release::query()->where('id', $this->_release->id)->update(['jpgstatus' => 1]);
}
}
private function getVideoTime(string $videoLocation): string
{
// Get the real duration of the file.
if ($this->ffprobe->isValid($videoLocation)) {
$time = $this->ffprobe->format($videoLocation)->get('duration');
}
if (empty($time) || ! preg_match('/time=(\d{1,2}:\d{1,2}:)?(\d{1,2})\.(\d{1,2})\s*bitrate=/i', $time, $numbers)) {
return '';
}
// Reduce the last number by 1, this is to make sure we don't ask avconv/ffmpeg for non existing data.
if ($numbers[3] > 0) {
$numbers[3]--;
} elseif ($numbers[1] > 0) {
$numbers[2]--;
$numbers[3] = '99';
}
// Manually pad the numbers in case they are 1 number. to get 02 for example instead of 2.
return '00:00:'.str_pad($numbers[2], 2, '0', STR_PAD_LEFT).'.'.str_pad($numbers[3], 2, '0', STR_PAD_LEFT);
}
/**
* @throws \Exception
*/
protected function _getSample(string $fileLocation): bool
{
if (! $this->_processThumbnails) {
return false;
}
if (File::isFile($fileLocation)) {
// Create path to temp file.
$fileName = ($this->tmpPath.'zzzz'.random_int(5, 12).random_int(5, 12).'.jpg');
$time = $this->getVideoTime($fileLocation);
// Create the image.
if ($this->ffprobe->isValid($fileLocation)) {
try {
$this->ffmpeg->open($fileLocation)->frame(TimeCode::fromString($time === '' ? '00:00:03:00' : $time))->save($fileName);
} catch (\RuntimeException $runtimeException) {
if (config('app.debug') === true) {
Log::error($runtimeException->getTraceAsString());
}
//We show no error we just log it, we failed to save the frame and move on
} catch (\InvalidArgumentException $e) {
if (config('app.debug') === true) {
Log::error($e->getTraceAsString());
}
//We do nothing, just prevent displaying errors because the file cannot be open(corrupted or incomplete file)
} catch (\Throwable $e) {
if (config('app.debug') === true) {
Log::error($e->getTraceAsString());
}
//Again we do nothing, we just want to catch the error
}
}
// Check if the file exists.
if (File::isFile($fileName)) {
// Try to resize/move the image.
$saved = $this->_releaseImage->saveImage(
$this->_release->guid.'_thumb',
$fileName,
$this->_releaseImage->imgSavePath,
800,
600
);
// Delete the temp file we created.
File::delete($fileName);
// Check if it saved.
if ($saved === 1) {
if ($this->_echoCLI) {
$this->_echo('s', 'primaryOver');
}
return true;
}
}
}
return false;
}
/**
* @throws \Exception
*/
protected function _getVideo(string $fileLocation): bool
{
if (! $this->_processVideo) {
return false;
}
// Try to find an avi file.
if (File::isFile($fileLocation)) {
// Create a filename to store the temp file.
$fileName = ($this->tmpPath.'zzzz'.$this->_release->guid.'.ogv');
$newMethod = false;
// If wanted sample length is less than 60, try to get sample from the end of the video.
if ($this->_ffMPEGDuration < 60) {
// Get the real duration of the file.
$time = $this->getVideoTime($fileLocation);
if ($time !== '' && preg_match('/(\d{2}).(\d{2})/', $time, $numbers)) {
$newMethod = true;
// Get the lowest time we can start making the video at based on how many seconds the admin wants the video to be.
if ($numbers[1] <= $this->_ffMPEGDuration) {
// If the clip is shorter than the length we want.
// The lowest we want is 0.
$lowestLength = '00:00:00.00';
} else {
// If the clip is longer than the length we want.
// The lowest we want is the the difference between the max video length and our wanted total time.
$lowestLength = ($numbers[1] - $this->_ffMPEGDuration);
// Form the time string.
$end = '.'.$numbers[2];
$lowestLength = match (\strlen($lowestLength)) {
1 => ('00:00:0'.$lowestLength.$end),
2 => ('00:00:'.$lowestLength.$end),
default => '00:00:60.00',
};
}
// Try to get the sample (from the end instead of the start).
if ($this->ffprobe->isValid($fileLocation)) {
try {
$video = $this->ffmpeg->open($fileLocation);
$videoSample = $video->clip(TimeCode::fromString($lowestLength), TimeCode::fromSeconds($this->_ffMPEGDuration));
$format = new Ogg();
$format->setAudioCodec('libvorbis');
$videoSample->filters()->resize(new Dimension(320, -1), ResizeFilter::RESIZEMODE_SCALE_HEIGHT);
$videoSample->save($format, $fileName);
} catch (\InvalidArgumentException $e) {
if (config('app.debug') === true) {
Log::error($e->getTraceAsString());
}
//We do nothing, just prevent displaying errors because the file cannot be open(corrupted or incomplete file)
}
}
}
}
// If longer than 60 or we could not get the video length, run the old way.
if (! $newMethod && $this->ffprobe->isValid($fileLocation)) {
try {
$video = $this->ffmpeg->open($fileLocation);
$videoSample = $video->clip(TimeCode::fromSeconds(0), TimeCode::fromSeconds($this->_ffMPEGDuration));
$format = new Ogg();
$format->setAudioCodec('libvorbis');
$videoSample->filters()->resize(new Dimension(320, -1), ResizeFilter::RESIZEMODE_SCALE_HEIGHT);
$videoSample->save($format, $fileName);
} catch (\InvalidArgumentException $e) {
if (config('app.debug') === true) {
Log::error($e->getTraceAsString());
}
//We do nothing, just prevent displaying errors because the file cannot be open(corrupted or incomplete file)
}
}
// Until we find the video file.
if (File::isFile($fileName)) {
// Create a path to where the file should be moved.
$newFile = ($this->_releaseImage->vidSavePath.$this->_release->guid.'.ogv');
// Try to move the file to the new path.
// If we couldn't rename it, try to copy it.
if (! @File::move($fileName, $newFile)) {
$copied = @File::copy($fileName, $newFile);
// Delete the old file.
File::delete($fileName);
// If it didn't copy, continue.
if (! $copied) {
return false;
}
}
// Change the permissions.
@chmod($newFile, 0764);
// Update query to say we got the video.
Release::query()->where('guid', $this->_release->guid)->update(['videostatus' => 1]);
if ($this->_echoCLI) {
$this->_echo('v', 'primaryOver');
}
return true;
}
}
return false;
}
/**
* @throws \Exception
*/
protected function _getMediaInfo($fileLocation): bool
{
if (! $this->_processMediaInfo) {
return false;
}
// Look for the video file.
if (File::isFile($fileLocation)) {
try {
$xmlArray = $this->mediaInfo->getInfo($fileLocation, true);
// Check if we got it.
if ($xmlArray === null) {
return false;
}
// Insert it into the DB.
$this->_releaseExtra->addFull($this->_release->id, $xmlArray);
$this->_releaseExtra->addFromXml($this->_release->id, $xmlArray);
if ($this->_echoCLI) {
$this->_echo('m', 'primaryOver');
}
return true;
} catch (\RuntimeException $e) {
Log::debug($e->getMessage());
return false;
} catch (\TypeError $e) {
Log::debug($e->getMessage());
return false;
} catch (\ErrorException $e) {
Log::debug($e->getMessage());
return false;
}
}
return false;
}
/**
* @throws \Exception
*/
protected function _siftPAR2Info($fileLocation): void
{
$this->_par2Info->open($fileLocation);
if ($this->_par2Info->error) {
return;
}
$releaseInfo = Release::query()->where('id', $this->_release->id)->select(['postdate', 'proc_pp'])->first();
if ($releaseInfo === null) {
return;
}
$postDate = Carbon::createFromFormat('Y-m-d H:i:s', $releaseInfo->postdate)->getTimestamp();
// Only get a new name if the category is OTHER.
$foundName = true;
if ((int) $releaseInfo->proc_pp === 0 && config('nntmux.rename_par2') &&
\in_array(
(int) $this->_release->categories_id,
Category::OTHERS_GROUP,
false
)
) {
$foundName = false;
}
$filesAdded = 0;
foreach ($this->_par2Info->getFileList() as $file) {
if (! isset($file['name'])) {
continue;
}
// If we found a name and added 10 files, stop.
if ($foundName && $filesAdded > 10) {
break;
}
// Add to release files.
if ($this->_addPAR2Files) {
if ($filesAdded < 11 && ReleaseFile::query()->where(['releases_id' => $this->_release->id, 'name' => $file['name']])->first() === null
) {
// Try to add the files to the DB.
if (ReleaseFile::addReleaseFiles($this->_release->id, $file['name'], $file['size'], $postDate, 0, $file['hash_16K'])) {
$filesAdded++;
}
}
} else {
$filesAdded++;
}
// Try to get a new name.
if (! $foundName) {
$this->_release->textstring = $file['name'];
$this->_release->releases_id = $this->_release->id;
if ($this->_nameFixer->checkName($this->_release, $this->_echoCLI, 'PAR2, ', 1, 1)) {
$foundName = true;
}
}
}
// Update the file count with the new file count + old file count.
Release::query()->where('id', $this->_release->id)->increment('rarinnerfilecount', $filesAdded);
$this->_foundPAR2Info = true;
}
/**
* @throws \Exception
*/
protected function _processNfoFile($fileLocation): void
{
$data = @File::get($fileLocation);
if ($data != false && $this->_nfo->isNFO($data, $this->_release->guid) && $this->_nfo->addAlternateNfo($data, $this->_release, $this->_nntp)) {
$this->_releaseHasNoNFO = false;
}
}
/**
* @throws \Exception
*/
protected function _processVideoFile($fileLocation): void
{
// Try to get a sample with it.
if (! $this->_foundSample) {
$this->_foundSample = $this->_getSample($fileLocation);
}
/* Try to get a video with it.
* Don't get it here if _sampleMessageIDs is empty
* or has 1 message-id (Saves downloading another part).
*/
if (! $this->_foundVideo && \count($this->_sampleMessageIDs) < 2) {
$this->_foundVideo = $this->_getVideo($fileLocation);
}
// Try to get media info with it.
if (! $this->_foundMediaInfo) {
$this->_foundMediaInfo = $this->_getMediaInfo($fileLocation);
}
}
/**
* Comparison function for uSort, for sorting NZB files.
*/
protected function _sortNZB(array|string|null $a, array|string|null $b): int
{
$pos = 0;
$af = $bf = false;
$a = preg_replace('/\d+[ ._-]?(\/|\||[o0]f)[ ._-]?\d+?(?![ ._-]\d)/i', ' ', $a['title']);
$b = preg_replace('/\d+[ ._-]?(\/|\||[o0]f)[ ._-]?\d+?(?![ ._-]\d)/i', ' ', $b['title']);
if (preg_match('/\.(part\d+|[r|z]\d+)(\s*\.rar)*($|[ ")\]-])/i', $a)) {
$af = true;
}
if (preg_match('/\.(part\d+|[r|z]\d+)(\s*\.rar)*($|[ ")\]-])/i', $b)) {
$bf = true;
}
if (! $af && preg_match('/\.rar($|[ ")\]-])/i', $a)) {
$a = preg_replace('/\.rar(?:$|[ ")\]-])/i', '.*rar', $a);
$af = true;
}
if (! $bf && preg_match('/\.rar($|[ ")\]-])/i', $b)) {
$b = preg_replace('/\.rar(?:$|[ ")\]-])/i', '.*rar', $b);
$bf = true;
}
if (! $af && ! $bf) {
return strnatcasecmp($a, $b);
}
if (! $bf) {
return -1;
}
if (! $af) {
return 1;
}
if ($af && $bf) {
return strnatcasecmp($a, $b);
}
if ($af) {
return -1;
}
if ($bf) {
return 1;
}
return $pos;
}
/**
* Reset some variables for the current release.
*/
protected function _resetReleaseStatus(): void
{
// Only process for samples, previews and images if not disabled.
$this->_foundVideo = ! $this->_processVideo;
$this->_foundMediaInfo = ! $this->_processMediaInfo;
$this->_foundAudioInfo = ! $this->_processAudioInfo;
$this->_foundAudioSample = ! $this->_processAudioSample;
$this->_foundJPGSample = ! $this->_processJPGSample;
$this->_foundSample = ! $this->_processThumbnails;
$this->_foundPAR2Info = false;
$this->_passwordStatus = Releases::PASSWD_NONE;
$this->_releaseHasPassword = false;
$this->_releaseGroupName = UsenetGroup::getNameByID($this->_release->groups_id);
$this->_releaseHasNoNFO = false;
// Make sure we don't already have an nfo.
if ((int) $this->_release->nfostatus !== 1) {
$this->_releaseHasNoNFO = true;
}
$this->_NZBHasCompressedFile = false;
$this->_sampleMessageIDs = $this->_JPGMessageIDs = $this->_MediaInfoMessageIDs = [];
$this->_AudioInfoMessageIDs = $this->_RARFileMessageIDs = [];
$this->_AudioInfoExtension = '';
$this->_addedFileInfo = 0;
$this->_totalFileInfo = 0;
$this->_compressedFilesChecked = 0;
}
/**
* Echo a string to CLI.
*
* @param string $string String to echo.
* @param string $type Method type.
*
* @void
*/
protected function _echo(string $string, string $type): void
{
if ($this->_echoCLI) {
(new ColorCLI())->$type($string);
}
}
/**
* Echo a string to CLI. For debugging.
*
*
* @void
*/
protected function _debug(string $string): void
{
$this->_echo('DEBUG: '.$string, 'debug');
}
}