starred/newznab-tmux
1
0
Fork 0
mirror of https://github.com/NNTmux/newznab-tmux.git synced 2026-08-29 17:28:55 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
4ab1ed9b6cb39bb66093f12f9aa6162501ed0f14
newznab-tmux/lib/IRCScraper.php
T
DariusIII 4ab1ed9b6c Updates to IRCScraper, added nukes
2014-03-31 23:30:28 +02:00

800 lines
24 KiB
PHP
Raw Blame History

<?php
require_once(dirname(__FILE__)."/../bin/config.php");
require_once(dirname(__FILE__)."/Net_SmartIRC/Net/SmartIRC.php");
require_once(WWW_DIR."/lib/framework/db.php");
require_once(WWW_DIR."/lib/groups.php");
require_once("functions.php");
/**
* Class IRCScraperRun
*/
class IRCScraper
{
/**
* Pre is not nuked.
* @const
*/
const NO_NUKE = 0;
/**
* Pre was un nuked.
* @const
*/
const UN_NUKE = 1;
/**
* Pre is nuked.
* @const
*/
const NUKE = 2;
/**
* Nuke reason was modified.
* @const
*/
const MOD_NUKE = 3;
/**
* Pre was re nuked.
* @const
*/
const RE_NUKE = 4;
/**
* Pre is nuked for being old.
* @const
*/
const OLD_NUKE = 5;
/**
* Array of current pre info.
* @var array
*/
protected $CurPre;
/**
* Array of old pre info.
* @var array
*/
protected $OldPre;
/**
* List of groups and their ID's
* @var array
*/
protected $groupList;
/**
* @var Net_SmartIRC
*/
protected $IRC = null;
/**
* Current server.
* efnet | corrupt | zenet
* @var string
*/
protected $serverType;
/**
* Run this in silent mode (no text output).
* @var bool
*/
protected $silent;
/**
* Is this pre nuked or un nuked?
* @var bool
*/
protected $nuked;
/**
* Construct
*
* @param Net_SmartIRC $irc Instance of class Net_SmartIRC
* @param string $serverType efnet | corrupt | zenet
* @param bool $silent Run this in silent mode (no text output).
* @param bool $debug Turn on Net_SmartIRC debug?
* @param bool $socket Use real sockets or fsock?
*/
public function __construct(&$irc, $serverType, &$silent = false, &$debug = false, &$socket = true)
{
$this->db = new DB();
$this->groups = new Groups();
$this->functions = new Functions();
$this->groupList = array();
$this->IRC = $irc;
if ($debug) {
$this->IRC->setDebug(SMARTIRC_DEBUG_ALL);
}
$this->serverType = $serverType;
$this->silent = $silent;
$this->resetPreVariables();
$this->startScraping($socket);
}
/**
* Destruct
*/
public function __destruct()
{
// Disconnect from IRC cleanly.
if (!is_null($this->IRC)) {
if (!$this->silent) {
echo
'Disconnecting from ' .
$this->serverType .
'.' .
PHP_EOL;
}
$this->IRC->disconnect();
}
}
/**
* Main method for scraping.
*
* @param bool $socket Use real sockets or fsock?
*/
protected function startScraping(&$socket)
{
switch($this->serverType) {
case 'efnet':
$server = SCRAPE_IRC_EFNET_SERVER;
$port = SCRAPE_IRC_EFNET_PORT;
$nickname = SCRAPE_IRC_EFNET_NICKNAME;
$username = SCRAPE_IRC_EFNET_USERNAME;
$realname = SCRAPE_IRC_EFNET_REALNAME;
$password = SCRAPE_IRC_EFNET_PASSWORD;
$channelList = array(
// Channel Password.
'#alt.binaries.inner-sanctum' => null,
'#alt.binaries.cd.image' => null,
'#alt.binaries.movies.divx' => null,
'#alt.binaries.sounds.mp3.complete_cd' => null,
'#alt.binaries.warez' => null,
'#alt.binaries.teevee' => 'teevee',
'#alt.binaries.moovee' => 'moovee',
'#alt.binaries.erotica' => 'erotica',
'#alt.binaries.flac' => 'flac',
'#alt.binaries.foreign' => 'foreign',
'#alt.binaries.console.ps3' => null
);
// Check if the user is ignoring channels.
if (defined('SCRAPE_IRC_EFNET_IGNORED_CHANNELS') && SCRAPE_IRC_EFNET_IGNORED_CHANNELS != '') {
$ignored = explode(',', SCRAPE_IRC_EFNET_IGNORED_CHANNELS);
$newList = array();
foreach($channelList as $channel => $password) {
if (!in_array($channel, $ignored)) {
$newList[$channel] = $password;
}
}
if (empty($newList)) {
exit('ERROR: You have ignored every group there is to scrape!' . PHP_EOL);
}
$channelList = $newList;
unset($newList);
}
$regex =
// Simple regex, more advanced regex below when doing the real checks.
'/' .
'FILLED.*Pred.*ago' . // a.b.inner-sanctum
'|' .
'Thank.*you.*Req.*Id.*Request' . // a.b.cd.image, a.b.movies.divx, a.b.sounds.mp3.complete_cd, a.b.warez
'|' .
'Thank.*?You.*?Request.*?Filled!.*?ReqId' . // a.b.moovee a.b.foreign a.b.flac a.b.teevee
'|' .
'That.*?was.*?awesome.*?Shall.*?ReqId' . // a.b.erotica
'|' .
'person.*?filling.*?request.*?for:.*?ReqID:' . // a.b.console.ps3
'|' .
'^\[(MOD|OLD|RE|UN)?NUKE\]' . // Nukes.
'/i';
break;
case 'corrupt':
$server = SCRAPE_IRC_CORRUPT_SERVER;
$port = SCRAPE_IRC_CORRUPT_PORT;
$nickname = SCRAPE_IRC_CORRUPT_NICKNAME;
$username = SCRAPE_IRC_CORRUPT_USERNAME;
$realname = SCRAPE_IRC_CORRUPT_REALNAME;
$password = SCRAPE_IRC_CORRUPT_PASSWORD;
$channelList = array('#pre' => null);
$regex = '/PRE:.+?\[.+?\]|^(MOD|OLD|RE|UN)?NUKE:\s+/i'; // #pre
break;
case 'zenet':
$server = SCRAPE_IRC_ZENET_SERVER;
$port = SCRAPE_IRC_ZENET_PORT;
$nickname = SCRAPE_IRC_ZENET_NICKNAME;
$username = SCRAPE_IRC_ZENET_USERNAME;
$realname = SCRAPE_IRC_ZENET_REALNAME;
$password = SCRAPE_IRC_ZENET_PASSWORD;
$channelList = array('#Pre' => null);
$regex = '/^\(PRE\)\s+\(|^\((MOD|OLD|RE|UN)?NUKE\)\s+/i'; // #Pre
break;
default:
return;
}
// Use real sockets instead of fsock.
$this->IRC->setUseSockets($socket);
// This will scan channel messages for the regex above.
$this->IRC->registerActionhandler(SMARTIRC_TYPE_CHANNEL, $regex, $this, 'check_type');
// If there's a problem during connection, try to reconnect.
$this->IRC->setAutoRetry(true);
// If problem connecting, wait 5 seconds before reconnecting.
$this->IRC->setReconnectdelay(50000);
// Try 4 times before giving up.
$this->IRC->setAutoRetryMax(4);
// If a network error happens, automatically reconnect.
$this->IRC->setAutoReconnect(true);
// Connect to IRC.
$connection = $this->IRC->connect($server, $port);
if ($connection === false) {
exit (
'Error connecting to (' .
$server .
':' .
$port .
'). Please verify your server information and try again.' .
PHP_EOL
);
}
// Login to IRC.
$this->IRC->login(
// Nick name.
$nickname,
// Real name.
$realname,
// User mode.
0,
// User name.
$username,
// Password.
(empty($password) ? null : $password)
);
// Join channels.
$this->IRC->join($channelList);
if (!$this->silent) {
echo
'[' .
date('r') .
'] [Scraping of IRC channels for (' .
$server .
':' .
$port .
') (' .
$nickname .
') started.]' .
PHP_EOL;
}
// Wait for action handlers.
$this->IRC->listen();
// If we return from action handlers, disconnect from IRC.
$this->IRC->disconnect();
}
/**
* Check channel and poster, send to right method.
*
* @param object $irc
* @param object $data
*/
public function check_type($irc, $data)
{
$channel = strtolower($data->channel);
$poster = strtolower($data->nick);
switch ($poster) {
case 'sanctum':
if ($channel === '#alt.binaries.inner-sanctum') {
$this->inner_sanctum($data->message);
}
break;
case 'alt-bin':
$this->alt_bin($data->message, $channel);
break;
case 'pr3':
$this->corrupt_pre($data->message);
break;
case 'abflac':
if ($channel === '#alt.binaries.flac') {
$this->ab_flac($data->message);
}
break;
case 'abking':
if ($channel === '#alt.binaries.moovee') {
$this->ab_moovee($data->message);
}
break;
case 'ginger':
if ($channel === '#alt.binaries.erotica') {
$this->ab_erotica($data->message);
}
break;
case 'abgod':
if ($channel === '#alt.binaries.teevee') {
$this->ab_teevee($data->message);
}
break;
case 'theannouncer':
if ($channel === '#pre') {
$this->zenet_pre($data->message);
}
break;
case 'abqueen':
if ($channel === '#alt.binaries.foreign') {
$this->ab_foreign($data->message);
}
break;
case 'binarybot':
if ($channel === '#alt.binaries.console.ps3') {
$this->ab_console_ps3($data->message);
}
break;
default:
break;
}
}
/**
* Get pre date from wD xH yM zS ago string.
*
* @param $agoString
*/
protected function getTimeFromAgo($agoString)
{
$predate = 0;
// Get pre date from this format : 10m 54s
if (preg_match('/((?P<day>\d+)d)?\s*((?P<hour>\d+)h)?\s*((?P<min>\d+)m)?\s*((?P<sec>\d+)s)?/i', $agoString, $matches)) {
if (!empty($matches['day'])) {
$predate += ((int)($matches['day']) * 86400);
}
if (!empty($matches['hour'])) {
$predate += ((int)($matches['hour']) * 3600);
}
if (!empty($matches['min'])) {
$predate += ((int)($matches['min']) * 60);
}
if (!empty($matches['sec'])) {
$predate += (int)$matches['sec'];
}
if ($predate !== 0) {
$predate = (time() - $predate);
}
}
$this->CurPre['predate'] = ($predate === 0 ? '' : $this->functions->from_unixtime($predate));
}
/**
* Go through regex matches, find PRE info.
*
* @param array $matches
*/
protected function siftMatches(&$matches)
{
$this->CurPre['md5'] = $this->db->escapeString(md5($matches['title']));
$this->CurPre['title'] = $matches['title'];
if (isset($matches['reqid'])) {
$this->CurPre['reqid'] = $matches['reqid'];
}
if (isset($matches['size'])) {
$this->CurPre['size'] = $matches['size'];
}
if (isset($matches['predago'])) {
$this->getTimeFromAgo($matches['predago']);
}
if (isset($matches['category'])) {
$this->CurPre['category'] = $matches['category'];
}
if (isset($matches['nuke'])) {
$this->nuked = true;
switch ($matches['nuke']) {
case 'NUKE':
$this->CurPre['nuked'] = self::NUKE;
break;
case 'UNNUKE':
$this->CurPre['nuked'] = self::UN_NUKE;
break;
case 'MODNUKE':
$this->CurPre['nuked'] = self::MOD_NUKE;
break;
case 'RENUKE':
$this->CurPre['nuked'] = self::RE_NUKE;
break;
case 'OLDNUKE':
$this->CurPre['nuked'] = self::OLD_NUKE;
break;
}
}
if (isset($matches['reason'])) {
$this->CurPre['reason'] = substr($matches['reason'], 0, 255);
}
if (isset($matches['files'])) {
$this->CurPre['files'] = substr($matches['files'], 0, 50);
}
$this->checkForDupe();
}
/**
* Gets new PRE from #a.b.erotica
*
* @param string $message The IRC message to parse.
*/
protected function ab_erotica(&$message)
{
//That was awesome [*Anonymous*] Shall we do it again? ReqId:[326264] [HD-Clip] [FULL 16x50MB TeenSexMovs.14.03.30.Daniela.XXX.720p.WMV-iaK] Filenames:[iak-teensexmovs-140330] Comments:[0] Watchers:[0] Total Size:[753MB] Points Earned:[54] [Pred 3m 20s ago]
//That was awesome [*Anonymous*] Shall we do it again? ReqId:[326663] [x264] [FULL 53x100MB Young.Ripe.Mellons.10.XXX.720P.WEBRIP.X264-GUSH] Filenames:[gush.yrmellons10] Comments:[1] Watchers:[0] Total Size:[4974MB] Points Earned:[354] [Pred 7m 5s ago] [NUKED]
if (preg_match('/ReqId:\[(?P<reqid>\d+)\]\s+\[.+?\]\s+\[FULL\s+(?P<files>\d+x\d+[KMGTP]?B)\s+(?P<title>.+?)\].+?Size:\[(?P<size>.+?)\](.+?\[Pred\s+(?P<predago>.+?)\s+ago\])?(.+?\[(?P<nuke>(MOD|OLD|RE|UN)?NUKE)D\])?/i', $message, $matches)) {
$this->CurPre['source'] = '#a.b.erotica';
$this->CurPre['groupid'] = $this->getGroupID('alt.binaries.erotica');
$this->CurPre['category'] = 'XXX';
$this->siftMatches($matches);
}
}
/**
* Gets new PRE from #a.b.flac
*
* @param string $message The IRC message to parse.
*/
protected function ab_flac(&$message)
{
//Thank You [*Anonymous*] Request Filled! ReqId:[42614] [FULL 10x15MB You_Blew_It-Keep_Doing_What_Youre_Doing-CD-FLAC-2014-WRE] Requested by:[*Anonymous* 21s ago] Comments:[0] Watchers:[0] Points Earned:[10] [Pred 3m 16s ago]
if (preg_match('/Request\s+Filled!\s+ReqId:\[(?P<reqid>\d+)\]\s+\[FULL\s+(?P<files>\d+x\d+[KMGTP]?B)\s+(?P<title>.+?)\].+?\[Pred\s+(?P<predago>.+?)\s+ago\]/i', $message, $matches)) {
$this->CurPre['source'] = '#a.b.flac';
$this->CurPre['groupid'] = $this->getGroupID('alt.binaries.sounds.flac');
$this->CurPre['category'] = 'FLAC';
$this->siftMatches($matches);
}
}
/**
* Gets new PRE from #a.b.moovee
*
* @param string $message The IRC message to parse.
*/
protected function ab_moovee(&$message)
{
//Thank You [*Anonymous*] Request Filled! ReqId:[140445] [FULL 94x50MB Burning.Daylight.2010.720p.BluRay.x264-SADPANDA] Requested by:[*Anonymous* 3h 29m ago] Comments:[0] Watchers:[0] Points Earned:[314] [Pred 4h 29m ago]
if (preg_match('/ReqId:\[(?P<reqid>\d+)\]\s+\[FULL\s+(?P<files>\d+x\d+[MGPTK]?B)\s+(?P<title>.+?)\]\s+.+?\[Pred\s+(?P<predago>.+?)\s+ago\]/i', $message, $matches)) {
$this->CurPre['source'] = '#a.b.moovee';
$this->CurPre['groupid'] = $this->getGroupID('alt.binaries.moovee');
$this->CurPre['category'] = 'Movies';
$this->siftMatches($matches);
}
}
/**
* Gets new PRE from #a.b.foreign
*
* @param string $message The IRC message to parse.
*/
protected function ab_foreign(&$message)
{
//Thank You [*Anonymous*] Request Filled! ReqId:[61525] [Movie] [FULL 95x50MB Wadjda.2012.PAL.MULTI.DVDR-VIAZAC] Requested by:[*Anonymous* 5m 13s ago] Comments:[0] Watchers:[0] Points Earned:[317] [Pred 8m 27s ago]
if (preg_match('/ReqId:\[(?P<reqid>\d+)\]\s+\[(?P<category>.+?)\]\s+\[FULL\s+(?P<files>\d+x\d+[MGPTK]?B)\s+(?P<title>.+?)\]\s+.+?\[Pred\s+(?P<predago>.+?)\s+ago\]/i', $message, $matches)) {
$this->CurPre['source'] = '#a.b.foreign';
$this->CurPre['groupid'] = $this->getGroupID('alt.binaries.mom');
$this->siftMatches($matches);
}
}
/**
* Gets new PRE from #a.b.teevee
*
* @param string $message The IRC message to parse.
*/
protected function ab_teevee(&$message)
{
//Thank You [*Anonymous*] Request Filled! ReqId:[183520] [FULL 19x50MB Louis.Therouxs.LA.Stories.S01E02.720p.HDTV.x264-FTP] Requested by:[*Anonymous* 53s ago] Comments:[0] Watchers:[0] Points Earned:[64] [Pred 3m 45s ago]
if (preg_match('/Request\s+Filled!\s+ReqId:\[(?P<reqid>\d+)\]\s+\[FULL\s+(?P<files>\d+x\d+[KMGPT]?B)\s+(?P<title>.+?)\].+?\[Pred\s+(?P<predago>.+?)\s+ago\]/i', $message, $matches)) {
$this->CurPre['source'] = '#a.b.teevee';
$this->CurPre['groupid'] = $this->getGroupID('alt.binaries.teevee');
$this->CurPre['category'] = 'TV';
$this->siftMatches($matches);
//[NUKE] ReqId:[183497] [From.Dusk.Till.Dawn.S01E01.720p.HDTV.x264-BATV] Reason:[bad.ivtc.causing.jerky.playback.due.to.dupe.and.missing.frames.in.segment.from.16m.to.30m]
//[UNNUKE] ReqId:[183449] [The.Biggest.Loser.AU.S09E29.PDTV.x264-RTA] Reason:[get.samplefix]
} else if (preg_match('/\[(?P<nuke>(MOD|OLD|RE|UN)?NUKE)\]\s+ReqId:\[(?P<reqid>\d+)\]\s+\[(?P<title>.+?)\]\s+Reason:\[(?P<reason>.+?)\]/i', $message, $matches)) {
$this->CurPre['source'] = '#a.b.teevee';
$this->CurPre['groupid'] = $this->getGroupID('alt.binaries.teevee');
$this->CurPre['category'] = 'TV';
$this->siftMatches($matches);
}
}
/**
* Gets new PRE from #a.b.console.ps3
*
* @param string $message The IRC message to parse.
*/
protected function ab_console_ps3(&$message)
{
//[Anonymous person filling request for: FULL 56 Ragnarok.Odyssey.ACE.PS3-iMARS NTSC BLURAY imars-ragodyace-ps3 56x100MB by Khaine13 on 2014-03-29 13:14:12][ReqID: 4888][You get a bonus of 6 for a total points earning of: 62 for filling with 10% par2s!][Your score will be adjusted once you have -filled 4888]
if (preg_match('/\s+FULL\s+\d+\s+(?P<title>.+?)\s+(?P<files>\d+x\d+[KMGTP]?B)\s+.+?\]\[ReqID:\s+(?P<reqid>\d+)\]\[/i', $message, $matches)) {
$this->CurPre['source'] = '#a.b.console.ps3';
$this->CurPre['groupid'] = $this->getGroupID('alt.binaries.console.ps3');
$this->CurPre['category'] = 'PS3';
$this->siftMatches($matches);
}
}
/**
* Gets new PRE from #Pre on zenet
*
* @param string $message The IRC message to parse.
*/
protected function zenet_pre(&$message)
{
//(PRE) (XXX) (The.Golden.Age.Of.Porn.Candy.Samples.XXX.WEBRIP.WMV-GUSH)
if (preg_match('/^\(PRE\)\s+\((?P<category>.+?)\)\s+\((?P<title>.+?)\)$/i', $message, $matches)) {
$this->CurPre['source'] = '#Pre@zenet';
$this->siftMatches($matches);
//(NUKE) (German_TOP100_Single_Charts_31_03_2014-MCG) (selfmade.compilations.not.allowed)
//(UNNUKE) (The.Biggest.Loser.AU.S09E29.PDTV.x264-RTA) (get.samplefix)
} else if (preg_match('/\((?P<nuke>(MOD|OLD|RE|UN)?NUKE)\)\s+\((?P<title>.+?)\)\s+\((?P<reason>.+?)\)/i', $message, $matches)) {
$this->CurPre['source'] = '#Pre@zenet';
$this->siftMatches($matches);
}
}
/**
* Gets new PRE from #pre on Corrupt-net
*
* @param string $message The IRC message to parse.
*/
protected function corrupt_pre(&$message)
{
//PRE: [TV-X264] Tinga.Tinga.Fabeln.S02E11.Warum.Bienen.stechen.GERMAN.WS.720p.HDTV.x264-RFG
if (preg_match('/^PRE:\s+\[(?P<category>.+?)\]\s+(?P<title>.+)$/i', $message, $matches)) {
$this->CurPre['source'] = '#pre@corrupt';
$this->siftMatches($matches);
//NUKE: Miclini-Sunday_Morning_P1-DIRFIX-DAB-03-30-2014-G4E [dirfix.must.state.name.of.release.being.fixed] [EthNet]
//UNNUKE: Youssoupha-Sur_Les_Chemins_De_Retour-FR-CD-FLAC-2009-0MNi [flac.rule.4.12.states.ENGLISH.artist.and.title.must.be.correct.and.this.is.not.ENGLISH] [LocalNet]
//MODNUKE: Miclini-Sunday_Morning_P1-DIRFIX-DAB-03-30-2014-G4E [nfo.must.state.name.of.release.being.fixed] [EthNet]
} else if (preg_match('/(?P<nuke>(MOD|OLD|RE|UN)?NUKE):\s+(?P<title>.+?)\s+\[(?P<reason>.+?)\]/i', $message, $matches)) {
$this->CurPre['source'] = '#pre@corrupt';
$this->siftMatches($matches);
}
}
/**
* Gets new PRE from #a.b.inner-sanctum.
*
* @param string $message The IRC message to parse.
*/
protected function inner_sanctum(&$message)
{
//[FILLED] [ 341953 | Emilie_Simon-Mue-CD-FR-2014-JUST | 16x79 | MP3 | *Anonymous* ] [ Pred 10m 54s ago ]
if (preg_match('/FILLED\]\s+\[\s+(?P<reqid>\d+)\s+\|\s+(?P<title>.+?)\s+\|\s+(?P<files>\d+x\d+)\s+\|\s+(?P<category>.+?)\s+\|\s+.+?\s+\]\s+\[\s+Pred\s+(?P<predago>.+?)\s+ago\s+\]/i', $message, $matches)) {
$this->CurPre['source'] = '#a.b.inner-sanctum';
$this->CurPre['groupid'] = $this->getGroupID('alt.binaries.inner-sanctum');
$this->siftMatches($matches);
}
}
/**
* Get new PRE from Alt-Bin groups.
*
* @param string $message The IRC message from the bot.
* @param string $channel The IRC channel name.
*/
protected function alt_bin(&$message, &$channel)
{
//Thank you<Bijour> Req Id<137732> Request<The_Blueprint-Phenomenology-(Retail)-2004-KzT *Pars Included*> Files<19> Dates<Req:2014-03-24 Filling:2014-03-29> Points<Filled:1393 Score:25604>
if (preg_match('/Req.+?Id.*?<.*?(?P<reqid>\d+).*?>.*?Request.*?<\d{0,2}(?P<title>.+?)(\s+\*Pars\s+Included\*\d{0,2}>|\d{0,2}>)\s+Files<(?P<files>\d+)>/i', $message, $matches)) {
$this->CurPre['source'] = str_replace('#alt.binaries', '#a.b', $channel);
$this->CurPre['groupid'] = $this->getGroupID(str_replace('#', '', $channel));
$this->siftMatches($matches);
}
}
/**
* Check if we already have the PRE.
*
* @return bool True if we already have, false if we don't.
*/
protected function checkForDupe()
{
$this->OldPre = $this->db->queryOneRow(sprintf('SELECT ID FROM prehash WHERE md5 = %s', $this->CurPre['md5']));
if ($this->OldPre === false) {
$this->insertNewPre();
} else {
$this->updatePre();
}
}
/**
* Insert new PRE into the DB.
*/
protected function insertNewPre()
{
if (empty($this->CurPre['title'])) {
return;
}
$query = 'INSERT INTO prehash (';
$query .= (!empty($this->CurPre['size']) ? 'size, ' : '');
$query .= (!empty($this->CurPre['category']) ? 'category, ' : '');
$query .= (!empty($this->CurPre['source']) ? 'source, ' : '');
$query .= (!empty($this->CurPre['reason']) ? 'reason, ' : '');
$query .= (!empty($this->CurPre['files']) ? 'files, ' : '');
$query .= (!empty($this->CurPre['reqid']) ? 'requestID, ' : '');
$query .= (!empty($this->CurPre['groupid']) ? 'groupID, ' : '');
$query .= (!empty($this->CurPre['nuked']) ? 'nuked, ' : '');
$query .= 'predate, md5, title, adddate) VALUES (';
$query .= (!empty($this->CurPre['size']) ? $this->db->escapeString($this->CurPre['size']) . ', ' : '');
$query .= (!empty($this->CurPre['category']) ? $this->db->escapeString($this->CurPre['category']) . ', ' : '');
$query .= (!empty($this->CurPre['source']) ? $this->db->escapeString($this->CurPre['source']) . ', ' : '');
$query .= (!empty($this->CurPre['reqid']) ? $this->CurPre['reqid'] . ', ' : '');
$query .= (!empty($this->CurPre['groupid']) ? $this->CurPre['groupid'] . ', ' : '');
$query .= (!empty($this->CurPre['predate']) ? $this->CurPre['predate'] . ', ' : 'NOW(), ');
$query .= '%s, %s, NOW())';
$this->db->exec(
sprintf(
$query,
$this->CurPre['md5'],
$this->db->escapeString($this->CurPre['title'])
)
);
$this->doEcho(true);
$this->resetPreVariables();
}
/**
* Updates PRE data in the DB.
*/
protected function updatePre()
{
if (empty($this->CurPre['title'])) {
return;
}
$query = 'UPDATE prehash SET ';
$query .= (!empty($this->CurPre['size']) ? 'size = ' . $this->db->escapeString($this->CurPre['size']) . ', ' : '');
$query .= (!empty($this->CurPre['source']) ? 'source = ' . $this->db->escapeString($this->CurPre['source']) . ', ' : '');
$query .= (!empty($this->CurPre['files']) ? 'files = ' . $this->db->escapeString($this->CurPre['files']) . ', ' : '');
$query .= (!empty($this->CurPre['reason']) ? 'nukereason = ' . $this->db->escapeString($this->CurPre['reason']) . ', ' : '');
$query .= (!empty($this->CurPre['reqid']) ? 'requestID = ' . $this->CurPre['reqid'] . ', ' : '');
$query .= (!empty($this->CurPre['groupid']) ? 'groupID = ' . $this->CurPre['groupid'] . ', ' : '');
$query .= (!empty($this->CurPre['predate']) ? 'predate = ' . $this->CurPre['predate'] . ', ' : '');
$query .= (!empty($this->CurPre['nuked']) ? 'nuked = ' . $this->CurPre['nuked'] . ', ' : '');
$query .= (empty($this->OldPre['category']) && !empty($this->CurPre['category']) ? 'category = ' . $this->db->escapeString($this->CurPre['category']) . ', ' : '');
if ($query === 'UPDATE prehash SET '){
return;
}
$query .= 'title = ' . $this->db->escapeString($this->CurPre['title']);
$query .= ' WHERE md5 = ' . $this->CurPre['md5'];
$this->db->exec($query);
$this->doEcho(false);
$this->resetPreVariables();
}
/**
* Echo new or update pre to CLI.
*
* @param bool $new
*/
protected function doEcho($new = true)
{
if (!$this->silent) {
$nukeString = '';
if ($this->nuked !== false) {
switch((int)$this->CurPre['nuked']) {
case self::NUKE:
$nukeString = '[ NUKED ] ';
break;
case self::UN_NUKE:
$nukeString = '[UNNUKED] ';
break;
case self::MOD_NUKE:
$nukeString = '[MODNUKE] ';
break;
case self::OLD_NUKE:
$nukeString = '[OLDNUKE] ';
break;
case self::RE_NUKE:
$nukeString = '[RENUKED] ';
break;
default:
break;
}
$nukeString .= ' [' . $this->CurPre['reason'] . '] ';
}
echo
'[' .
date('r') .
($new ? '] [ Added Pre ] [' : '] [Updated Pre] [') .
$this->CurPre['source'] .
'] ' .
($nukeString === '' ? '' : $nukeString) .
'[' .
$this->CurPre['title'] .
']' .
(!empty($this->CurPre['category']) ? ' [' . $this->CurPre['category'] . ']' : '') .
PHP_EOL;
}
}
/**
* Get a group ID for a group name.
*
* @param string $groupName
*
* @return mixed
*/
protected function getGroupID($groupName)
{
if (!isset($this->groupList[$groupName])) {
$this->groupList[$groupName] = $this->functions->getIDByName($groupName);
}
return $this->groupList[$groupName];
}
/**
* After updating or inserting new PRE, reset these.
*/
protected function resetPreVariables()
{
$this->nuked = false;
$this->OldPre = array();
$this->CurPre =
array(
'title' => '',
'md5' => '',
'size' => '',
'predate' => '',
'category' => '',
'source' => '',
'groupid' => '',
'reqid' => '',
'nuked' => '',
'reason' => '',
'files' => ''
);
}
}
Reference in New Issue View Git Blame Copy Permalink