ArchivedFile.php

Go to the documentation of this file.

<?php
class ArchivedFile {
    private $id;
 
    private $name;
 
    private $group;
 
    private $key;
 
    private $size;
 
    private $bits;
 
    private $width;
 
    private $height;
 
    private $metadata;
 
    private $mime;
 
    private $media_type;
 
    private $description;
 
    private $user;
 
    private $timestamp;
 
    private $dataLoaded;
 
    private $deleted;
 
    private $sha1;
 
    private $pageCount;
 
    private $archive_name;
 
    protected $handler;
 
    protected $title; # image title
 
    function __construct( $title, $id = 0, $key = '', $sha1 = '' ) {
        $this->id = -1;
        $this->title = false;
        $this->name = false;
        $this->group = 'deleted'; // needed for direct use of constructor
        $this->key = '';
        $this->size = 0;
        $this->bits = 0;
        $this->width = 0;
        $this->height = 0;
        $this->metadata = '';
        $this->mime = "unknown/unknown";
        $this->media_type = '';
        $this->description = '';
        $this->user = null;
        $this->timestamp = null;
        $this->deleted = 0;
        $this->dataLoaded = false;
        $this->exists = false;
        $this->sha1 = '';
 
        if ( $title instanceof Title ) {
            $this->title = File::normalizeTitle( $title, 'exception' );
            $this->name = $title->getDBkey();
        }
 
        if ( $id ) {
            $this->id = $id;
        }
 
        if ( $key ) {
            $this->key = $key;
        }
 
        if ( $sha1 ) {
            $this->sha1 = $sha1;
        }
 
        if ( !$id && !$key && !( $title instanceof Title ) && !$sha1 ) {
            throw new MWException( "No specifications provided to ArchivedFile constructor." );
        }
    }
 
    public function load() {
        if ( $this->dataLoaded ) {
            return true;
        }
        $conds = [];
 
        if ( $this->id > 0 ) {
            $conds['fa_id'] = $this->id;
        }
        if ( $this->key ) {
            $conds['fa_storage_group'] = $this->group;
            $conds['fa_storage_key'] = $this->key;
        }
        if ( $this->title ) {
            $conds['fa_name'] = $this->title->getDBkey();
        }
        if ( $this->sha1 ) {
            $conds['fa_sha1'] = $this->sha1;
        }
 
        if ( !count( $conds ) ) {
            throw new MWException( "No specific information for retrieving archived file" );
        }
 
        if ( !$this->title || $this->title->getNamespace() == NS_FILE ) {
            $this->dataLoaded = true; // set it here, to have also true on miss
            $dbr = wfGetDB( DB_REPLICA );
            $fileQuery = self::getQueryInfo();
            $row = $dbr->selectRow(
                $fileQuery['tables'],
                $fileQuery['fields'],
                $conds,
                __METHOD__,
                [ 'ORDER BY' => 'fa_timestamp DESC' ],
                $fileQuery['joins']
            );
            if ( !$row ) {
                // this revision does not exist?
                return null;
            }
 
            // initialize fields for filestore image object
            $this->loadFromRow( $row );
        } else {
            throw new MWException( 'This title does not correspond to an image page.' );
        }
        $this->exists = true;
 
        return true;
    }
 
    public static function newFromRow( $row ) {
        $file = new ArchivedFile( Title::makeTitle( NS_FILE, $row->fa_name ) );
        $file->loadFromRow( $row );
 
        return $file;
    }
 
    static function selectFields() {
        global $wgActorTableSchemaMigrationStage;
 
        if ( $wgActorTableSchemaMigrationStage > MIGRATION_WRITE_BOTH ) {
            // If code is using this instead of self::getQueryInfo(), there's a
            // decent chance it's going to try to directly access
            // $row->fa_user or $row->fa_user_text and we can't give it
            // useful values here once those aren't being written anymore.
            throw new BadMethodCallException(
                'Cannot use ' . __METHOD__ . ' when $wgActorTableSchemaMigrationStage > MIGRATION_WRITE_BOTH'
            );
        }
 
        wfDeprecated( __METHOD__, '1.31' );
        return [
            'fa_id',
            'fa_name',
            'fa_archive_name',
            'fa_storage_key',
            'fa_storage_group',
            'fa_size',
            'fa_bits',
            'fa_width',
            'fa_height',
            'fa_metadata',
            'fa_media_type',
            'fa_major_mime',
            'fa_minor_mime',
            'fa_user',
            'fa_user_text',
            'fa_actor' => $wgActorTableSchemaMigrationStage > MIGRATION_OLD ? 'fa_actor' : null,
            'fa_timestamp',
            'fa_deleted',
            'fa_deleted_timestamp', /* Used by LocalFileRestoreBatch */
            'fa_sha1',
        ] + CommentStore::getStore()->getFields( 'fa_description' );
    }
 
    public static function getQueryInfo() {
        $commentQuery = CommentStore::getStore()->getJoin( 'fa_description' );
        $actorQuery = ActorMigration::newMigration()->getJoin( 'fa_user' );
        return [
            'tables' => [ 'filearchive' ] + $commentQuery['tables'] + $actorQuery['tables'],
            'fields' => [
                'fa_id',
                'fa_name',
                'fa_archive_name',
                'fa_storage_key',
                'fa_storage_group',
                'fa_size',
                'fa_bits',
                'fa_width',
                'fa_height',
                'fa_metadata',
                'fa_media_type',
                'fa_major_mime',
                'fa_minor_mime',
                'fa_timestamp',
                'fa_deleted',
                'fa_deleted_timestamp', /* Used by LocalFileRestoreBatch */
                'fa_sha1',
            ] + $commentQuery['fields'] + $actorQuery['fields'],
            'joins' => $commentQuery['joins'] + $actorQuery['joins'],
        ];
    }
 
    public function loadFromRow( $row ) {
        $this->id = intval( $row->fa_id );
        $this->name = $row->fa_name;
        $this->archive_name = $row->fa_archive_name;
        $this->group = $row->fa_storage_group;
        $this->key = $row->fa_storage_key;
        $this->size = $row->fa_size;
        $this->bits = $row->fa_bits;
        $this->width = $row->fa_width;
        $this->height = $row->fa_height;
        $this->metadata = $row->fa_metadata;
        $this->mime = "$row->fa_major_mime/$row->fa_minor_mime";
        $this->media_type = $row->fa_media_type;
        $this->description = CommentStore::getStore()
            // Legacy because $row may have come from self::selectFields()
            ->getCommentLegacy( wfGetDB( DB_REPLICA ), 'fa_description', $row )->text;
        $this->user = User::newFromAnyId( $row->fa_user, $row->fa_user_text, $row->fa_actor );
        $this->timestamp = $row->fa_timestamp;
        $this->deleted = $row->fa_deleted;
        if ( isset( $row->fa_sha1 ) ) {
            $this->sha1 = $row->fa_sha1;
        } else {
            // old row, populate from key
            $this->sha1 = LocalRepo::getHashFromKey( $this->key );
        }
        if ( !$this->title ) {
            $this->title = Title::makeTitleSafe( NS_FILE, $row->fa_name );
        }
    }
 
    public function getTitle() {
        if ( !$this->title ) {
            $this->load();
        }
        return $this->title;
    }
 
    public function getName() {
        if ( $this->name === false ) {
            $this->load();
        }
 
        return $this->name;
    }
 
    public function getID() {
        $this->load();
 
        return $this->id;
    }
 
    public function exists() {
        $this->load();
 
        return $this->exists;
    }
 
    public function getKey() {
        $this->load();
 
        return $this->key;
    }
 
    public function getStorageKey() {
        return $this->getKey();
    }
 
    public function getGroup() {
        return $this->group;
    }
 
    public function getWidth() {
        $this->load();
 
        return $this->width;
    }
 
    public function getHeight() {
        $this->load();
 
        return $this->height;
    }
 
    public function getMetadata() {
        $this->load();
 
        return $this->metadata;
    }
 
    public function getSize() {
        $this->load();
 
        return $this->size;
    }
 
    public function getBits() {
        $this->load();
 
        return $this->bits;
    }
 
    public function getMimeType() {
        $this->load();
 
        return $this->mime;
    }
 
    function getHandler() {
        if ( !isset( $this->handler ) ) {
            $this->handler = MediaHandler::getHandler( $this->getMimeType() );
        }
 
        return $this->handler;
    }
 
    function pageCount() {
        if ( !isset( $this->pageCount ) ) {
            // @FIXME: callers expect File objects
            if ( $this->getHandler() && $this->handler->isMultiPage( $this ) ) {
                $this->pageCount = $this->handler->pageCount( $this );
            } else {
                $this->pageCount = false;
            }
        }
 
        return $this->pageCount;
    }
 
    public function getMediaType() {
        $this->load();
 
        return $this->media_type;
    }
 
    public function getTimestamp() {
        $this->load();
 
        return wfTimestamp( TS_MW, $this->timestamp );
    }
 
    function getSha1() {
        $this->load();
 
        return $this->sha1;
    }
 
    public function getUser( $type = 'text' ) {
        $this->load();
 
        if ( $type === 'object' ) {
            return $this->user;
        } elseif ( $type === 'text' ) {
            return $this->user ? $this->user->getName() : '';
        } elseif ( $type === 'id' ) {
            return $this->user ? $this->user->getId() : 0;
        }
 
        throw new MWException( "Unknown type '$type'." );
    }
 
    public function getDescription() {
        $this->load();
        if ( $this->isDeleted( File::DELETED_COMMENT ) ) {
            return 0;
        } else {
            return $this->description;
        }
    }
 
    public function getRawUser() {
        return $this->getUser( 'id' );
    }
 
    public function getRawUserText() {
        return $this->getUser( 'text' );
    }
 
    public function getRawDescription() {
        $this->load();
 
        return $this->description;
    }
 
    public function getVisibility() {
        $this->load();
 
        return $this->deleted;
    }
 
    public function isDeleted( $field ) {
        $this->load();
 
        return ( $this->deleted & $field ) == $field;
    }
 
    public function userCan( $field, User $user = null ) {
        $this->load();
 
        $title = $this->getTitle();
        return Revision::userCanBitfield( $this->deleted, $field, $user, $title ?: null );
    }
}