* $wgDebugTidy feature
[lhc/web/wiklou.git] / includes / api / ApiQueryAllpages.php
index a69ad43..8f7daec 100644 (file)
@@ -1,12 +1,11 @@
 <?php
 
-
 /*
  * Created on Sep 25, 2006
  *
  * API for MediaWiki 1.8+
  *
- * Copyright (C) 2006 Yuri Astrakhan <FirstnameLastname@gmail.com>
+ * Copyright (C) 2006 Yuri Astrakhan <Firstname><Lastname>@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
 
 if (!defined('MEDIAWIKI')) {
        // Eclipse helper - will be ignored in production
-       require_once ("ApiQueryBase.php");
+       require_once ('ApiQueryBase.php');
 }
 
-class ApiQueryAllpages extends ApiQueryBase {
+/**
+ * Query module to enumerate all available pages.
+ * 
+ * @addtogroup API
+ */
+class ApiQueryAllpages extends ApiQueryGeneratorBase {
 
-       public function __construct($query, $moduleName, $generator = false) {
-               parent :: __construct($query, $moduleName, $generator);
+       public function __construct($query, $moduleName) {
+               parent :: __construct($query, $moduleName, 'ap');
        }
 
-       public function Execute() {
-               $aplimit = $apfrom = $apnamespace = $apfilterredir = null;
-               extract($this->ExtractRequestParams());
+       public function execute() {
+               $this->run();
+       }
 
-               $db = $this->GetDB();
-               $where = array (
-                       'page_namespace' => $apnamespace
-               );
-               if (isset ($apfrom))
-                       $where[] = 'page_title>=' . $db->addQuotes(ApiQueryBase :: TitleToKey($apfrom));
-
-               if ($apfilterredir === 'redirects')
-                       $where['page_is_redirect'] = 1;
-               else
-                       if ($apfilterredir === 'nonredirects')
-                               $where['page_is_redirect'] = 0;
-
-               $res = $db->select('page', array (
-                       'page_id',
-                       'page_namespace',
-                       'page_title'
-               ), $where, __CLASS__ . '::' . __METHOD__, array (
-                       'USE INDEX' => 'name_title',
-                       'LIMIT' => $aplimit +1,
-                       'ORDER BY' => 'page_namespace, page_title'
-               ));
+       public function executeGenerator($resultPageSet) {
+               if ($resultPageSet->isResolvingRedirects())
+                       $this->dieUsage('Use "gapfilterredir=nonredirects" option instead of "redirects" when using allpages as a generator', 'params');
+
+               $this->run($resultPageSet);
+       }
+
+       private function run($resultPageSet = null) {
+
+               $db = $this->getDB();
+
+               $params = $this->extractRequestParams();
+               
+               // Page filters
+               if (!$this->addWhereIf('page_is_redirect = 1', $params['filterredir'] === 'redirects'))
+                       $this->addWhereIf('page_is_redirect = 0', $params['filterredir'] === 'nonredirects');
+               $this->addWhereFld('page_namespace', $params['namespace']);
+               if (!is_null($params['from']))
+                       $this->addWhere('page_title>=' . $db->addQuotes(ApiQueryBase :: titleToKey($params['from'])));
+               if (isset ($params['prefix']))
+                       $this->addWhere("page_title LIKE '" . $db->escapeLike(ApiQueryBase :: titleToKey($params['prefix'])) . "%'");
+
+               $forceNameTitleIndex = true;
+               if (isset ($params['minsize'])) {
+                       $this->addWhere('page_len>=' . intval($params['minsize']));
+                       $forceNameTitleIndex = false;
+               }
+               
+               if (isset ($params['maxsize'])) {
+                       $this->addWhere('page_len<=' . intval($params['maxsize']));
+                       $forceNameTitleIndex = false;
+               }
+       
+               // Page protection filtering
+               if (isset ($params['prtype'])) {
+                       $this->addTables('page_restrictions');
+                       $this->addWhere('page_id=pr_page');
+                       $this->addWhere('pr_expiry>' . $db->addQuotes($db->timestamp()));
+                       $this->addWhereFld('pr_type', $params['prtype']);
+
+                       $prlevel = $params['prlevel'];
+                       if (!is_null($prlevel) && $prlevel != '' && $prlevel != '*')
+                               $this->addWhereFld('pr_level', $prlevel);
+                               
+                       $this->addOption('DISTINCT');
+
+                       $forceNameTitleIndex = false;
+
+               } else if (isset ($params['prlevel'])) {
+                       $this->dieUsage('prlevel may not be used without prtype', 'params');
+               }
+               
+               $this->addTables('page');
+               if ($forceNameTitleIndex)
+                       $this->addOption('USE INDEX', 'name_title');
+               
+
+               if (is_null($resultPageSet)) {
+                       $this->addFields(array (
+                               'page_id',
+                               'page_namespace',
+                               'page_title'
+                       ));
+               } else {
+                       $this->addFields($resultPageSet->getPageTableFields());
+               }
+
+               $limit = $params['limit'];
+               $this->addOption('LIMIT', $limit+1);
+               $this->addOption('ORDER BY', 'page_namespace, page_title' .
+                                               ($params['dir'] == 'ZtoA' ? ' DESC' : ''));
+
+               $res = $this->select(__METHOD__);
 
                $data = array ();
-               $data['_element'] = 'p';
                $count = 0;
                while ($row = $db->fetchObject($res)) {
-                       if (++ $count > $aplimit) {
+                       if (++ $count > $limit) {
                                // We've reached the one extra which shows that there are additional pages to be had. Stop here...
-                               $msg = array (
-                                       'continue' => 'apfrom=' . ApiQueryBase :: KeyToTitle($row->page_title
-                               ));
-                               $this->GetResult()->AddMessage('query-status', 'allpages', $msg);
+                               // TODO: Security issue - if the user has no right to view next title, it will still be shown
+                               $this->setContinueEnumParameter('from', ApiQueryBase :: keyToTitle($row->page_title));
                                break;
                        }
 
-                       $title = Title :: makeTitle($row->page_namespace, $row->page_title);
-                       // skip any pages that user has no rights to read
-                       if ($title->userCanRead()) {
-
-                               $id = intval($row->page_id);
-                               $pagedata = array ();
-                               $pagedata['id'] = $id;
-                               if ($title->getNamespace() !== 0)
-                                       $pagedata['ns'] = $title->getNamespace();
-                               $pagedata['title'] = $title->getPrefixedText();
-                               $pagedata['*'] = '';
-
-                               $data[$id] = $pagedata;
+                       if (is_null($resultPageSet)) {
+                               $title = Title :: makeTitle($row->page_namespace, $row->page_title);
+                               $data[] = array(
+                                       'pageid' => intval($row->page_id),
+                                       'ns' => intval($title->getNamespace()),
+                                       'title' => $title->getPrefixedText());
+                       } else {
+                               $resultPageSet->processDbRow($row);
                        }
                }
                $db->freeResult($res);
-               $this->GetResult()->AddMessage('query', 'allpages', $data);
-       }
-
-       protected function GetAllowedParams() {
 
-               global $wgContLang;
-               $validNamespaces = array ();
-               foreach (array_keys($wgContLang->getNamespaces()) as $ns) {
-                       if ($ns >= 0)
-                               $validNamespaces[] = $ns; // strval($ns);               
+               if (is_null($resultPageSet)) {
+                       $result = $this->getResult();
+                       $result->setIndexedTagName($data, 'p');
+                       $result->addValue('query', $this->getModuleName(), $data);
                }
+       }
 
+       protected function getAllowedParams() {
+               global $wgRestrictionTypes, $wgRestrictionLevels;
+               
                return array (
-                       'apfrom' => null,
-                       'apnamespace' => array (
-                               GN_ENUM_DFLT => 0,
-                               GN_ENUM_TYPE => $validNamespaces
+                       'from' => null,
+                       'prefix' => null,
+                       'namespace' => array (
+                               ApiBase :: PARAM_DFLT => 0,
+                               ApiBase :: PARAM_TYPE => 'namespace',
                        ),
-                       'apfilterredir' => array (
-                               GN_ENUM_DFLT => 'all',
-                               GN_ENUM_TYPE => array (
+                       'filterredir' => array (
+                               ApiBase :: PARAM_DFLT => 'all',
+                               ApiBase :: PARAM_TYPE => array (
                                        'all',
                                        'redirects',
                                        'nonredirects'
                                )
                        ),
-                       'aplimit' => array (
-                               GN_ENUM_DFLT => 10,
-                               GN_ENUM_TYPE => 'limit',
-                               GN_ENUM_MIN => 1,
-                               GN_ENUM_MAX1 => 500,
-                               GN_ENUM_MAX2 => 5000
+                       'minsize' => array (
+                               ApiBase :: PARAM_TYPE => 'integer',
+                       ), 
+                       'maxsize' => array (
+                               ApiBase :: PARAM_TYPE => 'integer',
+                       ), 
+                       'prtype' => array (
+                               ApiBase :: PARAM_TYPE => $wgRestrictionTypes,
+                               ApiBase :: PARAM_ISMULTI => true
+                       ),
+                       'prlevel' => array (
+                               ApiBase :: PARAM_TYPE => $wgRestrictionLevels,
+                               ApiBase :: PARAM_ISMULTI => true
+                       ),
+                       'limit' => array (
+                               ApiBase :: PARAM_DFLT => 10,
+                               ApiBase :: PARAM_TYPE => 'limit',
+                               ApiBase :: PARAM_MIN => 1,
+                               ApiBase :: PARAM_MAX => ApiBase :: LIMIT_BIG1,
+                               ApiBase :: PARAM_MAX2 => ApiBase :: LIMIT_BIG2
+                       ),
+                       'dir' => array (
+                               ApiBase :: PARAM_DFLT => 'AtoZ',
+                               ApiBase :: PARAM_TYPE => array (
+                                       'AtoZ',
+                                       'ZtoA'
+                               )
                        )
                );
        }
 
-       protected function GetParamDescription() {
-               return array ();
+       protected function getParamDescription() {
+               return array (
+                       'from' => 'The page title to start enumerating from.',
+                       'prefix' => 'Search for all page titles that begin with this value.',
+                       'namespace' => 'The namespace to enumerate.',
+                       'filterredir' => 'Which pages to list.',
+                       'dir' => 'The direction in which to list',
+                       'minsize' => 'Limit to pages with at least this many bytes',
+                       'maxsize' => 'Limit to pages with at most this many bytes',
+                       'prtype' => 'Limit to protected pages only',
+                       'prlevel' => 'The protection level (must be used with apprtype= parameter)',
+                       'limit' => 'How many total pages to return.'
+               );
        }
 
-       protected function GetDescription() {
+       protected function getDescription() {
                return 'Enumerate all pages sequentially in a given namespace';
        }
 
-       protected function GetExamples() {
+       protected function getExamples() {
                return array (
-                       'api.php?action=query&list=allpages',
-                       'api.php?action=query&list=allpages&apfrom=B&aplimit=5'
+                       'Simple Use',
+                       ' Show a list of pages starting at the letter "B"',
+                       '  api.php?action=query&list=allpages&apfrom=B',
+                       'Using as Generator',
+                       ' Show info about 4 pages starting at the letter "T"',
+                       '  api.php?action=query&generator=allpages&gaplimit=4&gapfrom=T&prop=info',
+                       ' Show content of first 2 non-redirect pages begining at "Re"',
+                       '  api.php?action=query&generator=allpages&gaplimit=2&gapfilterredir=nonredirects&gapfrom=Re&prop=revisions&rvprop=content'
                );
        }
-       public function GetCanGenerate() {
-               return true;
+
+       public function getVersion() {
+               return __CLASS__ . ': $Id$';
        }
 }
-?>
\ No newline at end of file
+