bedfe133c7e82f94a27f706c30df5adc931080b3
[lhc/web/wiklou.git] / includes / export / XmlDumpWriter.php
1 <?php
2 /**
3 * XmlDumpWriter
4 *
5 * Copyright © 2003, 2005, 2006 Brion Vibber <brion@pobox.com>
6 * https://www.mediawiki.org/
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 use MediaWiki\MediaWikiServices;
26 use MediaWiki\Revision\RevisionRecord;
27 use MediaWiki\Revision\RevisionStore;
28 use MediaWiki\Revision\SlotRecord;
29 use MediaWiki\Revision\SuppressedDataException;
30 use MediaWiki\Storage\SqlBlobStore;
31 use Wikimedia\Assert\Assert;
32
33 /**
34 * @ingroup Dump
35 */
36 class XmlDumpWriter {
37
38 /** Output serialized revision content. */
39 const WRITE_CONTENT = 0;
40
41 /** Only output subs for revision content. */
42 const WRITE_STUB = 1;
43
44 /**
45 * Only output subs for revision content, indicating that the content has been
46 * deleted/suppressed. For internal use only.
47 */
48 const WRITE_STUB_DELETED = 2;
49
50 /**
51 * @var string[] the schema versions supported for output
52 * @final
53 */
54 public static $supportedSchemas = [
55 XML_DUMP_SCHEMA_VERSION_10,
56 XML_DUMP_SCHEMA_VERSION_11
57 ];
58
59 /**
60 * @var string which schema version the generated XML should comply to.
61 * One of the values from self::$supportedSchemas, using the SCHEMA_VERSION_XX
62 * constants.
63 */
64 private $schemaVersion;
65
66 /**
67 * Title of the currently processed page
68 *
69 * @var Title|null
70 */
71 private $currentTitle = null;
72
73 /**
74 * @var int Whether to output revision content or just stubs. WRITE_CONTENT or WRITE_STUB.
75 */
76 private $contentMode;
77
78 /**
79 * XmlDumpWriter constructor.
80 *
81 * @param int $contentMode WRITE_CONTENT or WRITE_STUB.
82 * @param string $schemaVersion which schema version the generated XML should comply to.
83 * One of the values from self::$supportedSchemas, using the XML_DUMP_SCHEMA_VERSION_XX
84 * constants.
85 */
86 public function __construct(
87 $contentMode = self::WRITE_CONTENT,
88 $schemaVersion = XML_DUMP_SCHEMA_VERSION_11
89 ) {
90 Assert::parameter(
91 in_array( $contentMode, [ self::WRITE_CONTENT, self::WRITE_STUB ] ),
92 '$contentMode',
93 'must be one of the following constants: WRITE_CONTENT or WRITE_STUB.'
94 );
95
96 Assert::parameter(
97 in_array( $schemaVersion, self::$supportedSchemas ),
98 '$schemaVersion',
99 'must be one of the following schema versions: '
100 . implode( ',', self::$supportedSchemas )
101 );
102
103 $this->contentMode = $contentMode;
104 $this->schemaVersion = $schemaVersion;
105 }
106
107 /**
108 * Opens the XML output stream's root "<mediawiki>" element.
109 * This does not include an xml directive, so is safe to include
110 * as a subelement in a larger XML stream. Namespace and XML Schema
111 * references are included.
112 *
113 * Output will be encoded in UTF-8.
114 *
115 * @return string
116 */
117 function openStream() {
118 $ver = $this->schemaVersion;
119 return Xml::element( 'mediawiki', [
120 'xmlns' => "http://www.mediawiki.org/xml/export-$ver/",
121 'xmlns:xsi' => "http://www.w3.org/2001/XMLSchema-instance",
122 /*
123 * When a new version of the schema is created, it needs staging on mediawiki.org.
124 * This requires a change in the operations/mediawiki-config git repo.
125 *
126 * Create a changeset like https://gerrit.wikimedia.org/r/#/c/149643/ in which
127 * you copy in the new xsd file.
128 *
129 * After it is reviewed, merged and deployed (sync-docroot), the index.html needs purging.
130 * echo "https://www.mediawiki.org/xml/index.html" | mwscript purgeList.php --wiki=aawiki
131 */
132 'xsi:schemaLocation' => "http://www.mediawiki.org/xml/export-$ver/ " .
133 "http://www.mediawiki.org/xml/export-$ver.xsd",
134 'version' => $ver,
135 'xml:lang' => MediaWikiServices::getInstance()->getContentLanguage()->getHtmlCode() ],
136 null ) .
137 "\n" .
138 $this->siteInfo();
139 }
140
141 /**
142 * @return string
143 */
144 function siteInfo() {
145 $info = [
146 $this->sitename(),
147 $this->dbname(),
148 $this->homelink(),
149 $this->generator(),
150 $this->caseSetting(),
151 $this->namespaces() ];
152 return " <siteinfo>\n " .
153 implode( "\n ", $info ) .
154 "\n </siteinfo>\n";
155 }
156
157 /**
158 * @return string
159 */
160 function sitename() {
161 global $wgSitename;
162 return Xml::element( 'sitename', [], $wgSitename );
163 }
164
165 /**
166 * @return string
167 */
168 function dbname() {
169 global $wgDBname;
170 return Xml::element( 'dbname', [], $wgDBname );
171 }
172
173 /**
174 * @return string
175 */
176 function generator() {
177 global $wgVersion;
178 return Xml::element( 'generator', [], "MediaWiki $wgVersion" );
179 }
180
181 /**
182 * @return string
183 */
184 function homelink() {
185 return Xml::element( 'base', [], Title::newMainPage()->getCanonicalURL() );
186 }
187
188 /**
189 * @return string
190 */
191 function caseSetting() {
192 global $wgCapitalLinks;
193 // "case-insensitive" option is reserved for future
194 $sensitivity = $wgCapitalLinks ? 'first-letter' : 'case-sensitive';
195 return Xml::element( 'case', [], $sensitivity );
196 }
197
198 /**
199 * @return string
200 */
201 function namespaces() {
202 $spaces = "<namespaces>\n";
203 $nsInfo = MediaWikiServices::getInstance()->getNamespaceInfo();
204 foreach (
205 MediaWikiServices::getInstance()->getContentLanguage()->getFormattedNamespaces()
206 as $ns => $title
207 ) {
208 $spaces .= ' ' .
209 Xml::element( 'namespace',
210 [
211 'key' => $ns,
212 'case' => $nsInfo->isCapitalized( $ns )
213 ? 'first-letter' : 'case-sensitive',
214 ], $title ) . "\n";
215 }
216 $spaces .= " </namespaces>";
217 return $spaces;
218 }
219
220 /**
221 * Closes the output stream with the closing root element.
222 * Call when finished dumping things.
223 *
224 * @return string
225 */
226 function closeStream() {
227 return "</mediawiki>\n";
228 }
229
230 /**
231 * Opens a "<page>" section on the output stream, with data
232 * from the given database row.
233 *
234 * @param object $row
235 * @return string
236 */
237 public function openPage( $row ) {
238 $out = " <page>\n";
239 $this->currentTitle = Title::newFromRow( $row );
240 $canonicalTitle = self::canonicalTitle( $this->currentTitle );
241 $out .= ' ' . Xml::elementClean( 'title', [], $canonicalTitle ) . "\n";
242 $out .= ' ' . Xml::element( 'ns', [], strval( $row->page_namespace ) ) . "\n";
243 $out .= ' ' . Xml::element( 'id', [], strval( $row->page_id ) ) . "\n";
244 if ( $row->page_is_redirect ) {
245 $page = WikiPage::factory( $this->currentTitle );
246 $redirect = $page->getRedirectTarget();
247 if ( $redirect instanceof Title && $redirect->isValidRedirectTarget() ) {
248 $out .= ' ';
249 $out .= Xml::element( 'redirect', [ 'title' => self::canonicalTitle( $redirect ) ] );
250 $out .= "\n";
251 }
252 }
253
254 if ( $row->page_restrictions != '' ) {
255 $out .= ' ' . Xml::element( 'restrictions', [],
256 strval( $row->page_restrictions ) ) . "\n";
257 }
258
259 Hooks::run( 'XmlDumpWriterOpenPage', [ $this, &$out, $row, $this->currentTitle ] );
260
261 return $out;
262 }
263
264 /**
265 * Closes a "<page>" section on the output stream.
266 *
267 * @private
268 * @return string
269 */
270 function closePage() {
271 if ( $this->currentTitle !== null ) {
272 $linkCache = MediaWikiServices::getInstance()->getLinkCache();
273 // In rare cases, link cache has the same key for some pages which
274 // might be read as part of the same batch. T220424 and T220316
275 $linkCache->clearLink( $this->currentTitle );
276 }
277 return " </page>\n";
278 }
279
280 /**
281 * @return RevisionStore
282 */
283 private function getRevisionStore() {
284 return MediaWikiServices::getInstance()->getRevisionStore();
285 }
286
287 /**
288 * @return SqlBlobStore
289 */
290 private function getBlobStore() {
291 return MediaWikiServices::getInstance()->getBlobStore();
292 }
293
294 /**
295 * Dumps a "<revision>" section on the output stream, with
296 * data filled in from the given database row.
297 *
298 * @param object $row
299 * @param null|object[] $slotRows
300 *
301 * @return string
302 * @throws FatalError
303 * @throws MWException
304 * @private
305 */
306 function writeRevision( $row, $slotRows = null ) {
307 $rev = $this->getRevisionStore()->newRevisionFromRowAndSlots(
308 $row,
309 $slotRows,
310 0,
311 $this->currentTitle
312 );
313
314 $out = " <revision>\n";
315 $out .= " " . Xml::element( 'id', null, strval( $rev->getId() ) ) . "\n";
316
317 if ( $rev->getParentId() ) {
318 $out .= " " . Xml::element( 'parentid', null, strval( $rev->getParentId() ) ) . "\n";
319 }
320
321 $out .= $this->writeTimestamp( $rev->getTimestamp() );
322
323 if ( $rev->isDeleted( Revision::DELETED_USER ) ) {
324 $out .= " " . Xml::element( 'contributor', [ 'deleted' => 'deleted' ] ) . "\n";
325 } else {
326 // empty values get written out as uid 0, see T224221
327 $user = $rev->getUser();
328 $out .= $this->writeContributor(
329 $user ? $user->getId() : 0,
330 $user ? $user->getName() : ''
331 );
332 }
333
334 if ( $rev->isMinor() ) {
335 $out .= " <minor/>\n";
336 }
337 if ( $rev->isDeleted( Revision::DELETED_COMMENT ) ) {
338 $out .= " " . Xml::element( 'comment', [ 'deleted' => 'deleted' ] ) . "\n";
339 } else {
340 $out .= " "
341 . Xml::elementClean( 'comment', [], strval( $rev->getComment()->text ) )
342 . "\n";
343 }
344
345 $contentMode = $rev->isDeleted( Revision::DELETED_TEXT ) ? self::WRITE_STUB_DELETED
346 : $this->contentMode;
347
348 foreach ( $rev->getSlots()->getSlots() as $slot ) {
349 $out .= $this->writeSlot( $slot, $contentMode );
350 }
351
352 if ( $rev->isDeleted( Revision::DELETED_TEXT ) ) {
353 $out .= " <sha1/>\n";
354 } else {
355 $out .= " " . Xml::element( 'sha1', null, strval( $rev->getSha1() ) ) . "\n";
356 }
357
358 // Avoid PHP 7.1 warning from passing $this by reference
359 $writer = $this;
360 $text = $rev->getContent( SlotRecord::MAIN, RevisionRecord::RAW );
361 Hooks::run( 'XmlDumpWriterWriteRevision', [ &$writer, &$out, $row, $text, $rev ] );
362
363 $out .= " </revision>\n";
364
365 return $out;
366 }
367
368 /**
369 * @param SlotRecord $slot
370 * @param int $contentMode see the WRITE_XXX constants
371 *
372 * @return string
373 */
374 private function writeSlot( SlotRecord $slot, $contentMode ) {
375 $isMain = $slot->getRole() === SlotRecord::MAIN;
376 $isV11 = $this->schemaVersion >= XML_DUMP_SCHEMA_VERSION_11;
377
378 if ( !$isV11 && !$isMain ) {
379 // ignore extra slots
380 return '';
381 }
382
383 $out = '';
384 $indent = ' ';
385
386 if ( !$isMain ) {
387 // non-main slots are wrapped into an additional element.
388 $out .= ' ' . Xml::openElement( 'content' ) . "\n";
389 $indent .= ' ';
390 $out .= $indent . Xml::element( 'role', null, strval( $slot->getRole() ) ) . "\n";
391 }
392
393 if ( $isV11 ) {
394 $out .= $indent . Xml::element( 'origin', null, strval( $slot->getOrigin() ) ) . "\n";
395 }
396
397 $contentModel = $slot->getModel();
398 $contentHandler = ContentHandler::getForModelID( $contentModel );
399 $contentFormat = $contentHandler->getDefaultFormat();
400
401 // XXX: The content format is only relevant when actually outputting serialized content.
402 // It should probably be an attribute on the text tag.
403 $out .= $indent . Xml::element( 'model', null, strval( $contentModel ) ) . "\n";
404 $out .= $indent . Xml::element( 'format', null, strval( $contentFormat ) ) . "\n";
405
406 $textAttributes = [
407 'xml:space' => 'preserve',
408 'bytes' => $slot->getSize(),
409 ];
410
411 if ( $isV11 ) {
412 $textAttributes['sha1'] = $slot->getSha1();
413 }
414
415 if ( $contentMode === self::WRITE_CONTENT ) {
416 try {
417 // write <text> tag
418 $out .= $this->writeText( $slot->getContent(), $textAttributes, $indent );
419 } catch ( SuppressedDataException $ex ) {
420 // NOTE: this shouldn't happen, since the caller is supposed to have checked
421 // for suppressed content!
422 // write <text> placeholder tag
423 $textAttributes['deleted'] = 'deleted';
424 $out .= $indent . Xml::element( 'text', $textAttributes ) . "\n";
425 }
426 catch ( Exception $ex ) {
427 if ( $ex instanceof MWException || $ex instanceof RuntimeException ) {
428 // there's no provision in the schema for an attribute that will let
429 // the user know this element was unavailable due to error; an empty
430 // tag is the best we can do
431 $out .= $indent . Xml::element( 'text' ) . "\n";
432 wfLogWarning(
433 'failed to load content slot ' . $slot->getRole() . ' for revision '
434 . $slot->getRevision() . "\n"
435 );
436 } else {
437 throw $ex;
438 }
439 }
440 } elseif ( $contentMode === self::WRITE_STUB_DELETED ) {
441 // write <text> placeholder tag
442 $textAttributes['deleted'] = 'deleted';
443 $out .= $indent . Xml::element( 'text', $textAttributes ) . "\n";
444 } else {
445 // write <text> stub tag
446 if ( $isV11 ) {
447 $textAttributes['location'] = $slot->getAddress();
448 }
449
450 // Output the numerical text ID if possible, for backwards compatibility.
451 // Note that this is currently the ONLY reason we have a BlobStore here at all.
452 // When removing this line, check whether the BlobStore has become unused.
453 $textId = $this->getBlobStore()->getTextIdFromAddress( $slot->getAddress() );
454 if ( $textId ) {
455 $textAttributes['id'] = $textId;
456 } elseif ( !$isV11 ) {
457 throw new InvalidArgumentException(
458 'Cannot produce stubs for non-text-table content blobs with schema version '
459 . $this->schemaVersion
460 );
461 }
462
463 $out .= $indent . Xml::element( 'text', $textAttributes ) . "\n";
464 }
465
466 if ( !$isMain ) {
467 $out .= ' ' . Xml::closeElement( 'content' ) . "\n";
468 }
469
470 return $out;
471 }
472
473 /**
474 * @param Content $content
475 * @param string[] $textAttributes
476 * @param string $indent
477 *
478 * @return string
479 */
480 private function writeText( Content $content, $textAttributes, $indent ) {
481 $out = '';
482
483 $contentHandler = $content->getContentHandler();
484 $contentFormat = $contentHandler->getDefaultFormat();
485
486 if ( $content instanceof TextContent ) {
487 // HACK: For text based models, bypass the serialization step. This allows extensions (like Flow)
488 // that use incompatible combinations of serialization format and content model.
489 $data = $content->getNativeData();
490 } else {
491 $data = $content->serialize( $contentFormat );
492 }
493
494 $data = $contentHandler->exportTransform( $data, $contentFormat );
495 $textAttributes['bytes'] = $size = strlen( $data ); // make sure to use the actual size
496 $out .= $indent . Xml::elementClean( 'text', $textAttributes, strval( $data ) ) . "\n";
497
498 return $out;
499 }
500
501 /**
502 * Dumps a "<logitem>" section on the output stream, with
503 * data filled in from the given database row.
504 *
505 * @param object $row
506 * @return string
507 * @private
508 */
509 function writeLogItem( $row ) {
510 $out = " <logitem>\n";
511 $out .= " " . Xml::element( 'id', null, strval( $row->log_id ) ) . "\n";
512
513 $out .= $this->writeTimestamp( $row->log_timestamp, " " );
514
515 if ( $row->log_deleted & LogPage::DELETED_USER ) {
516 $out .= " " . Xml::element( 'contributor', [ 'deleted' => 'deleted' ] ) . "\n";
517 } else {
518 $out .= $this->writeContributor( $row->log_user, $row->user_name, " " );
519 }
520
521 if ( $row->log_deleted & LogPage::DELETED_COMMENT ) {
522 $out .= " " . Xml::element( 'comment', [ 'deleted' => 'deleted' ] ) . "\n";
523 } else {
524 $comment = CommentStore::getStore()->getComment( 'log_comment', $row )->text;
525 if ( $comment != '' ) {
526 $out .= " " . Xml::elementClean( 'comment', null, strval( $comment ) ) . "\n";
527 }
528 }
529
530 $out .= " " . Xml::element( 'type', null, strval( $row->log_type ) ) . "\n";
531 $out .= " " . Xml::element( 'action', null, strval( $row->log_action ) ) . "\n";
532
533 if ( $row->log_deleted & LogPage::DELETED_ACTION ) {
534 $out .= " " . Xml::element( 'text', [ 'deleted' => 'deleted' ] ) . "\n";
535 } else {
536 $title = Title::makeTitle( $row->log_namespace, $row->log_title );
537 $out .= " " . Xml::elementClean( 'logtitle', null, self::canonicalTitle( $title ) ) . "\n";
538 $out .= " " . Xml::elementClean( 'params',
539 [ 'xml:space' => 'preserve' ],
540 strval( $row->log_params ) ) . "\n";
541 }
542
543 $out .= " </logitem>\n";
544
545 return $out;
546 }
547
548 /**
549 * @param string $timestamp
550 * @param string $indent Default to six spaces
551 * @return string
552 */
553 function writeTimestamp( $timestamp, $indent = " " ) {
554 $ts = wfTimestamp( TS_ISO_8601, $timestamp );
555 return $indent . Xml::element( 'timestamp', null, $ts ) . "\n";
556 }
557
558 /**
559 * @param int $id
560 * @param string $text
561 * @param string $indent Default to six spaces
562 * @return string
563 */
564 function writeContributor( $id, $text, $indent = " " ) {
565 $out = $indent . "<contributor>\n";
566 if ( $id || !IP::isValid( $text ) ) {
567 $out .= $indent . " " . Xml::elementClean( 'username', null, strval( $text ) ) . "\n";
568 $out .= $indent . " " . Xml::element( 'id', null, strval( $id ) ) . "\n";
569 } else {
570 $out .= $indent . " " . Xml::elementClean( 'ip', null, strval( $text ) ) . "\n";
571 }
572 $out .= $indent . "</contributor>\n";
573 return $out;
574 }
575
576 /**
577 * Warning! This data is potentially inconsistent. :(
578 * @param object $row
579 * @param bool $dumpContents
580 * @return string
581 */
582 function writeUploads( $row, $dumpContents = false ) {
583 if ( $row->page_namespace == NS_FILE ) {
584 $img = MediaWikiServices::getInstance()->getRepoGroup()->getLocalRepo()
585 ->newFile( $row->page_title );
586 if ( $img && $img->exists() ) {
587 $out = '';
588 foreach ( array_reverse( $img->getHistory() ) as $ver ) {
589 $out .= $this->writeUpload( $ver, $dumpContents );
590 }
591 $out .= $this->writeUpload( $img, $dumpContents );
592 return $out;
593 }
594 }
595 return '';
596 }
597
598 /**
599 * @param File $file
600 * @param bool $dumpContents
601 * @return string
602 */
603 function writeUpload( $file, $dumpContents = false ) {
604 if ( $file->isOld() ) {
605 $archiveName = " " .
606 Xml::element( 'archivename', null, $file->getArchiveName() ) . "\n";
607 } else {
608 $archiveName = '';
609 }
610 if ( $dumpContents ) {
611 $be = $file->getRepo()->getBackend();
612 # Dump file as base64
613 # Uses only XML-safe characters, so does not need escaping
614 # @todo Too bad this loads the contents into memory (script might swap)
615 $contents = ' <contents encoding="base64">' .
616 chunk_split( base64_encode(
617 $be->getFileContents( [ 'src' => $file->getPath() ] ) ) ) .
618 " </contents>\n";
619 } else {
620 $contents = '';
621 }
622 if ( $file->isDeleted( File::DELETED_COMMENT ) ) {
623 $comment = Xml::element( 'comment', [ 'deleted' => 'deleted' ] );
624 } else {
625 $comment = Xml::elementClean( 'comment', null, strval( $file->getDescription() ) );
626 }
627 return " <upload>\n" .
628 $this->writeTimestamp( $file->getTimestamp() ) .
629 $this->writeContributor( $file->getUser( 'id' ), $file->getUser( 'text' ) ) .
630 " " . $comment . "\n" .
631 " " . Xml::element( 'filename', null, $file->getName() ) . "\n" .
632 $archiveName .
633 " " . Xml::element( 'src', null, $file->getCanonicalUrl() ) . "\n" .
634 " " . Xml::element( 'size', null, $file->getSize() ) . "\n" .
635 " " . Xml::element( 'sha1base36', null, $file->getSha1() ) . "\n" .
636 " " . Xml::element( 'rel', null, $file->getRel() ) . "\n" .
637 $contents .
638 " </upload>\n";
639 }
640
641 /**
642 * Return prefixed text form of title, but using the content language's
643 * canonical namespace. This skips any special-casing such as gendered
644 * user namespaces -- which while useful, are not yet listed in the
645 * XML "<siteinfo>" data so are unsafe in export.
646 *
647 * @param Title $title
648 * @return string
649 * @since 1.18
650 */
651 public static function canonicalTitle( Title $title ) {
652 if ( $title->isExternal() ) {
653 return $title->getPrefixedText();
654 }
655
656 $prefix = MediaWikiServices::getInstance()->getContentLanguage()->
657 getFormattedNsText( $title->getNamespace() );
658
659 // @todo Emit some kind of warning to the user if $title->getNamespace() !==
660 // NS_MAIN and $prefix === '' (viz. pages in an unregistered namespace)
661
662 if ( $prefix !== '' ) {
663 $prefix .= ':';
664 }
665
666 return $prefix . $title->getText();
667 }
668 }