PortableInfobox/querypage/AllinfoboxesQueryPage.php

<?php

class AllinfoboxesQueryPage extends PageQueryPage {

	const LIMIT = 1000;
	const ALL_INFOBOXES_TYPE = 'AllInfoboxes';
	private static $subpagesBlacklist = [ 'doc', 'draft', 'test' ];

	function __construct() {
		parent::__construct( self::ALL_INFOBOXES_TYPE );
	}

	public function isListed() {
		return false;
	}

	public function sortDescending() {
		return true;
	}

	public function isExpensive() {
		return true;
	}

	/**
	 * A wrapper for calling the querycache table
	 *
	 * @param bool $offset
	 * @param int $limit
	 *
	 * @return ResultWrapper
	 */
	public function doQuery( $offset = false, $limit = self::LIMIT ) {
		return $this->fetchFromCache( $limit, $offset );
	}

	/**
	 * Update the querycache table
	 *
	 * @param bool $limit Only for consistency
	 * @param bool $ignoreErrors Only for consistency
	 *
	 * @return int number of rows updated
	 */
	public function recache( $limit = false, $ignoreErrors = true ) {
		$dbw = wfGetDB( DB_MASTER );

		$infoboxes = $this->reallyDoQuery();

		( new WikiaSQL() )
			->DELETE( 'querycache' )
			->WHERE( 'qc_type' )->EQUAL_TO( $this->getName() )
			->run( $dbw );

		if ( !empty( $infoboxes ) ) {
			( new WikiaSQL() )
				->INSERT()->INTO( 'querycache', [
					'qc_type',
					'qc_value',
					'qc_namespace',
					'qc_title'
				] )
				->VALUES( $infoboxes )
				->run( $dbw );
		}

		wfRunHooks( 'AllInfoboxesQueryRecached' );

		return count( $infoboxes );
	}

	/**
	 * Queries all templates and get only those with portable infoboxes
	 *
	 * @param bool $limit Only for consistency
	 * @param bool $offset Only for consistency
	 *
	 * @return bool|mixed
	 */
	public function reallyDoQuery( $limit = false, $offset = false ) {
		$dbr = wfGetDB( DB_SLAVE, [ $this->getName(), __METHOD__, 'vslow' ] );
		$result = ( new WikiaSQL() )
			->SELECT( 'page_id', 'page_title', 'page_namespace' )
			->FROM( 'page' )
			->WHERE( 'page_namespace' )->EQUAL_TO( NS_TEMPLATE )
			->AND_( 'page_is_redirect' )->EQUAL_TO( 0 )
			->run( $dbr, function ( ResultWrapper $result ) {
				$out = [ ];
				while ( $row = $result->fetchRow() ) {
					$out[] = [ 'type' => $this->getName(),
							   'pageid' => $row[ 'page_id' ],
							   'ns' => $row[ 'page_namespace' ],
							   'title' => $row[ 'page_title' ] ];
				}

				return $out;
			} );

		return array_filter( $result, [ $this, 'filterInfoboxes' ] );
	}

	public function addTitleToCache( Title $title ) {
		if ( !$this->hasInfobox( $title ) ) {
			return;
		}

		$dbw = wfGetDB( DB_MASTER );
		( new WikiaSQL() )
			->INSERT()->INTO( 'querycache', [
				'qc_type',
				'qc_value',
				'qc_namespace',
				'qc_title'
			] )
			->VALUES( [ [
				$this->getName(),
				$title->getArticleID(),
				$title->getNamespace(),
				$title->getDBkey(),
			] ] )
			->run( $dbw );

		wfRunHooks( 'AllInfoboxesQueryRecached' );
	}

	private function hasInfobox( Title $title ) {
		// omit subages from blacklist
		return !(
				$title->isSubpage() &&
				in_array( mb_strtolower( $title->getSubpageText() ), self::$subpagesBlacklist )
			) &&
			!empty( PortableInfoboxDataService::newFromTitle( $title )->getData() );
	}

	private function filterInfoboxes( $tmpl ) {
		$title = Title::newFromID( $tmpl[ 'pageid' ] );

		return $title &&
			$title->exists() &&
			$this->hasInfobox( $title );
	}
}