wiki.techinc.nl/tests/phpunit/includes/specials/SpecialPageExecutor.php
Lucas Werkmeister (WMDE) 08ea47ce7a Revert "Revert "Default to qqx in SpecialPageExecutor""
This reverts commit 05e0582d86, 
reinstating commit a2f186a968.

Reason for revert: issues in other extensions should be resolved now.

Depends-On: Ie4736b25e5433ac0ffffd44dc9622d5f373e8424
Change-Id: I09844c22bfb74a28197da3df5ebb9406839f4a65
2020-03-18 18:25:28 +00:00

126 lines
3.2 KiB
PHP

<?php
/**
* @author Addshore
*
* @since 1.27
*/
class SpecialPageExecutor {
/**
* @param SpecialPage $page The special page to execute
* @param string $subPage The subpage parameter to call the page with
* @param WebRequest|null $request Web request that may contain URL parameters, etc
* @param Language|string|null $language The language which should be used in the context;
* if not specified, the pseudo-code 'qqx' is used
* @param User|null $user The user which should be used in the context of this special page
*
* @throws Exception
* @return array [ string, WebResponse ] A two-elements array containing the HTML output
* generated by the special page as well as the response object.
*/
public function executeSpecialPage(
SpecialPage $page,
$subPage = '',
WebRequest $request = null,
$language = null,
User $user = null
) {
$context = $this->newContext( $request, $language, $user );
$output = new OutputPage( $context );
$context->setOutput( $output );
$page->setContext( $context );
$output->setTitle( $page->getPageTitle() );
$html = $this->getHTMLFromSpecialPage( $page, $subPage );
$response = $context->getRequest()->response();
if ( $response instanceof FauxResponse ) {
$code = $response->getStatusCode();
if ( $code > 0 ) {
$response->header( 'Status: ' . $code . ' ' . HttpStatus::getMessage( $code ) );
}
}
return [ $html, $response ];
}
/**
* @param WebRequest|null $request
* @param Language|string|null $language Defaults to 'qqx'
* @param User|null $user
*
* @return DerivativeContext
*/
private function newContext(
WebRequest $request = null,
$language = null,
User $user = null
) {
$context = new DerivativeContext( RequestContext::getMain() );
$context->setRequest( $request ?: new FauxRequest() );
$context->setLanguage( $language ?: 'qqx' );
if ( $user !== null ) {
$context->setUser( $user );
}
$this->setEditTokenFromUser( $context );
// Make sure the skin context is correctly set https://phabricator.wikimedia.org/T200771
$context->getSkin()->setContext( $context );
return $context;
}
/**
* If we are trying to edit and no token is set, supply one.
*
* @param DerivativeContext $context
*/
private function setEditTokenFromUser( DerivativeContext $context ) {
$request = $context->getRequest();
// Edits via GET are a security issue and should not succeed. On the other hand, not all
// POST requests are edits, but should ignore unused parameters.
if ( !$request->getCheck( 'wpEditToken' ) && $request->wasPosted() ) {
$request->setVal( 'wpEditToken', $context->getUser()->getEditToken() );
}
}
/**
* @param SpecialPage $page
* @param string $subPage
*
* @throws Exception
* @return string HTML
*/
private function getHTMLFromSpecialPage( SpecialPage $page, $subPage ) {
ob_start();
try {
$page->execute( $subPage );
$output = $page->getOutput();
if ( $output->getRedirect() !== '' ) {
$output->output();
$html = ob_get_contents();
} elseif ( $output->isDisabled() ) {
$html = ob_get_contents();
} else {
$html = $output->getHTML();
}
} finally {
ob_end_clean();
}
return $html;
}
}