PublishStashedFileJob.php

Go to the documentation of this file.

<?php
class PublishStashedFileJob extends Job {
        public function __construct( $title, $params, $id = 0 ) {
                parent::__construct( 'PublishStashedFile', $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' => 'publish', 'status' => Status::newGood() )
                        );

                        $upload = new UploadFromStash( $user );
                        // @TODO: initialize() causes a GET, ideally we could frontload the antivirus
                        // checks and anything else to the stash stage (which includes concatenation and
                        // the local file is thus already there). That way, instead of GET+PUT, there could
                        // just be a COPY operation from the stash to the public zone.
                        $upload->initialize( $this->params['filekey'], $this->params['filename'] );

                        // Check if the local file checks out (this is generally a no-op)
                        $verification = $upload->verifyUpload();
                        if ( $verification['status'] !== UploadBase::OK ) {
                                $status = Status::newFatal( 'verification-error' );
                                $status->value = array( 'verification' => $verification );
                                UploadBase::setSessionStatus(
                                        $this->params['filekey'],
                                        array( 'result' => 'Failure', 'stage' => 'publish', 'status' => $status )
                                );
                                $this->setLastError( "Could not verify upload." );
                                return false;
                        }

                        // Upload the stashed file to a permanent location
                        $status = $upload->performUpload(
                                $this->params['comment'],
                                $this->params['text'],
                                $this->params['watch'],
                                $user
                        );
                        if ( !$status->isGood() ) {
                                UploadBase::setSessionStatus(
                                        $this->params['filekey'],
                                        array( 'result' => 'Failure', 'stage' => 'publish', 'status' => $status )
                                );
                                $this->setLastError( $status->getWikiText() );
                                return false;
                        }

                        // 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'     => 'publish',
                                        'filename'  => $upload->getLocalFile()->getName(),
                                        'imageinfo' => $imageInfo,
                                        'status'    => Status::newGood()
                                )
                        );
                } catch ( MWException $e ) {
                        UploadBase::setSessionStatus(
                                $this->params['filekey'],
                                array(
                                        'result' => 'Failure',
                                        'stage'  => 'publish',
                                        'status' => Status::newFatal( 'api-error-publishfailed' )
                                )
                        );
                        $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;
        }
}