getVal( 'mode' ) ) ? CategoryTree::decodeMode( $mode ) : $wgCategoryTreeCategoryPageMode; } if ( $wgCategoryTreeForceHeaders ) { $wgHooks['BeforePageDisplay'][] = 'CategoryTreeHooks::addHeaders'; } else { $wgHooks['OutputPageParserOutput'][] = 'CategoryTreeHooks::parserOutput'; } $wgHooks['MakeGlobalVariablesScript'][] = 'CategoryTreeHooks::getConfigVars'; } /** * @param $parser Parser * @return bool */ public static function setHooks( $parser ) { $parser->setHook( 'categorytree', 'CategoryTreeHooks::parserHook' ); $parser->setFunctionHook( 'categorytree', 'CategoryTreeHooks::parserFunction' ); return true; } /** * Entry point for the {{#categorytree}} tag parser function. * This is a wrapper around CategoryTreeHooks::parserHook * @param $parser Parser * @return array|string */ public static function parserFunction( $parser ) { $params = func_get_args(); array_shift( $params ); // first is $parser, strip it // first user-supplied parameter must be category name if ( !$params ) { return ''; // no category specified, return nothing } $cat = array_shift( $params ); // build associative arguments from flat parameter list $argv = []; foreach ( $params as $p ) { if ( preg_match( '/^\s*(\S.*?)\s*=\s*(.*?)\s*$/', $p, $m ) ) { $k = $m[1]; $v = preg_replace( '/^"\s*(.*?)\s*"$/', '$1', $m[2] ); // strip any quotes enclusing the value } else { $k = trim( $p ); $v = true; } $argv[$k] = $v; } // now handle just like a tag $html = self::parserHook( $cat, $argv, $parser ); return [ $html, 'noparse' => true, 'isHTML' => true ]; } /** * Hook implementation for injecting a category tree into the sidebar. * Registered automatically if $wgCategoryTreeSidebarRoot is set to a category name. * @param $skin * @param $tpl SkinTemplate * @return bool */ public static function skinTemplateOutputPageBeforeExec( $skin, $tpl ) { global $wgCategoryTreeSidebarRoot, $wgCategoryTreeSidebarOptions; $html = self::parserHook( $wgCategoryTreeSidebarRoot, $wgCategoryTreeSidebarOptions ); if ( $html ) { $tpl->data['sidebar']['categorytree-portlet'] = $html; } return true; } /** * Entry point for the tag parser hook. * This loads CategoryTreeFunctions.php and calls CategoryTree::getTag() * @param $cat * @param $argv * @param $parser Parser * @param $allowMissing bool * @return bool|string */ public static function parserHook( $cat, $argv, $parser = null, $allowMissing = false ) { global $wgOut; if ( $parser ) { $parser->mOutput->mCategoryTreeTag = true; # flag for use by CategoryTreeHooks::parserOutput } else { CategoryTree::setHeaders( $wgOut ); } $ct = new CategoryTree( $argv ); $attr = Sanitizer::validateTagAttributes( $argv, 'div' ); $hideroot = isset( $argv['hideroot'] ) ? CategoryTree::decodeBoolean( $argv['hideroot'] ) : null; $onlyroot = isset( $argv['onlyroot'] ) ? CategoryTree::decodeBoolean( $argv['onlyroot'] ) : null; $depthArg = isset( $argv['depth'] ) ? (int)$argv['depth'] : null; $depth = CategoryTree::capDepth( $ct->getOption( 'mode' ), $depthArg ); if ( $onlyroot ) { $depth = 0; } return $ct->getTag( $parser, $cat, $hideroot, $attr, $depth, $allowMissing ); } /** * Hook callback that injects messages and things into the tag, * if needed in the current page. * Does nothing if $parserOutput->mCategoryTreeTag is not set * @param $outputPage OutputPage * @param $parserOutput ParserOutput * @return bool */ public static function parserOutput( $outputPage, $parserOutput ) { if ( !empty( $parserOutput->mCategoryTreeTag ) ) { CategoryTree::setHeaders( $outputPage ); } return true; } /** * BeforePageDisplay hook. This hook is set when $wgCategoryTreeForceHeaders * is set. * Otherwise similar to CategoryTreeHooks::parserOutput. * @param $out OutputPage * @param $skin Skin * @return bool */ public static function addHeaders( OutputPage $out, Skin $skin ) { CategoryTree::setHeaders( $out ); return true; } /** * ArticleFromTitle hook, override category page handling * * @param $title Title * @param $article Article * @return bool */ public static function articleFromTitle( $title, &$article ) { if ( $title->getNamespace() == NS_CATEGORY ) { $article = new CategoryTreeCategoryPage( $title ); } return true; } /** * OutputPageMakeCategoryLinks hook, override category links * @param $out * @param $categories * @param $links * @return bool */ public static function outputPageMakeCategoryLinks( $out, &$categories, &$links ) { global $wgCategoryTreePageCategoryOptions; foreach ( $categories as $category => $type ) { $links[$type][] = self::parserHook( $category, $wgCategoryTreePageCategoryOptions, null, true ); } return false; } /** * @param $skin * @param $links * @param $result * @return bool */ public static function skinJoinCategoryLinks( $skin, &$links, &$result ) { $embed = '

'; $pop = '

'; $sep = ' '; $result = $embed . implode( "{$pop} {$sep} {$embed}", $links ) . $pop; return false; } /** * @param $vars * @return bool */ public static function getConfigVars( &$vars ) { global $wgCategoryTreeCategoryPageOptions; // Look this is pretty bad but Category tree is just whacky, it needs to be rewritten $ct = new CategoryTree( $wgCategoryTreeCategoryPageOptions ); $vars['wgCategoryTreePageCategoryOptions'] = $ct->getOptionsAsJsStructure(); return true; } /** * Hook handler for the SpecialTrackingCategories::preprocess hook * @param SpecialPage $specialPage SpecialTrackingCategories object * @param array $trackingCategories [ 'msg' => Title, 'cats' => Title[] ] */ public static function onSpecialTrackingCategoriesPreprocess( $specialPage, $trackingCategories ) { $categoryDbKeys = []; foreach ( $trackingCategories as $catMsg => $data ) { foreach ( $data['cats'] as $catTitle ) { $categoryDbKeys[] = $catTitle->getDbKey(); } } $categories = []; if ( $categoryDbKeys ) { $dbr = wfGetDB( DB_REPLICA ); $res = $dbr->select( 'category', [ 'cat_id', 'cat_title', 'cat_pages', 'cat_subcats', 'cat_files' ], [ 'cat_title' => array_unique( $categoryDbKeys ) ], __METHOD__ ); foreach ( $res as $row ) { $categories[$row->cat_title] = Category::newFromRow( $row ); } } $specialPage->categoryTreeCategories = $categories; } /** * Hook handler for the SpecialTrackingCategories::generateCatLink hook * @param SpecialPage $specialPage SpecialTrackingCategories object * @param Title $catTitle Title object of the linked category * @param string &$html Result html */ public static function onSpecialTrackingCategoriesGenerateCatLink( $specialPage, $catTitle, &$html ) { if ( !isset( $specialPage->categoryTreeCategories ) ) { return; } $cat = null; if ( isset( $specialPage->categoryTreeCategories[$catTitle->getDbKey()] ) ) { $cat = $specialPage->categoryTreeCategories[$catTitle->getDbKey()]; } $html .= CategoryTree::createCountString( $specialPage->getContext(), $cat, 0 ); } }