maintenance/installExtension.php

   1 <?php
   2 /**
   3  * Copyright (C) 2006 Daniel Kinzler, brightbyte.de
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18  * http://www.gnu.org/copyleft/gpl.html
  19  *
  20  * @package MediaWiki
  21  * @subpackage Maintenance
  22  */
  23
  24 $optionsWithArgs = array( 'target' );
  25
  26 require_once( 'commandLine.inc' );
  27
  28 class ExtensionInstaller {
  29         var $source;
  30         var $target;
  31         var $name;
  32         var $dir;
  33
  34         function ExtensionInstaller( $name, $source, $target ) {
  35                 $this->name = $name;
  36                 $this->source = $source;
  37                 $this->target = realpath( $target );
  38                 $this->extdir = "$target/extensions";
  39                 $this->dir = "{$this->extdir}/$name";
  40                 $this->incpath = "extensions/$name";
  41
  42                 #TODO: allow a subdir different from "extensions"
  43                 #TODO: allow a config file different from "LocalSettings.php"
  44         }
  45
  46         function note( $msg ) {
  47                 print "$msg\n";
  48         }
  49
  50         function warn( $msg ) {
  51                 print "WARNING: $msg\n";
  52         }
  53
  54         function error( $msg ) {
  55                 print "ERROR: $msg\n";
  56         }
  57
  58         function prompt( $msg ) {
  59                 if ( function_exists( 'readline' ) ) {
  60                         $s = readline( $msg );
  61                 }
  62                 else {
  63                         if ( !@$this->stdin ) $this->stdin = fopen( 'php://stdin', 'r' );
  64                         if ( !$this->stdin ) die( "Failed to open stdin for user interaction!\n" );
  65
  66                         print $msg;
  67                         flush();
  68
  69                         $s = fgets( $this->stdin );
  70                 }
  71
  72                 $s = trim( $s );
  73                 return $s;
  74         }
  75
  76         function confirm( $msg ) {
  77                 while ( true ) {
  78                         $s = $this->prompt( $msg . " [yes/no]: ");
  79                         $s = strtolower( trim($s) );
  80
  81                         if ( $s == 'yes' || $s == 'y' ) return true;
  82                         else if ( $s == 'no' || $s == 'n' ) return false;
  83                         else print "bad response: $s\n";
  84                 }
  85         }
  86
  87         function deleteContents( $dir ) {
  88                 $ff = glob( $dir . "/*" );
  89                 if ( !$ff ) return;
  90
  91                 foreach ( $ff as $f ) {
  92                         if ( is_dir( $f ) ) $this->deleteContents( $f );
  93                         unlink( $f );
  94                 }
  95         }
  96
  97         function copyDir( $dir, $tgt ) {
  98                 $d = $tgt . '/' . basename( $dir );
  99
 100                 if ( !file_exists( $d ) ) {
 101                         $ok = mkdir( $d );
 102                         if ( !$ok ) {
 103                                 $this->error( "failed to create director $d" );
 104                                 return false;
 105                         }
 106                 }
 107
 108                 $ff = glob( $dir . "/*" );
 109                 if ( $ff === false || $ff === NULL ) return false;
 110
 111                 foreach ( $ff as $f ) {
 112                         if ( is_dir( $f ) ) {
 113                                 $ok = $this->copyDir( $f, $d );
 114                                 if ( !$ok ) return false;
 115                         }
 116                         else {
 117                                 $t = $d . '/' . basename( $f );
 118                                 $ok = copy( $f, $t );
 119
 120                                 if ( !$ok ) {
 121                                         $this->error( "failed to copy $f to $t" );
 122                                         return false;
 123                                 }
 124                         }
 125                 }
 126
 127                 return true;
 128         }
 129
 130         function fetchExtension( ) {
 131                 if ( file_exists( $this->dir ) && glob( $this->dir . "/*" )
 132                      && realpath( $this->source ) != $this->dir ) {
 133
 134                         if ( $this->confirm( "{$this->dir} exists and is not empty.\nDelete all files in that directory?" ) ) {
 135                                 $this->deleteContents( $this->dir );
 136                         }
 137                         else {
 138                                 return false;
 139                         }
 140                 }
 141
 142                 preg_match( '!([-\w]+://)?.*?(\.[-\w\d.]+)?$!', $this->source, $m );
 143                 $proto = @$m[1];
 144                 $ext = @$m[2];
 145                 if ( $ext ) $ext = strtolower( $ext );
 146
 147                 $src = $this->source;
 148
 149                 #TODO: check that the required program is available.
 150                 #may be used: tar, unzip, svn
 151
 152                 if ( $proto && $ext ) { #remote file
 153                         $tmp = wfTempDir() . '/' . basename( $src );
 154
 155                         $this->note( "fetching {$this->source}..." );
 156                         $ok = copy( $src, $tmp );
 157
 158                         if ( !$ok ) {
 159                                 $this->error( "failed to download {$src}" );
 160                                 return false;
 161                         }
 162
 163                         $src = $tmp;
 164                         $proto = NULL;
 165                 }
 166
 167                 if ( $proto ) { #assume SVN repository
 168                         $this->note( "SVN checkout of $src..." );
 169                         wfShellExec( 'svn co ' . escapeshellarg( $src ) . ' ' . escapeshellarg( $this->dir ), $code );
 170
 171                         if ( $code !== 0 ) {
 172                                 $this->error( "checkout failed for $src!" );
 173                                 return false;
 174                         }
 175                 }
 176                 else { #local file or directory
 177                         $src = realpath ( $src );
 178
 179                         if ( !file_exists( $src ) ) {
 180                                 $this->error( "file not found: {$this->source}" );
 181                                 return false;
 182                         }
 183
 184                         if ( $ext === NULL || $ext === '') { #local dir
 185                                 if ( $src == $this->dir ) {
 186                                         $this->note( "files are already in the extension dir" );
 187                                         return true;
 188                                 }
 189
 190                                 $this->copyDir( $src, $this->extdir );
 191                         }
 192                         else if ( $ext == '.tgz' || $ext == '.tar.gz' ) { #tgz file
 193                                 $this->note( "extracting $src..." );
 194                                 wfShellExec( 'tar zxvf ' . escapeshellarg( $src ) . ' -C ' . escapeshellarg( $this->extdir ), $code );
 195
 196                                 if ( $code !== 0 ) {
 197                                         $this->error( "failed to extract $src!" );
 198                                         return false;
 199                                 }
 200                         }
 201                         else if ( $ext == '.zip' ) { #zip file
 202                                 $this->note( "extracting $src..." );
 203                                 wfShellExec( 'unzip ' . escapeshellarg( $src ) . ' -d ' . escapeshellarg( $this->extdir ) , $code );
 204
 205                                 if ( $code !== 0 ) {
 206                                         $this->error( "failed to extract $src!" );
 207                                         return false;
 208                                 }
 209                         }
 210                         else {
 211                                 $this->error( "unknown file extension: $ext" );
 212                                 return false;
 213                         }
 214                 }
 215
 216                 if ( !file_exists( $this->dir ) && glob( $this->dir . "/*" ) ) {
 217                         $this->error( "{$this->dir} does not exist or is empty. Something went wrong, sorry." );
 218                         return false;
 219                 }
 220
 221                 #TODO: set permissions.... somehow. Copy from extension dir??
 222
 223                 $this->note( "fetched extension to {$this->dir}" );
 224                 return true;
 225         }
 226
 227         function patchLocalSettings( ) {
 228                 #NOTE: if we get a better way to hook up extensions, that should be used instead.
 229
 230                 $f = $this->dir . '/install.settings';
 231                 $t = $this->target . '/LocalSettings.php';
 232
 233                 #TODO: assert version ?!
 234                 #TODO: allow custom installer scripts + sql patches
 235
 236                 if ( !file_exists( $f ) ) {
 237                         $this->warn( "No install.settings file provided! Please read the instructions and edit LocalSettings.php manually." );
 238                         return '?';
 239                 }
 240
 241                 $settings = file_get_contents( $f );
 242
 243                 if ( !$settings ) {
 244                         $this->error( "failed to read settings from $f!" );
 245                         return false;
 246                 }
 247
 248                 $settings = str_replace( '{{path}}', $this->incpath, $settings );
 249
 250                 #NOTE: keep php extension for backup file!
 251                 $bak = $this->target . '/LocalSettings.install-' . $this->name . '-' . wfTimestamp(TS_MW) . '.bak.php';
 252
 253                 $ok = copy( $t, $bak );
 254
 255                 if ( !$ok ) {
 256                         $this->warn( "failed to create backup of LocalSettings.php!" );
 257                         return false;
 258                 }
 259                 else {
 260                         $this->note( "created backup of LocalSettings.php at $bak" );
 261                 }
 262
 263                 $localsettings = file_get_contents( $t );
 264
 265                 if ( !$settings ) {
 266                         $this->error( "failed to read $t for patching!" );
 267                         return false;
 268                 }
 269
 270                 $marker = "<@< extension {$this->name} >@>";
 271                 $blockpattern = "/\n\s*#\s*BEGIN\s*$marker.*END\s*$marker\s*/smi";
 272
 273                 if ( preg_match( $blockpattern, $localsettings ) ) {
 274                         $localsettings = preg_replace( $blockpattern, "\n", $localsettings );
 275                         $this->warn( "removed old configuration block for extension {$this->name}!" );
 276                 }
 277
 278                 $newblock= "\n# BEGIN $marker\n$settings\n# END $marker\n";
 279
 280                 $localsettings = preg_replace( "/\?>\s*$/si", "$newblock?>", $localsettings );
 281
 282                 $ok = file_put_contents( $t, $localsettings );
 283
 284                 if ( !$ok ) {
 285                         $this->error( "failed to patch $t!" );
 286                         return false;
 287                 }
 288                 else {
 289                         $this->note( "successfully patched LocalSettings.php" );
 290                 }
 291
 292                 return true;
 293         }
 294
 295         function printNotices( ) {
 296                 $files = array();
 297
 298                 if ( file_exists( $this->dir . '/README' ) ) $files[] = 'README';
 299                 if ( file_exists( $this->dir . '/INSTALL' ) ) $files[] = 'INSTALL';
 300
 301                 if ( !$files ) {
 302                         $this->note( "no information files found in {$this->dir}" );
 303                 }
 304                 else {
 305                         $this->note( "" );
 306
 307                         $this->note( "Please have a look at the following files in {$this->dir}," );
 308                         $this->note( "they may contain important information about {$this->name}." );
 309
 310                         $this->note( "" );
 311
 312                         foreach ( $files as $f ) {
 313                                 $this->note ( "\t* $f" );
 314                         }
 315
 316                         $this->note( "" );
 317                 }
 318
 319                 return true;
 320         }
 321
 322         /* static */ function listRepository( $repos ) {
 323                 preg_match( '!([-\w]+://)?.*?(\.[-\w\d.]+)?$!', $repos, $m );
 324                 $proto = @$m[1];
 325
 326                 #TODO: right now, this basically lists filenames, so it's not terribly useful.
 327                 #In future, there should be a "repository + logical name" scheme
 328
 329                 if ( $proto == 'http://' ) { #HTML directory listing
 330                         ExtensionInstaller::note( "listing index from $repos..." );
 331
 332                         $txt = file_get_contents( $repos );
 333
 334                         $ok = preg_match_all( '!<a\s[^>]*href\s*=\s*['."'".'"]([^/'."'".'"]+)['."'".'"][^>]*>.*?</a>!si', $txt, $m, PREG_SET_ORDER );
 335                         if ( !$ok ) {
 336                                 ExtensionInstaller::error( "listing index from $repos failed!" );
 337                                 print ( $txt );
 338                                 return false;
 339                         }
 340
 341                         foreach ( $m as $l ) {
 342                                 $n = $l[1];
 343
 344                                 if ( preg_match('!^[./?]!', $n) ) continue;
 345
 346                                 ExtensionInstaller::note( "\t$n" );
 347                         }
 348                 }
 349                 else if ( !$proto ) { #local directory
 350                         ExtensionInstaller::note( "listing directory $repos..." );
 351
 352                         $ff = glob( "$repos/*" );
 353                         if ( $ff === false || $ff === NULL ) {
 354                                 ExtensionInstaller::error( "listing directory $repos failed!" );
 355                                 return false;
 356                         }
 357
 358                         foreach ( $ff as $f ) {
 359                                 $n = basename($f);
 360
 361                                 ExtensionInstaller::note( "\t$n" );
 362                         }
 363                 }
 364                 else { #assume svn
 365                         ExtensionInstaller::note( "SVN list $repos..." );
 366                         $txt = wfShellExec( 'svn ls ' . escapeshellarg( $repos ), $code );
 367
 368                         if ( $code !== 0 ) {
 369                                 ExtensionInstaller::error( "svn list for $repos failed!" );
 370                                 return false;
 371                         }
 372
 373                         $ll = preg_split('/(\s*[\r\n]\s*)+/', $txt);
 374
 375                         foreach ( $ll as $line ) {
 376                                 if ( !preg_match('!^(.*)/$!', $line, $m) ) continue;
 377
 378                                 ExtensionInstaller::note( "\t{$m[1]}" );
 379                         }
 380                 }
 381         }
 382 }
 383
 384 if ( isset( $options['list'] ) ) {
 385         $repos = $options['list'];
 386         if ( $repos === true || $repos === 1 ) {
 387                 # Default to SVN trunk. Perhaps change that to use the version of the present install,
 388                 # and/or use bundles at an official download location.
 389                 $repos = 'http://svn.wikimedia.org/svnroot/mediawiki/trunk/extensions/';
 390         }
 391
 392         ExtensionInstaller::listRepository( $repos );
 393
 394         exit(0);
 395 }
 396
 397 if( !isset( $args[0] ) ) {
 398         die( "USAGE: installExtension.php [options] name [source]\n" .
 399                 "OPTIONS: \n" .
 400                 "    --target=<dir>    mediawiki installation directory\n" .
 401                 "    --nopatch         don't touch LocalSettings.php\n" .
 402                 "SOURCE: \n" .
 403                 "    May be a local file (tgz or zip) or directory.\n" .
 404                 "    May be the URL of a remote file (tgz or zip).\n" .
 405                 "    May be a SVN repository\n"
 406                 );
 407 }
 408
 409 $name = $args[0];
 410
 411 # Default to SVN trunk. Perhaps change that to use the version of the present install,
 412 # and/or use bundles at an official download location.
 413 $defsrc = "http://svn.wikimedia.org/svnroot/mediawiki/trunk/extensions/" . urlencode($name);
 414
 415 $src = isset ( $args[1] ) ? $args[1] : $defsrc;
 416
 417 $tgt = isset ( $options['target'] ) ? $options['target'] : $IP;
 418
 419 if ( !file_exists( "$tgt/LocalSettings.php" ) ) {
 420         die("can't find $tgt/LocalSettings.php\n");
 421 }
 422
 423 if ( !is_writable( "$tgt/LocalSettings.php" ) ) {
 424         die("can't write to  $tgt/LocalSettings.php\n");
 425 }
 426
 427 if ( !file_exists( "$tgt/extensions" ) ) {
 428         die("can't find $tgt/extensions\n");
 429 }
 430
 431 if ( !is_writable( "$tgt/extensions" ) ) {
 432         die("can't write to  $tgt/extensions\n");
 433 }
 434
 435 $installer = new ExtensionInstaller( $name, $src, $tgt );
 436
 437 $installer->note( "Installing extension {$installer->name} from {$installer->source} to {$installer->dir}" );
 438
 439 print "\n";
 440 print "\tTHIS TOOL IS EXPERIMENTAL!\n";
 441 print "\tEXPECT THE UNEXPECTED!\n";
 442 print "\n";
 443
 444 if ( !$installer->confirm("continue") ) die("aborted\n");
 445
 446 $ok = $installer->fetchExtension();
 447
 448 if ( $ok ) {
 449         if ( isset( $options['nopatch'] ) ) $installer->note( "skipping patch phase; Please edit LocalSettings.php manually to activate the extension." );
 450         else $ok = $installer->patchLocalSettings();
 451 }
 452
 453 $ok = $installer->printNotices();
 454
 455 if ( $ok ) $installer->note( "$name extension was installed successfully" );
 456 ?>