mirror of
https://github.com/diocloid/LinkTitles.git
synced 2025-07-13 01:39:30 +02:00
Rename maintenance script.
This commit is contained in:
193
linktitles-cli.php
Executable file
193
linktitles-cli.php
Executable file
@ -0,0 +1,193 @@
|
||||
<?php
|
||||
/**
|
||||
* LinkTitles command line interface (CLI)/maintenance script
|
||||
*
|
||||
* Copyright 2012-2017 Daniel Kraus <bovender@bovender.de> @bovender
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation; either version 2 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with this program; if not, write to the Free Software
|
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||||
* MA 02110-1301, USA.
|
||||
*/
|
||||
namespace LinkTitles;
|
||||
|
||||
// Attempt to include the maintenance base class from:
|
||||
// $wgScriptPath/maintenance/Maintenance.php
|
||||
// Our script is normally located at:
|
||||
// $wgScriptPath/extensions/LinkTitles/LinkTitles_Maintenance.php
|
||||
$maintenanceScript = __DIR__ . "/../../maintenance/Maintenance.php";
|
||||
if ( file_exists( $maintenanceScript ) ) {
|
||||
require_once $maintenanceScript;
|
||||
}
|
||||
else
|
||||
{
|
||||
// Did not find the script where we expected it (maybe because we are a
|
||||
// symlinked file -- __DIR resolves symbolic links).
|
||||
$maintenanceScript = __DIR__ . "/Maintenance.php";
|
||||
if ( file_exists( $maintenanceScript ) ) {
|
||||
require_once $maintenanceScript;
|
||||
}
|
||||
else
|
||||
{
|
||||
die("FATAL: Could not locate Maintenance.php.\n" .
|
||||
"You may want to create a symbolic link named Maintenance.php in this directory\n" .
|
||||
"which points to <YOUR_MEDIAWIKI_ROOT_IN_FILESYSTEM>/extensions/Maintenance.php\n" .
|
||||
"Ex.: ln -s /var/www/wiki/maintenance/Maintenance.php\n\n");
|
||||
}
|
||||
};
|
||||
|
||||
require_once( __DIR__ . "/includes/Extension.php" );
|
||||
|
||||
/**
|
||||
* Core class of the maintanance script.
|
||||
* @note Note that the execution of maintenance scripts is prohibited for
|
||||
* an Apache web server due to a `.htaccess` file that declares `deny from
|
||||
* all`. Other webservers may exhibit different behavior. Be aware that
|
||||
* anybody who is able to execute this script may place a high load on the
|
||||
* server.
|
||||
* @ingroup batch
|
||||
*/
|
||||
class Cli extends \Maintenance {
|
||||
/**
|
||||
* Constructor.
|
||||
*/
|
||||
public function __construct() {
|
||||
parent::__construct();
|
||||
$this->addDescription("Iterates over wiki pages and automatically adds links to other pages.");
|
||||
$this->addOption(
|
||||
"start",
|
||||
"Set start index.",
|
||||
false, // not required
|
||||
true, // requires argument
|
||||
"s"
|
||||
);
|
||||
$this->addOption(
|
||||
"page",
|
||||
"page name to process",
|
||||
false, // not required
|
||||
true, // requires argument
|
||||
"p"
|
||||
);
|
||||
// TODO: Add back logging options.
|
||||
// TODO: Add configuration options.
|
||||
// $this->addOption(
|
||||
// "log",
|
||||
// "enables logging to console",
|
||||
// false, // not required
|
||||
// false, // requires no argument
|
||||
// "l"
|
||||
// );
|
||||
// $this->addOption(
|
||||
// "debug",
|
||||
// "enables debug logging to console",
|
||||
// false, // not required
|
||||
// false // requires no argument
|
||||
// );
|
||||
}
|
||||
|
||||
/*
|
||||
* Main function of the maintenance script.
|
||||
* Will iterate over all pages in the wiki (starting at a certain index,
|
||||
* if the `--start` option is given) and call LinkTitles::processPage() for
|
||||
* each page.
|
||||
*/
|
||||
public function execute() {
|
||||
// if ($this->hasOption('log'))
|
||||
// {
|
||||
// Extension::$ltConsoleOutput = true;
|
||||
// }
|
||||
// if ($this->hasOption('debug'))
|
||||
// {
|
||||
// Extension::$ltConsoleOutputDebug = true;
|
||||
// }
|
||||
if ( $this->hasOption('page') ) {
|
||||
if ( !$this->hasOption( 'start' ) ) {
|
||||
$this->singlePage();
|
||||
}
|
||||
else {
|
||||
$this->error( 'FATAL: Must not use --start option with --page option.', 2 );
|
||||
}
|
||||
}
|
||||
else {
|
||||
$startIndex = intval( $this->getOption( 'start', 0 ) );
|
||||
if ( $startIndex < 0 ) {
|
||||
$this->error( 'FATAL: Start index must be 0 or greater.', 1 );
|
||||
};
|
||||
$this->allPages( $startIndex );
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Processes a single page.
|
||||
* @return bool True on success, false on failure.
|
||||
*/
|
||||
private function singlePage() {
|
||||
$pageName = strval( $this->getOption( 'page' ) );
|
||||
$this->output( "Processing single page: '$pageName'\n" );
|
||||
$title = \Title::newFromText( $pageName );
|
||||
$success = Extension::processPage( $title, \RequestContext::getMain() );
|
||||
if ( $success ) {
|
||||
$this->output( "Finished.\n" );
|
||||
}
|
||||
else {
|
||||
$this->error( 'FATAL: There is no such page.', 3 );
|
||||
}
|
||||
return $success;
|
||||
}
|
||||
|
||||
/**
|
||||
* Process all pages in the Wiki.
|
||||
* @param integer $index Index of the start page.
|
||||
* @return bool True on success, false on failure.
|
||||
*/
|
||||
private function allPages( $index = 0 ) {
|
||||
$config = new Config();
|
||||
|
||||
// Retrieve page names from the database.
|
||||
$dbr = $this->getDB( DB_SLAVE );
|
||||
$nameSpacesClause = str_replace( '_', ' ','(' . implode( ', ', $config->nameSpaces ) . ')' );
|
||||
$res = $dbr->select(
|
||||
'page',
|
||||
array( 'page_title', 'page_namespace' ),
|
||||
array(
|
||||
'page_namespace IN ' . $nameSpacesClause,
|
||||
),
|
||||
__METHOD__,
|
||||
array(
|
||||
'LIMIT' => 999999999,
|
||||
'OFFSET' => $index
|
||||
)
|
||||
);
|
||||
$numPages = $res->numRows();
|
||||
$context = \RequestContext::getMain();
|
||||
$this->output( "Processing ${numPages} pages, starting at index ${index}...\n" );
|
||||
|
||||
foreach ( $res as $row ) {
|
||||
$index += 1; // at this point, $index is only needed for reporting to user
|
||||
$title = \Title::makeTitleSafe( $row->page_namespace, $row->page_title );
|
||||
$this->output( sprintf( "\rPage #%d (%02.0f%%) ", $index, $index / $numPages * 100 ) );
|
||||
Extension::processPage( $title, $context );
|
||||
}
|
||||
|
||||
$this->output( "\rFinished. \n" );
|
||||
}
|
||||
}
|
||||
|
||||
$maintClass = 'LinkTitles\Cli';
|
||||
if( defined('RUN_MAINTENANCE_IF_MAIN') ) {
|
||||
require_once( RUN_MAINTENANCE_IF_MAIN );
|
||||
} else {
|
||||
require_once( DO_MAINTENANCE );
|
||||
}
|
||||
|
||||
// vim: ts=2:sw=2:noet:comments^=\:///
|
Reference in New Issue
Block a user