mirror of
https://github.com/NNTmux/newznab-tmux.git
synced 2026-08-29 01:08:56 +00:00
DariusIII
1746 lines
63 KiB
PHP
Executable File
1746 lines
63 KiB
PHP
Executable File
<?php
|
|
|
|
namespace nntmux;
|
|
|
|
use nntmux\db\DB;
|
|
use App\Models\Group;
|
|
use App\Models\Settings;
|
|
use Illuminate\Support\Carbon;
|
|
use App\Models\BinaryBlacklist;
|
|
use App\Models\MultigroupPoster;
|
|
use Illuminate\Support\Facades\Cache;
|
|
use nntmux\processing\ProcessReleasesMultiGroup;
|
|
|
|
/**
|
|
* Class Binaries.
|
|
*/
|
|
class Binaries
|
|
{
|
|
public const OPTYPE_BLACKLIST = 1;
|
|
public const OPTYPE_WHITELIST = 2;
|
|
|
|
public const BLACKLIST_DISABLED = 0;
|
|
public const BLACKLIST_ENABLED = 1;
|
|
|
|
public const BLACKLIST_FIELD_SUBJECT = 1;
|
|
public const BLACKLIST_FIELD_FROM = 2;
|
|
public const BLACKLIST_FIELD_MESSAGEID = 3;
|
|
|
|
/**
|
|
* @var array
|
|
*/
|
|
public $blackList = [];
|
|
|
|
/**
|
|
* @var array
|
|
*/
|
|
public $whiteList = [];
|
|
|
|
/**
|
|
* @var int
|
|
*/
|
|
public $messageBuffer;
|
|
|
|
/**
|
|
* @var \nntmux\ColorCLI
|
|
*/
|
|
protected $_colorCLI;
|
|
|
|
/**
|
|
* @var \nntmux\CollectionsCleaning
|
|
*/
|
|
protected $_collectionsCleaning;
|
|
|
|
/**
|
|
* @var \nntmux\NNTP
|
|
*/
|
|
protected $_nntp;
|
|
|
|
/**
|
|
* Should we use header compression?
|
|
*
|
|
* @var bool
|
|
*/
|
|
protected $_compressedHeaders;
|
|
|
|
/**
|
|
* Should we use part repair?
|
|
*
|
|
* @var bool
|
|
*/
|
|
protected $_partRepair;
|
|
|
|
/**
|
|
* @var \nntmux\db\DB
|
|
*/
|
|
protected $_pdo;
|
|
|
|
/**
|
|
* How many days to go back on a new group?
|
|
*
|
|
* @var bool
|
|
*/
|
|
protected $_newGroupScanByDays;
|
|
|
|
/**
|
|
* How many headers to download on new groups?
|
|
*
|
|
* @var int
|
|
*/
|
|
protected $_newGroupMessagesToScan;
|
|
|
|
/**
|
|
* How many days to go back on new groups?
|
|
*
|
|
* @var int
|
|
*/
|
|
protected $_newGroupDaysToScan;
|
|
|
|
/**
|
|
* How many headers to download per run of part repair?
|
|
*
|
|
* @var int
|
|
*/
|
|
protected $_partRepairLimit;
|
|
|
|
/**
|
|
* Should we show dropped yEnc to CLI?
|
|
*
|
|
* @var bool
|
|
*/
|
|
protected $_showDroppedYEncParts;
|
|
|
|
/**
|
|
* Echo to cli?
|
|
*
|
|
* @var bool
|
|
*/
|
|
protected $_echoCLI;
|
|
|
|
/**
|
|
* @var bool
|
|
*/
|
|
protected $_debug = false;
|
|
|
|
/**
|
|
* Max tries to download headers.
|
|
* @var int
|
|
*/
|
|
protected $_partRepairMaxTries;
|
|
|
|
/**
|
|
* An array of binaryblacklist IDs that should have their activity date updated.
|
|
* @var array(int)
|
|
*/
|
|
protected $_binaryBlacklistIdsToUpdate = [];
|
|
|
|
/**
|
|
* @var float microseconds time of cleaning process start
|
|
*/
|
|
protected $startCleaning;
|
|
|
|
/**
|
|
* @var float microseconds time of the start of the scan function
|
|
*/
|
|
protected $startLoop;
|
|
|
|
/**
|
|
* @var bool Is this retrieved header a multigroup one?
|
|
*/
|
|
protected $multiGroup;
|
|
|
|
/**
|
|
* @var bool
|
|
*/
|
|
protected $allAsMgr;
|
|
|
|
/**
|
|
* @var string How long it took in seconds to download headers
|
|
*/
|
|
protected $timeHeaders;
|
|
|
|
/**
|
|
* @var string How long it took in seconds to clean/parse headers
|
|
*/
|
|
protected $timeCleaning;
|
|
|
|
/**
|
|
* @var float microseconds time part repair was started
|
|
*/
|
|
protected $startPR;
|
|
|
|
/**
|
|
* @var array The CBP/MGR tables names
|
|
*/
|
|
protected $tableNames;
|
|
|
|
/**
|
|
* @var float microseconds time header update was started
|
|
*/
|
|
protected $startUpdate;
|
|
|
|
/**
|
|
* @var string The time it took to insert the headers
|
|
*/
|
|
protected $timeInsert;
|
|
|
|
/**
|
|
* @var array the header currently being scanned
|
|
*/
|
|
protected $header;
|
|
|
|
/**
|
|
* @var bool Should we add parts to part repair queue?
|
|
*/
|
|
protected $addToPartRepair;
|
|
|
|
/**
|
|
* @var array Numbers of Headers received from the USP
|
|
*/
|
|
protected $headersReceived;
|
|
|
|
/**
|
|
* @var array The current newsgroup information being updated
|
|
*/
|
|
protected $groupMySQL;
|
|
|
|
/**
|
|
* @var int the last article number in the range
|
|
*/
|
|
protected $last;
|
|
|
|
/**
|
|
* @var int the first article number in the range
|
|
*/
|
|
protected $first;
|
|
|
|
/**
|
|
* @var int How many received headers were not yEnc encoded
|
|
*/
|
|
protected $notYEnc;
|
|
|
|
/**
|
|
* @var int How many received headers were blacklist matched
|
|
*/
|
|
protected $headersBlackListed;
|
|
|
|
/**
|
|
* @var array Header numbers that were not inserted
|
|
*/
|
|
protected $headersNotInserted;
|
|
|
|
/**
|
|
* Constructor.
|
|
*
|
|
* @param array $options Class instances / echo to CLI?
|
|
*
|
|
* @throws \Exception
|
|
*/
|
|
public function __construct(array $options = [])
|
|
{
|
|
$defaults = [
|
|
'Echo' => true,
|
|
'CollectionsCleaning' => null,
|
|
'ColorCLI' => null,
|
|
'Logger' => null,
|
|
'Groups' => null,
|
|
'NNTP' => null,
|
|
'Settings' => null,
|
|
];
|
|
$options += $defaults;
|
|
|
|
$this->_echoCLI = ($options['Echo'] && NN_ECHOCLI);
|
|
|
|
$this->_pdo = ($options['Settings'] instanceof DB ? $options['Settings'] : new DB());
|
|
$this->_colorCLI = ($options['ColorCLI'] instanceof ColorCLI ? $options['ColorCLI'] : new ColorCLI());
|
|
$this->_nntp = ($options['NNTP'] instanceof NNTP ? $options['NNTP'] : new NNTP(['Echo' => $this->_colorCLI, 'Settings' => $this->_pdo, 'ColorCLI' => $this->_colorCLI]));
|
|
$this->_collectionsCleaning = ($options['CollectionsCleaning'] instanceof CollectionsCleaning ? $options['CollectionsCleaning'] : new CollectionsCleaning(['Settings' => $this->_pdo]));
|
|
|
|
$this->messageBuffer = Settings::settingValue('..maxmssgs') !== '' ?
|
|
(int) Settings::settingValue('..maxmssgs') : 20000;
|
|
$this->_compressedHeaders = (int) Settings::settingValue('..compressedheaders') === 1;
|
|
$this->_partRepair = (int) Settings::settingValue('..partrepair') === 1;
|
|
$this->_newGroupScanByDays = (int) Settings::settingValue('..newgroupscanmethod') === 1;
|
|
$this->_newGroupMessagesToScan = Settings::settingValue('..newgroupmsgstoscan') !== '' ? (int) Settings::settingValue('..newgroupmsgstoscan') : 50000;
|
|
$this->_newGroupDaysToScan = Settings::settingValue('..newgroupdaystoscan') !== '' ? (int) Settings::settingValue('..newgroupdaystoscan') : 3;
|
|
$this->_partRepairLimit = Settings::settingValue('..maxpartrepair') !== '' ? (int) Settings::settingValue('..maxpartrepair') : 15000;
|
|
$this->_partRepairMaxTries = (Settings::settingValue('..partrepairmaxtries') !== '' ? (int) Settings::settingValue('..partrepairmaxtries') : 3);
|
|
$this->_showDroppedYEncParts = (int) Settings::settingValue('..showdroppedyencparts') === 1;
|
|
$this->allAsMgr = (int) Settings::settingValue('..allasmgr') === 1;
|
|
|
|
$this->blackList = $this->whiteList = [];
|
|
}
|
|
|
|
/**
|
|
* Download new headers for all active groups.
|
|
*
|
|
* @param int $maxHeaders (Optional) How many headers to download max.
|
|
*
|
|
* @return void
|
|
* @throws \Exception
|
|
*/
|
|
public function updateAllGroups($maxHeaders = 100000): void
|
|
{
|
|
$groups = Group::getActive();
|
|
|
|
$groupCount = \count($groups);
|
|
if ($groupCount > 0) {
|
|
$counter = 1;
|
|
$allTime = microtime(true);
|
|
|
|
$this->log(
|
|
'Updating: '.$groupCount.' group(s) - Using compression? '.($this->_compressedHeaders ? 'Yes' : 'No'),
|
|
__FUNCTION__,
|
|
'header'
|
|
);
|
|
|
|
// Loop through groups.
|
|
foreach ($groups as $group) {
|
|
$this->log(
|
|
'Starting group '.$counter.' of '.$groupCount,
|
|
__FUNCTION__,
|
|
'header'
|
|
);
|
|
$this->updateGroup($group, $maxHeaders);
|
|
$counter++;
|
|
}
|
|
|
|
$this->log(
|
|
'Updating completed in '.number_format(microtime(true) - $allTime, 2).' seconds.',
|
|
__FUNCTION__,
|
|
'primary'
|
|
);
|
|
} else {
|
|
$this->log(
|
|
'No groups specified. Ensure groups are added to NNTmux\'s database for updating.',
|
|
__FUNCTION__,
|
|
'warning'
|
|
);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* When the indexer is started, log the date/time.
|
|
*/
|
|
public function logIndexerStart(): void
|
|
{
|
|
Settings::query()->where('setting', '=', 'last_run_time')->update(['value' => Carbon::now()]);
|
|
}
|
|
|
|
/**
|
|
* Download new headers for a single group.
|
|
*
|
|
* @param array $groupMySQL Array of MySQL results for a single group.
|
|
* @param int $maxHeaders (Optional) How many headers to download max.
|
|
*
|
|
* @return void
|
|
* @throws \Exception
|
|
*/
|
|
public function updateGroup($groupMySQL, $maxHeaders = 0): void
|
|
{
|
|
$startGroup = microtime(true);
|
|
|
|
$this->logIndexerStart();
|
|
|
|
// Select the group on the NNTP server, gets the latest info on it.
|
|
$groupNNTP = $this->_nntp->selectGroup($groupMySQL['name']);
|
|
if ($this->_nntp->isError($groupNNTP)) {
|
|
$groupNNTP = $this->_nntp->dataError($this->_nntp, $groupMySQL['name']);
|
|
|
|
if (isset($groupNNTP['code']) && (int) $groupNNTP['code'] === 411) {
|
|
Group::disableIfNotExist($groupMySQL['id']);
|
|
}
|
|
if ($this->_nntp->isError($groupNNTP)) {
|
|
return;
|
|
}
|
|
}
|
|
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(ColorCLI::primary('Processing '.$groupMySQL['name']), true);
|
|
}
|
|
|
|
// Attempt to repair any missing parts before grabbing new ones.
|
|
if ((int) $groupMySQL['last_record'] !== 0) {
|
|
if ($this->_partRepair) {
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(ColorCLI::primary('Part repair enabled. Checking for missing parts.'), true);
|
|
}
|
|
$this->partRepair($groupMySQL);
|
|
|
|
$mgrPosters = $this->getMultiGroupPosters();
|
|
if ($this->allAsMgr === true || ! empty($mgrPosters)) {
|
|
$tableNames = ProcessReleasesMultiGroup::tableNames();
|
|
$this->partRepair($groupMySQL, $tableNames);
|
|
}
|
|
} elseif ($this->_echoCLI) {
|
|
ColorCLI::doEcho(ColorCLI::primary('Part repair disabled by user.'), true);
|
|
}
|
|
}
|
|
|
|
// Generate postdate for first record, for those that upgraded.
|
|
if ($groupMySQL['first_record_postdate'] === null && (int) $groupMySQL['first_record'] !== 0) {
|
|
$groupMySQL['first_record_postdate'] = $this->postdate($groupMySQL['first_record'], $groupNNTP);
|
|
Group::query()->where('id', $groupMySQL['id'])->update(['first_record_postdate' => Carbon::createFromTimestamp($groupMySQL['first_record_postdate'])]);
|
|
}
|
|
|
|
// Get first article we want aka the oldest.
|
|
if ((int) $groupMySQL['last_record'] === 0) {
|
|
if ($this->_newGroupScanByDays) {
|
|
// For new newsgroups - determine here how far we want to go back using date.
|
|
$first = $this->daytopost($this->_newGroupDaysToScan, $groupNNTP);
|
|
} elseif ($groupNNTP['first'] >= ($groupNNTP['last'] - ($this->_newGroupMessagesToScan + $this->messageBuffer))) {
|
|
// If what we want is lower than the groups first article, set the wanted first to the first.
|
|
$first = $groupNNTP['first'];
|
|
} else {
|
|
// Or else, use the newest article minus how much we should get for new groups.
|
|
$first = (string) ($groupNNTP['last'] - ($this->_newGroupMessagesToScan + $this->messageBuffer));
|
|
}
|
|
|
|
// We will use this to subtract so we leave articles for the next time (in case the server doesn't have them yet)
|
|
$leaveOver = $this->messageBuffer;
|
|
|
|
// If this is not a new group, go from our newest to the servers newest.
|
|
} else {
|
|
// Set our oldest wanted to our newest local article.
|
|
$first = $groupMySQL['last_record'];
|
|
|
|
// This is how many articles we will grab. (the servers newest minus our newest).
|
|
$totalCount = (string) ($groupNNTP['last'] - $first);
|
|
|
|
// Check if the server has more articles than our loop limit x 2.
|
|
if ($totalCount > ($this->messageBuffer * 2)) {
|
|
// Get the remainder of $totalCount / $this->message buffer
|
|
$leaveOver = round($totalCount % $this->messageBuffer, 0, PHP_ROUND_HALF_DOWN) + $this->messageBuffer;
|
|
} else {
|
|
// Else get half of the available.
|
|
$leaveOver = round($totalCount / 2, 0, PHP_ROUND_HALF_DOWN);
|
|
}
|
|
}
|
|
|
|
// The last article we want, aka the newest.
|
|
$last = $groupLast = (string) ($groupNNTP['last'] - $leaveOver);
|
|
|
|
// If the newest we want is older than the oldest we want somehow.. set them equal.
|
|
if ($last < $first) {
|
|
$last = $groupLast = $first;
|
|
}
|
|
|
|
// This is how many articles we are going to get.
|
|
$total = (string) ($groupLast - $first);
|
|
// This is how many articles are available (without $leaveOver).
|
|
$realTotal = (string) ($groupNNTP['last'] - $first);
|
|
|
|
// Check if we should limit the amount of fetched new headers.
|
|
if ($maxHeaders > 0) {
|
|
if ($maxHeaders < ($groupLast - $first)) {
|
|
$groupLast = $last = (string) ($first + $maxHeaders);
|
|
}
|
|
$total = (string) ($groupLast - $first);
|
|
}
|
|
|
|
// If total is bigger than 0 it means we have new parts in the newsgroup.
|
|
if ($total > 0) {
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::primary(
|
|
(
|
|
(int) $groupMySQL['last_record'] === 0
|
|
? 'New group '.$groupNNTP['group'].' starting with '.
|
|
(
|
|
$this->_newGroupScanByDays
|
|
? $this->_newGroupDaysToScan.' days'
|
|
: number_format($this->_newGroupMessagesToScan).' messages'
|
|
).' worth.'
|
|
: 'Group '.$groupNNTP['group'].' has '.number_format($realTotal).' new articles.'
|
|
).
|
|
' Leaving '.number_format($leaveOver).
|
|
" for next pass.\nServer oldest: ".number_format($groupNNTP['first']).
|
|
' Server newest: '.number_format($groupNNTP['last']).
|
|
' Local newest: '.number_format($groupMySQL['last_record'])
|
|
),
|
|
true
|
|
);
|
|
}
|
|
|
|
$done = false;
|
|
// Get all the parts (in portions of $this->messageBuffer to not use too much memory).
|
|
while ($done === false) {
|
|
|
|
// Increment last until we reach $groupLast (group newest article).
|
|
if ($total > $this->messageBuffer) {
|
|
if ((string) ($first + $this->messageBuffer) > $groupLast) {
|
|
$last = $groupLast;
|
|
} else {
|
|
$last = (string) ($first + $this->messageBuffer);
|
|
}
|
|
}
|
|
// Increment first so we don't get an article we already had.
|
|
$first++;
|
|
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::header(
|
|
PHP_EOL.'Getting '.number_format($last - $first + 1).' articles ('.number_format($first).
|
|
' to '.number_format($last).') from '.$groupMySQL['name'].' - ('.
|
|
number_format($groupLast - $last).' articles in queue).'
|
|
)
|
|
);
|
|
}
|
|
|
|
// Get article headers from newsgroup.
|
|
$scanSummary = $this->scan($groupMySQL, $first, $last);
|
|
|
|
// Check if we fetched headers.
|
|
if (! empty($scanSummary)) {
|
|
|
|
// If new group, update first record & postdate
|
|
if ($groupMySQL['first_record_postdate'] === null && (int) $groupMySQL['first_record'] === 0) {
|
|
$groupMySQL['first_record'] = $scanSummary['firstArticleNumber'];
|
|
|
|
if (isset($scanSummary['firstArticleDate'])) {
|
|
$groupMySQL['first_record_postdate'] = strtotime($scanSummary['firstArticleDate']);
|
|
} else {
|
|
$groupMySQL['first_record_postdate'] = $this->postdate($groupMySQL['first_record'], $groupNNTP);
|
|
}
|
|
|
|
Group::query()
|
|
->where('id', $groupMySQL['id'])
|
|
->update(
|
|
[
|
|
'first_record' => $scanSummary['firstArticleNumber'],
|
|
'first_record_postdate' => Carbon::createFromTimestamp(
|
|
$groupMySQL['first_record_postdate']
|
|
),
|
|
]
|
|
);
|
|
}
|
|
|
|
$scanSummary['lastArticleDate'] = (isset($scanSummary['lastArticleDate']) ? strtotime($scanSummary['lastArticleDate']) : false);
|
|
if (! is_numeric($scanSummary['lastArticleDate'])) {
|
|
$scanSummary['lastArticleDate'] = $this->postdate($scanSummary['lastArticleNumber'], $groupNNTP);
|
|
}
|
|
|
|
Group::query()
|
|
->where('id', $groupMySQL['id'])
|
|
->update(
|
|
[
|
|
'last_record' => $scanSummary['lastArticleNumber'],
|
|
'last_record_postdate' => Carbon::createFromTimestamp($scanSummary['lastArticleDate']),
|
|
'last_updated' => Carbon::now(),
|
|
]
|
|
);
|
|
} else {
|
|
// If we didn't fetch headers, update the record still.
|
|
Group::query()
|
|
->where('id', $groupMySQL['id'])
|
|
->update(
|
|
[
|
|
'last_record' => $last,
|
|
'last_updated' => Carbon::now(),
|
|
]
|
|
);
|
|
}
|
|
|
|
if ((int) $last === (int) $groupLast) {
|
|
$done = true;
|
|
} else {
|
|
$first = $last;
|
|
}
|
|
}
|
|
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::primary(
|
|
PHP_EOL.'Group '.$groupMySQL['name'].' processed in '.
|
|
number_format(microtime(true) - $startGroup, 2).' seconds.'
|
|
),
|
|
true
|
|
);
|
|
}
|
|
} elseif ($this->_echoCLI) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::primary(
|
|
'No new articles for '.$groupMySQL['name'].' (first '.number_format($first).
|
|
', last '.number_format($last).', grouplast '.number_format($groupMySQL['last_record']).
|
|
', total '.number_format($total).")\n".'Server oldest: '.number_format($groupNNTP['first']).
|
|
' Server newest: '.number_format($groupNNTP['last']).' Local newest: '.number_format($groupMySQL['last_record'])
|
|
),
|
|
true
|
|
);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Loop over range of wanted headers, insert headers into DB.
|
|
*
|
|
* @param array $groupMySQL The group info from mysql.
|
|
* @param int $first The oldest wanted header.
|
|
* @param int $last The newest wanted header.
|
|
* @param string $type Is this partrepair or update or backfill?
|
|
* @param null|array $missingParts If we are running in partrepair, the list of missing article numbers.
|
|
*
|
|
* @return array Empty on failure.
|
|
* @throws \Exception
|
|
*/
|
|
public function scan($groupMySQL, $first, $last, $type = 'update', $missingParts = null): array
|
|
{
|
|
// Start time of scan method and of fetching headers.
|
|
$this->startLoop = microtime(true);
|
|
$this->groupMySQL = $groupMySQL;
|
|
$this->last = $last;
|
|
$this->first = $first;
|
|
|
|
$this->notYEnc = $this->headersBlackListed = 0;
|
|
|
|
// Check if MySQL tables exist, create if they do not, get their names at the same time.
|
|
$this->tableNames = (new Group())->getCBPTableNames($this->groupMySQL['id']);
|
|
|
|
$mgrPosters = $this->getMultiGroupPosters();
|
|
|
|
if ($this->allAsMgr === true || ! empty($mgrPosters)) {
|
|
$mgrActive = true;
|
|
$mgrPosters = ! empty($mgrPosters) ? array_flip(array_column($mgrPosters, 'poster')) : '';
|
|
} else {
|
|
$mgrActive = false;
|
|
}
|
|
|
|
$returnArray = $stdHeaders = $mgrHeaders = [];
|
|
|
|
$partRepair = ($type === 'partrepair');
|
|
$this->addToPartRepair = ($type === 'update' && $this->_partRepair);
|
|
|
|
// Download the headers.
|
|
if ($partRepair === true) {
|
|
// This is slower but possibly is better with missing headers.
|
|
$headers = $this->_nntp->getOverview($this->first.'-'.$this->last, true, false);
|
|
} else {
|
|
$headers = $this->_nntp->getXOVER($this->first.'-'.$this->last);
|
|
}
|
|
|
|
// If there was an error, try to reconnect.
|
|
if ($this->_nntp->isError($headers)) {
|
|
|
|
// Increment if part repair and return false.
|
|
if ($partRepair === true) {
|
|
$this->_pdo->queryExec(
|
|
sprintf(
|
|
'UPDATE %s SET attempts = attempts + 1 WHERE groups_id = %d AND numberid %s',
|
|
$this->tableNames['prname'],
|
|
$this->groupMySQL['id'],
|
|
((int) $this->first === (int) $this->last ? '= '.$this->first : 'IN ('.implode(',', range($this->first, $this->last)).')')
|
|
)
|
|
);
|
|
|
|
return $returnArray;
|
|
}
|
|
|
|
// This is usually a compression error, so try disabling compression.
|
|
$this->_nntp->doQuit();
|
|
if ($this->_nntp->doConnect(false) !== true) {
|
|
return $returnArray;
|
|
}
|
|
|
|
// Re-select group, download headers again without compression and re-enable compression.
|
|
$this->_nntp->selectGroup($this->groupMySQL['name']);
|
|
$headers = $this->_nntp->getXOVER($this->first.'-'.$this->last);
|
|
$this->_nntp->enableCompression();
|
|
|
|
// Check if the non-compression headers have an error.
|
|
if ($this->_nntp->isError($headers)) {
|
|
$message = ((int) $headers->code === 0 ? 'Unknown error' : $headers->message);
|
|
$this->log(
|
|
"Code {$headers->code}: $message\nSkipping group: {$this->groupMySQL['name']}",
|
|
__FUNCTION__,
|
|
'error'
|
|
);
|
|
|
|
return $returnArray;
|
|
}
|
|
}
|
|
|
|
// Start of processing headers.
|
|
$this->startCleaning = microtime(true);
|
|
|
|
// End of the getting data from usenet.
|
|
$this->timeHeaders = number_format($this->startCleaning - $this->startLoop, 2);
|
|
|
|
// Check if we got headers.
|
|
$msgCount = \count($headers);
|
|
|
|
if ($msgCount < 1) {
|
|
return $returnArray;
|
|
}
|
|
|
|
$this->getHighLowArticleInfo($returnArray, $headers, $msgCount);
|
|
|
|
$headersRepaired = $rangeNotReceived = $this->headersReceived = $this->headersNotInserted = [];
|
|
|
|
foreach ($headers as $header) {
|
|
|
|
// Check if we got the article or not.
|
|
if (isset($header['Number'])) {
|
|
$this->headersReceived[] = $header['Number'];
|
|
} else {
|
|
if ($this->addToPartRepair) {
|
|
$rangeNotReceived[] = $header['Number'];
|
|
}
|
|
continue;
|
|
}
|
|
|
|
// If set we are running in partRepair mode.
|
|
if ($partRepair === true && $missingParts !== null) {
|
|
if (! \in_array($header['Number'], $missingParts, false)) {
|
|
// If article isn't one that is missing skip it.
|
|
continue;
|
|
}
|
|
// We got the part this time. Remove article from part repair.
|
|
$headersRepaired[] = $header['Number'];
|
|
}
|
|
|
|
/*
|
|
* Find part / total parts. Ignore if no part count found.
|
|
*
|
|
* \s* Trims the leading space.
|
|
* (?!"Usenet Index Post) ignores these types of articles, they are useless.
|
|
* (.+) Fetches the subject.
|
|
* \s+ Trims trailing space after the subject.
|
|
* \((\d+)\/(\d+)\) Gets the part count.
|
|
* No ending ($) as there are cases of subjects with extra data after the part count.
|
|
*/
|
|
if (preg_match('/^\s*(?!"Usenet Index Post)(.+)\s+\((\d+)\/(\d+)\)/', $header['Subject'], $header['matches'])) {
|
|
// Add yEnc to subjects that do not have them, but have the part number at the end of the header.
|
|
if (stripos($header['Subject'], 'yEnc') === false) {
|
|
$header['matches'][1] .= ' yEnc';
|
|
}
|
|
} else {
|
|
if ($this->_showDroppedYEncParts === true && strpos($header['Subject'], '"Usenet Index Post') !== 0) {
|
|
file_put_contents(
|
|
NN_LOGS.'not_yenc'.$this->groupMySQL['name'].'.dropped.log',
|
|
$header['Subject'].PHP_EOL,
|
|
FILE_APPEND
|
|
);
|
|
}
|
|
$this->notYEnc++;
|
|
continue;
|
|
}
|
|
|
|
// Filter subject based on black/white list.
|
|
if ($this->isBlackListed($header, $this->groupMySQL['name'])) {
|
|
$this->headersBlackListed++;
|
|
continue;
|
|
}
|
|
|
|
if (! isset($header['Bytes'])) {
|
|
$header['Bytes'] = (isset($this->header[':bytes']) ? $header[':bytes'] : 0);
|
|
}
|
|
$header['Bytes'] = (int) $header['Bytes'];
|
|
|
|
if ($this->allAsMgr === true || ($mgrActive === true && array_key_exists($header['From'], $mgrPosters))) {
|
|
$mgrHeaders[] = $header;
|
|
} else {
|
|
$stdHeaders[] = $header;
|
|
}
|
|
}
|
|
|
|
unset($headers); // Reclaim memory now that headers are split.
|
|
|
|
if (! empty($this->_binaryBlacklistIdsToUpdate)) {
|
|
$this->updateBlacklistUsage();
|
|
}
|
|
|
|
if ($this->_echoCLI && $partRepair === false) {
|
|
$this->outputHeaderInitial();
|
|
}
|
|
|
|
// MGR headers goes first
|
|
if (! empty($mgrHeaders)) {
|
|
$this->tableNames = ProcessReleasesMultiGroup::tableNames();
|
|
$this->storeHeaders($mgrHeaders, true);
|
|
}
|
|
unset($mgrHeaders);
|
|
|
|
// Standard headers go second so we can switch tableNames back and do part repair to standard group tables
|
|
if (! empty($stdHeaders)) {
|
|
$this->tableNames = (new Group())->getCBPTableNames($this->groupMySQL['id']);
|
|
$this->storeHeaders($stdHeaders, false);
|
|
}
|
|
unset($stdHeaders);
|
|
|
|
// Start of part repair.
|
|
$this->startPR = microtime(true);
|
|
|
|
// End of inserting.
|
|
$this->timeInsert = number_format($this->startPR - $this->startUpdate, 2);
|
|
|
|
if ($partRepair && \count($headersRepaired) > 0) {
|
|
$this->removeRepairedParts($headersRepaired, $this->tableNames['prname'], $this->groupMySQL['id']);
|
|
}
|
|
unset($headersRepaired);
|
|
|
|
if ($this->addToPartRepair) {
|
|
$notInsertedCount = \count($this->headersNotInserted);
|
|
if ($notInsertedCount > 0) {
|
|
$this->addMissingParts($this->headersNotInserted, $this->tableNames['prname'], $this->groupMySQL['id']);
|
|
|
|
$this->log(
|
|
$notInsertedCount.' articles failed to insert!',
|
|
__FUNCTION__,
|
|
'warning'
|
|
);
|
|
}
|
|
unset($this->headersNotInserted);
|
|
|
|
// Check if we have any missing headers.
|
|
if (($this->last - $this->first - $this->notYEnc - $this->headersBlackListed + 1) > \count($this->headersReceived)) {
|
|
$rangeNotReceived = array_merge($rangeNotReceived, array_diff(range($this->first, $this->last), $this->headersReceived));
|
|
}
|
|
$notReceivedCount = \count($rangeNotReceived);
|
|
if ($notReceivedCount > 0) {
|
|
$this->addMissingParts($rangeNotReceived, $this->tableNames['prname'], $this->groupMySQL['id']);
|
|
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::alternate(
|
|
'Server did not return '.$notReceivedCount.
|
|
' articles from '.$this->groupMySQL['name'].'.'
|
|
),
|
|
true
|
|
);
|
|
}
|
|
}
|
|
unset($rangeNotReceived);
|
|
}
|
|
|
|
$this->outputHeaderDuration();
|
|
|
|
return $returnArray;
|
|
}
|
|
|
|
/**
|
|
* Parse headers into collections/binaries and store header data as parts.
|
|
*
|
|
* @param array $headers The retrieved headers
|
|
* @param bool $multiGroup Is this task being run in MGR mode?
|
|
*
|
|
* @throws \Exception
|
|
*/
|
|
protected function storeHeaders(array $headers, $multiGroup): void
|
|
{
|
|
$this->multiGroup = $multiGroup;
|
|
$binariesUpdate = $collectionIDs = $articles = [];
|
|
|
|
$this->_pdo->beginTransaction();
|
|
|
|
$partsQuery = $partsCheck =
|
|
"INSERT IGNORE INTO {$this->tableNames['pname']} (binaries_id, number, messageid, partnumber, size) VALUES ";
|
|
|
|
// Loop articles, figure out files/parts.
|
|
foreach ($headers as $this->header) {
|
|
// Set up the info for inserting into parts/binaries/collections tables.
|
|
if (! isset($articles[$this->header['matches'][1]])) {
|
|
|
|
// check whether file count should be ignored (XXX packs for now only).
|
|
$whitelistMatch = false;
|
|
if ($this->_ignoreFileCount($this->groupMySQL['name'], $this->header['matches'][1])) {
|
|
$whitelistMatch = true;
|
|
$fileCount[1] = $fileCount[3] = 0;
|
|
}
|
|
|
|
// Attempt to find the file count. If it is not found, set it to 0.
|
|
if (! $whitelistMatch && ! preg_match('/[[(\s](\d{1,5})(\/|[\s_]of[\s_]|-)(\d{1,5})[])\s$:]/i', $this->header['matches'][1], $fileCount)) {
|
|
$fileCount[1] = $fileCount[3] = 0;
|
|
if ($this->_showDroppedYEncParts === true) {
|
|
file_put_contents(
|
|
NN_LOGS.'no_files'.$this->groupMySQL['name'].'.log',
|
|
$this->header['Subject'].PHP_EOL,
|
|
FILE_APPEND
|
|
);
|
|
}
|
|
}
|
|
|
|
if ($this->multiGroup) {
|
|
$ckName = '';
|
|
$ckId = '';
|
|
} else {
|
|
$ckName = $this->groupMySQL['name'];
|
|
$ckId = $this->groupMySQL['id'];
|
|
}
|
|
|
|
$collMatch = $this->_collectionsCleaning->collectionsCleaner(
|
|
$this->header['matches'][1],
|
|
$ckName
|
|
);
|
|
|
|
// Used to group articles together when forming the release. MGR requires this to be group irrespective
|
|
$this->header['CollectionKey'] = $collMatch['name'].$ckId.$fileCount[3];
|
|
|
|
// If this header's collection key isn't in memory, attempt to insert the collection
|
|
if (! isset($collectionIDs[$this->header['CollectionKey']])) {
|
|
|
|
/* Date from header should be a string this format:
|
|
* 31 Mar 2014 15:36:04 GMT or 6 Oct 1998 04:38:40 -0500
|
|
* Still make sure it's not unix time, convert it to unix time if it is.
|
|
*/
|
|
$this->header['Date'] = (is_numeric($this->header['Date']) ? $this->header['Date'] : strtotime($this->header['Date']));
|
|
|
|
// Get the current unixtime from PHP.
|
|
$now = Carbon::now()->timestamp;
|
|
|
|
$xref = ($this->multiGroup === true ? sprintf('xref = CONCAT(xref, "\\n"%s ),', $this->_pdo->escapeString(substr($this->header['Xref'], 2, 255))) : '');
|
|
$date = $this->header['Date'] > $now ? $now : $this->header['Date'];
|
|
$unixtime = is_numeric($this->header['Date']) ? $date : $now;
|
|
|
|
$random = random_bytes(16);
|
|
|
|
$collectionID = $this->_pdo->queryInsert(
|
|
sprintf(
|
|
"
|
|
INSERT INTO %s (subject, fromname, date, xref, groups_id,
|
|
totalfiles, collectionhash, collection_regexes_id, dateadded)
|
|
VALUES (%s, %s, FROM_UNIXTIME(%s), %s, %d, %d, '%s', %d, NOW())
|
|
ON DUPLICATE KEY UPDATE %s dateadded = NOW(), noise = '%s'",
|
|
$this->tableNames['cname'],
|
|
$this->_pdo->escapeString(substr(utf8_encode($this->header['matches'][1]), 0, 255)),
|
|
$this->_pdo->escapeString(utf8_encode($this->header['From'])),
|
|
$unixtime,
|
|
$this->_pdo->escapeString(substr($this->header['Xref'], 0, 255)),
|
|
$this->groupMySQL['id'],
|
|
$fileCount[3],
|
|
sha1($this->header['CollectionKey']),
|
|
$collMatch['id'],
|
|
$xref,
|
|
sodium_bin2hex($random)
|
|
)
|
|
);
|
|
|
|
if ($collectionID === false) {
|
|
if ($this->addToPartRepair) {
|
|
$this->headersNotInserted[] = $this->header['Number'];
|
|
}
|
|
$this->_pdo->Rollback();
|
|
$this->_pdo->beginTransaction();
|
|
continue;
|
|
}
|
|
$collectionIDs[$this->header['CollectionKey']] = $collectionID;
|
|
} else {
|
|
$collectionID = $collectionIDs[$this->header['CollectionKey']];
|
|
}
|
|
|
|
// MGR or Standard, Binary Hash should be unique to the group
|
|
$hash = md5($this->header['matches'][1].$this->header['From'].$this->groupMySQL['id']);
|
|
|
|
$binaryID = $this->_pdo->queryInsert(
|
|
sprintf(
|
|
"
|
|
INSERT INTO %s (binaryhash, name, collections_id, totalparts, currentparts, filenumber, partsize)
|
|
VALUES (UNHEX('%s'), %s, %d, %d, 1, %d, %d)
|
|
ON DUPLICATE KEY UPDATE currentparts = currentparts + 1, partsize = partsize + %d",
|
|
$this->tableNames['bname'],
|
|
$hash,
|
|
$this->_pdo->escapeString(utf8_encode($this->header['matches'][1])),
|
|
$collectionID,
|
|
$this->header['matches'][3],
|
|
$fileCount[1],
|
|
$this->header['Bytes'],
|
|
$this->header['Bytes']
|
|
)
|
|
);
|
|
|
|
if ($binaryID === false) {
|
|
if ($this->addToPartRepair) {
|
|
$this->headersNotInserted[] = $this->header['Number'];
|
|
}
|
|
$this->_pdo->Rollback();
|
|
$this->_pdo->beginTransaction();
|
|
continue;
|
|
}
|
|
|
|
$binariesUpdate[$binaryID]['Size'] = 0;
|
|
$binariesUpdate[$binaryID]['Parts'] = 0;
|
|
|
|
$articles[$this->header['matches'][1]]['CollectionID'] = $collectionID;
|
|
$articles[$this->header['matches'][1]]['BinaryID'] = $binaryID;
|
|
} else {
|
|
$binaryID = $articles[$this->header['matches'][1]]['BinaryID'];
|
|
$binariesUpdate[$binaryID]['Size'] += $this->header['Bytes'];
|
|
$binariesUpdate[$binaryID]['Parts']++;
|
|
}
|
|
|
|
// Strip the < and >, saves space in DB.
|
|
$this->header['Message-ID'][0] = "'";
|
|
|
|
$partsQuery .=
|
|
'('.$binaryID.','.$this->header['Number'].','.rtrim($this->header['Message-ID'], '>')."',".
|
|
$this->header['matches'][2].','.$this->header['Bytes'].'),';
|
|
}
|
|
|
|
unset($headers); // Reclaim memory.
|
|
|
|
// Start of inserting into SQL.
|
|
$this->startUpdate = microtime(true);
|
|
|
|
// End of processing headers.
|
|
$this->timeCleaning = number_format($this->startUpdate - $this->startCleaning, 2);
|
|
$binariesQuery = $binariesCheck = sprintf('INSERT INTO %s (id, partsize, currentparts) VALUES ', $this->tableNames['bname']);
|
|
foreach ($binariesUpdate as $binaryID => $binary) {
|
|
$binariesQuery .= '('.$binaryID.','.$binary['Size'].','.$binary['Parts'].'),';
|
|
}
|
|
$binariesEnd = ' ON DUPLICATE KEY UPDATE partsize = VALUES(partsize) + partsize, currentparts = VALUES(currentparts) + currentparts';
|
|
$binariesQuery = rtrim($binariesQuery, ',').$binariesEnd;
|
|
|
|
// Check if we got any binaries. If we did, try to insert them.
|
|
if (\strlen($binariesCheck.$binariesEnd) === \strlen($binariesQuery) ? true : $this->_pdo->queryExec($binariesQuery)) {
|
|
if ($this->_debug) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::debug(
|
|
'Sending '.round(\strlen($partsQuery) / 1024, 2).
|
|
' KB of'.($this->multiGroup ? ' MGR' : '').' parts to MySQL'
|
|
), true
|
|
);
|
|
}
|
|
if (\strlen($partsQuery) === \strlen($partsCheck) ? true : $this->_pdo->queryExec(rtrim($partsQuery, ','))) {
|
|
$this->_pdo->Commit();
|
|
} else {
|
|
if ($this->addToPartRepair) {
|
|
$this->headersNotInserted += $this->headersReceived;
|
|
}
|
|
$this->_pdo->Rollback();
|
|
}
|
|
} else {
|
|
if ($this->addToPartRepair) {
|
|
$this->headersNotInserted += $this->headersReceived;
|
|
}
|
|
$this->_pdo->Rollback();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Gets the First and Last Article Number and Date for the received headers.
|
|
*
|
|
* @param array $returnArray
|
|
* @param array $headers
|
|
* @param int $msgCount
|
|
*/
|
|
protected function getHighLowArticleInfo(array &$returnArray, array $headers, int $msgCount): void
|
|
{
|
|
// Get highest and lowest article numbers/dates.
|
|
$iterator1 = 0;
|
|
$iterator2 = $msgCount - 1;
|
|
while (true) {
|
|
if (! isset($returnArray['firstArticleNumber']) && isset($headers[$iterator1]['Number'])) {
|
|
$returnArray['firstArticleNumber'] = $headers[$iterator1]['Number'];
|
|
$returnArray['firstArticleDate'] = $headers[$iterator1]['Date'];
|
|
}
|
|
|
|
if (! isset($returnArray['lastArticleNumber']) && isset($headers[$iterator2]['Number'])) {
|
|
$returnArray['lastArticleNumber'] = $headers[$iterator2]['Number'];
|
|
$returnArray['lastArticleDate'] = $headers[$iterator2]['Date'];
|
|
}
|
|
|
|
// Break if we found non empty articles.
|
|
if (isset($returnArray['firstArticleNumber, lastArticleNumber'])) {
|
|
break;
|
|
}
|
|
|
|
// Break out if we couldn't find anything.
|
|
if ($iterator1++ >= $msgCount - 1 || $iterator2-- <= 0) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Updates Blacklist Regex Timers in DB to reflect last usage.
|
|
*/
|
|
protected function updateBlacklistUsage(): void
|
|
{
|
|
BinaryBlacklist::query()->whereIn('id', $this->_binaryBlacklistIdsToUpdate)->update(['last_activity' => Carbon::now()]);
|
|
$this->_binaryBlacklistIdsToUpdate = [];
|
|
}
|
|
|
|
/**
|
|
* Outputs the initial header scan results after yEnc check and blacklist routines.
|
|
*/
|
|
protected function outputHeaderInitial(): void
|
|
{
|
|
ColorCLI::doEcho(
|
|
ColorCLI::primary(
|
|
'Received '.\count($this->headersReceived).
|
|
' articles of '.number_format($this->last - $this->first + 1).' requested, '.
|
|
$this->headersBlackListed.' blacklisted, '.$this->notYEnc.' not yEnc.'
|
|
), true
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Outputs speed metrics of the scan function to CLI.
|
|
*/
|
|
protected function outputHeaderDuration(): void
|
|
{
|
|
$currentMicroTime = microtime(true);
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::alternateOver($this->timeHeaders.'s').
|
|
ColorCLI::primaryOver(' to download articles, ').
|
|
ColorCLI::alternateOver($this->timeCleaning.'s').
|
|
ColorCLI::primaryOver(' to process collections, ').
|
|
ColorCLI::alternateOver($this->timeInsert.'s').
|
|
ColorCLI::primaryOver(' to insert binaries/parts, ').
|
|
ColorCLI::alternateOver(number_format($currentMicroTime - $this->startPR, 2).'s').
|
|
ColorCLI::primaryOver(' for part repair, ').
|
|
ColorCLI::alternateOver(number_format($currentMicroTime - $this->startLoop, 2).'s').
|
|
ColorCLI::primary(' total.')
|
|
, true);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* If we failed to insert Collections/Binaries/Parts, rollback the transaction and add the parts to part repair.
|
|
*
|
|
* @param array $headers Array of headers containing sub-arrays with parts.
|
|
*
|
|
* @return array Array of article numbers to add to part repair.
|
|
*/
|
|
protected function _rollbackAddToPartRepair(array $headers): array
|
|
{
|
|
$headersNotInserted = [];
|
|
foreach ($headers as $header) {
|
|
foreach ($header as $file) {
|
|
$headersNotInserted[] = $file['Parts']['number'];
|
|
}
|
|
}
|
|
$this->_pdo->Rollback();
|
|
|
|
return $headersNotInserted;
|
|
}
|
|
|
|
/**
|
|
* Attempt to get missing article headers.
|
|
*
|
|
* @param array|string $tables
|
|
* @param array $groupArr The info for this group from mysql.
|
|
*
|
|
* @return void
|
|
* @throws \Exception
|
|
*/
|
|
public function partRepair($groupArr, $tables = ''): void
|
|
{
|
|
$tableNames = $tables;
|
|
|
|
if ($tableNames === '') {
|
|
$tableNames = (new Group())->getCBPTableNames($groupArr['id']);
|
|
}
|
|
// Get all parts in partrepair table.
|
|
$missingParts = $this->_pdo->query(
|
|
sprintf(
|
|
'
|
|
SELECT * FROM %s
|
|
WHERE groups_id = %d AND attempts < %d
|
|
ORDER BY numberid ASC LIMIT %d',
|
|
$tableNames['prname'],
|
|
$groupArr['id'],
|
|
$this->_partRepairMaxTries,
|
|
$this->_partRepairLimit
|
|
)
|
|
);
|
|
|
|
$missingCount = \count($missingParts);
|
|
if ($missingCount > 0) {
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::primary(
|
|
'Attempting to repair '.
|
|
number_format($missingCount).
|
|
' parts.'
|
|
),
|
|
true
|
|
);
|
|
}
|
|
|
|
// Loop through each part to group into continuous ranges with a maximum range of messagebuffer/4.
|
|
$ranges = $partList = [];
|
|
$firstPart = $lastNum = $missingParts[0]['numberid'];
|
|
|
|
foreach ($missingParts as $part) {
|
|
if (($part['numberid'] - $firstPart) > ($this->messageBuffer / 4)) {
|
|
$ranges[] = [
|
|
'partfrom' => $firstPart,
|
|
'partto' => $lastNum,
|
|
'partlist' => $partList,
|
|
];
|
|
|
|
$firstPart = $part['numberid'];
|
|
$partList = [];
|
|
}
|
|
$partList[] = $part['numberid'];
|
|
$lastNum = $part['numberid'];
|
|
}
|
|
|
|
$ranges[] = [
|
|
'partfrom' => $firstPart,
|
|
'partto' => $lastNum,
|
|
'partlist' => $partList,
|
|
];
|
|
|
|
// Download missing parts in ranges.
|
|
foreach ($ranges as $range) {
|
|
$partFrom = $range['partfrom'];
|
|
$partTo = $range['partto'];
|
|
$partList = $range['partlist'];
|
|
|
|
if ($this->_echoCLI) {
|
|
echo \chr(random_int(45, 46)).PHP_EOL;
|
|
}
|
|
|
|
// Get article headers from newsgroup.
|
|
$this->scan($groupArr, $partFrom, $partTo, 'missed_parts', $partList);
|
|
}
|
|
|
|
// Calculate parts repaired
|
|
$result = $this->_pdo->queryOneRow(
|
|
sprintf(
|
|
'
|
|
SELECT COUNT(id) AS num
|
|
FROM %s
|
|
WHERE groups_id = %d
|
|
AND numberid <= %d',
|
|
$tableNames['prname'],
|
|
$groupArr['id'],
|
|
$missingParts[$missingCount - 1]['numberid']
|
|
)
|
|
);
|
|
|
|
$partsRepaired = 0;
|
|
if ($result !== false) {
|
|
$partsRepaired = ($missingCount - $result['num']);
|
|
}
|
|
|
|
// Update attempts on remaining parts for active group
|
|
if (isset($missingParts[$missingCount - 1]['id'])) {
|
|
$this->_pdo->queryExec(
|
|
sprintf(
|
|
'
|
|
UPDATE %s
|
|
SET attempts = attempts + 1
|
|
WHERE groups_id = %d
|
|
AND numberid <= %d',
|
|
$tableNames['prname'],
|
|
$groupArr['id'],
|
|
$missingParts[$missingCount - 1]['numberid']
|
|
)
|
|
);
|
|
}
|
|
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::primary(
|
|
PHP_EOL.
|
|
number_format($partsRepaired).
|
|
' parts repaired.'
|
|
),
|
|
true
|
|
);
|
|
}
|
|
}
|
|
|
|
// Remove articles that we cant fetch after x attempts.
|
|
$this->_pdo->queryExec(
|
|
sprintf(
|
|
'DELETE FROM %s WHERE attempts >= %d AND groups_id = %d',
|
|
$tableNames['prname'],
|
|
$this->_partRepairMaxTries,
|
|
$groupArr['id']
|
|
)
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Returns unix time for an article number.
|
|
*
|
|
* @param int $post The article number to get the time from.
|
|
* @param array $groupData Usenet group info from NNTP selectGroup method.
|
|
*
|
|
* @return int Timestamp.
|
|
* @throws \Exception
|
|
*/
|
|
public function postdate($post, array $groupData): int
|
|
{
|
|
// Set table names
|
|
$groupID = Group::getIDByName($groupData['group']);
|
|
$group = [];
|
|
if ($groupID !== '') {
|
|
$group = (new Group())->getCBPTableNames($groupID);
|
|
}
|
|
|
|
$currentPost = $post;
|
|
|
|
$attempts = $date = 0;
|
|
do {
|
|
// Try to get the article date locally first.
|
|
if ($groupID !== '') {
|
|
// Try to get locally.
|
|
$local = $this->_pdo->queryOneRow(
|
|
sprintf(
|
|
'
|
|
SELECT c.date AS date
|
|
FROM %s c
|
|
INNER JOIN %s b ON(c.id=b.collections_id)
|
|
INNER JOIN %s p ON(b.id=p.binaries_id)
|
|
WHERE p.number = %s',
|
|
$group['cname'],
|
|
$group['bname'],
|
|
$group['pname'],
|
|
$currentPost
|
|
)
|
|
);
|
|
if ($local !== false) {
|
|
$date = $local['date'];
|
|
break;
|
|
}
|
|
}
|
|
|
|
// If we could not find it locally, try usenet.
|
|
$header = $this->_nntp->getXOVER($currentPost);
|
|
if (! $this->_nntp->isError($header)) {
|
|
// Check if the date is set.
|
|
if (isset($header[0]['Date']) && \strlen($header[0]['Date']) > 0) {
|
|
$date = $header[0]['Date'];
|
|
break;
|
|
}
|
|
}
|
|
|
|
// Try to get a different article number.
|
|
if (abs($currentPost - $groupData['first']) > abs($groupData['last'] - $currentPost)) {
|
|
$tempPost = round($currentPost / (random_int(1005, 1012) / 1000), 0, PHP_ROUND_HALF_UP);
|
|
if ($tempPost < $groupData['first']) {
|
|
$tempPost = $groupData['first'];
|
|
}
|
|
} else {
|
|
$tempPost = round((random_int(1005, 1012) / 1000) * $currentPost, 0, PHP_ROUND_HALF_UP);
|
|
if ($tempPost > $groupData['last']) {
|
|
$tempPost = $groupData['last'];
|
|
}
|
|
}
|
|
// If we got the same article number as last time, give up.
|
|
if ($tempPost === $currentPost) {
|
|
break;
|
|
}
|
|
$currentPost = $tempPost;
|
|
|
|
if ($this->_debug) {
|
|
ColorCLI::doEcho(ColorCLI::debug('Postdate retried '.$attempts.' time(s).'), true);
|
|
}
|
|
} while ($attempts++ <= 20);
|
|
|
|
// If we didn't get a date, set it to now.
|
|
if (! $date) {
|
|
$date = time();
|
|
} else {
|
|
$date = strtotime($date);
|
|
}
|
|
|
|
return $date;
|
|
}
|
|
|
|
/**
|
|
* Returns article number based on # of days.
|
|
*
|
|
* @param int $days How many days back we want to go.
|
|
* @param array $data Group data from usenet.
|
|
*
|
|
* @return string
|
|
* @throws \Exception
|
|
*/
|
|
public function daytopost($days, $data): string
|
|
{
|
|
$goalTime = Carbon::now()->subDays($days)->timestamp;
|
|
// The time we want = current unix time (ex. 1395699114) - minus 86400 (seconds in a day)
|
|
// times days wanted. (ie 1395699114 - 2592000 (30days)) = 1393107114
|
|
|
|
// The servers oldest date.
|
|
$firstDate = $this->postdate($data['first'], $data);
|
|
if ($goalTime < $firstDate) {
|
|
// If the date we want is older than the oldest date in the group return the groups oldest article.
|
|
return $data['first'];
|
|
}
|
|
|
|
// The servers newest date.
|
|
$lastDate = $this->postdate($data['last'], $data);
|
|
if ($goalTime > $lastDate) {
|
|
// If the date we want is newer than the groups newest date, return the groups newest article.
|
|
return $data['last'];
|
|
}
|
|
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::primary(
|
|
'Searching for an approximate article number for group '.$data['group'].' '.$days.' days back.'
|
|
), true
|
|
);
|
|
}
|
|
|
|
// Pick the middle to start with
|
|
$wantedArticle = round(($data['last'] + $data['first']) / 2);
|
|
$aMax = $data['last'];
|
|
$aMin = $data['first'];
|
|
$reallyOldArticle = $oldArticle = $articleTime = null;
|
|
|
|
while (true) {
|
|
// Article exists outside of available range, this shouldn't happen
|
|
if ($wantedArticle <= $data['first'] || $wantedArticle >= $data['last']) {
|
|
break;
|
|
}
|
|
|
|
// Keep a note of the last articles we checked
|
|
$reallyOldArticle = $oldArticle;
|
|
$oldArticle = $wantedArticle;
|
|
|
|
// Get the date of this article
|
|
$articleTime = $this->postdate($wantedArticle, $data);
|
|
|
|
// Article doesn't exist, start again with something random
|
|
if (! $articleTime) {
|
|
$wantedArticle = random_int($aMin, $aMax);
|
|
$articleTime = $this->postdate($wantedArticle, $data);
|
|
}
|
|
|
|
if ($articleTime < $goalTime) {
|
|
// Article is older than we want
|
|
$aMin = $oldArticle;
|
|
$wantedArticle = round(($aMax + $oldArticle) / 2);
|
|
if ($this->_echoCLI) {
|
|
echo '-';
|
|
}
|
|
} elseif ($articleTime > $goalTime) {
|
|
// Article is newer than we want
|
|
$aMax = $oldArticle;
|
|
$wantedArticle = round(($aMin + $oldArticle) / 2);
|
|
if ($this->_echoCLI) {
|
|
echo '+';
|
|
}
|
|
} elseif ($articleTime === $goalTime) {
|
|
// Exact match. We did it! (this will likely never happen though)
|
|
break;
|
|
}
|
|
|
|
// We seem to be flip-flopping between 2 articles, assume we're out of articles to check.
|
|
// End on an article more recent than our oldest so that we don't miss any releases.
|
|
if ($reallyOldArticle === $wantedArticle && ($goalTime - $articleTime) <= 0) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
$wantedArticle = (int) $wantedArticle;
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::primary(
|
|
PHP_EOL.'Found article #'.$wantedArticle.' which has a date of '.date('r', $articleTime).
|
|
', vs wanted date of '.date('r', $goalTime).'. Difference from goal is '.Carbon::createFromTimestamp($goalTime)->diffInDays(Carbon::createFromTimestamp($articleTime)).'days.'
|
|
), true
|
|
);
|
|
}
|
|
|
|
return $wantedArticle;
|
|
}
|
|
|
|
/**
|
|
* Convert unix time to days ago.
|
|
*
|
|
*
|
|
* @param $timestamp
|
|
* @return int
|
|
*/
|
|
private function daysOld($timestamp): int
|
|
{
|
|
return Carbon::createFromTimestamp($timestamp)->diffInDays();
|
|
}
|
|
|
|
/**
|
|
* Add article numbers from missing headers to DB.
|
|
*
|
|
* @param array $numbers The article numbers of the missing headers.
|
|
* @param string $tableName Name of the partrepair table to insert into.
|
|
* @param int $groupID The ID of this groups.
|
|
*
|
|
* @return bool
|
|
*/
|
|
private function addMissingParts($numbers, $tableName, $groupID): bool
|
|
{
|
|
$insertStr = 'INSERT INTO '.$tableName.' (numberid, groups_id) VALUES ';
|
|
foreach ($numbers as $number) {
|
|
$insertStr .= '('.$number.','.$groupID.'),';
|
|
}
|
|
|
|
return $this->_pdo->queryInsert(rtrim($insertStr, ',').' ON DUPLICATE KEY UPDATE attempts=attempts+1');
|
|
}
|
|
|
|
/**
|
|
* Clean up part repair table.
|
|
*
|
|
* @param array $numbers The article numbers.
|
|
* @param string $tableName Name of the part repair table to work on.
|
|
* @param int $groupID The ID of the group.
|
|
*
|
|
* @return void
|
|
*/
|
|
private function removeRepairedParts(array $numbers, $tableName, $groupID): void
|
|
{
|
|
$sql = 'DELETE FROM '.$tableName.' WHERE numberid in (';
|
|
foreach ($numbers as $number) {
|
|
$sql .= $number.',';
|
|
}
|
|
$this->_pdo->queryExec(rtrim($sql, ',').') AND groups_id = '.$groupID);
|
|
}
|
|
|
|
/**
|
|
* Are white or black lists loaded for a group name?
|
|
* @var array
|
|
*/
|
|
protected $_listsFound = [];
|
|
|
|
/**
|
|
* Get blacklist and cache it. Return if already cached.
|
|
*
|
|
* @param string $groupName
|
|
*
|
|
* @return void
|
|
*/
|
|
protected function _retrieveBlackList($groupName): void
|
|
{
|
|
if (! isset($this->blackList[$groupName])) {
|
|
$this->blackList[$groupName] = $this->getBlacklist(true, self::OPTYPE_BLACKLIST, $groupName, true);
|
|
}
|
|
if (! isset($this->whiteList[$groupName])) {
|
|
$this->whiteList[$groupName] = $this->getBlacklist(true, self::OPTYPE_WHITELIST, $groupName, true);
|
|
}
|
|
$this->_listsFound[$groupName] = ($this->blackList[$groupName] || $this->whiteList[$groupName]);
|
|
}
|
|
|
|
/**
|
|
* Check if an article is blacklisted.
|
|
*
|
|
* @param array $msg The article header (OVER format).
|
|
* @param string $groupName The group name.
|
|
*
|
|
* @return bool
|
|
*/
|
|
public function isBlackListed($msg, $groupName): bool
|
|
{
|
|
if (! isset($this->_listsFound[$groupName])) {
|
|
$this->_retrieveBlackList($groupName);
|
|
}
|
|
if (! $this->_listsFound[$groupName]) {
|
|
return false;
|
|
}
|
|
|
|
$blackListed = false;
|
|
|
|
$field = [
|
|
self::BLACKLIST_FIELD_SUBJECT => $msg['Subject'],
|
|
self::BLACKLIST_FIELD_FROM => $msg['From'],
|
|
self::BLACKLIST_FIELD_MESSAGEID => $msg['Message-ID'],
|
|
];
|
|
|
|
// Try white lists first.
|
|
if ($this->whiteList[$groupName]) {
|
|
// There are white lists for this group, so anything that doesn't match a white list should be considered black listed.
|
|
$blackListed = true;
|
|
foreach ($this->whiteList[$groupName] as $whiteList) {
|
|
if (preg_match('/'.$whiteList['regex'].'/i', $field[$whiteList['msgcol']])) {
|
|
// This field matched a white list, so it might not be black listed.
|
|
$blackListed = false;
|
|
$this->_binaryBlacklistIdsToUpdate[$whiteList['id']] = $whiteList['id'];
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Check if the field is black listed.
|
|
if (! $blackListed && $this->blackList[$groupName]) {
|
|
foreach ($this->blackList[$groupName] as $blackList) {
|
|
if (preg_match('/'.$blackList['regex'].'/i', $field[$blackList['msgcol']])) {
|
|
$blackListed = true;
|
|
$this->_binaryBlacklistIdsToUpdate[$blackList['id']] = $blackList['id'];
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
return $blackListed;
|
|
}
|
|
|
|
/**
|
|
* Return all blacklists.
|
|
*
|
|
* @param bool $activeOnly Only display active blacklists ?
|
|
* @param int|string $opType Optional, get white or black lists (use Binaries constants).
|
|
* @param string $groupName Optional, group.
|
|
* @param bool $groupRegex Optional Join groups / binaryblacklist using regexp for equals.
|
|
*
|
|
* @return array
|
|
*/
|
|
public function getBlacklist($activeOnly = true, $opType = -1, $groupName = '', $groupRegex = false): array
|
|
{
|
|
switch ($opType) {
|
|
case self::OPTYPE_BLACKLIST:
|
|
$opType = 'AND bb.optype = '.self::OPTYPE_BLACKLIST;
|
|
break;
|
|
case self::OPTYPE_WHITELIST:
|
|
$opType = 'AND bb.optype = '.self::OPTYPE_WHITELIST;
|
|
break;
|
|
default:
|
|
$opType = '';
|
|
break;
|
|
}
|
|
|
|
return $this->_pdo->query(
|
|
sprintf(
|
|
'
|
|
SELECT
|
|
bb.id, bb.optype, bb.status, bb.description,
|
|
bb.groupname AS groupname, bb.regex, g.id AS group_id, bb.msgcol,
|
|
bb.last_activity as last_activity
|
|
FROM binaryblacklist bb
|
|
LEFT OUTER JOIN groups g ON g.name %s bb.groupname
|
|
WHERE 1=1 %s %s %s
|
|
ORDER BY coalesce(groupname,\'zzz\')',
|
|
($groupRegex ? 'REGEXP' : '='),
|
|
($activeOnly ? 'AND bb.status = 1' : ''),
|
|
$opType,
|
|
($groupName ? ('AND g.name REGEXP '.$this->_pdo->escapeString($groupName)) : '')
|
|
)
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Return the specified blacklist.
|
|
*
|
|
* @param int $id The blacklist ID.
|
|
*
|
|
* @return \Illuminate\Database\Eloquent\Model|null|static
|
|
*/
|
|
public function getBlacklistByID($id)
|
|
{
|
|
return BinaryBlacklist::query()->where('id', $id)->first();
|
|
}
|
|
|
|
/**
|
|
* Delete a blacklist.
|
|
*
|
|
* @param int $id The ID of the blacklist.
|
|
*/
|
|
public function deleteBlacklist($id): void
|
|
{
|
|
BinaryBlacklist::query()->where('id', $id)->delete();
|
|
}
|
|
|
|
/**
|
|
* @param $blacklistArray
|
|
*/
|
|
public function updateBlacklist($blacklistArray): void
|
|
{
|
|
BinaryBlacklist::query()->where('id', $blacklistArray['id'])->update(
|
|
[
|
|
'groupname' => $blacklistArray['groupname'] === '' ? 'null' : preg_replace('/a\.b\./i', 'alt.binaries.', $blacklistArray['groupname']),
|
|
'regex' => $blacklistArray['regex'],
|
|
'status' => $blacklistArray['status'],
|
|
'description' => $blacklistArray['description'],
|
|
'optype' => $blacklistArray['optype'],
|
|
'msgcol' => $blacklistArray['msgcol'],
|
|
]
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Adds a new blacklist from binary blacklist edit admin web page.
|
|
*
|
|
* @param array $blacklistArray
|
|
*/
|
|
public function addBlacklist($blacklistArray): void
|
|
{
|
|
BinaryBlacklist::query()->insert(
|
|
[
|
|
'groupname' => $blacklistArray['groupname'] === '' ? 'null' : preg_replace('/a\.b\./i', 'alt.binaries.', $blacklistArray['groupname']),
|
|
'regex' => $blacklistArray['regex'],
|
|
'status' => $blacklistArray['status'],
|
|
'description' => $blacklistArray['description'],
|
|
'optype' => $blacklistArray['optype'],
|
|
'msgcol' => $blacklistArray['msgcol'],
|
|
]
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Delete Collections/Binaries/Parts for a Collection ID.
|
|
*
|
|
* @param int $collectionID Collections table ID
|
|
*
|
|
* @note A trigger automatically deletes the parts/binaries.
|
|
*
|
|
* @return void
|
|
*/
|
|
public function delete($collectionID): void
|
|
{
|
|
$this->_pdo->queryExec(sprintf('DELETE FROM collections WHERE id = %d', $collectionID));
|
|
}
|
|
|
|
/**
|
|
* Log / Echo message.
|
|
*
|
|
* @param string $message Message to log.
|
|
* @param string $method Method that called this.
|
|
* @param string $color ColorCLI method name.
|
|
*/
|
|
private function log($message, $method, $color): void
|
|
{
|
|
if ($this->_echoCLI) {
|
|
ColorCLI::doEcho(
|
|
ColorCLI::$color($message.' ['.__CLASS__."::$method]"),
|
|
true
|
|
);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Check if we should ignore the file count and return true or false.
|
|
*
|
|
* @param string $groupName
|
|
* @param string $subject
|
|
*
|
|
* @return bool
|
|
*/
|
|
protected function _ignoreFileCount($groupName, $subject): bool
|
|
{
|
|
$ignore = false;
|
|
switch ($groupName) {
|
|
case 'alt.binaries.erotica':
|
|
if (preg_match('/^\[\d+\]-\[FULL\]-\[#a\.b\.erotica@EFNet\]-\[ \d{2,3}_/', $subject)) {
|
|
$ignore = true;
|
|
}
|
|
break;
|
|
}
|
|
|
|
return $ignore;
|
|
}
|
|
|
|
/**
|
|
* Returns all multigroup poster entries from the database.
|
|
*
|
|
*
|
|
* @return \Illuminate\Database\Eloquent\Collection|static[]
|
|
*/
|
|
protected function getMultiGroupPosters()
|
|
{
|
|
$poster = Cache::get('mgrposter');
|
|
if ($poster !== null) {
|
|
return $poster;
|
|
}
|
|
|
|
$poster = MultigroupPoster::query()->get(['poster'])->toArray();
|
|
$expiresAt = Carbon::now()->addSeconds(NN_CACHE_EXPIRY_SHORT);
|
|
Cache::put('mgrposter', $poster, $expiresAt);
|
|
|
|
return $poster;
|
|
}
|
|
}
|