mirror of
https://github.com/NNTmux/newznab-tmux.git
synced 2026-08-29 09:18:54 +00:00
Darko
363 lines
8.8 KiB
PHP
363 lines
8.8 KiB
PHP
<?php
|
|
require_once(dirname(__FILE__) . "/../bin/config.php");
|
|
require_once(WWW_DIR."/lib/framework/db.php");
|
|
require_once(WWW_DIR."/lib/site.php");
|
|
require_once(WWW_DIR."/lib/category.php");
|
|
|
|
/**
|
|
* Class for reading and writing NZB files on the hard disk,
|
|
* building folder paths to store the NZB files.
|
|
*/
|
|
class Enzebe
|
|
{
|
|
/**
|
|
* Instance of class site.
|
|
* @var object
|
|
* @access private
|
|
*/
|
|
private $s;
|
|
|
|
/**
|
|
* Site settings.
|
|
* @var object
|
|
* @access private
|
|
*/
|
|
public $site;
|
|
|
|
|
|
/**
|
|
* Group ID when writing NZBs.
|
|
* @var int
|
|
* @access protected
|
|
*/
|
|
protected $groupID;
|
|
|
|
/**
|
|
* Instance of class db.
|
|
*
|
|
* @var $db
|
|
* @access protected
|
|
*/
|
|
protected $db;
|
|
|
|
/**
|
|
* Date when writing NZBs.
|
|
* @var string
|
|
* @access protected
|
|
*/
|
|
protected $writeDate;
|
|
|
|
/*
|
|
* Table names for TPG.
|
|
*/
|
|
protected $bname;
|
|
protected $pname;
|
|
|
|
const NZB_NONE = 0; // Release has no NZB file yet.
|
|
const NZB_ADDED = 1; // Release had an NZB file created.
|
|
|
|
/**
|
|
* Default constructor.
|
|
*
|
|
* @access public
|
|
*/
|
|
public function __construct()
|
|
{
|
|
$s = new Sites();
|
|
$this->site = $s->get();
|
|
}
|
|
|
|
/**
|
|
* Initiate class vars when writing NZB's.
|
|
*
|
|
* @param $db
|
|
* @param string $date
|
|
* @param int $groupID
|
|
*
|
|
* @access public
|
|
*/
|
|
public function initiateForWrite($db, $date, $groupID)
|
|
{
|
|
$this->db = $db;
|
|
$this->writeDate = $date;
|
|
$this->groupID = $groupID;
|
|
$this->bname = 'binaries';
|
|
$this->pname = 'parts';
|
|
}
|
|
|
|
/**
|
|
* Clean up class vars when done writing NZB's.
|
|
*
|
|
* @access public
|
|
*/
|
|
public function cleanForWrite()
|
|
{
|
|
$this->writeDate = null;
|
|
$this->groupID = null;
|
|
}
|
|
|
|
/**
|
|
* 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?
|
|
*
|
|
* @access public
|
|
*/
|
|
public function writeNZBforReleaseId($relID, $relGuid, $name, $cTitle)
|
|
{
|
|
$path = $this->getNZBPath($relGuid, true);
|
|
$this->site = new Sites();
|
|
$this->bname = 'binaries';
|
|
$this->pname = 'parts';
|
|
$this->db = new DB();
|
|
$fp = gzopen($path, 'w5');
|
|
if ($fp) {
|
|
$nzb_guid = '';
|
|
gzwrite($fp,
|
|
'<?xml version="1.0" encoding="UTF-8"?>' .
|
|
"\n" .
|
|
'<!DOCTYPE nzb PUBLIC "-//newzBin//DTD NZB 1.1//EN" "http://www.newzbin.com/DTD/nzb/nzb-1.1.dtd">' .
|
|
"\n" .
|
|
'<!-- NZB Generated by: newznab ' . $this->site->version() . ' ' . $this->writeDate . ' -->' .
|
|
"\n" .
|
|
"<nzb xmlns=\"http://www.newzbin.com/DTD/2003/nzb\">\n<head>\n <meta type=\"category\">" .
|
|
htmlspecialchars($cTitle, ENT_QUOTES, 'utf-8') .
|
|
"</meta>\n <meta type=\"name\">" .
|
|
htmlspecialchars($name, ENT_QUOTES, 'utf-8') .
|
|
"</meta>\n</head>\n\n"
|
|
);
|
|
|
|
$result =
|
|
$this->db->query(
|
|
sprintf('
|
|
SELECT %s.*, %s.date AS udate, groups.name AS groupname
|
|
FROM %s
|
|
INNER JOIN groups ON %s.groupID = groups.ID
|
|
WHERE %s.releaseID = %d',
|
|
$this->bname,
|
|
$this->bname,
|
|
$this->bname,
|
|
$this->bname,
|
|
$this->bname,
|
|
$relID
|
|
)
|
|
);
|
|
|
|
foreach ($result as $binRow) {
|
|
$unixDate = strtotime($binRow['udate']);
|
|
$poster = htmlspecialchars($binRow['fromname'], ENT_QUOTES, 'utf-8');
|
|
$result2 = $this->db->query(
|
|
sprintf(
|
|
'SELECT ID, name, totalparts FROM %s WHERE releaseID = %d ORDER BY name',
|
|
$this->bname,
|
|
$binRow['ID']
|
|
)
|
|
);
|
|
|
|
foreach ($result2 as $binRow2) {
|
|
gzwrite($fp,
|
|
'<file poster="' . $poster .
|
|
'" date="' . $unixDate .
|
|
'" subject="' .
|
|
htmlspecialchars($binRow2['name'], ENT_QUOTES, 'utf-8') .
|
|
' (1/' . $binRow2['totalparts'] .
|
|
")\">\n <groups>\n <group>" . $binRow['groupname'] .
|
|
"</group>\n </groups>\n <segments>\n"
|
|
);
|
|
|
|
$resParts = $this->db->query(
|
|
sprintf('
|
|
SELECT DISTINCT(messageID), size, partnumber FROM %s WHERE binaryID = %d ORDER BY partnumber',
|
|
$this->pname,
|
|
$binRow2['ID']
|
|
)
|
|
);
|
|
|
|
foreach ($resParts as $partsRow) {
|
|
if ($nzb_guid === '') {
|
|
$nzb_guid = $partsRow['messageID'];
|
|
}
|
|
|
|
gzwrite($fp, ' <segment bytes="' . $partsRow['size']
|
|
. '" number="' . $partsRow['partnumber'] . '">'
|
|
. htmlspecialchars($partsRow['messageID'], ENT_QUOTES, 'utf-8')
|
|
. "</segment>\n"
|
|
);
|
|
}
|
|
gzwrite($fp, " </segments>\n</file>\n");
|
|
}
|
|
}
|
|
gzwrite($fp, '</nzb>');
|
|
gzclose($fp);
|
|
|
|
if (is_file($path)) {
|
|
$this->db->exec(
|
|
sprintf('
|
|
UPDATE releases SET nzbstatus = %d %s WHERE ID = %d',
|
|
Enzebe::NZB_ADDED,
|
|
($nzb_guid === '' ? '' : ', nzb_guid = ' . $this->db->escapestring(md5($nzb_guid))),
|
|
$relID
|
|
)
|
|
);
|
|
|
|
// Chmod to fix issues some users have with file permissions.
|
|
chmod($path, 0777);
|
|
return true;
|
|
} else {
|
|
echo "ERROR: $path does not exist.\n";
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Build a folder path on the hard drive where the NZB file will be stored.
|
|
*
|
|
* @param string $releaseGuid The guid of the release.
|
|
* @param bool $createIfNotExist Create the folder if it doesn't exist.
|
|
*
|
|
* @return string $nzbpath The path to store the NZB file.
|
|
*
|
|
* @access public
|
|
*/
|
|
private function buildNZBPath($releaseGuid, $createIfNotExist)
|
|
{
|
|
$s = new Sites();
|
|
$this->site = $s->get();
|
|
$siteNzbPath = $this->site->nzbpath;
|
|
$nzbPath = '';
|
|
$i = 0;
|
|
$nzbPath .= substr($releaseGuid, $i, 1) . '/';
|
|
$nzbPath = $siteNzbPath . $nzbPath;
|
|
|
|
if ($createIfNotExist && !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 bool $createIfNotExist Create the folder if it doesn't exist. (optional)
|
|
*
|
|
* @return string Path+filename.
|
|
*
|
|
* @access public
|
|
*/
|
|
public function getNZBPath($releaseGuid, $createIfNotExist = false)
|
|
{
|
|
return ($this->buildNZBPath($releaseGuid, $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.
|
|
*
|
|
* @access public
|
|
*/
|
|
public function NZBPath($releaseGuid)
|
|
{
|
|
$nzbFile = $this->getNZBPath($releaseGuid);
|
|
return !is_file($nzbFile) ? false : $nzbFile;
|
|
}
|
|
|
|
/**
|
|
* 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.
|
|
*
|
|
* @return array $result Empty if not an NZB or the contents of the NZB.
|
|
*
|
|
* @access public
|
|
*/
|
|
public function nzbFileList($nzb)
|
|
{
|
|
$num_pars = $i = 0;
|
|
$result = array();
|
|
|
|
$nzb = str_replace("\x0F", '', $nzb);
|
|
$xml = @simplexml_load_string($nzb);
|
|
if (!$xml || strtolower($xml->getName()) !== 'nzb') {
|
|
return $result;
|
|
}
|
|
|
|
foreach ($xml->file as $file) {
|
|
// Subject.
|
|
$title = $file->attributes()->subject;
|
|
|
|
// Amount of pars.
|
|
if (stripos($title, '.par2')) {
|
|
$num_pars++;
|
|
}
|
|
|
|
$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', $file->attributes()->subject, $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'] = array();
|
|
}
|
|
|
|
// File size.
|
|
foreach ($file->segments->segment as $segment) {
|
|
array_push($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'] = array();
|
|
}
|
|
foreach ($file->groups->group as $g) {
|
|
array_push($result[$i]['groups'], (string) $g);
|
|
}
|
|
|
|
unset($result[$i]['segments']['@attributes']);
|
|
$i++;
|
|
}
|
|
return $result;
|
|
}
|
|
|
|
}
|