AssembleUploadChunksJob.php

Go to the documentation of this file.

<?php
class AssembleUploadChunksJob extends Job {
        public function __construct( $title, $params, $id = 0 ) {
                parent::__construct( 'AssembleUploadChunks', $title, $params, $id );
                $this->removeDuplicates = true;
        }

        public function run() {
                $scope = RequestContext::importScopedSession( $this->params['session'] );
                $context = RequestContext::getMain();
                try {
                        $user = $context->getUser();
                        if ( !$user->isLoggedIn() ) {
                                $this->setLastError( "Could not load the author user from session." );
                                return false;
                        }

                        UploadBase::setSessionStatus(
                                $this->params['filekey'],
                                array( 'result' => 'Poll', 'stage' => 'assembling', 'status' => Status::newGood() )
                        );

                        $upload = new UploadFromChunks( $user );
                        $upload->continueChunks(
                                $this->params['filename'],
                                $this->params['filekey'],
                                $context->getRequest()
                        );

                        // 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(
                                        $this->params['filekey'],
                                        array( 'result' => 'Failure', 'stage' => 'assembling', 'status' => $status )
                                );
                                $this->setLastError( $status->getWikiText() );
                                return false;
                        }

                        // We have a new filekey for the fully concatenated file
                        $newFileKey = $upload->getLocalFile()->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(
                                $this->params['filekey'],
                                array(
                                        'result'    => 'Success',
                                        'stage'     => 'assembling',
                                        'filekey'   => $newFileKey,
                                        'imageinfo' => $imageInfo,
                                        'status'    => Status::newGood()
                                )
                        );
                } catch ( MWException $e ) {
                        UploadBase::setSessionStatus(
                                $this->params['filekey'],
                                array(
                                        'result' => 'Failure',
                                        'stage'  => 'assembling',
                                        'status' => Status::newFatal( 'api-error-stashfailed' )
                                )
                        );
                        $this->setLastError( get_class( $e ) . ": " . $e->getText() );
                        return false;
                }
                return true;
        }

        public function getDeduplicationInfo() {
                $info = parent::getDeduplicationInfo();
                if ( is_array( $info['params'] ) ) {
                        $info['params'] = array( 'filekey' => $info['params']['filekey'] );
                }
                return $info;
        }

        public function allowRetries() {
                return false;
        }
}