ResourceLoaderFileModule.php

Go to the documentation of this file.

<?php
class ResourceLoaderFileModule extends ResourceLoaderModule {

    /* Protected Members */

    protected $localBasePath = '';
    protected $remoteBasePath = '';
    protected $scripts = array();
    protected $languageScripts = array();
    protected $skinScripts = array();
    protected $debugScripts = array();
    protected $loaderScripts = array();
    protected $styles = array();
    protected $skinStyles = array();
    protected $dependencies = array();
    protected $messages = array();
    protected $group;
    protected $position = 'bottom';
    protected $debugRaw = true;
    protected $raw = false;
    protected $targets = array( 'desktop' );

    protected $hasGeneratedStyles = false;

    protected $modifiedTime = array();
    protected $localFileRefs = array();

    /* Methods */

    public function __construct( $options = array(), $localBasePath = null,
        $remoteBasePath = null
    ) {
        global $IP, $wgScriptPath, $wgResourceBasePath;
        $this->localBasePath = $localBasePath === null ? $IP : $localBasePath;
        if ( $remoteBasePath !== null ) {
            $this->remoteBasePath = $remoteBasePath;
        } else {
            $this->remoteBasePath = $wgResourceBasePath === null ? $wgScriptPath : $wgResourceBasePath;
        }

        if ( isset( $options['remoteExtPath'] ) ) {
            global $wgExtensionAssetsPath;
            $this->remoteBasePath = $wgExtensionAssetsPath . '/' . $options['remoteExtPath'];
        }

        foreach ( $options as $member => $option ) {
            switch ( $member ) {
                // Lists of file paths
                case 'scripts':
                case 'debugScripts':
                case 'loaderScripts':
                case 'styles':
                    $this->{$member} = (array)$option;
                    break;
                // Collated lists of file paths
                case 'languageScripts':
                case 'skinScripts':
                case 'skinStyles':
                    if ( !is_array( $option ) ) {
                        throw new MWException(
                            "Invalid collated file path list error. " .
                            "'$option' given, array expected."
                        );
                    }
                    foreach ( $option as $key => $value ) {
                        if ( !is_string( $key ) ) {
                            throw new MWException(
                                "Invalid collated file path list key error. " .
                                "'$key' given, string expected."
                            );
                        }
                        $this->{$member}[$key] = (array)$value;
                    }
                    break;
                // Lists of strings
                case 'dependencies':
                case 'messages':
                case 'targets':
                    $this->{$member} = (array)$option;
                    break;
                // Single strings
                case 'group':
                case 'position':
                case 'localBasePath':
                case 'remoteBasePath':
                    $this->{$member} = (string)$option;
                    break;
                // Single booleans
                case 'debugRaw':
                case 'raw':
                    $this->{$member} = (bool)$option;
                    break;
            }
        }
        // Make sure the remote base path is a complete valid URL,
        // but possibly protocol-relative to avoid cache pollution
        $this->remoteBasePath = wfExpandUrl( $this->remoteBasePath, PROTO_RELATIVE );
    }

    public function getScript( ResourceLoaderContext $context ) {
        $files = $this->getScriptFiles( $context );
        return $this->readScriptFiles( $files );
    }

    public function getScriptURLsForDebug( ResourceLoaderContext $context ) {
        $urls = array();
        foreach ( $this->getScriptFiles( $context ) as $file ) {
            $urls[] = $this->getRemotePath( $file );
        }
        return $urls;
    }

    public function supportsURLLoading() {
        return $this->debugRaw;
    }

    public function getLoaderScript() {
        if ( count( $this->loaderScripts ) == 0 ) {
            return false;
        }
        return $this->readScriptFiles( $this->loaderScripts );
    }

    public function getStyles( ResourceLoaderContext $context ) {
        $styles = $this->readStyleFiles(
            $this->getStyleFiles( $context ),
            $this->getFlip( $context )
        );
        // Collect referenced files
        $this->localFileRefs = array_unique( $this->localFileRefs );
        // If the list has been modified since last time we cached it, update the cache
        try {
            if ( $this->localFileRefs !== $this->getFileDependencies( $context->getSkin() ) ) {
                $dbw = wfGetDB( DB_MASTER );
                $dbw->replace( 'module_deps',
                    array( array( 'md_module', 'md_skin' ) ), array(
                        'md_module' => $this->getName(),
                        'md_skin' => $context->getSkin(),
                        'md_deps' => FormatJson::encode( $this->localFileRefs ),
                    )
                );
            }
        } catch ( Exception $e ) {
            wfDebugLog( 'resourceloader', __METHOD__ . ": failed to update DB: $e" );
        }
        return $styles;
    }

    public function getStyleURLsForDebug( ResourceLoaderContext $context ) {
        if ( $this->hasGeneratedStyles ) {
            // Do the default behaviour of returning a url back to load.php
            // but with only=styles.
            return parent::getStyleURLsForDebug( $context );
        }
        // Our module consists entirely of real css files,
        // in debug mode we can load those directly.
        $urls = array();
        foreach ( $this->getStyleFiles( $context ) as $mediaType => $list ) {
            $urls[$mediaType] = array();
            foreach ( $list as $file ) {
                $urls[$mediaType][] = $this->getRemotePath( $file );
            }
        }
        return $urls;
    }

    public function getMessages() {
        return $this->messages;
    }

    public function getGroup() {
        return $this->group;
    }

    public function getPosition() {
        return $this->position;
    }

    public function getDependencies() {
        return $this->dependencies;
    }

    public function isRaw() {
        return $this->raw;
    }

    public function getModifiedTime( ResourceLoaderContext $context ) {
        if ( isset( $this->modifiedTime[$context->getHash()] ) ) {
            return $this->modifiedTime[$context->getHash()];
        }
        wfProfileIn( __METHOD__ );

        $files = array();

        // Flatten style files into $files
        $styles = self::collateFilePathListByOption( $this->styles, 'media', 'all' );
        foreach ( $styles as $styleFiles ) {
            $files = array_merge( $files, $styleFiles );
        }
        $skinFiles = self::tryForKey(
            self::collateFilePathListByOption( $this->skinStyles, 'media', 'all' ),
            $context->getSkin(),
            'default'
        );
        foreach ( $skinFiles as $styleFiles ) {
            $files = array_merge( $files, $styleFiles );
        }

        // Final merge, this should result in a master list of dependent files
        $files = array_merge(
            $files,
            $this->scripts,
            $context->getDebug() ? $this->debugScripts : array(),
            self::tryForKey( $this->languageScripts, $context->getLanguage() ),
            self::tryForKey( $this->skinScripts, $context->getSkin(), 'default' ),
            $this->loaderScripts
        );
        $files = array_map( array( $this, 'getLocalPath' ), $files );
        // File deps need to be treated separately because they're already prefixed
        $files = array_merge( $files, $this->getFileDependencies( $context->getSkin() ) );

        // If a module is nothing but a list of dependencies, we need to avoid
        // giving max() an empty array
        if ( count( $files ) === 0 ) {
            wfProfileOut( __METHOD__ );
            return $this->modifiedTime[$context->getHash()] = 1;
        }

        wfProfileIn( __METHOD__ . '-filemtime' );
        $filesMtime = max( array_map( array( __CLASS__, 'safeFilemtime' ), $files ) );
        wfProfileOut( __METHOD__ . '-filemtime' );
        $this->modifiedTime[$context->getHash()] = max(
            $filesMtime,
            $this->getMsgBlobMtime( $context->getLanguage() ) );

        wfProfileOut( __METHOD__ );
        return $this->modifiedTime[$context->getHash()];
    }

    /* Protected Methods */

    protected function getLocalPath( $path ) {
        return "{$this->localBasePath}/$path";
    }

    protected function getRemotePath( $path ) {
        return "{$this->remoteBasePath}/$path";
    }

    public function getStyleSheetLang( $path ) {
        return preg_match( '/\.less$/i', $path ) ? 'less' : 'css';
    }

    protected static function collateFilePathListByOption( array $list, $option, $default ) {
        $collatedFiles = array();
        foreach ( (array)$list as $key => $value ) {
            if ( is_int( $key ) ) {
                // File name as the value
                if ( !isset( $collatedFiles[$default] ) ) {
                    $collatedFiles[$default] = array();
                }
                $collatedFiles[$default][] = $value;
            } elseif ( is_array( $value ) ) {
                // File name as the key, options array as the value
                $optionValue = isset( $value[$option] ) ? $value[$option] : $default;
                if ( !isset( $collatedFiles[$optionValue] ) ) {
                    $collatedFiles[$optionValue] = array();
                }
                $collatedFiles[$optionValue][] = $key;
            }
        }
        return $collatedFiles;
    }

    protected static function tryForKey( array $list, $key, $fallback = null ) {
        if ( isset( $list[$key] ) && is_array( $list[$key] ) ) {
            return $list[$key];
        } elseif ( is_string( $fallback )
            && isset( $list[$fallback] )
            && is_array( $list[$fallback] ) )
        {
            return $list[$fallback];
        }
        return array();
    }

    protected function getScriptFiles( ResourceLoaderContext $context ) {
        $files = array_merge(
            $this->scripts,
            self::tryForKey( $this->languageScripts, $context->getLanguage() ),
            self::tryForKey( $this->skinScripts, $context->getSkin(), 'default' )
        );
        if ( $context->getDebug() ) {
            $files = array_merge( $files, $this->debugScripts );
        }

        return array_unique( $files );
    }

    protected function getStyleFiles( ResourceLoaderContext $context ) {
        return array_merge_recursive(
            self::collateFilePathListByOption( $this->styles, 'media', 'all' ),
            self::collateFilePathListByOption(
                self::tryForKey( $this->skinStyles, $context->getSkin(), 'default' ), 'media', 'all'
            )
        );
    }

    public function getAllStyleFiles() {
        $files = array();
        foreach( (array)$this->styles as $key => $value ) {
            if ( is_array( $value ) ) {
                $path = $key;
            } else {
                $path = $value;
            }
            $files[] = $this->getLocalPath( $path );
        }
        return $files;
    }

    protected function readScriptFiles( array $scripts ) {
        global $wgResourceLoaderValidateStaticJS;
        if ( empty( $scripts ) ) {
            return '';
        }
        $js = '';
        foreach ( array_unique( $scripts ) as $fileName ) {
            $localPath = $this->getLocalPath( $fileName );
            if ( !file_exists( $localPath ) ) {
                throw new MWException( __METHOD__ . ": script file not found: \"$localPath\"" );
            }
            $contents = file_get_contents( $localPath );
            if ( $wgResourceLoaderValidateStaticJS ) {
                // Static files don't really need to be checked as often; unlike
                // on-wiki module they shouldn't change unexpectedly without
                // admin interference.
                $contents = $this->validateScriptFile( $fileName, $contents );
            }
            $js .= $contents . "\n";
        }
        return $js;
    }

    protected function readStyleFiles( array $styles, $flip ) {
        if ( empty( $styles ) ) {
            return array();
        }
        foreach ( $styles as $media => $files ) {
            $uniqueFiles = array_unique( $files );
            $styles[$media] = implode(
                "\n",
                array_map(
                    array( $this, 'readStyleFile' ),
                    $uniqueFiles,
                    array_fill( 0, count( $uniqueFiles ), $flip )
                )
            );
        }
        return $styles;
    }

    protected function readStyleFile( $path, $flip ) {
        $localPath = $this->getLocalPath( $path );
        if ( !file_exists( $localPath ) ) {
            $msg = __METHOD__ . ": style file not found: \"$localPath\"";
            wfDebugLog( 'resourceloader', $msg );
            throw new MWException( $msg );
        }

        if ( $this->getStyleSheetLang( $path ) === 'less' ) {
            $style = $this->compileLESSFile( $localPath );
            $this->hasGeneratedStyles = true;
        } else {
            $style = file_get_contents( $localPath );
        }

        if ( $flip ) {
            $style = CSSJanus::transform( $style, true, false );
        }
        $dirname = dirname( $path );
        if ( $dirname == '.' ) {
            // If $path doesn't have a directory component, don't prepend a dot
            $dirname = '';
        }
        $dir = $this->getLocalPath( $dirname );
        $remoteDir = $this->getRemotePath( $dirname );
        // Get and register local file references
        $this->localFileRefs = array_merge(
            $this->localFileRefs,
            CSSMin::getLocalFileReferences( $style, $dir )
        );
        return CSSMin::remap(
            $style, $dir, $remoteDir, true
        );
    }

    public function getFlip( $context ) {
        return $context->getDirection() === 'rtl';
    }

    public function getTargets() {
        return $this->targets;
    }

    protected static function getLESSCacheKey( $fileName ) {
        $vars = json_encode( ResourceLoader::getLESSVars() );
        $hash = md5( $fileName . $vars );
        return wfMemcKey( 'resourceloader', 'less', $hash );
    }

    protected function compileLESSFile( $fileName ) {
        $key = self::getLESSCacheKey( $fileName );
        $cache = wfGetCache( CACHE_ANYTHING );

        // The input to lessc. Either an associative array representing the
        // cached results of a previous compilation, or the string file name if
        // no cache result exists.
        $source = $cache->get( $key );
        if ( !is_array( $source ) || !isset( $source['root'] ) ) {
            $source = $fileName;
        }

        $compiler = ResourceLoader::getLessCompiler();
        $result = null;

        $result = $compiler->cachedCompile( $source );

        if ( !is_array( $result ) ) {
            throw new MWException( 'LESS compiler result has type ' . gettype( $result ) . '; array expected.' );
        }

        $this->localFileRefs += array_keys( $result['files'] );
        $cache->set( $key, $result );
        return $result['compiled'];
    }
}