MainSlotRoleHandler.php

Go to the documentation of this file.

<?php
namespace MediaWiki\Revision;
 
use MediaWiki\Content\IContentHandlerFactory;
use MediaWiki\Content\UnknownContentModelException;
use MediaWiki\HookContainer\HookContainer;
use MediaWiki\HookContainer\HookRunner;
use MediaWiki\Linker\LinkTarget;
use MediaWiki\Page\PageIdentity;
use MediaWiki\Page\PageReference;
use MediaWiki\Title\TitleFactory;
 
class MainSlotRoleHandler extends SlotRoleHandler {
 
    private $namespaceContentModels;
 
    private $contentHandlerFactory;
 
    private $hookRunner;
 
    private $titleFactory;
 
    public function __construct(
        array $namespaceContentModels,
        IContentHandlerFactory $contentHandlerFactory,
        HookContainer $hookContainer,
        TitleFactory $titleFactory
    ) {
        parent::__construct( SlotRecord::MAIN, CONTENT_MODEL_WIKITEXT );
        $this->namespaceContentModels = $namespaceContentModels;
        $this->contentHandlerFactory = $contentHandlerFactory;
        $this->hookRunner = new HookRunner( $hookContainer );
        $this->titleFactory = $titleFactory;
    }
 
    public function supportsArticleCount() {
        return true;
    }
 
    public function isAllowedModel( $model, PageIdentity $page ) {
        $title = $this->titleFactory->newFromPageIdentity( $page );
        $handler = $this->contentHandlerFactory->getContentHandler( $model );
 
        return $handler->canBeUsedOn( $title );
    }
 
    public function getDefaultModel( $page ) {
        // NOTE: this method must not rely on $title->getContentModel() directly or indirectly,
        //       because it is used to initialize the mContentModel member.
 
        $ext = '';
        $ns = $page->getNamespace();
        $model = $this->namespaceContentModels[$ns] ?? null;
 
        // Hook can determine default model
        if ( $page instanceof PageReference ) {
            $title = $this->titleFactory->newFromPageReference( $page );
        } else {
            $title = $this->titleFactory->newFromLinkTarget( $page );
        }
        // @phan-suppress-next-line PhanTypeMismatchArgument Type mismatch on pass-by-ref args
        if ( !$this->hookRunner->onContentHandlerDefaultModelFor( $title, $model ) && $model !== null ) {
            return $model;
        }
 
        // Could this page contain code based on the title?
        $isCodePage = $ns === NS_MEDIAWIKI && preg_match( '!\.(css|js|json|vue)$!u', $title->getText(), $m );
        if ( $isCodePage ) {
            $ext = $m[1];
        }
 
        // Is this a user subpage containing code?
        $isCodeSubpage = $ns === NS_USER
            && !$isCodePage
            && preg_match( "/\\/.*\\.(js|css|json|vue)$/", $title->getText(), $m );
 
        if ( $isCodeSubpage ) {
            $ext = $m[1];
        }
 
        // Is this wikitext, according to $wgNamespaceContentModels or the DefaultModelFor hook?
        $isWikitext = $model === null || $model == CONTENT_MODEL_WIKITEXT;
        $isWikitext = $isWikitext && !$isCodePage && !$isCodeSubpage;
 
        if ( !$isWikitext ) {
            switch ( $ext ) {
                case 'js':
                    return CONTENT_MODEL_JAVASCRIPT;
                case 'css':
                    return CONTENT_MODEL_CSS;
                case 'json':
                    return CONTENT_MODEL_JSON;
                case 'vue':
                    return CONTENT_MODEL_VUE;
                default:
                    return $model ?? CONTENT_MODEL_TEXT;
            }
        }
 
        // We established that it must be wikitext
 
        return CONTENT_MODEL_WIKITEXT;
    }
 
}