0f8a17fba2a76a51dd1f8ad5187fcf75c36b833b
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();
77 public static function titleToKey($title) {
78 return str_replace(' ', '_', $title);
81 public static function keyToTitle($key) {
82 return str_replace('_', ' ', $key);
85 public static function getBaseVersion() {
86 return __CLASS__
. ': $Id$';
90 abstract class ApiQueryGeneratorBase
extends ApiQueryBase
{
92 private $mIsGenerator;
94 public function __construct($query, $moduleName, $paramPrefix = '') {
95 parent
:: __construct($query, $moduleName, $paramPrefix);
96 $mIsGenerator = false;
99 public function setGeneratorMode() {
100 $this->mIsGenerator
= true;
104 * Overrides base class to prepend 'g' to every generator parameter
106 public function encodeParamName($paramName) {
107 if ($this->mIsGenerator
)
108 return 'g' . parent
:: encodeParamName($paramName);
110 return parent
:: encodeParamName($paramName);
114 * Execute this module as a generator
115 * @param $resultPageSet PageSet: All output should be appended to this object
117 public abstract function executeGenerator($resultPageSet);