* Code cleanup per TimStarling's suggestions
[lhc/web/wiklou.git] / includes / api / ApiQuery.php
index fe7dd82..458007b 100644 (file)
-<?php\r
-\r
-\r
-/*\r
- * Created on Sep 7, 2006\r
- *\r
- * API for MediaWiki 1.8+\r
- *\r
- * Copyright (C) 2006 Yuri Astrakhan <FirstnameLastname@gmail.com>\r
- *\r
- * This program is free software; you can redistribute it and/or modify\r
- * it under the terms of the GNU General Public License as published by\r
- * the Free Software Foundation; either version 2 of the License, or\r
- * (at your option) any later version.\r
- *\r
- * This program is distributed in the hope that it will be useful,\r
- * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\r
- * GNU General Public License for more details.\r
- *\r
- * You should have received a copy of the GNU General Public License along\r
- * with this program; if not, write to the Free Software Foundation, Inc.,\r
- * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.\r
- * http://www.gnu.org/copyleft/gpl.html\r
- */\r
-\r
-if (!defined('MEDIAWIKI')) {\r
-       // Eclipse helper - will be ignored in production\r
-       require_once ('ApiBase.php');\r
-}\r
-\r
-class ApiQuery extends ApiBase {\r
-\r
-       private $mMetaModuleNames, $mPropModuleNames, $mListModuleNames;\r
-       private $mData;\r
-\r
-       private $mQueryMetaModules = array (\r
-               'siteinfo' => 'ApiQuerySiteinfo'\r
-       );\r
-       //      'userinfo' => 'ApiQueryUserinfo',\r
-\r
-       private $mQueryPropModules = array (\r
-               'info' => 'ApiQueryInfo',\r
-               'revisions' => 'ApiQueryRevisions'\r
-       );\r
-       //      'categories' => 'ApiQueryCategories',\r
-       //      'imageinfo' => 'ApiQueryImageinfo',\r
-       //      'langlinks' => 'ApiQueryLanglinks',\r
-       //      'links' => 'ApiQueryLinks',\r
-       //      'templates' => 'ApiQueryTemplates',\r
-\r
-       private $mQueryListModules = array (\r
-               'allpages' => 'ApiQueryAllpages'\r
-       );\r
-       //      'backlinks' => 'ApiQueryBacklinks',\r
-       //      'categorymembers' => 'ApiQueryCategorymembers',\r
-       //      'embeddedin' => 'ApiQueryEmbeddedin',\r
-       //      'imagelinks' => 'ApiQueryImagelinks',\r
-       //      'logevents' => 'ApiQueryLogevents',\r
-       //      'recentchanges' => 'ApiQueryRecentchanges',\r
-       //      'usercontribs' => 'ApiQueryUsercontribs',\r
-       //      'users' => 'ApiQueryUsers',\r
-       //      'watchlist' => 'ApiQueryWatchlist',\r
-\r
-       private $mSlaveDB = null;\r
-\r
-       public function __construct($main, $action) {\r
-               parent :: __construct($main);\r
-               $this->mMetaModuleNames = array_keys($this->mQueryMetaModules);\r
-               $this->mPropModuleNames = array_keys($this->mQueryPropModules);\r
-               $this->mListModuleNames = array_keys($this->mQueryListModules);\r
-\r
-               // Allow the entire list of modules at first,\r
-               // but during module instantiation check if it can be used as a generator.\r
-               $this->mAllowedGenerators = array_merge($this->mListModuleNames, $this->mPropModuleNames);\r
-       }\r
-\r
-       public function getDB() {\r
-               if (!isset ($this->mSlaveDB))\r
-                       $this->mSlaveDB = & wfGetDB(DB_SLAVE);\r
-               return $this->mSlaveDB;\r
-       }\r
-\r
-       public function getData() {\r
-               return $this->mData;\r
-       }\r
-\r
-       /**\r
-        * Query execution happens in the following steps:\r
-        * #1 Create a PageSet object with any pages requested by the user\r
-        * #2 If using generator, execute it to get a new PageSet object\r
-        * #3 Instantiate all requested modules. \r
-        *    This way the PageSet object will know what shared data is required,\r
-        *    and minimize DB calls. \r
-        * #4 Output all normalization and redirect resolution information\r
-        * #5 Execute all requested modules\r
-        */\r
-       public function execute() {\r
-               $meta = $prop = $list = $generator = null;\r
-               extract($this->extractRequestParams());\r
-\r
-               //\r
-               // Create and initialize PageSet\r
-               //\r
-               $this->mData = new ApiPageSet($this);\r
-               $this->mData->profileIn();\r
-               $this->mData->execute();\r
-               $this->mData->profileOut();\r
-\r
-               //\r
-               // If generator is provided, get a new dataset to work on\r
-               //\r
-               if (isset ($generator))\r
-                       $this->executeGenerator($generator);\r
-\r
-               // Instantiate required modules\r
-               // During instantiation, modules may optimize data requests through the $this->mData object \r
-               // $this->mData will be lazy loaded when modules begin to request data during execution\r
-               $modules = array ();\r
-               if (isset ($meta))\r
-                       foreach ($meta as $moduleName)\r
-                               $modules[] = new $this->mQueryMetaModules[$moduleName] ($this, $moduleName);\r
-               if (isset ($prop))\r
-                       foreach ($prop as $moduleName)\r
-                               $modules[] = new $this->mQueryPropModules[$moduleName] ($this, $moduleName);\r
-               if (isset ($list))\r
-                       foreach ($list as $moduleName)\r
-                               $modules[] = new $this->mQueryListModules[$moduleName] ($this, $moduleName);\r
-\r
-               // Title normalizations\r
-               $normValues = array ();\r
-               foreach ($this->mData->getNormalizedTitles() as $rawTitleStr => $titleStr) {\r
-                       $normValues[] = array (\r
-                               'from' => $rawTitleStr,\r
-                               'to' => $titleStr\r
-                       );\r
-               }\r
-\r
-               if (!empty ($normValues)) {\r
-                       ApiResult :: setIndexedTagName($normValues, 'n');\r
-                       $this->getResult()->addValue('query', 'normalized', $normValues);\r
-               }\r
-\r
-               // Show redirect information\r
-               $redirValues = array ();\r
-               foreach ($this->mData->getRedirectTitles() as $titleStrFrom => $titleStrTo) {\r
-                       $redirValues[] = array (\r
-                               'from' => $titleStrFrom,\r
-                               'to' => $titleStrTo\r
-                       );\r
-               }\r
-\r
-               if (!empty ($redirValues)) {\r
-                       ApiResult :: setIndexedTagName($redirValues, 'r');\r
-                       $this->getResult()->addValue('query', 'redirects', $redirValues);\r
-               }\r
-\r
-               // Execute all requested modules.\r
-               foreach ($modules as $module) {\r
-                       $module->profileIn();\r
-                       $module->execute();\r
-                       $module->profileOut();\r
-               }\r
-\r
-               // Ensure that pages are shown as '<page>' elements\r
-               $data = & $this->getResultData();\r
-               if (isset ($data['query']['pages'])) {\r
-                       ApiResult :: setIndexedTagName($data['query']['pages'], 'page');\r
-               }\r
-       }\r
-\r
-       protected function executeGenerator($generator) {\r
-\r
-               // Find class that implements requested generator\r
-               if (isset ($this->mQueryListModules[$generator]))\r
-                       $className = $this->mQueryListModules[$generator];\r
-               else\r
-                       if (isset ($this->mQueryPropModules[$generator]))\r
-                               $className = $this->mQueryPropModules[$generator];\r
-                       else\r
-                               ApiBase :: dieDebug("Unknown generator=$generator");\r
-\r
-               $module = new $className ($this, $generator, true);\r
-\r
-               // change $this->mData\r
-\r
-               // TODO: implement\r
-               $this->dieUsage('Generator execution has not been implemented', 'notimplemented');\r
-       }\r
-\r
-       protected function getAllowedParams() {\r
-               return array (\r
-                       'meta' => array (\r
-                               GN_ENUM_ISMULTI => true,\r
-                               GN_ENUM_TYPE => $this->mMetaModuleNames\r
-                       ),\r
-                       'prop' => array (\r
-                               GN_ENUM_ISMULTI => true,\r
-                               GN_ENUM_TYPE => $this->mPropModuleNames\r
-                       ),\r
-                       'list' => array (\r
-                               GN_ENUM_ISMULTI => true,\r
-                               GN_ENUM_TYPE => $this->mListModuleNames\r
-                       )\r
-                       //                      'generator' => array (\r
-                       //                              GN_ENUM_TYPE => $this->mAllowedGenerators\r
-                       //                      ),\r
-\r
-                       \r
-               );\r
-       }\r
-\r
-       /**\r
-        * Override the parent to generate help messages for all available query modules.\r
-        */\r
-       public function makeHelpMsg() {\r
-\r
-               // Use parent to make default message for the query module\r
-               $msg = parent :: makeHelpMsg();\r
-\r
-               // Make sure the internal object is empty\r
-               // (just in case a sub-module decides to optimize during instantiation)\r
-               $this->mData = null;\r
-\r
-               $astriks = str_repeat('--- ', 8);\r
-               $msg .= "\n$astriks Query: Meta  $astriks\n\n";\r
-               $msg .= $this->makeHelpMsgHelper($this->mQueryMetaModules, 'meta');\r
-               $msg .= "\n$astriks Query: Prop  $astriks\n\n";\r
-               $msg .= $this->makeHelpMsgHelper($this->mQueryPropModules, 'prop');\r
-               $msg .= "\n$astriks Query: List  $astriks\n\n";\r
-               $msg .= $this->makeHelpMsgHelper($this->mQueryListModules, 'list');\r
-\r
-               return $msg;\r
-       }\r
-\r
-       private function makeHelpMsgHelper($moduleList, $paramName) {\r
-               $msg = '';\r
-\r
-               foreach ($moduleList as $moduleName => $moduleClass) {\r
-                       $msg .= "* $paramName=$moduleName *";\r
-                       $module = new $moduleClass ($this, $moduleName, null);\r
-                       $msg2 = $module->makeHelpMsg();\r
-                       if ($msg2 !== false)\r
-                               $msg .= $msg2;\r
-                       $msg .= "\n";\r
-                       if ($module->getCanGenerate())\r
-                               $msg .= "  * Can be used as a generator\n";\r
-               }\r
-\r
-               return $msg;\r
-       }\r
-\r
-       /**\r
-        * Override to add extra parameters from PageSet\r
-        */\r
-       public function makeHelpMsgParameters() {\r
-               $module = new ApiPageSet($this);\r
-               return $module->makeHelpMsgParameters() . parent :: makeHelpMsgParameters();\r
-       }\r
-\r
-       protected function getParamDescription() {\r
-               return array (\r
-                       'meta' => 'Which meta data to get about the site',\r
-                       'prop' => 'Which properties to get for the titles/revisions/pageids',\r
-                       'list' => 'Which lists to get',\r
-                       'generator' => 'Use the output of a list as the input for other prop/list/meta items',\r
-\r
-                       \r
-               );\r
-       }\r
-\r
-       protected function getDescription() {\r
-               return array (\r
-                       'Query API module allows applications to get needed pieces of data from the MediaWiki databases,',\r
-                       'and is loosely based on the Query API interface currently available on all MediaWiki servers.',\r
-                       'All data modifications will first have to use query to acquire a token to prevent abuse from malicious sites.'\r
-               );\r
-       }\r
-\r
-       protected function getExamples() {\r
-               return array (\r
-                       'api.php?action=query&what=content&titles=ArticleA|ArticleB'\r
-               );\r
-       }\r
-}\r
-?>\r
+<?php
+
+
+/*
+ * Created on Sep 7, 2006
+ *
+ * API for MediaWiki 1.8+
+ *
+ * Copyright (C) 2006 Yuri Astrakhan <FirstnameLastname@gmail.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+ * http://www.gnu.org/copyleft/gpl.html
+ */
+
+if (!defined('MEDIAWIKI')) {
+       // Eclipse helper - will be ignored in production
+       require_once ('ApiBase.php');
+}
+
+class ApiQuery extends ApiBase {
+
+       private $mMetaModuleNames, $mPropModuleNames, $mListModuleNames;
+       private $mData;
+
+       private $mQueryMetaModules = array (
+               'siteinfo' => 'ApiQuerySiteinfo'
+       );
+       //      'userinfo' => 'ApiQueryUserinfo',
+
+       private $mQueryPropModules = array (
+               'info' => 'ApiQueryInfo',
+               'revisions' => 'ApiQueryRevisions'
+       );
+       //      'categories' => 'ApiQueryCategories',
+       //      'imageinfo' => 'ApiQueryImageinfo',
+       //      'langlinks' => 'ApiQueryLanglinks',
+       //      'links' => 'ApiQueryLinks',
+       //      'templates' => 'ApiQueryTemplates',
+
+       private $mQueryListModules = array (
+               'allpages' => 'ApiQueryAllpages'
+       );
+       //      'backlinks' => 'ApiQueryBacklinks',
+       //      'categorymembers' => 'ApiQueryCategorymembers',
+       //      'embeddedin' => 'ApiQueryEmbeddedin',
+       //      'imagelinks' => 'ApiQueryImagelinks',
+       //      'logevents' => 'ApiQueryLogevents',
+       //      'recentchanges' => 'ApiQueryRecentchanges',
+       //      'usercontribs' => 'ApiQueryUsercontribs',
+       //      'users' => 'ApiQueryUsers',
+       //      'watchlist' => 'ApiQueryWatchlist',
+
+       private $mSlaveDB = null;
+
+       public function __construct($main, $action) {
+               parent :: __construct($main);
+               $this->mMetaModuleNames = array_keys($this->mQueryMetaModules);
+               $this->mPropModuleNames = array_keys($this->mQueryPropModules);
+               $this->mListModuleNames = array_keys($this->mQueryListModules);
+
+               // Allow the entire list of modules at first,
+               // but during module instantiation check if it can be used as a generator.
+               $this->mAllowedGenerators = array_merge($this->mListModuleNames, $this->mPropModuleNames);
+       }
+
+       public function getDB() {
+               if (!isset ($this->mSlaveDB))
+                       $this->mSlaveDB = & wfGetDB(DB_SLAVE);
+               return $this->mSlaveDB;
+       }
+
+       public function getData() {
+               return $this->mData;
+       }
+
+       /**
+        * Query execution happens in the following steps:
+        * #1 Create a PageSet object with any pages requested by the user
+        * #2 If using generator, execute it to get a new PageSet object
+        * #3 Instantiate all requested modules. 
+        *    This way the PageSet object will know what shared data is required,
+        *    and minimize DB calls. 
+        * #4 Output all normalization and redirect resolution information
+        * #5 Execute all requested modules
+        */
+       public function execute() {
+               $meta = $prop = $list = $generator = null;
+               extract($this->extractRequestParams());
+
+               //
+               // Create PageSet
+               //
+               $this->mData = new ApiPageSet($this);
+
+               //
+               // If generator is provided, get a new dataset to work on
+               //
+               if (isset ($generator))
+                       $this->executeGenerator($generator);
+
+               // Instantiate required modules
+               // During instantiation, modules may optimize data requests through the $this->mData object 
+               // $this->mData will be lazy loaded when modules begin to request data during execution
+               $modules = array ();
+               if (isset ($meta))
+                       foreach ($meta as $moduleName)
+                               $modules[] = new $this->mQueryMetaModules[$moduleName] ($this, $moduleName);
+               if (isset ($prop))
+                       foreach ($prop as $moduleName)
+                               $modules[] = new $this->mQueryPropModules[$moduleName] ($this, $moduleName);
+               if (isset ($list))
+                       foreach ($list as $moduleName)
+                               $modules[] = new $this->mQueryListModules[$moduleName] ($this, $moduleName);
+
+               //
+               // Get page information for the given pageSet
+               //
+               $this->mData->profileIn();
+               $this->mData->execute();
+               $this->mData->profileOut();
+
+               // Title normalizations
+               $normValues = array ();
+               foreach ($this->mData->getNormalizedTitles() as $rawTitleStr => $titleStr) {
+                       $normValues[] = array (
+                               'from' => $rawTitleStr,
+                               'to' => $titleStr
+                       );
+               }
+
+               if (!empty ($normValues)) {
+                       ApiResult :: setIndexedTagName($normValues, 'n');
+                       $this->getResult()->addValue('query', 'normalized', $normValues);
+               }
+
+               // Show redirect information
+               $redirValues = array ();
+               foreach ($this->mData->getRedirectTitles() as $titleStrFrom => $titleStrTo) {
+                       $redirValues[] = array (
+                               'from' => $titleStrFrom,
+                               'to' => $titleStrTo
+                       );
+               }
+
+               if (!empty ($redirValues)) {
+                       ApiResult :: setIndexedTagName($redirValues, 'r');
+                       $this->getResult()->addValue('query', 'redirects', $redirValues);
+               }
+
+               // Execute all requested modules.
+               foreach ($modules as $module) {
+                       $module->profileIn();
+                       $module->execute();
+                       $module->profileOut();
+               }
+
+               // Ensure that pages are shown as '<page>' elements
+               $data = & $this->getResultData();
+               if (isset ($data['query']['pages'])) {
+                       ApiResult :: setIndexedTagName($data['query']['pages'], 'page');
+               }
+       }
+
+       protected function executeGenerator($generator) {
+
+               // Find class that implements requested generator
+               if (isset ($this->mQueryListModules[$generator]))
+                       $className = $this->mQueryListModules[$generator];
+               else
+                       if (isset ($this->mQueryPropModules[$generator]))
+                               $className = $this->mQueryPropModules[$generator];
+                       else
+                               ApiBase :: dieDebug("Unknown generator=$generator");
+
+               $module = new $className ($this, $generator, true);
+
+               // execute pageSet here to get the data required by the generator module
+               $this->mData->profileIn();
+               $this->mData->execute();
+               $this->mData->profileOut();
+
+               // change $this->mData
+
+               // TODO: implement
+               $this->dieUsage('Generator execution has not been implemented', 'notimplemented');
+       }
+
+       protected function getAllowedParams() {
+               return array (
+                       'meta' => array (
+                               ApiBase::PARAM_ISMULTI => true,
+                               ApiBase::PARAM_TYPE => $this->mMetaModuleNames
+                       ),
+                       'prop' => array (
+                               ApiBase::PARAM_ISMULTI => true,
+                               ApiBase::PARAM_TYPE => $this->mPropModuleNames
+                       ),
+                       'list' => array (
+                               ApiBase::PARAM_ISMULTI => true,
+                               ApiBase::PARAM_TYPE => $this->mListModuleNames
+                       )
+                       //                      'generator' => array (
+                       //                              ApiBase::PARAM_TYPE => $this->mAllowedGenerators
+                       //                      ),
+
+                       
+               );
+       }
+
+       /**
+        * Override the parent to generate help messages for all available query modules.
+        */
+       public function makeHelpMsg() {
+
+               // Use parent to make default message for the query module
+               $msg = parent :: makeHelpMsg();
+
+               // Make sure the internal object is empty
+               // (just in case a sub-module decides to optimize during instantiation)
+               $this->mData = null;
+
+               $astriks = str_repeat('--- ', 8);
+               $msg .= "\n$astriks Query: Meta  $astriks\n\n";
+               $msg .= $this->makeHelpMsgHelper($this->mQueryMetaModules, 'meta');
+               $msg .= "\n$astriks Query: Prop  $astriks\n\n";
+               $msg .= $this->makeHelpMsgHelper($this->mQueryPropModules, 'prop');
+               $msg .= "\n$astriks Query: List  $astriks\n\n";
+               $msg .= $this->makeHelpMsgHelper($this->mQueryListModules, 'list');
+
+               return $msg;
+       }
+
+       private function makeHelpMsgHelper($moduleList, $paramName) {
+               $msg = '';
+
+               foreach ($moduleList as $moduleName => $moduleClass) {
+                       $msg .= "* $paramName=$moduleName *";
+                       $module = new $moduleClass ($this, $moduleName, null);
+                       $msg2 = $module->makeHelpMsg();
+                       if ($msg2 !== false)
+                               $msg .= $msg2;
+                       $msg .= "\n";
+                       if ($module->getCanGenerate())
+                               $msg .= "  * Can be used as a generator\n";
+               }
+
+               return $msg;
+       }
+
+       /**
+        * Override to add extra parameters from PageSet
+        */
+       public function makeHelpMsgParameters() {
+               $module = new ApiPageSet($this);
+               return $module->makeHelpMsgParameters() . parent :: makeHelpMsgParameters();
+       }
+
+       protected function getParamDescription() {
+               return array (
+                       'meta' => 'Which meta data to get about the site',
+                       'prop' => 'Which properties to get for the titles/revisions/pageids',
+                       'list' => 'Which lists to get',
+                       'generator' => 'Use the output of a list as the input for other prop/list/meta items',
+
+                       
+               );
+       }
+
+       protected function getDescription() {
+               return array (
+                       'Query API module allows applications to get needed pieces of data from the MediaWiki databases,',
+                       'and is loosely based on the Query API interface currently available on all MediaWiki servers.',
+                       'All data modifications will first have to use query to acquire a token to prevent abuse from malicious sites.'
+               );
+       }
+
+       protected function getExamples() {
+               return array (
+                       'api.php?action=query&prop=revisions&meta=siteinfo&titles=Main%20Page&rvprop=user|comment'
+               );
+       }
+}
+?>