UploadFromStash.php

Go to the documentation of this file.

<?php
use MediaWiki\MediaWikiServices;
use MediaWiki\Request\WebRequest;
use MediaWiki\User\UserIdentity;
 
class UploadFromStash extends UploadBase {
    protected $mFileKey;
    protected $mVirtualTempPath;
    protected $mSourceType;
 
    private $stash;
 
    private $repo;
 
    public function __construct( ?UserIdentity $user = null, $stash = false, $repo = false ) {
        if ( $repo ) {
            $this->repo = $repo;
        } else {
            $this->repo = MediaWikiServices::getInstance()->getRepoGroup()->getLocalRepo();
        }
 
        if ( $stash ) {
            $this->stash = $stash;
        } else {
            if ( $user ) {
                wfDebug( __METHOD__ . " creating new UploadStash instance for " . $user->getId() );
            } else {
                wfDebug( __METHOD__ . " creating new UploadStash instance with no user" );
            }
 
            $this->stash = new UploadStash( $this->repo, $user );
        }
    }
 
    public static function isValidKey( $key ) {
        // this is checked in more detail in UploadStash
        return (bool)preg_match( UploadStash::KEY_FORMAT_REGEX, $key );
    }
 
    public static function isValidRequest( $request ) {
        // this passes wpSessionKey to getText() as a default when wpFileKey isn't set.
        // wpSessionKey has no default which guarantees failure if both are missing
        // (though that should have been caught earlier)
        return self::isValidKey( $request->getText( 'wpFileKey', $request->getText( 'wpSessionKey' ) ) );
    }
 
    public function initialize( $key, $name = 'upload_file', $initTempFile = true ) {
        $metadata = $this->stash->getMetadata( $key );
        $this->initializePathInfo( $name,
            $initTempFile ? $this->getRealPath( $metadata['us_path'] ) : false,
            $metadata['us_size'],
            false
        );
 
        $this->mFileKey = $key;
        $this->mVirtualTempPath = $metadata['us_path'];
        $this->mFileProps = $this->stash->getFileProps( $key );
        $this->mSourceType = $metadata['us_source_type'];
    }
 
    public function initializeFromRequest( &$request ) {
        // sends wpSessionKey as a default when wpFileKey is missing
        $fileKey = $request->getText( 'wpFileKey', $request->getText( 'wpSessionKey' ) );
 
        // chooses one of wpDestFile, wpUploadFile, filename in that order.
        $desiredDestName = $request->getText(
            'wpDestFile',
            $request->getText( 'wpUploadFile', $request->getText( 'filename' ) )
        );
 
        $this->initialize( $fileKey, $desiredDestName );
    }
 
    public function getSourceType() {
        return $this->mSourceType;
    }
 
    public function getTempFileSha1Base36() {
        // phan doesn't like us accessing this directly since in
        // parent class this can be null, however we always set this in
        // this class so it is safe. Add a check to keep phan happy.
        if ( !is_array( $this->mFileProps ) ) {
            throw new LogicException( "mFileProps should never be null" );
        } else {
            return $this->mFileProps['sha1'];
        }
    }
 
    public function unsaveUploadedFile() {
        return $this->stash->removeFile( $this->mFileKey );
    }
 
    public function postProcessUpload() {
        parent::postProcessUpload();
        $this->unsaveUploadedFile();
    }
}