Merge "Make DBAccessBase use DBConnRef, rename $wiki, and hide getLoadBalancer()"
[lhc/web/wiklou.git] / includes / api / ApiQueryDeletedRevisions.php
1 <?php
2 /**
3 * Copyright © 2014 Wikimedia Foundation and contributors
4 *
5 * Heavily based on ApiQueryDeletedrevs,
6 * Copyright © 2007 Roan Kattouw "<Firstname>.<Lastname>@gmail.com"
7 *
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, write to the Free Software Foundation, Inc.,
20 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
21 * http://www.gnu.org/copyleft/gpl.html
22 *
23 * @file
24 */
25
26 use MediaWiki\MediaWikiServices;
27 use MediaWiki\Revision\RevisionRecord;
28 use MediaWiki\Storage\NameTableAccessException;
29
30 /**
31 * Query module to enumerate deleted revisions for pages.
32 *
33 * @ingroup API
34 */
35 class ApiQueryDeletedRevisions extends ApiQueryRevisionsBase {
36
37 public function __construct( ApiQuery $query, $moduleName ) {
38 parent::__construct( $query, $moduleName, 'drv' );
39 }
40
41 protected function run( ApiPageSet $resultPageSet = null ) {
42 $user = $this->getUser();
43 // Before doing anything at all, let's check permissions
44 $this->checkUserRightsAny( 'deletedhistory' );
45
46 $pageSet = $this->getPageSet();
47 $pageMap = $pageSet->getGoodAndMissingTitlesByNamespace();
48 $pageCount = count( $pageSet->getGoodAndMissingTitles() );
49 $revCount = $pageSet->getRevisionCount();
50 if ( $revCount === 0 && $pageCount === 0 ) {
51 // Nothing to do
52 return;
53 }
54 if ( $revCount !== 0 && count( $pageSet->getDeletedRevisionIDs() ) === 0 ) {
55 // Nothing to do, revisions were supplied but none are deleted
56 return;
57 }
58
59 $params = $this->extractRequestParams( false );
60
61 $db = $this->getDB();
62 $revisionStore = MediaWikiServices::getInstance()->getRevisionStore();
63
64 $this->requireMaxOneParameter( $params, 'user', 'excludeuser' );
65
66 if ( $resultPageSet === null ) {
67 $this->parseParameters( $params );
68 $arQuery = $revisionStore->getArchiveQueryInfo();
69 $this->addTables( $arQuery['tables'] );
70 $this->addFields( $arQuery['fields'] );
71 $this->addJoinConds( $arQuery['joins'] );
72 $this->addFields( [ 'ar_title', 'ar_namespace' ] );
73 } else {
74 $this->limit = $this->getParameter( 'limit' ) ?: 10;
75 $this->addTables( 'archive' );
76 $this->addFields( [ 'ar_title', 'ar_namespace', 'ar_timestamp', 'ar_rev_id', 'ar_id' ] );
77 }
78
79 if ( $this->fld_tags ) {
80 $this->addFields( [ 'ts_tags' => ChangeTags::makeTagSummarySubquery( 'archive' ) ] );
81 }
82
83 if ( !is_null( $params['tag'] ) ) {
84 $this->addTables( 'change_tag' );
85 $this->addJoinConds(
86 [ 'change_tag' => [ 'JOIN', [ 'ar_rev_id=ct_rev_id' ] ] ]
87 );
88 $changeTagDefStore = MediaWikiServices::getInstance()->getChangeTagDefStore();
89 try {
90 $this->addWhereFld( 'ct_tag_id', $changeTagDefStore->getId( $params['tag'] ) );
91 } catch ( NameTableAccessException $exception ) {
92 // Return nothing.
93 $this->addWhere( '1=0' );
94 }
95 }
96
97 // This means stricter restrictions
98 if ( $this->fetchContent ) {
99 $this->checkUserRightsAny( [ 'deletedtext', 'undelete' ] );
100 }
101
102 $dir = $params['dir'];
103
104 if ( $revCount !== 0 ) {
105 $this->addWhere( [
106 'ar_rev_id' => array_keys( $pageSet->getDeletedRevisionIDs() )
107 ] );
108 } else {
109 // We need a custom WHERE clause that matches all titles.
110 $lb = new LinkBatch( $pageSet->getGoodAndMissingTitles() );
111 $where = $lb->constructSet( 'ar', $db );
112 $this->addWhere( $where );
113 }
114
115 if ( !is_null( $params['user'] ) ) {
116 // Don't query by user ID here, it might be able to use the ar_usertext_timestamp index.
117 $actorQuery = ActorMigration::newMigration()
118 ->getWhere( $db, 'ar_user', User::newFromName( $params['user'], false ), false );
119 $this->addTables( $actorQuery['tables'] );
120 $this->addJoinConds( $actorQuery['joins'] );
121 $this->addWhere( $actorQuery['conds'] );
122 } elseif ( !is_null( $params['excludeuser'] ) ) {
123 // Here there's no chance of using ar_usertext_timestamp.
124 $actorQuery = ActorMigration::newMigration()
125 ->getWhere( $db, 'ar_user', User::newFromName( $params['excludeuser'], false ) );
126 $this->addTables( $actorQuery['tables'] );
127 $this->addJoinConds( $actorQuery['joins'] );
128 $this->addWhere( 'NOT(' . $actorQuery['conds'] . ')' );
129 }
130
131 if ( !is_null( $params['user'] ) || !is_null( $params['excludeuser'] ) ) {
132 // Paranoia: avoid brute force searches (T19342)
133 // (shouldn't be able to get here without 'deletedhistory', but
134 // check it again just in case)
135 if ( !$this->getPermissionManager()->userHasRight( $user, 'deletedhistory' ) ) {
136 $bitmask = RevisionRecord::DELETED_USER;
137 } elseif ( !$this->getPermissionManager()
138 ->userHasAnyRight( $this->getUser(), 'suppressrevision', 'viewsuppressed' )
139 ) {
140 $bitmask = RevisionRecord::DELETED_USER | RevisionRecord::DELETED_RESTRICTED;
141 } else {
142 $bitmask = 0;
143 }
144 if ( $bitmask ) {
145 $this->addWhere( $db->bitAnd( 'ar_deleted', $bitmask ) . " != $bitmask" );
146 }
147 }
148
149 if ( !is_null( $params['continue'] ) ) {
150 $cont = explode( '|', $params['continue'] );
151 $op = ( $dir == 'newer' ? '>' : '<' );
152 if ( $revCount !== 0 ) {
153 $this->dieContinueUsageIf( count( $cont ) != 2 );
154 $rev = (int)$cont[0];
155 $this->dieContinueUsageIf( strval( $rev ) !== $cont[0] );
156 $ar_id = (int)$cont[1];
157 $this->dieContinueUsageIf( strval( $ar_id ) !== $cont[1] );
158 $this->addWhere( "ar_rev_id $op $rev OR " .
159 "(ar_rev_id = $rev AND " .
160 "ar_id $op= $ar_id)" );
161 } else {
162 $this->dieContinueUsageIf( count( $cont ) != 4 );
163 $ns = (int)$cont[0];
164 $this->dieContinueUsageIf( strval( $ns ) !== $cont[0] );
165 $title = $db->addQuotes( $cont[1] );
166 $ts = $db->addQuotes( $db->timestamp( $cont[2] ) );
167 $ar_id = (int)$cont[3];
168 $this->dieContinueUsageIf( strval( $ar_id ) !== $cont[3] );
169 $this->addWhere( "ar_namespace $op $ns OR " .
170 "(ar_namespace = $ns AND " .
171 "(ar_title $op $title OR " .
172 "(ar_title = $title AND " .
173 "(ar_timestamp $op $ts OR " .
174 "(ar_timestamp = $ts AND " .
175 "ar_id $op= $ar_id)))))" );
176 }
177 }
178
179 $this->addOption( 'LIMIT', $this->limit + 1 );
180
181 if ( $revCount !== 0 ) {
182 // Sort by ar_rev_id when querying by ar_rev_id
183 $this->addWhereRange( 'ar_rev_id', $dir, null, null );
184 } else {
185 // Sort by ns and title in the same order as timestamp for efficiency
186 // But only when not already unique in the query
187 if ( count( $pageMap ) > 1 ) {
188 $this->addWhereRange( 'ar_namespace', $dir, null, null );
189 }
190 $oneTitle = key( reset( $pageMap ) );
191 foreach ( $pageMap as $pages ) {
192 if ( count( $pages ) > 1 || key( $pages ) !== $oneTitle ) {
193 $this->addWhereRange( 'ar_title', $dir, null, null );
194 break;
195 }
196 }
197 $this->addTimestampWhereRange( 'ar_timestamp', $dir, $params['start'], $params['end'] );
198 }
199 // Include in ORDER BY for uniqueness
200 $this->addWhereRange( 'ar_id', $dir, null, null );
201
202 $res = $this->select( __METHOD__ );
203 $count = 0;
204 $generated = [];
205 foreach ( $res as $row ) {
206 if ( ++$count > $this->limit ) {
207 // We've had enough
208 $this->setContinueEnumParameter( 'continue',
209 $revCount
210 ? "$row->ar_rev_id|$row->ar_id"
211 : "$row->ar_namespace|$row->ar_title|$row->ar_timestamp|$row->ar_id"
212 );
213 break;
214 }
215
216 if ( $resultPageSet !== null ) {
217 $generated[] = $row->ar_rev_id;
218 } else {
219 if ( !isset( $pageMap[$row->ar_namespace][$row->ar_title] ) ) {
220 // Was it converted?
221 $title = Title::makeTitle( $row->ar_namespace, $row->ar_title );
222 $converted = $pageSet->getConvertedTitles();
223 if ( $title && isset( $converted[$title->getPrefixedText()] ) ) {
224 $title = Title::newFromText( $converted[$title->getPrefixedText()] );
225 if ( $title && isset( $pageMap[$title->getNamespace()][$title->getDBkey()] ) ) {
226 $pageMap[$row->ar_namespace][$row->ar_title] =
227 $pageMap[$title->getNamespace()][$title->getDBkey()];
228 }
229 }
230 }
231 if ( !isset( $pageMap[$row->ar_namespace][$row->ar_title] ) ) {
232 ApiBase::dieDebug(
233 __METHOD__,
234 "Found row in archive (ar_id={$row->ar_id}) that didn't get processed by ApiPageSet"
235 );
236 }
237
238 $fit = $this->addPageSubItem(
239 $pageMap[$row->ar_namespace][$row->ar_title],
240 $this->extractRevisionInfo( $revisionStore->newRevisionFromArchiveRow( $row ), $row ),
241 'rev'
242 );
243 if ( !$fit ) {
244 $this->setContinueEnumParameter( 'continue',
245 $revCount
246 ? "$row->ar_rev_id|$row->ar_id"
247 : "$row->ar_namespace|$row->ar_title|$row->ar_timestamp|$row->ar_id"
248 );
249 break;
250 }
251 }
252 }
253
254 if ( $resultPageSet !== null ) {
255 $resultPageSet->populateFromRevisionIDs( $generated );
256 }
257 }
258
259 public function getAllowedParams() {
260 return parent::getAllowedParams() + [
261 'start' => [
262 ApiBase::PARAM_TYPE => 'timestamp',
263 ],
264 'end' => [
265 ApiBase::PARAM_TYPE => 'timestamp',
266 ],
267 'dir' => [
268 ApiBase::PARAM_TYPE => [
269 'newer',
270 'older'
271 ],
272 ApiBase::PARAM_DFLT => 'older',
273 ApiBase::PARAM_HELP_MSG => 'api-help-param-direction',
274 ],
275 'tag' => null,
276 'user' => [
277 ApiBase::PARAM_TYPE => 'user'
278 ],
279 'excludeuser' => [
280 ApiBase::PARAM_TYPE => 'user'
281 ],
282 'continue' => [
283 ApiBase::PARAM_HELP_MSG => 'api-help-param-continue',
284 ],
285 ];
286 }
287
288 protected function getExamplesMessages() {
289 return [
290 'action=query&prop=deletedrevisions&titles=Main%20Page|Talk:Main%20Page&' .
291 'drvslots=*&drvprop=user|comment|content'
292 => 'apihelp-query+deletedrevisions-example-titles',
293 'action=query&prop=deletedrevisions&revids=123456'
294 => 'apihelp-query+deletedrevisions-example-revids',
295 ];
296 }
297
298 public function getHelpUrls() {
299 return 'https://www.mediawiki.org/wiki/Special:MyLanguage/API:Deletedrevisions';
300 }
301 }