49639273586ae45609f3abf7c6752a7c1c9a9b7f
5 * Created on Sep 7, 2006
7 * API for MediaWiki 1.8+
9 * Copyright (C) 2006 Yuri Astrakhan <FirstnameLastname@gmail.com>
11 * This program is free software; you can redistribute it and/or modify
12 * it under the terms of the GNU General Public License as published by
13 * the Free Software Foundation; either version 2 of the License, or
14 * (at your option) any later version.
16 * This program is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 * GNU General Public License for more details.
21 * You should have received a copy of the GNU General Public License along
22 * with this program; if not, write to the Free Software Foundation, Inc.,
23 * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
24 * http://www.gnu.org/copyleft/gpl.html
27 if (!defined('MEDIAWIKI')) {
28 // Eclipse helper - will be ignored in production
29 require_once ('ApiBase.php');
32 abstract class ApiQueryBase
extends ApiBase
{
34 private $mQueryModule;
36 public function __construct($query, $moduleName, $paramPrefix = '') {
37 parent
:: __construct($query->getMain(), $moduleName, $paramPrefix);
38 $this->mQueryModule
= $query;
42 * Override this method to request extra fields from the pageSet
43 * using $this->getPageSet()->requestField('fieldName')
45 public function requestExtraData() {
49 * Get the main Query module
51 public function getQuery() {
52 return $this->mQueryModule
;
55 protected function setContinueEnumParameter($paramName, $paramValue) {
57 $this->encodeParamName($paramName
59 $this->getResult()->addValue('query-continue', $this->getModuleName(), $msg);
63 * Get the Query database connection (readonly)
65 protected function getDB() {
66 return $this->getQuery()->getDB();
70 * Get the PageSet object to work on
71 * @return ApiPageSet data
73 protected function getPageSet() {
74 return $this->mQueryModule
->getPageSet();
78 * This is a very simplistic utility function
79 * to convert a non-namespaced title string to a db key.
80 * It will replace all ' ' with '_'
82 public static function titleToKey($title) {
83 return str_replace(' ', '_', $title);
86 public static function keyToTitle($key) {
87 return str_replace('_', ' ', $key);
90 public static function getBaseVersion() {
91 return __CLASS__
. ': $Id$';
95 abstract class ApiQueryGeneratorBase
extends ApiQueryBase
{
97 private $mIsGenerator;
99 public function __construct($query, $moduleName, $paramPrefix = '') {
100 parent
:: __construct($query, $moduleName, $paramPrefix);
101 $this->mIsGenerator
= false;
104 public function setGeneratorMode() {
105 $this->mIsGenerator
= true;
109 * Overrides base class to prepend 'g' to every generator parameter
111 public function encodeParamName($paramName) {
112 if ($this->mIsGenerator
)
113 return 'g' . parent
:: encodeParamName($paramName);
115 return parent
:: encodeParamName($paramName);
119 * Execute this module as a generator
120 * @param $resultPageSet PageSet: All output should be appended to this object
122 public abstract function executeGenerator($resultPageSet);