3 * This program is free software; you can redistribute it and/or modify
4 * it under the terms of the GNU General Public License as published by
5 * the Free Software Foundation; either version 2 of the License, or
6 * (at your option) any later version.
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
13 * You should have received a copy of the GNU General Public License along
14 * with this program; if not, write to the Free Software Foundation, Inc.,
15 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 * http://www.gnu.org/copyleft/gpl.html
21 use MediaWiki\FileRepo\File\FileSelectQueryBuilder
;
22 use MediaWiki\MediaWikiServices
;
23 use MediaWiki\Permissions\Authority
;
24 use MediaWiki\Revision\RevisionRecord
;
25 use MediaWiki\Title\Title
;
26 use MediaWiki\User\UserIdentity
;
27 use Wikimedia\Rdbms\Blob
;
28 use Wikimedia\Rdbms\IReadableDatabase
;
29 use Wikimedia\Rdbms\SelectQueryBuilder
;
32 * Deleted file in the 'filearchive' table.
35 * @ingroup FileAbstraction
39 // Audience options for ::getDescription() and ::getUploader()
40 public const FOR_PUBLIC
= 1;
41 public const FOR_THIS_USER
= 2;
44 /** @var string Metadata serialization: empty string. This is a compact non-legacy format. */
45 private const MDS_EMPTY
= 'empty';
47 /** @var string Metadata serialization: some other string */
48 private const MDS_LEGACY
= 'legacy';
50 /** @var string Metadata serialization: PHP serialize() */
51 private const MDS_PHP
= 'php';
53 /** @var string Metadata serialization: JSON */
54 private const MDS_JSON
= 'json';
56 /** @var int Filearchive row ID */
59 /** @var string|false File name */
62 /** @var string FileStore storage group */
65 /** @var string FileStore SHA-1 key */
68 /** @var int File size in bytes */
71 /** @var int Bitdepth */
80 /** @var array Unserialized metadata */
81 protected $metadataArray = [];
83 /** @var bool Whether or not lazy-loaded data has been loaded from the database */
84 protected $extraDataLoaded = false;
87 * One of the MDS_* constants, giving the format of the metadata as stored
88 * in the DB, or null if the data was not loaded from the DB.
92 protected $metadataSerializationFormat;
94 /** @var string[] Map of metadata item name to blob address */
95 protected $metadataBlobs = [];
98 * Map of metadata item name to blob address for items that exist but
99 * have not yet been loaded into $this->metadataArray
103 protected $unloadedMetadataBlobs = [];
105 /** @var string MIME type */
108 /** @var string Media type */
111 /** @var string Upload description */
112 private $description;
114 /** @var UserIdentity|null Uploader */
117 /** @var string|null Time of upload */
120 /** @var bool Whether or not all this has been loaded from the database (loadFromXxx) */
123 /** @var int Bitfield akin to rev_deleted */
126 /** @var string SHA-1 hash of file content */
129 /** @var int|false Number of pages of a multipage document, or false for
130 * documents which aren't multipage documents
134 /** @var string Original base filename */
135 private $archive_name;
137 /** @var MediaHandler */
140 /** @var Title|null */
141 protected $title; # image title
146 /** @var LocalRepo */
149 /** @var MetadataStorageHelper */
150 private $metadataStorageHelper;
154 * @param Title|null $title
157 * @param string $sha1
159 public function __construct( $title, $id = 0, $key = '', $sha1 = '' ) {
163 $this->group
= 'deleted'; // needed for direct use of constructor
169 $this->mime
= "unknown/unknown";
170 $this->media_type
= '';
171 $this->description
= '';
173 $this->timestamp
= null;
175 $this->dataLoaded
= false;
176 $this->exists
= false;
179 if ( $title instanceof Title
) {
180 $this->title
= File
::normalizeTitle( $title, 'exception' );
181 $this->name
= $title->getDBkey();
196 if ( !$id && !$key && !( $title instanceof Title
) && !$sha1 ) {
197 throw new BadMethodCallException( "No specifications provided to ArchivedFile constructor." );
200 $this->repo
= MediaWikiServices
::getInstance()->getRepoGroup()->getLocalRepo();
201 $this->metadataStorageHelper
= new MetadataStorageHelper( $this->repo
);
205 * Loads a file object from the filearchive table
206 * @stable to override
207 * @return bool|null True on success or null
209 public function load() {
210 if ( $this->dataLoaded
) {
215 if ( $this->id
> 0 ) {
216 $conds['fa_id'] = $this->id
;
219 $conds['fa_storage_group'] = $this->group
;
220 $conds['fa_storage_key'] = $this->key
;
222 if ( $this->title
) {
223 $conds['fa_name'] = $this->title
->getDBkey();
226 $conds['fa_sha1'] = $this->sha1
;
229 if ( $conds === [] ) {
230 throw new RuntimeException( "No specific information for retrieving archived file" );
233 if ( !$this->title ||
$this->title
->getNamespace() === NS_FILE
) {
234 $this->dataLoaded
= true; // set it here, to have also true on miss
235 $dbr = $this->repo
->getReplicaDB();
236 $queryBuilder = FileSelectQueryBuilder
::newForArchivedFile( $dbr );
237 $row = $queryBuilder->where( $conds )
238 ->orderBy( 'fa_timestamp', SelectQueryBuilder
::SORT_DESC
)
239 ->caller( __METHOD__
)->fetchRow();
241 // this revision does not exist?
245 // initialize fields for filestore image object
246 $this->loadFromRow( $row );
248 throw new UnexpectedValueException( 'This title does not correspond to an image page.' );
255 * Loads a file object from the filearchive table
256 * @stable to override
258 * @param stdClass $row
259 * @return ArchivedFile
261 public static function newFromRow( $row ) {
262 $file = new ArchivedFile( Title
::makeTitle( NS_FILE
, $row->fa_name
) );
263 $file->loadFromRow( $row );
269 * Return the tables, fields, and join conditions to be selected to create
270 * a new archivedfile object.
272 * Since 1.34, fa_user and fa_user_text have not been present in the
273 * database, but they continue to be available in query results as an
277 * @stable to override
278 * @deprecated since 1.41 use FileSelectQueryBuilder instead
279 * @return array[] With three keys:
280 * - tables: (string[]) to include in the `$table` to `IDatabase->select()` or `SelectQueryBuilder::tables`
281 * - fields: (string[]) to include in the `$vars` to `IDatabase->select()` or `SelectQueryBuilder::fields`
282 * - joins: (array) to include in the `$join_conds` to `IDatabase->select()` or `SelectQueryBuilder::joinConds`
283 * @phan-return array{tables:string[],fields:string[],joins:array}
285 public static function getQueryInfo() {
286 $dbr = MediaWikiServices
::getInstance()->getConnectionProvider()->getReplicaDatabase();
287 $queryInfo = ( FileSelectQueryBuilder
::newForArchivedFile( $dbr ) )->getQueryInfo();
289 'tables' => $queryInfo['tables'],
290 'fields' => $queryInfo['fields'],
291 'joins' => $queryInfo['join_conds'],
296 * Load ArchivedFile object fields from a DB row.
297 * @stable to override
299 * @param stdClass $row Object database row
302 public function loadFromRow( $row ) {
303 $this->id
= intval( $row->fa_id
);
304 $this->name
= $row->fa_name
;
305 $this->archive_name
= $row->fa_archive_name
;
306 $this->group
= $row->fa_storage_group
;
307 $this->key
= $row->fa_storage_key
;
308 $this->size
= $row->fa_size
;
309 $this->bits
= $row->fa_bits
;
310 $this->width
= $row->fa_width
;
311 $this->height
= $row->fa_height
;
312 $this->loadMetadataFromDbFieldValue(
313 $this->repo
->getReplicaDB(), $row->fa_metadata
);
314 $this->mime
= "$row->fa_major_mime/$row->fa_minor_mime";
315 $this->media_type
= $row->fa_media_type
;
316 $services = MediaWikiServices
::getInstance();
317 $this->description
= $services->getCommentStore()
318 // Legacy because $row may have come from self::selectFields()
319 ->getCommentLegacy( $this->repo
->getReplicaDB(), 'fa_description', $row )->text
;
320 $this->user
= $services->getUserFactory()
321 ->newFromAnyId( $row->fa_user
, $row->fa_user_text
, $row->fa_actor
);
322 $this->timestamp
= $row->fa_timestamp
;
323 $this->deleted
= $row->fa_deleted
;
324 if ( isset( $row->fa_sha1
) ) {
325 $this->sha1
= $row->fa_sha1
;
327 // old row, populate from key
328 $this->sha1
= LocalRepo
::getHashFromKey( $this->key
);
330 if ( !$this->title
) {
331 $this->title
= Title
::makeTitleSafe( NS_FILE
, $row->fa_name
);
333 $this->exists
= $row->fa_archive_name
!== '';
337 * Return the associated title object
341 public function getTitle() {
342 if ( !$this->title
) {
349 * Return the file name
353 public function getName() {
354 if ( $this->name
=== false ) {
364 public function getID() {
373 public function exists() {
376 return $this->exists
;
380 * Return the FileStore key
383 public function getKey() {
390 * Return the FileStore key (overriding base File class)
393 public function getStorageKey() {
394 return $this->getKey();
398 * Return the FileStore storage group
401 public function getGroup() {
406 * Return the width of the image
409 public function getWidth() {
416 * Return the height of the image
419 public function getHeight() {
422 return $this->height
;
426 * Get handler-specific metadata as a serialized string
428 * @deprecated since 1.37 use getMetadataArray() or getMetadataItem()
431 public function getMetadata() {
432 $data = $this->getMetadataArray();
435 } elseif ( array_keys( $data ) === [ '_error' ] ) {
436 // Legacy error encoding
437 return $data['_error'];
439 return serialize( $this->getMetadataArray() );
444 * Get unserialized handler-specific metadata
449 public function getMetadataArray(): array {
451 if ( $this->unloadedMetadataBlobs
) {
452 return $this->getMetadataItems(
453 array_unique( array_merge(
454 array_keys( $this->metadataArray
),
455 array_keys( $this->unloadedMetadataBlobs
)
459 return $this->metadataArray
;
462 public function getMetadataItems( array $itemNames ): array {
466 foreach ( $itemNames as $itemName ) {
467 if ( array_key_exists( $itemName, $this->metadataArray
) ) {
468 $result[$itemName] = $this->metadataArray
[$itemName];
469 } elseif ( isset( $this->unloadedMetadataBlobs
[$itemName] ) ) {
470 $addresses[$itemName] = $this->unloadedMetadataBlobs
[$itemName];
475 $resultFromBlob = $this->metadataStorageHelper
->getMetadataFromBlobStore( $addresses );
476 foreach ( $addresses as $itemName => $address ) {
477 unset( $this->unloadedMetadataBlobs
[$itemName] );
478 $value = $resultFromBlob[$itemName] ??
null;
479 if ( $value !== null ) {
480 $result[$itemName] = $value;
481 $this->metadataArray
[$itemName] = $value;
489 * Serialize the metadata array for insertion into img_metadata, oi_metadata
492 * If metadata splitting is enabled, this may write blobs to the database,
493 * returning their addresses.
496 * @param IReadableDatabase $db
497 * @return string|Blob
499 public function getMetadataForDb( IReadableDatabase
$db ) {
501 if ( !$this->metadataArray
&& !$this->metadataBlobs
) {
503 } elseif ( $this->repo
->isJsonMetadataEnabled() ) {
504 $s = $this->getJsonMetadata();
506 $s = serialize( $this->getMetadataArray() );
508 if ( !is_string( $s ) ) {
509 throw new RuntimeException( 'Could not serialize image metadata value for DB' );
511 return $db->encodeBlob( $s );
515 * Get metadata in JSON format ready for DB insertion, optionally splitting
516 * items out to BlobStore.
520 private function getJsonMetadata() {
521 // Directly store data that is not already in BlobStore
523 'data' => array_diff_key( $this->metadataArray
, $this->metadataBlobs
)
526 // Also store the blob addresses
527 if ( $this->metadataBlobs
) {
528 $envelope['blobs'] = $this->metadataBlobs
;
531 [ $s, $blobAddresses ] = $this->metadataStorageHelper
->getJsonMetadata( $this, $envelope );
533 // Repeated calls to this function should not keep inserting more blobs
534 $this->metadataBlobs +
= $blobAddresses;
540 * Unserialize a metadata blob which came from the database and store it
544 * @param IReadableDatabase $db
545 * @param string|Blob $metadataBlob
547 protected function loadMetadataFromDbFieldValue( IReadableDatabase
$db, $metadataBlob ) {
548 $this->loadMetadataFromString( $db->decodeBlob( $metadataBlob ) );
552 * Unserialize a metadata string which came from some non-DB source, or is
553 * the return value of IReadableDatabase::decodeBlob().
556 * @param string $metadataString
558 protected function loadMetadataFromString( $metadataString ) {
559 $this->extraDataLoaded
= true;
560 $this->metadataArray
= [];
561 $this->metadataBlobs
= [];
562 $this->unloadedMetadataBlobs
= [];
563 $metadataString = (string)$metadataString;
564 if ( $metadataString === '' ) {
565 $this->metadataSerializationFormat
= self
::MDS_EMPTY
;
568 if ( $metadataString[0] === '{' ) {
569 $envelope = $this->metadataStorageHelper
->jsonDecode( $metadataString );
571 // Legacy error encoding
572 $this->metadataArray
= [ '_error' => $metadataString ];
573 $this->metadataSerializationFormat
= self
::MDS_LEGACY
;
575 $this->metadataSerializationFormat
= self
::MDS_JSON
;
576 if ( isset( $envelope['data'] ) ) {
577 $this->metadataArray
= $envelope['data'];
579 if ( isset( $envelope['blobs'] ) ) {
580 $this->metadataBlobs
= $this->unloadedMetadataBlobs
= $envelope['blobs'];
584 // phpcs:ignore Generic.PHP.NoSilencedErrors.Discouraged
585 $data = @unserialize
( $metadataString );
586 if ( !is_array( $data ) ) {
587 // Legacy error encoding
588 $data = [ '_error' => $metadataString ];
589 $this->metadataSerializationFormat
= self
::MDS_LEGACY
;
591 $this->metadataSerializationFormat
= self
::MDS_PHP
;
593 $this->metadataArray
= $data;
598 * Return the size of the image file, in bytes
601 public function getSize() {
610 public function getBits() {
617 * Returns the MIME type of the file.
620 public function getMimeType() {
627 * Get a MediaHandler instance for this file
628 * @return MediaHandler
630 private function getHandler() {
631 if ( !$this->handler
) {
632 $this->handler
= MediaHandler
::getHandler( $this->getMimeType() );
635 return $this->handler
;
639 * Returns the number of pages of a multipage document, or false for
640 * documents which aren't multipage documents
641 * @stable to override
644 public function pageCount() {
645 if ( $this->pageCount
=== null ) {
646 // @FIXME: callers expect File objects
647 // @phan-suppress-next-line PhanTypeMismatchArgument
648 if ( $this->getHandler() && $this->handler
->isMultiPage( $this ) ) {
649 // @phan-suppress-next-line PhanTypeMismatchArgument
650 $this->pageCount
= $this->handler
->pageCount( $this );
652 $this->pageCount
= false;
656 return $this->pageCount
;
660 * Return the type of the media in the file.
661 * Use the value returned by this function with the MEDIATYPE_xxx constants.
664 public function getMediaType() {
667 return $this->media_type
;
671 * Return upload timestamp.
675 public function getTimestamp() {
678 return wfTimestamp( TS_MW
, $this->timestamp
);
682 * Get the SHA-1 base 36 hash of the file
687 public function getSha1() {
695 * @stable to override
696 * @param int $audience One of:
697 * File::FOR_PUBLIC to be displayed to all users
698 * File::FOR_THIS_USER to be displayed to the given user
699 * File::RAW get the description regardless of permissions
700 * @param Authority|null $performer to check for, only if FOR_THIS_USER is
701 * passed to the $audience parameter
702 * @return UserIdentity|null
704 public function getUploader( int $audience = self
::FOR_PUBLIC
, ?Authority
$performer = null ): ?UserIdentity
{
706 if ( $audience === self
::FOR_PUBLIC
&& $this->isDeleted( File
::DELETED_USER
) ) {
708 } elseif ( $audience === self
::FOR_THIS_USER
&& !$this->userCan( File
::DELETED_USER
, $performer ) ) {
716 * Return upload description.
718 * @since 1.37 the method takes $audience and $performer parameters.
719 * @param int $audience One of:
720 * File::FOR_PUBLIC to be displayed to all users
721 * File::FOR_THIS_USER to be displayed to the given user
722 * File::RAW get the description regardless of permissions
723 * @param Authority|null $performer to check for, only if FOR_THIS_USER is
724 * passed to the $audience parameter
727 public function getDescription( int $audience = self
::FOR_PUBLIC
, ?Authority
$performer = null ): string {
729 if ( $audience === self
::FOR_PUBLIC
&& $this->isDeleted( File
::DELETED_COMMENT
) ) {
731 } elseif ( $audience === self
::FOR_THIS_USER
&& !$this->userCan( File
::DELETED_COMMENT
, $performer ) ) {
734 return $this->description
;
739 * Returns the deletion bitfield
742 public function getVisibility() {
745 return $this->deleted
;
749 * for file or revision rows
751 * @param int $field One of DELETED_* bitfield constants
754 public function isDeleted( $field ) {
757 return ( $this->deleted
& $field ) == $field;
761 * Determine if the current user is allowed to view a particular
762 * field of this FileStore image file, if it's marked as deleted.
764 * @param Authority $performer
767 public function userCan( $field, Authority
$performer ) {
769 $title = $this->getTitle();
771 return RevisionRecord
::userCanBitfield(