mediawiki-extensions-TextExtracts/includes/Hooks.php

<?php

namespace TextExtracts;


use ApiMain;
use ApiResult;
use ConfigFactory;
use FauxRequest;

class Hooks {

	// Unit test hook
	public static function onUnitTestsList( &$files ) {
		$files = array_merge( $files, glob( __DIR__ . '/tests/ExtractFormatterTest.php' ) );
		return true;
	}

	/**
	 * ApiOpenSearchSuggest hook handler
	 * @param array $results
	 * @return bool
	 */
	public static function onApiOpenSearchSuggest( &$results ) {
		$config = ConfigFactory::getDefaultInstance()->makeConfig( 'textextracts' );
		if ( !$config->get( 'ExtractsExtendOpenSearchXml' ) || !count( $results ) ) {
			return true;
		}
		$pageIds = array_keys( $results );
		$api = new ApiMain( new FauxRequest(
			array(
				'action' => 'query',
				'prop' => 'extracts',
				'explaintext' => true,
				'exintro' => true,
				'exlimit' => count( $results ),
				'pageids' => implode( '|', $pageIds ),
			) )
		);
		$api->execute();
		if ( defined( 'ApiResult::META_CONTENT' ) ) {
			$data = $api->getResult()->getResultData( array( 'query', 'pages' ) );
			foreach ( $pageIds as $id ) {
				$contentKey = isset( $data[$id]['extract'][ApiResult::META_CONTENT] )
					? $data[$id]['extract'][ApiResult::META_CONTENT]
					: '*';
				if ( isset( $data[$id]['extract'][$contentKey] ) ) {
					$results[$id]['extract'] = $data[$id]['extract'][$contentKey];
					$results[$id]['extract trimmed'] = false;
				}
			}
		} else {
			$data = $api->getResultData();
			foreach ( $pageIds as $id ) {
				if ( isset( $data['query']['pages'][$id]['extract']['*'] ) ) {
					$results[$id]['extract'] = $data['query']['pages'][$id]['extract']['*'];
					$results[$id]['extract trimmed'] = false;
				}
			}
		}
		return true;
	}
}