mirror of
https://github.com/NNTmux/newznab-tmux.git
synced 2026-08-28 17:01:16 +00:00
Darko
315 lines
8.2 KiB
PHP
315 lines
8.2 KiB
PHP
<?php
|
|
|
|
use newznab\db\Settings;
|
|
use \newznab\processing\PProcess;
|
|
use newznab\utility\Utility;
|
|
|
|
/**
|
|
* Gets information contained within the NZB.
|
|
*
|
|
* Class NZBContents
|
|
*/
|
|
Class NZBContents
|
|
{
|
|
/**
|
|
* @var newznab\db\Settings
|
|
* @access protected
|
|
*/
|
|
public $pdo;
|
|
|
|
/**
|
|
* @var NNTP
|
|
* @access protected
|
|
*/
|
|
protected $nntp;
|
|
|
|
/**
|
|
* @var Info
|
|
* @access protected
|
|
*/
|
|
protected $nfo;
|
|
|
|
/**
|
|
* @var PProcess
|
|
* @access protected
|
|
*/
|
|
protected $pp;
|
|
|
|
/**
|
|
* @var NZB
|
|
* @access protected
|
|
*/
|
|
protected $nzb;
|
|
|
|
/**
|
|
* @var bool|stdClass
|
|
* @access protected
|
|
*/
|
|
protected $site;
|
|
|
|
/**
|
|
* @var bool
|
|
* @access protected
|
|
*/
|
|
protected $lookuppar2;
|
|
|
|
/**
|
|
* @var bool
|
|
* @access protected
|
|
*/
|
|
protected $echooutput;
|
|
|
|
/**
|
|
* Construct.
|
|
*
|
|
* @param array $options
|
|
* array(
|
|
* 'Echo' => bool ; To echo to CLI or not.
|
|
* 'NNTP' => NNTP ; Class NNTP.
|
|
* 'Nfo' => Nfo ; Class Info.
|
|
* 'NZB' => NZB ; Class NZB.
|
|
* 'Settings' => DB ; Class newznab\db\Settings.
|
|
* 'PostProcess' => PProcess ; Class PProcess.
|
|
* )
|
|
*
|
|
* @access public
|
|
*/
|
|
public function __construct(array $options = array())
|
|
{
|
|
$defaults = [
|
|
'Echo' => false,
|
|
'NNTP' => null,
|
|
'Nfo' => null,
|
|
'NZB' => null,
|
|
'Settings' => null,
|
|
'PostProcess' => null,
|
|
];
|
|
$options += $defaults;
|
|
|
|
$this->echooutput = ($options['Echo'] && NN_ECHOCLI);
|
|
$this->pdo = ($options['Settings'] instanceof Settings ? $options['Settings'] : new Settings());
|
|
$this->nntp = ($options['NNTP'] instanceof \NNTP ? $options['NNTP'] : new \NNTP(['Echo' => $this->echooutput, 'Settings' => $this->pdo]));
|
|
$this->nfo = ($options['Nfo'] instanceof \Info ? $options['Nfo'] : new \Info(['Echo' => $this->echooutput, 'Settings' => $this->pdo]));
|
|
$this->pp = (
|
|
$options['PostProcess'] instanceof PProcess
|
|
? $options['PostProcess']
|
|
: new PProcess(['Echo' => $this->echooutput, 'Nfo' => $this->nfo, 'Settings' => $this->pdo])
|
|
);
|
|
$this->nzb = ($options['NZB'] instanceof \NZB ? $options['NZB'] : new \NZB($this->pdo));
|
|
$t = new Tmux();
|
|
$this->tmux = $t->get();
|
|
$s = new Sites();
|
|
$this->site = $s->get();
|
|
$this->lookuppar2 = ($this->tmux->lookuppar2 == 1 ? true : false);
|
|
$this->alternateNNTP = ($this->pdo->getSetting('alternate_nntp') == 1 ? true : false);
|
|
}
|
|
|
|
/**
|
|
* Look for an .nfo file in the NZB, return the NFO message id.
|
|
* Gets the NZB completion.
|
|
* Looks for PAR2 files in the NZB.
|
|
*
|
|
* @param string $guid
|
|
* @param string $relID
|
|
* @param int $groupID
|
|
* @param string $groupName
|
|
*
|
|
* @return bool
|
|
*
|
|
* @access public
|
|
*/
|
|
public function getNfoFromNZB($guid, $relID, $groupID, $groupName)
|
|
{
|
|
$fetchedBinary = false;
|
|
|
|
$messageID = $this->parseNZB($guid, $relID, $groupID, true);
|
|
if ($messageID !== false) {
|
|
$fetchedBinary = $this->nntp->getMessages($groupName, $messageID['id'], $this->alternateNNTP);
|
|
if ($this->nntp->isError($fetchedBinary)) {
|
|
// NFO download failed, increment attempts.
|
|
$this->pdo->queryExec(sprintf('UPDATE releases SET nfostatus = nfostatus - 1 WHERE id = %d', $relID));
|
|
if ($this->echooutput) {
|
|
echo 'f';
|
|
}
|
|
return false;
|
|
}
|
|
if ($this->nfo->isNFO($fetchedBinary, $guid) === true) {
|
|
if ($this->echooutput) {
|
|
echo ($messageID['hidden'] === false ? '+' : '*');
|
|
}
|
|
} else {
|
|
if ($this->echooutput) {
|
|
echo '-';
|
|
}
|
|
$this->pdo->queryExec(sprintf('UPDATE releases SET nfostatus = %d WHERE id = %d', Info::NFO_NONFO, $relID));
|
|
$fetchedBinary = false;
|
|
}
|
|
} else {
|
|
if ($this->echooutput) {
|
|
echo '-';
|
|
}
|
|
$this->pdo->queryExec(sprintf('UPDATE releases SET nfostatus = %d WHERE id = %d', Info::NFO_NONFO, $relID));
|
|
}
|
|
|
|
return $fetchedBinary;
|
|
}
|
|
|
|
/**
|
|
* Gets the completion from the NZB, optionally looks if there is an NFO/PAR2 file.
|
|
*
|
|
* @param string $guid
|
|
* @param int $relID
|
|
* @param int $groupID
|
|
* @param bool $nfoCheck
|
|
*
|
|
* @return array|bool
|
|
*
|
|
* @access public
|
|
*/
|
|
public function parseNZB($guid, $relID, $groupID, $nfoCheck = false)
|
|
{
|
|
$nzbFile = $this->LoadNZB($guid);
|
|
if ($nzbFile !== false) {
|
|
$messageID = $hiddenID = '';
|
|
$actualParts = $artificialParts = 0;
|
|
$foundPAR2 = ($this->lookuppar2 === false ? true : false);
|
|
$foundNFO = $hiddenNFO = ($nfoCheck === false ? true : false);
|
|
|
|
foreach ($nzbFile->file as $nzbcontents) {
|
|
foreach ($nzbcontents->segments->segment as $segment) {
|
|
$actualParts++;
|
|
}
|
|
|
|
$subject = (string)$nzbcontents->attributes()->subject;
|
|
if (preg_match('/(\d+)\)$/', $subject, $parts)) {
|
|
$artificialParts += $parts[1];
|
|
}
|
|
|
|
if ($foundNFO === false) {
|
|
if (preg_match('/\.\b(nfo|inf|ofn)\b(?![ .-])/i', $subject)) {
|
|
$messageID = (string)$nzbcontents->segments->segment;
|
|
$foundNFO = true;
|
|
}
|
|
}
|
|
|
|
if ($foundNFO === false && $hiddenNFO === false) {
|
|
if (preg_match('/\(1\/1\)$/i', $subject) &&
|
|
!preg_match('/\.(apk|bat|bmp|cbr|cbz|cfg|css|csv|cue|db|dll|doc|epub|exe|gif|htm|ico|idx|ini' .
|
|
'|jpg|lit|log|m3u|mid|mobi|mp3|nib|nzb|odt|opf|otf|par|par2|pdf|psd|pps|png|ppt|r\d{2,4}' .
|
|
'|rar|sfv|srr|sub|srt|sql|rom|rtf|tif|torrent|ttf|txt|vb|vol\d+\+\d+|wps|xml|zip)/i',
|
|
$subject))
|
|
{
|
|
$hiddenID = (string)$nzbcontents->segments->segment;
|
|
$hiddenNFO = true;
|
|
}
|
|
}
|
|
|
|
if ($foundPAR2 === false) {
|
|
if (preg_match('/\.(par[2" ]|\d{2,3}").+\(1\/1\)$/i', $subject)) {
|
|
if ($this->pp->parsePAR2((string)$nzbcontents->segments->segment, $relID, $groupID, $this->nntp, 1) === true) {
|
|
$this->pdo->queryExec(sprintf('UPDATE releases SET proc_par2 = 1 WHERE id = %d', $relID));
|
|
$foundPAR2 = true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if ($artificialParts <= 0 || $actualParts <= 0) {
|
|
$completion = 0;
|
|
} else {
|
|
$completion = ($actualParts / $artificialParts) * 100;
|
|
}
|
|
if ($completion > 100) {
|
|
$completion = 100;
|
|
}
|
|
|
|
$this->pdo->queryExec(sprintf('UPDATE releases SET completion = %d WHERE id = %d', $completion, $relID));
|
|
|
|
if ($foundNFO === true && strlen($messageID) > 1) {
|
|
return array('hidden' => false, 'id' => $messageID);
|
|
} elseif ($hiddenNFO === true && strlen($hiddenID) > 1) {
|
|
return array('hidden' => true, 'id' => $hiddenID);
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Decompress a NZB, load it into simplexml and return.
|
|
*
|
|
* @param string $guid Release guid.
|
|
*
|
|
* @return bool|SimpleXMLElement
|
|
*
|
|
* @access public
|
|
*/
|
|
public function LoadNZB($guid)
|
|
{
|
|
// Fetch the NZB location using the GUID.
|
|
$nzbPath = $this->nzb->NZBPath($guid);
|
|
if ($nzbPath === false) {
|
|
if ($this->echooutput) {
|
|
echo PHP_EOL . $guid . ' appears to be missing the nzb file, skipping.' . PHP_EOL;
|
|
}
|
|
return false;
|
|
}
|
|
$nzbContents = Utility::unzipGzipFile($nzbPath);
|
|
if (!$nzbContents) {
|
|
if ($this->echooutput) {
|
|
echo
|
|
PHP_EOL .
|
|
'Unable to decompress: ' .
|
|
$nzbPath .
|
|
' - ' .
|
|
fileperms($nzbPath) .
|
|
' - may have bad file permissions, skipping.' .
|
|
PHP_EOL;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
$nzbFile = @simplexml_load_string($nzbContents);
|
|
if (!$nzbFile) {
|
|
if ($this->echooutput) {
|
|
echo PHP_EOL . "Unable to load NZB: $guid appears to be an invalid NZB, skipping." . PHP_EOL;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
return $nzbFile;
|
|
}
|
|
|
|
/**
|
|
* Attempts to get the releasename from a par2 file
|
|
*
|
|
* @param string $guid
|
|
* @param int $relID
|
|
* @param int $groupID
|
|
* @param int $nameStatus
|
|
* @param int $show
|
|
*
|
|
* @return bool
|
|
*
|
|
* @access public
|
|
*/
|
|
public function checkPAR2($guid, $relID, $groupID, $nameStatus, $show)
|
|
{
|
|
$nzbFile = $this->LoadNZB($guid);
|
|
if ($nzbFile !== false) {
|
|
foreach ($nzbFile->file as $nzbContents) {
|
|
if (preg_match('/\.(par[2" ]|\d{2,3}").+\(1\/1\)$/i', (string)$nzbContents->attributes()->subject)) {
|
|
if ($this->pp->parsePAR2((string)$nzbContents->segments->segment, $relID, $groupID, $this->nntp, $show) === true && $nameStatus === 1) {
|
|
$this->pdo->queryExec(sprintf('UPDATE releases SET proc_par2 = 1 WHERE id = %d', $relID));
|
|
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
if ($nameStatus === 1) {
|
|
$this->pdo->queryExec(sprintf('UPDATE releases SET proc_par2 = 1 WHERE id = %d', $relID));
|
|
}
|
|
|
|
return false;
|
|
}
|
|
} |