mirror of
https://github.com/NNTmux/newznab-tmux.git
synced 2026-08-28 17:01:16 +00:00
DariusIII
473 lines
15 KiB
PHP
Executable File
473 lines
15 KiB
PHP
Executable File
<?php
|
|
|
|
namespace nntmux;
|
|
|
|
use nntmux\db\DB;
|
|
use App\Models\Settings;
|
|
use nntmux\utility\Utility;
|
|
use App\Extensions\util\Versions;
|
|
|
|
/**
|
|
* Class for reading and writing NZB files on the hard disk,
|
|
* building folder paths to store the NZB files.
|
|
*/
|
|
class NZB
|
|
{
|
|
const NZB_NONE = 0; // Release has no NZB file yet.
|
|
const NZB_ADDED = 1; // Release had an NZB file created.
|
|
|
|
const NZB_DTD_NAME = 'nzb';
|
|
const NZB_DTD_PUBLIC = '-//newzBin//DTD NZB 1.1//EN';
|
|
const NZB_DTD_EXTERNAL = 'http://www.newzbin.com/DTD/nzb/nzb-1.1.dtd';
|
|
const NZB_XML_NS = 'http://www.newzbin.com/DTD/2003/nzb';
|
|
|
|
/**
|
|
* Levels deep to store NZB files.
|
|
*
|
|
* @var int
|
|
*/
|
|
protected $nzbSplitLevel;
|
|
|
|
/**
|
|
* Path to store NZB files.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $siteNzbPath;
|
|
|
|
/**
|
|
* Group id when writing NZBs.
|
|
*
|
|
* @var int
|
|
*/
|
|
protected $groupID;
|
|
|
|
/**
|
|
* @var \nntmux\db\DB
|
|
*/
|
|
public $pdo;
|
|
|
|
/**
|
|
* @var bool
|
|
*/
|
|
protected $_debug = false;
|
|
|
|
/**
|
|
* Base query for selecting collection data for writing NZB files.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $_collectionsQuery;
|
|
|
|
/**
|
|
* Base query for selecting binary data for writing NZB files.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $_binariesQuery;
|
|
|
|
/**
|
|
* Base query for selecting parts data for writing NZB files.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $_partsQuery;
|
|
|
|
/**
|
|
* String used for head in NZB XML file.
|
|
*
|
|
* @var string
|
|
*/
|
|
protected $_nzbCommentString;
|
|
|
|
/**
|
|
* Names of CBP tables.
|
|
*
|
|
* @var array [string => string]
|
|
*/
|
|
protected $_tableNames;
|
|
|
|
/**
|
|
* @var string
|
|
*/
|
|
protected $_siteCommentString;
|
|
|
|
/**
|
|
* NZB constructor.
|
|
*
|
|
* @param array $options
|
|
* @throws \Exception
|
|
*/
|
|
public function __construct(array $options = [])
|
|
{
|
|
$defaults = [
|
|
'Settings' => null,
|
|
];
|
|
$options += $defaults;
|
|
|
|
$this->pdo = ($options['Settings'] instanceof DB ? $options['Settings'] : new DB());
|
|
|
|
$nzbSplitLevel = Settings::settingValue('..nzbsplitlevel');
|
|
$this->nzbSplitLevel = $nzbSplitLevel ?? 1;
|
|
$this->siteNzbPath = (string) Settings::settingValue('..nzbpath');
|
|
if (substr($this->siteNzbPath, -1) !== DS) {
|
|
$this->siteNzbPath .= DS;
|
|
}
|
|
$this->_nzbCommentString = sprintf(
|
|
'NZB Generated by: NNTmux %s %s',
|
|
(new Versions())->getGitTagInFile(),
|
|
Utility::htmlfmt(date('F j, Y, g:i a O'))
|
|
);
|
|
$this->_siteCommentString = sprintf(
|
|
'NZB downloaded from %s',
|
|
Settings::settingValue('site.main.title')
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Initiate class vars when writing NZB's.
|
|
*
|
|
* @param int $groupID
|
|
*/
|
|
public function initiateForWrite($groupID)
|
|
{
|
|
$this->groupID = $groupID;
|
|
// Set table names
|
|
|
|
if ($this->groupID === '') {
|
|
exit("{$this->groupID} is missing\n");
|
|
}
|
|
$this->_tableNames = [
|
|
'cName' => 'collections_'.$this->groupID,
|
|
'bName' => 'binaries_'.$this->groupID,
|
|
'pName' => 'parts_'.$this->groupID,
|
|
];
|
|
$this->setQueries();
|
|
}
|
|
|
|
protected function setQueries(): void
|
|
{
|
|
$this->_collectionsQuery = "
|
|
SELECT c.*, UNIX_TIMESTAMP(c.date) AS udate,
|
|
g.name AS groupname
|
|
FROM {$this->_tableNames['cName']} c
|
|
INNER JOIN groups g ON c.groups_id = g.id
|
|
WHERE c.releases_id = ";
|
|
$this->_binariesQuery = "
|
|
SELECT b.id, b.name, b.totalparts
|
|
FROM {$this->_tableNames['bName']} b
|
|
WHERE b.collections_id = %d
|
|
ORDER BY b.name ASC";
|
|
$this->_partsQuery = "
|
|
SELECT DISTINCT(p.messageid), p.size, p.partnumber
|
|
FROM {$this->_tableNames['pName']} p
|
|
WHERE p.binaries_id = %d
|
|
ORDER BY p.partnumber ASC";
|
|
}
|
|
|
|
/**
|
|
* Write an NZB to the hard drive for a single release.
|
|
*
|
|
* @param int $relID The ID of the release in the DB.
|
|
* @param string $relGuid The guid of the release.
|
|
* @param string $name The name of the release.
|
|
* @param string $cTitle The name of the category this release is in.
|
|
*
|
|
* @return bool Have we successfully written the NZB to the hard drive?
|
|
*/
|
|
public function writeNZBforReleaseId($relID, $relGuid, $name, $cTitle): bool
|
|
{
|
|
$collections = $this->pdo->queryDirect($this->_collectionsQuery.$relID);
|
|
|
|
if (! $collections instanceof \Traversable) {
|
|
return false;
|
|
}
|
|
|
|
$XMLWriter = new \XMLWriter();
|
|
$XMLWriter->openMemory();
|
|
$XMLWriter->setIndent(true);
|
|
$XMLWriter->setIndentString(' ');
|
|
|
|
$nzb_guid = '';
|
|
|
|
$XMLWriter->startDocument('1.0', 'UTF-8');
|
|
$XMLWriter->startDTD(self::NZB_DTD_NAME, self::NZB_DTD_PUBLIC, self::NZB_DTD_EXTERNAL);
|
|
$XMLWriter->endDTD();
|
|
$XMLWriter->writeComment($this->_nzbCommentString);
|
|
|
|
$XMLWriter->startElement('nzb');
|
|
$XMLWriter->writeAttribute('xmlns', self::NZB_XML_NS);
|
|
$XMLWriter->startElement('head');
|
|
$XMLWriter->startElement('meta');
|
|
$XMLWriter->writeAttribute('type', 'category');
|
|
$XMLWriter->text($cTitle);
|
|
$XMLWriter->endElement();
|
|
$XMLWriter->startElement('meta');
|
|
$XMLWriter->writeAttribute('type', 'name');
|
|
$XMLWriter->text($name);
|
|
$XMLWriter->endElement();
|
|
$XMLWriter->endElement(); //head
|
|
|
|
foreach ($collections as $collection) {
|
|
$binaries = $this->pdo->queryDirect(sprintf($this->_binariesQuery, $collection['id']));
|
|
if ($binaries === false) {
|
|
return false;
|
|
}
|
|
|
|
$poster = $collection['fromname'];
|
|
|
|
foreach ($binaries as $binary) {
|
|
$parts = $this->pdo->queryDirect(sprintf($this->_partsQuery, $binary['id']));
|
|
if ($parts === false) {
|
|
return false;
|
|
}
|
|
|
|
$subject = $binary['name'].'(1/'.$binary['totalparts'].')';
|
|
$XMLWriter->startElement('file');
|
|
$XMLWriter->writeAttribute('poster', $poster);
|
|
$XMLWriter->writeAttribute('date', $collection['udate']);
|
|
$XMLWriter->writeAttribute('subject', $subject);
|
|
$XMLWriter->startElement('groups');
|
|
if (preg_match_all('#(\S+):\S+#', $collection['xref'], $matches)) {
|
|
$matches = array_unique($matches[1]);
|
|
foreach ($matches as $group) {
|
|
$XMLWriter->writeElement('group', $group);
|
|
}
|
|
} else {
|
|
return false;
|
|
}
|
|
$XMLWriter->endElement(); //groups
|
|
$XMLWriter->startElement('segments');
|
|
foreach ($parts as $part) {
|
|
if ($nzb_guid === '') {
|
|
$nzb_guid = $part['messageid'];
|
|
}
|
|
$XMLWriter->startElement('segment');
|
|
$XMLWriter->writeAttribute('bytes', $part['size']);
|
|
$XMLWriter->writeAttribute('number', $part['partnumber']);
|
|
$XMLWriter->text($part['messageid']);
|
|
$XMLWriter->endElement();
|
|
}
|
|
$XMLWriter->endElement(); //segments
|
|
$XMLWriter->endElement(); //file
|
|
}
|
|
}
|
|
$XMLWriter->writeComment($this->_siteCommentString);
|
|
$XMLWriter->endElement(); //nzb
|
|
$XMLWriter->endDocument();
|
|
$path = ($this->buildNZBPath($relGuid, $this->nzbSplitLevel, true).$relGuid.'.nzb.gz');
|
|
$fp = gzopen($path, 'wb7');
|
|
if (! $fp) {
|
|
return false;
|
|
}
|
|
gzwrite($fp, $XMLWriter->outputMemory());
|
|
gzclose($fp);
|
|
unset($XMLWriter);
|
|
if (! is_file($path)) {
|
|
echo "ERROR: $path does not exist.\n";
|
|
|
|
return false;
|
|
}
|
|
// Mark release as having NZB.
|
|
$this->pdo->queryExec(
|
|
sprintf(
|
|
'
|
|
UPDATE releases SET nzbstatus = %d %s WHERE id = %d',
|
|
self::NZB_ADDED,
|
|
($nzb_guid === '' ? '' : ', nzb_guid = UNHEX( '.$this->pdo->escapeString(md5($nzb_guid)).' )'),
|
|
$relID
|
|
)
|
|
);
|
|
// Delete CBP for release that has its NZB created.
|
|
$this->pdo->queryExec(
|
|
sprintf(
|
|
'
|
|
DELETE c, b, p FROM %s c JOIN %s b ON(c.id=b.collections_id) STRAIGHT_JOIN %s p ON(b.id=p.binaries_id) WHERE c.releases_id = %d',
|
|
$this->_tableNames['cName'],
|
|
$this->_tableNames['bName'],
|
|
$this->_tableNames['pName'],
|
|
$relID
|
|
)
|
|
);
|
|
// Chmod to fix issues some users have with file permissions.
|
|
chmod($path, 0777);
|
|
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* Build a folder path on the hard drive where the NZB file will be stored.
|
|
*
|
|
* @param string $releaseGuid The guid of the release.
|
|
* @param int $levelsToSplit How many sub-paths the folder will be in.
|
|
* @param bool $createIfNotExist Create the folder if it doesn't exist.
|
|
*
|
|
* @return string $nzbpath The path to store the NZB file.
|
|
*/
|
|
public function buildNZBPath($releaseGuid, $levelsToSplit, $createIfNotExist)
|
|
{
|
|
$nzbPath = '';
|
|
|
|
for ($i = 0; $i < $levelsToSplit && $i < 32; $i++) {
|
|
$nzbPath .= substr($releaseGuid, $i, 1).DS;
|
|
}
|
|
|
|
$nzbPath = $this->siteNzbPath.$nzbPath;
|
|
|
|
if ($createIfNotExist === true && ! is_dir($nzbPath)) {
|
|
mkdir($nzbPath, 0777, true);
|
|
}
|
|
|
|
return $nzbPath;
|
|
}
|
|
|
|
/**
|
|
* Retrieve path + filename of the NZB to be stored.
|
|
*
|
|
* @param string $releaseGuid The guid of the release.
|
|
* @param int $levelsToSplit How many sub-paths the folder will be in. (optional)
|
|
* @param bool $createIfNotExist Create the folder if it doesn't exist. (optional)
|
|
*
|
|
* @return string Path+filename.
|
|
*/
|
|
public function getNZBPath($releaseGuid, $levelsToSplit = 0, $createIfNotExist = false): string
|
|
{
|
|
if ($levelsToSplit === 0) {
|
|
$levelsToSplit = $this->nzbSplitLevel;
|
|
}
|
|
|
|
return $this->buildNZBPath($releaseGuid, $levelsToSplit, $createIfNotExist).$releaseGuid.'.nzb.gz';
|
|
}
|
|
|
|
/**
|
|
* Determine is an NZB exists, returning the path+filename, if not return false.
|
|
*
|
|
* @param string $releaseGuid The guid of the release.
|
|
*
|
|
* @return bool|string On success: (string) Path+file name of the nzb.
|
|
* On failure: (bool) False.
|
|
*/
|
|
public function NZBPath($releaseGuid)
|
|
{
|
|
$nzbFile = $this->getNZBPath($releaseGuid);
|
|
|
|
return is_file($nzbFile) ? $nzbFile : false;
|
|
}
|
|
|
|
/**
|
|
* Retrieve various information on a NZB file (the subject, # of pars,
|
|
* file extensions, file sizes, file completion, group names, # of parts).
|
|
*
|
|
* @param string $nzb The NZB contents in a string.
|
|
* @param array $options
|
|
* 'no-file-key' => True - use numeric array key; False - Use filename as array key.
|
|
* 'strip-count' => True - Strip file/part count from file name to make the array key; False - Leave file name as is.
|
|
*
|
|
* @return array $result Empty if not an NZB or the contents of the NZB.
|
|
*/
|
|
public function nzbFileList($nzb, array $options = []): array
|
|
{
|
|
$defaults = [
|
|
'no-file-key' => true,
|
|
'strip-count' => false,
|
|
];
|
|
$options += $defaults;
|
|
|
|
$num_pars = $i = 0;
|
|
$result = [];
|
|
|
|
if (! $nzb) {
|
|
return $result;
|
|
}
|
|
|
|
$xml = @simplexml_load_string(str_replace("\x0F", '', $nzb));
|
|
if (! $xml || strtolower($xml->getName()) !== 'nzb') {
|
|
return $result;
|
|
}
|
|
|
|
foreach ($xml->file as $file) {
|
|
// Subject.
|
|
$title = (string) $file->attributes()->subject;
|
|
|
|
// Amount of pars.
|
|
if (stripos($title, '.par2')) {
|
|
$num_pars++;
|
|
}
|
|
|
|
if ($options['no-file-key'] === false) {
|
|
$i = $title;
|
|
if ($options['strip-count']) {
|
|
// Strip file / part count to get proper sorting.
|
|
$i = preg_replace('#\d+[- ._]?(/|\||[o0]f)[- ._]?\d+?(?![- ._]\d)#i', '', $i);
|
|
// Change .rar and .par2 to be sorted before .part0x.rar and .volxxx+xxx.par2
|
|
if (strpos($i, '.par2') !== false && ! preg_match('#\.vol\d+\+\d+\.par2#i', $i)) {
|
|
$i = str_replace('.par2', '.vol0.par2', $i);
|
|
} elseif (preg_match('#\.rar[^a-z0-9]#i', $i) && ! preg_match('#\.part\d+\.rar#i', $i)) {
|
|
$i = preg_replace('#\.rar(?:[^a-z0-9])#i', '.part0.rar', $i);
|
|
}
|
|
}
|
|
}
|
|
|
|
$result[$i]['title'] = $title;
|
|
|
|
// Extensions.
|
|
if (preg_match(
|
|
'/\.(\d{2,3}|7z|ace|ai7|srr|srt|sub|aiff|asc|avi|audio|bin|bz2|'
|
|
.'c|cfc|cfm|chm|class|conf|cpp|cs|css|csv|cue|deb|divx|doc|dot|'
|
|
.'eml|enc|exe|file|gif|gz|hlp|htm|html|image|iso|jar|java|jpeg|'
|
|
.'jpg|js|lua|m|m3u|mkv|mm|mov|mp3|mp4|mpg|nfo|nzb|odc|odf|odg|odi|odp|'
|
|
.'ods|odt|ogg|par2|parity|pdf|pgp|php|pl|png|ppt|ps|py|r\d{2,3}|'
|
|
.'ram|rar|rb|rm|rpm|rtf|sfv|sig|sql|srs|swf|sxc|sxd|sxi|sxw|tar|'
|
|
.'tex|tgz|txt|vcf|video|vsd|wav|wma|wmv|xls|xml|xpi|xvid|zip7|zip)'
|
|
.'[" ](?!(\)|\-))/i',
|
|
$title,
|
|
$ext
|
|
)
|
|
) {
|
|
if (preg_match('/\.r\d{2,3}/i', $ext[0])) {
|
|
$ext[1] = 'rar';
|
|
}
|
|
$result[$i]['ext'] = strtolower($ext[1]);
|
|
} else {
|
|
$result[$i]['ext'] = '';
|
|
}
|
|
|
|
$fileSize = $numSegments = 0;
|
|
|
|
// Parts.
|
|
if (! isset($result[$i]['segments'])) {
|
|
$result[$i]['segments'] = [];
|
|
}
|
|
|
|
// File size.
|
|
foreach ($file->segments->segment as $segment) {
|
|
$result[$i]['segments'][] = (string) $segment;
|
|
$fileSize += $segment->attributes()->bytes;
|
|
$numSegments++;
|
|
}
|
|
$result[$i]['size'] = $fileSize;
|
|
|
|
// File completion.
|
|
if (preg_match('/(\d+)\)$/', $title, $parts)) {
|
|
$result[$i]['partstotal'] = $parts[1];
|
|
}
|
|
$result[$i]['partsactual'] = $numSegments;
|
|
|
|
// Groups.
|
|
if (! isset($result[$i]['groups'])) {
|
|
$result[$i]['groups'] = [];
|
|
}
|
|
foreach ($file->groups->group as $g) {
|
|
$result[$i]['groups'][] = (string) $g;
|
|
}
|
|
|
|
unset($result[$i]['segments']['@attributes']);
|
|
if ($options['no-file-key']) {
|
|
$i++;
|
|
}
|
|
}
|
|
|
|
return $result;
|
|
}
|
|
}
|