includes/registration/ExtensionRegistry.php

   1 <?php
   2
   3 use MediaWiki\MediaWikiServices;
   4
   5 /**
   6  * ExtensionRegistry class
   7  *
   8  * The Registry loads JSON files, and uses a Processor
   9  * to extract information from them. It also registers
  10  * classes with the autoloader.
  11  *
  12  * @since 1.25
  13  */
  14 class ExtensionRegistry {
  15
  16         /**
  17          * "requires" key that applies to MediaWiki core/$wgVersion
  18          */
  19         const MEDIAWIKI_CORE = 'MediaWiki';
  20
  21         /**
  22          * Version of the highest supported manifest version
  23          */
  24         const MANIFEST_VERSION = 2;
  25
  26         /**
  27          * Version of the oldest supported manifest version
  28          */
  29         const OLDEST_MANIFEST_VERSION = 1;
  30
  31         /**
  32          * Bump whenever the registration cache needs resetting
  33          */
  34         const CACHE_VERSION = 4;
  35
  36         /**
  37          * Special key that defines the merge strategy
  38          *
  39          * @since 1.26
  40          */
  41         const MERGE_STRATEGY = '_merge_strategy';
  42
  43         /**
  44          * @var BagOStuff
  45          */
  46         protected $cache;
  47
  48         /**
  49          * Array of loaded things, keyed by name, values are credits information
  50          *
  51          * @var array
  52          */
  53         private $loaded = [];
  54
  55         /**
  56          * List of paths that should be loaded
  57          *
  58          * @var array
  59          */
  60         protected $queued = [];
  61
  62         /**
  63          * Whether we are done loading things
  64          *
  65          * @var bool
  66          */
  67         private $finished = false;
  68
  69         /**
  70          * Items in the JSON file that aren't being
  71          * set as globals
  72          *
  73          * @var array
  74          */
  75         protected $attributes = [];
  76
  77         /**
  78          * @var ExtensionRegistry
  79          */
  80         private static $instance;
  81
  82         /**
  83          * @return ExtensionRegistry
  84          */
  85         public static function getInstance() {
  86                 if ( self::$instance === null ) {
  87                         self::$instance = new self();
  88                 }
  89
  90                 return self::$instance;
  91         }
  92
  93         public function __construct() {
  94                 // We use a try/catch because we don't want to fail here
  95                 // if $wgObjectCaches is not configured properly for APC setup
  96                 try {
  97                         $this->cache = MediaWikiServices::getInstance()->getLocalServerObjectCache();
  98                 } catch ( MWException $e ) {
  99                         $this->cache = new EmptyBagOStuff();
 100                 }
 101         }
 102
 103         /**
 104          * @param string $path Absolute path to the JSON file
 105          */
 106         public function queue( $path ) {
 107                 global $wgExtensionInfoMTime;
 108
 109                 $mtime = $wgExtensionInfoMTime;
 110                 if ( $mtime === false ) {
 111                         if ( file_exists( $path ) ) {
 112                                 $mtime = filemtime( $path );
 113                         } else {
 114                                 throw new Exception( "$path does not exist!" );
 115                         }
 116                         if ( !$mtime ) {
 117                                 $err = error_get_last();
 118                                 throw new Exception( "Couldn't stat $path: {$err['message']}" );
 119                         }
 120                 }
 121                 $this->queued[$path] = $mtime;
 122         }
 123
 124         /**
 125          * @throws MWException If the queue is already marked as finished (no further things should
 126          *  be loaded then).
 127          */
 128         public function loadFromQueue() {
 129                 global $wgVersion;
 130                 if ( !$this->queued ) {
 131                         return;
 132                 }
 133
 134                 if ( $this->finished ) {
 135                         throw new MWException(
 136                                 "The following paths tried to load late: "
 137                                 . implode( ', ', array_keys( $this->queued ) )
 138                         );
 139                 }
 140
 141                 // A few more things to vary the cache on
 142                 $versions = [
 143                         'registration' => self::CACHE_VERSION,
 144                         'mediawiki' => $wgVersion
 145                 ];
 146
 147                 // See if this queue is in APC
 148                 $key = wfMemcKey(
 149                         'registration',
 150                         md5( json_encode( $this->queued + $versions ) )
 151                 );
 152                 $data = $this->cache->get( $key );
 153                 if ( $data ) {
 154                         $this->exportExtractedData( $data );
 155                 } else {
 156                         $data = $this->readFromQueue( $this->queued );
 157                         $this->exportExtractedData( $data );
 158                         // Do this late since we don't want to extract it since we already
 159                         // did that, but it should be cached
 160                         $data['globals']['wgAutoloadClasses'] += $data['autoload'];
 161                         unset( $data['autoload'] );
 162                         $this->cache->set( $key, $data, 60 * 60 * 24 );
 163                 }
 164                 $this->queued = [];
 165         }
 166
 167         /**
 168          * Get the current load queue. Not intended to be used
 169          * outside of the installer.
 170          *
 171          * @return array
 172          */
 173         public function getQueue() {
 174                 return $this->queued;
 175         }
 176
 177         /**
 178          * Clear the current load queue. Not intended to be used
 179          * outside of the installer.
 180          */
 181         public function clearQueue() {
 182                 $this->queued = [];
 183         }
 184
 185         /**
 186          * After this is called, no more extensions can be loaded
 187          *
 188          * @since 1.29
 189          */
 190         public function finish() {
 191                 $this->finished = true;
 192         }
 193
 194         /**
 195          * Process a queue of extensions and return their extracted data
 196          *
 197          * @param array $queue keys are filenames, values are ignored
 198          * @return array extracted info
 199          * @throws Exception
 200          */
 201         public function readFromQueue( array $queue ) {
 202                 global $wgVersion;
 203                 $autoloadClasses = [];
 204                 $autoloaderPaths = [];
 205                 $processor = new ExtensionProcessor();
 206                 $incompatible = [];
 207                 $versionParser = new VersionChecker();
 208                 $versionParser->setCoreVersion( $wgVersion );
 209                 foreach ( $queue as $path => $mtime ) {
 210                         $json = file_get_contents( $path );
 211                         if ( $json === false ) {
 212                                 throw new Exception( "Unable to read $path, does it exist?" );
 213                         }
 214                         $info = json_decode( $json, /* $assoc = */ true );
 215                         if ( !is_array( $info ) ) {
 216                                 throw new Exception( "$path is not a valid JSON file." );
 217                         }
 218
 219                         // Check any constraints against MediaWiki core
 220                         $requires = $processor->getRequirements( $info );
 221                         if ( $requires ) {
 222                                 $versionCheck = $versionParser->checkArray(
 223                                         [ $info['name'] => $requires ]
 224                                 );
 225                                 $incompatible = array_merge( $incompatible, $versionCheck );
 226                                 if ( $versionCheck ) {
 227                                         continue;
 228                                 }
 229                         }
 230
 231                         if ( !isset( $info['manifest_version'] ) ) {
 232                                 // For backwards-compatability, assume a version of 1
 233                                 $info['manifest_version'] = 1;
 234                         }
 235                         $version = $info['manifest_version'];
 236                         if ( $version < self::OLDEST_MANIFEST_VERSION || $version > self::MANIFEST_VERSION ) {
 237                                 throw new Exception( "$path: unsupported manifest_version: {$version}" );
 238                         }
 239
 240                         $autoload = $this->processAutoLoader( dirname( $path ), $info );
 241                         // Set up the autoloader now so custom processors will work
 242                         $GLOBALS['wgAutoloadClasses'] += $autoload;
 243                         $autoloadClasses += $autoload;
 244
 245                         // Get extra paths for later inclusion
 246                         $autoloaderPaths = array_merge( $autoloaderPaths,
 247                                 $processor->getExtraAutoloaderPaths( dirname( $path ), $info ) );
 248                         // Compatible, read and extract info
 249                         $processor->extractInfo( $path, $info, $version );
 250                 }
 251                 if ( $incompatible ) {
 252                         if ( count( $incompatible ) === 1 ) {
 253                                 throw new Exception( $incompatible[0] );
 254                         } else {
 255                                 throw new Exception( implode( "\n", $incompatible ) );
 256                         }
 257                 }
 258                 $data = $processor->getExtractedInfo();
 259                 // Need to set this so we can += to it later
 260                 $data['globals']['wgAutoloadClasses'] = [];
 261                 $data['autoload'] = $autoloadClasses;
 262                 $data['autoloaderPaths'] = $autoloaderPaths;
 263                 return $data;
 264         }
 265
 266         protected function exportExtractedData( array $info ) {
 267                 foreach ( $info['globals'] as $key => $val ) {
 268                         // If a merge strategy is set, read it and remove it from the value
 269                         // so it doesn't accidentally end up getting set.
 270                         if ( is_array( $val ) && isset( $val[self::MERGE_STRATEGY] ) ) {
 271                                 $mergeStrategy = $val[self::MERGE_STRATEGY];
 272                                 unset( $val[self::MERGE_STRATEGY] );
 273                         } else {
 274                                 $mergeStrategy = 'array_merge';
 275                         }
 276
 277                         // Optimistic: If the global is not set, or is an empty array, replace it entirely.
 278                         // Will be O(1) performance.
 279                         if ( !isset( $GLOBALS[$key] ) || ( is_array( $GLOBALS[$key] ) && !$GLOBALS[$key] ) ) {
 280                                 $GLOBALS[$key] = $val;
 281                                 continue;
 282                         }
 283
 284                         if ( !is_array( $GLOBALS[$key] ) || !is_array( $val ) ) {
 285                                 // config setting that has already been overridden, don't set it
 286                                 continue;
 287                         }
 288
 289                         switch ( $mergeStrategy ) {
 290                                 case 'array_merge_recursive':
 291                                         $GLOBALS[$key] = array_merge_recursive( $GLOBALS[$key], $val );
 292                                         break;
 293                                 case 'array_replace_recursive':
 294                                         $GLOBALS[$key] = array_replace_recursive( $GLOBALS[$key], $val );
 295                                         break;
 296                                 case 'array_plus_2d':
 297                                         $GLOBALS[$key] = wfArrayPlus2d( $GLOBALS[$key], $val );
 298                                         break;
 299                                 case 'array_plus':
 300                                         $GLOBALS[$key] += $val;
 301                                         break;
 302                                 case 'array_merge':
 303                                         $GLOBALS[$key] = array_merge( $val, $GLOBALS[$key] );
 304                                         break;
 305                                 default:
 306                                         throw new UnexpectedValueException( "Unknown merge strategy '$mergeStrategy'" );
 307                         }
 308                 }
 309
 310                 foreach ( $info['defines'] as $name => $val ) {
 311                         define( $name, $val );
 312                 }
 313                 foreach ( $info['autoloaderPaths'] as $path ) {
 314                         require_once $path;
 315                 }
 316
 317                 $this->loaded += $info['credits'];
 318                 if ( $info['attributes'] ) {
 319                         if ( !$this->attributes ) {
 320                                 $this->attributes = $info['attributes'];
 321                         } else {
 322                                 $this->attributes = array_merge_recursive( $this->attributes, $info['attributes'] );
 323                         }
 324                 }
 325
 326                 foreach ( $info['callbacks'] as $name => $cb ) {
 327                         call_user_func( $cb, $info['credits'][$name] );
 328                 }
 329         }
 330
 331         /**
 332          * Loads and processes the given JSON file without delay
 333          *
 334          * If some extensions are already queued, this will load
 335          * those as well.
 336          *
 337          * @param string $path Absolute path to the JSON file
 338          */
 339         public function load( $path ) {
 340                 $this->loadFromQueue(); // First clear the queue
 341                 $this->queue( $path );
 342                 $this->loadFromQueue();
 343         }
 344
 345         /**
 346          * Whether a thing has been loaded
 347          * @param string $name
 348          * @return bool
 349          */
 350         public function isLoaded( $name ) {
 351                 return isset( $this->loaded[$name] );
 352         }
 353
 354         /**
 355          * @param string $name
 356          * @return array
 357          */
 358         public function getAttribute( $name ) {
 359                 if ( isset( $this->attributes[$name] ) ) {
 360                         return $this->attributes[$name];
 361                 } else {
 362                         return [];
 363                 }
 364         }
 365
 366         /**
 367          * Get information about all things
 368          *
 369          * @return array
 370          */
 371         public function getAllThings() {
 372                 return $this->loaded;
 373         }
 374
 375         /**
 376          * Mark a thing as loaded
 377          *
 378          * @param string $name
 379          * @param array $credits
 380          */
 381         protected function markLoaded( $name, array $credits ) {
 382                 $this->loaded[$name] = $credits;
 383         }
 384
 385         /**
 386          * Register classes with the autoloader
 387          *
 388          * @param string $dir
 389          * @param array $info
 390          * @return array
 391          */
 392         protected function processAutoLoader( $dir, array $info ) {
 393                 if ( isset( $info['AutoloadClasses'] ) ) {
 394                         // Make paths absolute, relative to the JSON file
 395                         return array_map( function( $file ) use ( $dir ) {
 396                                 return "$dir/$file";
 397                         }, $info['AutoloadClasses'] );
 398                 } else {
 399                         return [];
 400                 }
 401         }
 402 }