61adc1e146
The un-namespaced \ScopedCallback is deprecated. Change-Id: Ie014d5a775ead66335a24acac9d339915884d1a4
138 lines
4 KiB
PHP
138 lines
4 KiB
PHP
<?php
|
|
/**
|
|
* Assemble the segments of a chunked upload.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
* http://www.gnu.org/copyleft/gpl.html
|
|
*
|
|
* @file
|
|
* @ingroup Upload
|
|
*/
|
|
use Wikimedia\ScopedCallback;
|
|
|
|
/**
|
|
* Assemble the segments of a chunked upload.
|
|
*
|
|
* @ingroup Upload
|
|
*/
|
|
class AssembleUploadChunksJob extends Job {
|
|
public function __construct( Title $title, array $params ) {
|
|
parent::__construct( 'AssembleUploadChunks', $title, $params );
|
|
$this->removeDuplicates = true;
|
|
}
|
|
|
|
public function run() {
|
|
$scope = RequestContext::importScopedSession( $this->params['session'] );
|
|
$this->addTeardownCallback( function () use ( &$scope ) {
|
|
ScopedCallback::consume( $scope ); // T126450
|
|
} );
|
|
|
|
$context = RequestContext::getMain();
|
|
$user = $context->getUser();
|
|
try {
|
|
if ( !$user->isLoggedIn() ) {
|
|
$this->setLastError( "Could not load the author user from session." );
|
|
|
|
return false;
|
|
}
|
|
|
|
UploadBase::setSessionStatus(
|
|
$user,
|
|
$this->params['filekey'],
|
|
[ 'result' => 'Poll', 'stage' => 'assembling', 'status' => Status::newGood() ]
|
|
);
|
|
|
|
$upload = new UploadFromChunks( $user );
|
|
$upload->continueChunks(
|
|
$this->params['filename'],
|
|
$this->params['filekey'],
|
|
new WebRequestUpload( $context->getRequest(), 'null' )
|
|
);
|
|
|
|
// Combine all of the chunks into a local file and upload that to a new stash file
|
|
$status = $upload->concatenateChunks();
|
|
if ( !$status->isGood() ) {
|
|
UploadBase::setSessionStatus(
|
|
$user,
|
|
$this->params['filekey'],
|
|
[ 'result' => 'Failure', 'stage' => 'assembling', 'status' => $status ]
|
|
);
|
|
$this->setLastError( $status->getWikiText( false, false, 'en' ) );
|
|
|
|
return false;
|
|
}
|
|
|
|
// We can only get warnings like 'duplicate' after concatenating the chunks
|
|
$status = Status::newGood();
|
|
$status->value = [ 'warnings' => $upload->checkWarnings() ];
|
|
|
|
// We have a new filekey for the fully concatenated file
|
|
$newFileKey = $upload->getStashFile()->getFileKey();
|
|
|
|
// Remove the old stash file row and first chunk file
|
|
$upload->stash->removeFileNoAuth( $this->params['filekey'] );
|
|
|
|
// Build the image info array while we have the local reference handy
|
|
$apiMain = new ApiMain(); // dummy object (XXX)
|
|
$imageInfo = $upload->getImageInfo( $apiMain->getResult() );
|
|
|
|
// Cleanup any temporary local file
|
|
$upload->cleanupTempFile();
|
|
|
|
// Cache the info so the user doesn't have to wait forever to get the final info
|
|
UploadBase::setSessionStatus(
|
|
$user,
|
|
$this->params['filekey'],
|
|
[
|
|
'result' => 'Success',
|
|
'stage' => 'assembling',
|
|
'filekey' => $newFileKey,
|
|
'imageinfo' => $imageInfo,
|
|
'status' => $status
|
|
]
|
|
);
|
|
} catch ( Exception $e ) {
|
|
UploadBase::setSessionStatus(
|
|
$user,
|
|
$this->params['filekey'],
|
|
[
|
|
'result' => 'Failure',
|
|
'stage' => 'assembling',
|
|
'status' => Status::newFatal( 'api-error-stashfailed' )
|
|
]
|
|
);
|
|
$this->setLastError( get_class( $e ) . ": " . $e->getMessage() );
|
|
// To be extra robust.
|
|
MWExceptionHandler::rollbackMasterChangesAndLog( $e );
|
|
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
public function getDeduplicationInfo() {
|
|
$info = parent::getDeduplicationInfo();
|
|
if ( is_array( $info['params'] ) ) {
|
|
$info['params'] = [ 'filekey' => $info['params']['filekey'] ];
|
|
}
|
|
|
|
return $info;
|
|
}
|
|
|
|
public function allowRetries() {
|
|
return false;
|
|
}
|
|
}
|