mediawiki-extensions-Replac.../ReplaceTextSearch.php
Mark A. Hershberger 133982dc59 Create CLI utility for ReplaceText jobs
This creates a new class (ReplaceTextSearch) to move the logic from
the special page so that it is available for CLI access.  Also creates
replace.php which can be used for search-replace tasks that need to be
automated.

Change-Id: I8ccec61f570f33043d8a8d00c52b40acd9d6894a
2015-04-20 14:02:28 -04:00

59 lines
1.7 KiB
PHP

<?php
class ReplaceTextSearch {
public static function doSearchQuery( $search, $namespaces, $category, $prefix, $use_regex = false ) {
$dbr = wfGetDB( DB_SLAVE );
$tables = array( 'page', 'revision', 'text' );
$vars = array( 'page_id', 'page_namespace', 'page_title', 'old_text' );
if ( $use_regex ) {
$comparisonCond = self::regexCond( $dbr, 'old_text', $search );
} else {
$any = $dbr->anyString();
$comparisonCond = 'old_text ' . $dbr->buildLike( $any, $search, $any );
}
$conds = array(
$comparisonCond,
'page_namespace' => $namespaces,
'rev_id = page_latest',
'rev_text_id = old_id'
);
self::categoryCondition( $category, $tables, $conds );
self::prefixCondition( $prefix, $conds );
$sort = array( 'ORDER BY' => 'page_namespace, page_title' );
return $dbr->select( $tables, $vars, $conds, __METHOD__ , $sort );
}
static protected function categoryCondition( $category, &$tables, &$conds ) {
if ( strval( $category ) !== '' ) {
$category = Title::newFromText( $category )->getDbKey();
$tables[] = 'categorylinks';
$conds[] = 'page_id = cl_from';
$conds['cl_to'] = $category;
}
}
static protected function prefixCondition( $prefix, &$conds ) {
if ( strval( $prefix ) === '' ) {
return;
}
$dbr = wfGetDB( DB_SLAVE );
$title = Title::newFromText( $prefix );
if ( !is_null( $title ) ) {
$prefix = $title->getDbKey();
}
$any = $dbr->anyString();
$conds[] = 'page_title ' . $dbr->buildLike( $prefix, $any );
}
static private function regexCond( $dbr, $column, $regex ) {
if ( $dbr instanceof DatabasePostgres ) {
$op = '~';
} else {
$op = 'REGEXP';
}
return "$column $op " . $dbr->addQuotes( $regex );
}
}