[Summary view] [Print] [Text view]

   1  <?php
   2  /**
   3   * Fixes any entries for protocol-relative URLs in the externallinks table,
   4   * replacing each protocol-relative entry with two entries, one for http
   5   * and one for https.
   6   *
   7   * This program is free software; you can redistribute it and/or modify
   8   * it under the terms of the GNU General Public License as published by
   9   * the Free Software Foundation; either version 2 of the License, or
  10   * (at your option) any later version.
  11   *
  12   * This program is distributed in the hope that it will be useful,
  13   * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15   * GNU General Public License for more details.
  16   *
  17   * You should have received a copy of the GNU General Public License along
  18   * with this program; if not, write to the Free Software Foundation, Inc.,
  19   * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  20   * http://www.gnu.org/copyleft/gpl.html
  21   *
  22   * @file
  23   * @ingroup Maintenance
  24   */
  25  
  26  require_once  __DIR__ . '/Maintenance.php';
  27  
  28  /**
  29   * Maintenance script that fixes any entriy for protocol-relative URLs
  30   * in the externallinks table.
  31   *
  32   * @ingroup Maintenance
  33   */
  34  class FixExtLinksProtocolRelative extends LoggedUpdateMaintenance {
  35  	public function __construct() {
  36          parent::__construct();
  37          $this->mDescription =
  38              "Fixes any entries in the externallinks table containing protocol-relative URLs";
  39      }
  40  
  41  	protected function getUpdateKey() {
  42          return 'fix protocol-relative URLs in externallinks';
  43      }
  44  
  45  	protected function updateSkippedMessage() {
  46          return 'protocol-relative URLs in externallinks table already fixed.';
  47      }
  48  
  49  	protected function doDBUpdates() {
  50          $db = wfGetDB( DB_MASTER );
  51          if ( !$db->tableExists( 'externallinks' ) ) {
  52              $this->error( "externallinks table does not exist" );
  53  
  54              return false;
  55          }
  56          $this->output( "Fixing protocol-relative entries in the externallinks table...\n" );
  57          $res = $db->select( 'externallinks', array( 'el_from', 'el_to', 'el_index' ),
  58              array( 'el_index' . $db->buildLike( '//', $db->anyString() ) ),
  59              __METHOD__
  60          );
  61          $count = 0;
  62          foreach ( $res as $row ) {
  63              $count++;
  64              if ( $count % 100 == 0 ) {
  65                  $this->output( $count . "\n" );
  66                  wfWaitForSlaves();
  67              }
  68              $db->insert( 'externallinks',
  69                  array(
  70                      array(
  71                          'el_id' => $db->nextSequenceValue( 'externallinks_el_id_seq' ),
  72                          'el_from' => $row->el_from,
  73                          'el_to' => $row->el_to,
  74                          'el_index' => "http:{$row->el_index}",
  75                      ),
  76                      array(
  77                          'el_id' => $db->nextSequenceValue( 'externallinks_el_id_seq' ),
  78                          'el_from' => $row->el_from,
  79                          'el_to' => $row->el_to,
  80                          'el_index' => "https:{$row->el_index}",
  81                      )
  82                  ), __METHOD__, array( 'IGNORE' )
  83              );
  84              $db->delete(
  85                  'externallinks',
  86                  array(
  87                      'el_index' => $row->el_index,
  88                      'el_from' => $row->el_from,
  89                      'el_to' => $row->el_to
  90                  ),
  91                  __METHOD__
  92              );
  93          }
  94          $this->output( "Done, $count rows updated.\n" );
  95  
  96          return true;
  97      }
  98  }
  99  
 100  $maintClass = "FixExtLinksProtocolRelative";
 101  require_once RUN_MAINTENANCE_IF_MAIN;