mirror of
https://github.com/NNTmux/newznab-tmux.git
synced 2026-08-29 01:08:56 +00:00
Darko
654 lines
16 KiB
PHP
654 lines
16 KiB
PHP
<?php
|
|
|
|
use newznab\db\Settings;
|
|
|
|
/**
|
|
* This class handles data access for groups.
|
|
*/
|
|
class Groups
|
|
{
|
|
|
|
/**
|
|
* @var newznab\db\Settings
|
|
*/
|
|
public $pdo;
|
|
|
|
/**
|
|
* Construct.
|
|
*
|
|
* @param array $options Class instances.
|
|
*/
|
|
public function __construct(array $options = [])
|
|
{
|
|
$defaults = [
|
|
'Settings' => null
|
|
];
|
|
$options += $defaults;
|
|
|
|
$this->pdo = ($options['Settings'] instanceof Settings ? $options['Settings'] : new Settings());
|
|
}
|
|
/**
|
|
* Get all group rows.
|
|
*/
|
|
public function getAll()
|
|
{
|
|
return $this->pdo->query(
|
|
"SELECT groups.*,
|
|
COALESCE(rel.num, 0) AS num_releases
|
|
FROM groups
|
|
LEFT OUTER JOIN
|
|
(SELECT groupid, COUNT(id) AS num FROM releases GROUP BY groupid) rel
|
|
ON rel.groupid = groups.id
|
|
ORDER BY groups.name"
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Get all group rows for use in a select list.
|
|
*/
|
|
public function getGroupsForSelect()
|
|
{
|
|
|
|
$categories = $this->pdo->query("SELECT * FROM groups WHERE active = 1 ORDER BY name");
|
|
$temp_array = array();
|
|
|
|
$temp_array[-1] = "--Please Select--";
|
|
|
|
foreach ($categories as $category)
|
|
$temp_array[$category["name"]] = $category["name"];
|
|
|
|
return $temp_array;
|
|
}
|
|
|
|
/**
|
|
* Get a group row by its id.
|
|
*/
|
|
public function getByID($id)
|
|
{
|
|
|
|
|
|
return $this->pdo->queryOneRow(sprintf("select * from groups where id = %d ", $id));
|
|
}
|
|
|
|
/**
|
|
* Get all active group rows.
|
|
*/
|
|
public function getActive()
|
|
{
|
|
|
|
return $this->pdo->query("SELECT * FROM groups WHERE active = 1 ORDER BY name");
|
|
}
|
|
|
|
/**
|
|
* Get a group row by name.
|
|
*/
|
|
public function getByName($grp)
|
|
{
|
|
|
|
|
|
return $this->pdo->queryOneRow(sprintf("SELECT * FROM groups WHERE name = %s", $this->pdo->escapeString($grp)));
|
|
}
|
|
|
|
/**
|
|
* Get a group name using its id.
|
|
*
|
|
* @param int|string $id The group id.
|
|
*
|
|
* @return string Empty string on failure, groupName on success.
|
|
*/
|
|
public function getByNameByID($id)
|
|
{
|
|
|
|
$res = $this->pdo->queryOneRow(sprintf("SELECT name FROM groups WHERE id = %d ", $id));
|
|
return ($res === false ? '' : $res["name"]);
|
|
}
|
|
|
|
/**
|
|
* Get a group name using its name.
|
|
*
|
|
* @param string $name The group name.
|
|
*
|
|
* @return string Empty string on failure, groupid on success.
|
|
*/
|
|
public function getIDByName($name)
|
|
{
|
|
|
|
$res = $this->pdo->queryOneRow(sprintf("SELECT id FROM groups WHERE name = %s", $this->pdo->escapeString($name)));
|
|
return ($res === false ? '' : $res["id"]);
|
|
}
|
|
|
|
/**
|
|
* Get count of all groups, filter by name.
|
|
*/
|
|
public function getCount($groupname = "", $activeonly = false)
|
|
{
|
|
|
|
|
|
$grpsql = '';
|
|
if ($groupname != "")
|
|
$grpsql .= sprintf("and groups.name like %s ", $this->pdo->escapeString("%" . $groupname . "%"));
|
|
|
|
if ($activeonly == true)
|
|
$grpsql .= "and active=1 ";
|
|
|
|
$res = $this->pdo->queryOneRow(sprintf("select count(id) as num from groups where 1=1 %s", $grpsql));
|
|
|
|
return $res["num"];
|
|
}
|
|
|
|
/**
|
|
* Get groups rows for browse list by limit.
|
|
*/
|
|
public function getRange($start, $num, $groupname = "", $activeonly = false)
|
|
{
|
|
|
|
if ($start === false)
|
|
$limit = "";
|
|
else
|
|
$limit = " LIMIT " . $start . "," . $num;
|
|
|
|
$grpsql = '';
|
|
if ($groupname != "")
|
|
$grpsql .= sprintf("and groups.name like %s ", $this->pdo->escapeString("%" . $groupname . "%"));
|
|
if ($activeonly == true)
|
|
$grpsql .= "and active=1 ";
|
|
|
|
$sql = sprintf("SELECT groups.*, COALESCE(rel.num, 0) AS num_releases
|
|
FROM groups
|
|
LEFT OUTER JOIN
|
|
(
|
|
SELECT groupid, COUNT(id) AS num FROM releases group by groupid
|
|
) rel ON rel.groupid = groups.id WHERE 1=1 %s ORDER BY groups.name " . $limit, $grpsql
|
|
);
|
|
|
|
return $this->pdo->query($sql);
|
|
}
|
|
|
|
/**
|
|
* Update an existing group.
|
|
*
|
|
* @param Array $group
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function update($group)
|
|
{
|
|
|
|
$minFileString =
|
|
($group["minfilestoformrelease"] == '' ?
|
|
"minfilestoformrelease = NULL," :
|
|
sprintf(" minfilestoformrelease = %d,", $this->formatNumberString($group["minfilestoformrelease"], false))
|
|
);
|
|
|
|
$minSizeString =
|
|
($group["minsizetoformrelease"] == '' ?
|
|
"minsizetoformrelease = NULL" :
|
|
sprintf(" minsizetoformrelease = %d", $this->formatNumberString($group["minsizetoformrelease"], false))
|
|
);
|
|
|
|
return $this->pdo->queryExec(
|
|
sprintf(
|
|
"UPDATE groups
|
|
SET name = %s, description = %s, backfill_target = %s, first_record = %s, last_record = %s,
|
|
last_updated = NOW(), active = %s, backfill = %s, %s %s
|
|
WHERE id = %d",
|
|
$this->pdo->escapeString(trim($group["name"])),
|
|
$this->pdo->escapeString(trim($group["description"])),
|
|
$this->formatNumberString($group["backfill_target"]),
|
|
$this->formatNumberString($group["first_record"]),
|
|
$this->formatNumberString($group["last_record"]),
|
|
$this->formatNumberString($group["active"]),
|
|
$this->formatNumberString($group["backfill"]),
|
|
$minFileString,
|
|
$minSizeString,
|
|
$group["id"]
|
|
)
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Add a new group.
|
|
*
|
|
* @param array $group
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function add($group)
|
|
{
|
|
$minFileString =
|
|
($group["minfilestoformrelease"] == '' ? "NULL" : sprintf("%d", $this->formatNumberString($group["minfilestoformrelease"], false))
|
|
);
|
|
|
|
$minSizeString =
|
|
($group["minsizetoformrelease"] == '' ? "NULL" : sprintf("%d", $this->formatNumberString($group["minsizetoformrelease"], false))
|
|
);
|
|
|
|
return $this->pdo->queryInsert(
|
|
sprintf("
|
|
INSERT INTO groups
|
|
(name, description, backfill_target, first_record, last_record, last_updated,
|
|
active, backfill, minfilestoformrelease, minsizetoformrelease)
|
|
VALUES (%s, %s, %s, %s, %s, NOW(), %s, %s, %s, %s)",
|
|
$this->pdo->escapeString(trim($group["name"])),
|
|
$this->pdo->escapeString(trim($group["description"])),
|
|
$this->formatNumberString($group["backfill_target"]),
|
|
$this->formatNumberString($group["first_record"]),
|
|
$this->formatNumberString($group["last_record"]),
|
|
$this->formatNumberString($group["active"]),
|
|
$this->formatNumberString($group["backfill"]),
|
|
$minFileString,
|
|
$minSizeString
|
|
)
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Format numeric string when adding/updating groups.
|
|
*
|
|
* @param string $setting
|
|
* @param bool $escape
|
|
*
|
|
* @return string|int
|
|
*/
|
|
protected function formatNumberString($setting, $escape=true)
|
|
{
|
|
$setting = trim($setting);
|
|
if ($setting === "0" || !is_numeric($setting)) {
|
|
$setting = '0';
|
|
}
|
|
|
|
return ($escape ? $this->pdo->escapeString($setting) : (int)$setting);
|
|
}
|
|
|
|
/**
|
|
* Delete a group.
|
|
*/
|
|
public function delete($id)
|
|
{
|
|
|
|
return $this->pdo->queryExec(sprintf("DELETE from groups where id = %d", $id));
|
|
}
|
|
|
|
/**
|
|
* Reset a group.
|
|
*
|
|
* @param string|int $id The group id.
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function reset($id)
|
|
{
|
|
// Remove rows from collections / binaries / parts.
|
|
(new \Binaries(['Groups' => $this, 'Settings' => $this->pdo]))->purgeGroup($id);
|
|
|
|
// Remove rows from part repair.
|
|
$this->pdo->queryExec(sprintf("DELETE FROM partrepair WHERE group_id = %d", $id));
|
|
|
|
$this->pdo->queryExec(sprintf('DROP TABLE IF EXISTS collections_%d', $id));
|
|
$this->pdo->queryExec(sprintf('DROP TABLE IF EXISTS binaries_%d', $id));
|
|
$this->pdo->queryExec(sprintf('DROP TABLE IF EXISTS parts_%d', $id));
|
|
$this->pdo->queryExec(sprintf('DROP TABLE IF EXISTS partrepair_%d', $id));
|
|
|
|
// Reset the group stats.
|
|
return $this->pdo->queryExec(
|
|
sprintf("
|
|
UPDATE groups
|
|
SET backfill_target = 0, first_record = 0, first_record_postdate = NULL, last_record = 0,
|
|
last_record_postdate = NULL, last_updated = NULL
|
|
WHERE id = %d", $id)
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Reset all groups.
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function resetall()
|
|
{
|
|
$this->pdo->queryExec("TRUNCATE TABLE collections");
|
|
$this->pdo->queryExec("TRUNCATE TABLE binaries");
|
|
$this->pdo->queryExec("TRUNCATE TABLE parts");
|
|
$this->pdo->queryExec("TRUNCATE TABLE partrepair");
|
|
$groups = $this->pdo->query("SELECT id FROM groups");
|
|
foreach ($groups as $group) {
|
|
$this->pdo->queryExec('DROP TABLE IF EXISTS collections_' . $group['id']);
|
|
$this->pdo->queryExec('DROP TABLE IF EXISTS binaries_' . $group['id']);
|
|
$this->pdo->queryExec('DROP TABLE IF EXISTS parts_' . $group['id']);
|
|
$this->pdo->queryExec('DROP TABLE IF EXISTS partrepair_' . $group['id']);
|
|
}
|
|
|
|
// Reset the group stats.
|
|
return $this->pdo->queryExec("
|
|
UPDATE groups
|
|
SET backfill_target = 0, first_record = 0, first_record_postdate = NULL, last_record = 0,
|
|
last_record_postdate = NULL, last_updated = NULL, active = 0"
|
|
);
|
|
}
|
|
|
|
|
|
/**
|
|
* Update the list of newsgroups from nntp provider matching a regex and return an array of messages.
|
|
*/
|
|
function addBulk($groupList, $active = 1, $backfill = 1)
|
|
{
|
|
|
|
$ret = array();
|
|
|
|
if ($groupList == "") {
|
|
$ret[] = "No group list provided.";
|
|
} else {
|
|
$nntp = new NNTP(['Echo' => false]);
|
|
if (!$nntp->doConnect()) {
|
|
$ret[] = "Failed to get NNTP connection";
|
|
|
|
return $ret;
|
|
}
|
|
$groups = $nntp->getGroups();
|
|
$nntp->doQuit();
|
|
|
|
$regfilter = "/(" . str_replace(array('.', '*'), array('\.', '.*?'), $groupList) . ")$/";
|
|
|
|
foreach ($groups AS $group) {
|
|
if (preg_match($regfilter, $group['group']) > 0) {
|
|
$res = $this->pdo->queryOneRow(sprintf("SELECT id FROM groups WHERE name = %s ", $this->pdo->escapeString($group['group'])));
|
|
if ($res) {
|
|
|
|
$this->pdo->queryExec(sprintf("update groups SET active = %d where id = %d", $active, $res["id"]));
|
|
$ret[] = array('group' => $group['group'], 'msg' => 'Updated');
|
|
} else {
|
|
$desc = "";
|
|
$this->pdo->queryInsert(sprintf("INSERT INTO groups (name, description, active, backfill) VALUES (%s, %s, %d, %s)", $this->pdo->escapeString($group['group']), $this->pdo->escapeString($desc), $active, $backfill));
|
|
$ret[] = array('group' => $group['group'], 'msg' => 'Created');
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return $ret;
|
|
}
|
|
|
|
/**
|
|
* @param $id
|
|
* @param int $status
|
|
*
|
|
* @return string
|
|
*/
|
|
public function updateGroupStatus($id, $status = 0)
|
|
{
|
|
$this->pdo->queryExec(sprintf("UPDATE groups SET active = %d WHERE id = %d", $status, $id));
|
|
return "Group $id has been " . (($status == 0) ? 'deactivated' : 'activated') . '.';
|
|
}
|
|
|
|
/**
|
|
* @param $id
|
|
* @param int $status
|
|
*
|
|
* @return string
|
|
*/
|
|
public function updateBackfillStatus($id, $status = 0)
|
|
{
|
|
$this->pdo->queryExec(sprintf("UPDATE groups SET backfill = %d WHERE id = %d", $status, $id));
|
|
return "Group $id has been " . (($status == 0) ? 'deactivated' : 'activated') . '.';
|
|
}
|
|
|
|
/**
|
|
* @return array
|
|
*/
|
|
public function getActiveBackfill()
|
|
{
|
|
return $this->pdo->query("SELECT * FROM groups WHERE backfill = 1 AND last_record != 0 ORDER BY name");
|
|
}
|
|
|
|
/**
|
|
* @return array
|
|
*/
|
|
public function getActiveByDateBackfill()
|
|
{
|
|
return $this->pdo->query("SELECT * FROM groups WHERE backfill = 1 AND last_record != 0 ORDER BY first_record_postdate DESC");
|
|
}
|
|
|
|
/**
|
|
* @var array
|
|
*/
|
|
private $cbppTableNames;
|
|
|
|
/**
|
|
* Get the names of the collections/binaries/parts/part repair tables.
|
|
* If TPG is on, try to create new tables for the group_id, if we fail, log the error and exit.
|
|
*
|
|
* @param bool $tpgSetting false, tpg is off in site setting, true tpg is on in site setting.
|
|
* @param int $groupID id of the group.
|
|
*
|
|
* @return array The table names.
|
|
*/
|
|
public function getCBPTableNames($tpgSetting, $groupID)
|
|
{
|
|
$groupKey = ($groupID . '_' . (int) $tpgSetting);
|
|
|
|
// Check if buffered and return. Prevents re-querying MySQL when TPG is on.
|
|
if (isset($this->cbppTableNames[$groupKey])) {
|
|
return $this->cbppTableNames[$groupKey];
|
|
}
|
|
|
|
$tables = [];
|
|
$tables['cname'] = 'collections';
|
|
$tables['bname'] = 'binaries';
|
|
$tables['pname'] = 'parts';
|
|
$tables['prname'] = 'partrepair';
|
|
|
|
if ($tpgSetting === true) {
|
|
if ($groupID == '') {
|
|
exit('Error: You must use .../misc/update_scripts/nix_scripts/multiprocessing/releases.php since you have enabled TPG!');
|
|
}
|
|
|
|
if ($this->createNewTPGTables($groupID) === false && NN_ECHOCLI) {
|
|
exit('There is a problem creating new TPG tables for this group id: ' . $groupID . PHP_EOL);
|
|
}
|
|
|
|
$groupEnding = '_' . $groupID;
|
|
$tables['cname'] .= $groupEnding;
|
|
$tables['bname'] .= $groupEnding;
|
|
$tables['pname'] .= $groupEnding;
|
|
$tables['prname'] .= $groupEnding;
|
|
}
|
|
|
|
// Buffer.
|
|
$this->cbppTableNames[$groupKey] = $tables;
|
|
|
|
return $tables;
|
|
}
|
|
|
|
/**
|
|
* @return array
|
|
*/
|
|
public function getActiveIDs()
|
|
{
|
|
return $this->pdo->query("SELECT id FROM groups WHERE active = 1 ORDER BY name");
|
|
}
|
|
|
|
/**
|
|
* Set the backfill to 0 when the group is backfilled to max.
|
|
*
|
|
* @param $name
|
|
*/
|
|
public function disableForPost($name)
|
|
{
|
|
$this->pdo->queryExec(sprintf("UPDATE groups SET backfill = 0 WHERE name = %s", $this->pdo->escapeString($name)));
|
|
}
|
|
|
|
/**
|
|
* Check if the tables exists for the group_id, make new tables for table per group.
|
|
*
|
|
* @param int $groupID
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function createNewTPGTables($groupID)
|
|
{
|
|
foreach (['collections', 'binaries', 'parts', 'partrepair'] as $tableName) {
|
|
if ($this->pdo->queryExec(sprintf('SELECT * FROM %s_%s LIMIT 1', $tableName, $groupID), true) === false) {
|
|
if ($this->pdo->queryExec(sprintf('CREATE TABLE %s_%s LIKE %s', $tableName, $groupID, $tableName), true) === false) {
|
|
return false;
|
|
} else {
|
|
if ($tableName === 'collections') {
|
|
$this->pdo->queryExec(
|
|
sprintf(
|
|
'CREATE TRIGGER delete_collections_%s BEFORE DELETE ON collections_%s FOR EACH ROW BEGIN' .
|
|
' DELETE FROM binaries_%s WHERE collection_id = OLD.id; DELETE FROM parts_%s WHERE collection_id = OLD.id; END',
|
|
$groupID, $groupID, $groupID, $groupID
|
|
)
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
/**
|
|
* Purge a single group or all groups.
|
|
*
|
|
* @param int|string|bool $id The group id. If false, purge all groups.
|
|
*/
|
|
public function purge($id = false)
|
|
{
|
|
if ($id === false) {
|
|
$this->resetall();
|
|
} else {
|
|
$this->reset($id);
|
|
}
|
|
|
|
$releaseArray = $this->pdo->queryDirect(
|
|
sprintf("SELECT id, guid FROM releases %s", ($id === false ? '' : 'WHERE groupid = ' . $id))
|
|
);
|
|
|
|
if ($releaseArray instanceof \Traversable) {
|
|
$releases = new \Releases(['Settings' => $this->pdo, 'Groups' => $this]);
|
|
$nzb = new \NZB($this->pdo);
|
|
$releaseImage = new \ReleaseImage($this->pdo);
|
|
foreach ($releaseArray as $release) {
|
|
$releases->deleteSingle(['g' => $release['guid'], 'i' => $release['id']], $nzb, $releaseImage);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param string $groupname
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function getCountActive($groupname="")
|
|
{
|
|
$res = $this->pdo->queryOneRow(
|
|
sprintf("
|
|
SELECT COUNT(id) AS num
|
|
FROM groups
|
|
WHERE 1 = 1 %s
|
|
AND active = 1",
|
|
($groupname !== ''
|
|
?
|
|
sprintf(
|
|
"AND groups.name LIKE %s ",
|
|
$this->pdo->escapeString("%".$groupname."%")
|
|
)
|
|
: ''
|
|
)
|
|
)
|
|
);
|
|
return $res["num"];
|
|
}
|
|
|
|
/**
|
|
* @param string $groupname
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function getCountInactive($groupname="")
|
|
{
|
|
$res = $this->pdo->queryOneRow(
|
|
sprintf("
|
|
SELECT COUNT(id) AS num
|
|
FROM groups
|
|
WHERE 1 = 1 %s
|
|
AND active = 0",
|
|
($groupname !== ''
|
|
?
|
|
sprintf(
|
|
"AND groups.name LIKE %s ",
|
|
$this->pdo->escapeString("%".$groupname."%")
|
|
)
|
|
: ''
|
|
)
|
|
)
|
|
);
|
|
return $res["num"];
|
|
}
|
|
|
|
/**
|
|
* @param $start
|
|
* @param $num
|
|
* @param string $groupname
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function getRangeActive($start, $num, $groupname="")
|
|
{
|
|
return $this->pdo->query(
|
|
sprintf("
|
|
SELECT groups.*, COALESCE(rel.num, 0) AS num_releases
|
|
FROM groups
|
|
LEFT OUTER JOIN
|
|
(SELECT groupid, COUNT(id) AS num
|
|
FROM releases
|
|
GROUP BY groupid
|
|
) rel
|
|
ON rel.groupid = groups.id
|
|
WHERE 1 = 1 %s
|
|
AND active = 1
|
|
ORDER BY groups.name " . ($start === false ? '' : " LIMIT " . $num . " OFFSET " .$start),
|
|
($groupname !== ''
|
|
?
|
|
sprintf(
|
|
"AND groups.name LIKE %s ",
|
|
$this->pdo->escapeString("%".$groupname."%")
|
|
)
|
|
: ''
|
|
)
|
|
)
|
|
);
|
|
}
|
|
|
|
/**
|
|
* @param $start
|
|
* @param $num
|
|
* @param string $groupname
|
|
*
|
|
* @return mixed
|
|
*/
|
|
public function getRangeInactive($start, $num, $groupname="")
|
|
{
|
|
return $this->pdo->query(
|
|
sprintf("
|
|
SELECT groups.*, COALESCE(rel.num, 0) AS num_releases
|
|
FROM groups
|
|
LEFT OUTER JOIN
|
|
(SELECT groupid, COUNT(id) AS num
|
|
FROM releases
|
|
GROUP BY groupid
|
|
) rel
|
|
ON rel.groupid = groups.id
|
|
WHERE 1 = 1 %s
|
|
AND active = 0
|
|
ORDER BY groups.name " . ($start === false ? '' : " LIMIT ".$num." OFFSET ".$start),
|
|
($groupname !== ''
|
|
? sprintf(
|
|
"AND groups.name LIKE %s ",
|
|
$this->pdo->escapeString("%".$groupname."%")
|
|
)
|
|
: ''
|
|
)
|
|
)
|
|
);
|
|
}
|
|
} |