UploadFromUrlJob.php

Go to the documentation of this file.

<?php
class UploadFromUrlJob extends Job {
    const SESSION_KEYNAME = 'wsUploadFromUrlJobData';

    public $upload;

    protected $user;

    public function __construct( $title, $params, $id = 0 ) {
        parent::__construct( 'uploadFromUrl', $title, $params, $id );
    }

    public function run() {
        global $wgCopyUploadAsyncTimeout;
        # Initialize this object and the upload object
        $this->upload = new UploadFromUrl();
        $this->upload->initialize(
            $this->title->getText(),
            $this->params['url'],
            false
        );
        $this->user = User::newFromName( $this->params['userName'] );

        # Fetch the file
        $opts = array();
        if ( $wgCopyUploadAsyncTimeout ) {
            $opts['timeout'] = $wgCopyUploadAsyncTimeout;
        }
        $status = $this->upload->fetchFile( $opts );
        if ( !$status->isOk() ) {
            $this->leaveMessage( $status );
            return true;
        }

        # Verify upload
        $result = $this->upload->verifyUpload();
        if ( $result['status'] != UploadBase::OK ) {
            $status = $this->upload->convertVerifyErrorToStatus( $result );
            $this->leaveMessage( $status );
            return true;
        }

        # Check warnings
        if ( !$this->params['ignoreWarnings'] ) {
            $warnings = $this->upload->checkWarnings();
            if ( $warnings ) {

                # Stash the upload
                $key = $this->upload->stashFile();

                if ( $this->params['leaveMessage'] ) {
                    $this->user->leaveUserMessage(
                        wfMessage( 'upload-warning-subj' )->text(),
                        wfMessage( 'upload-warning-msg',
                            $key,
                            $this->params['url'] )->text()
                    );
                } else {
                    wfSetupSession( $this->params['sessionId'] );
                    $this->storeResultInSession( 'Warning',
                        'warnings', $warnings );
                    session_write_close();
                }

                return true;
            }
        }

        # Perform the upload
        $status = $this->upload->performUpload(
            $this->params['comment'],
            $this->params['pageText'],
            $this->params['watch'],
            $this->user
        );
        $this->leaveMessage( $status );
        return true;

    }

    protected function leaveMessage( $status ) {
        if ( $this->params['leaveMessage'] ) {
            if ( $status->isGood() ) {
                $this->user->leaveUserMessage( wfMessage( 'upload-success-subj' )->text(),
                    wfMessage( 'upload-success-msg',
                        $this->upload->getTitle()->getText(),
                        $this->params['url']
                    )->text() );
            } else {
                $this->user->leaveUserMessage( wfMessage( 'upload-failure-subj' )->text(),
                    wfMessage( 'upload-failure-msg',
                        $status->getWikiText(),
                        $this->params['url']
                    )->text() );
            }
        } else {
            wfSetupSession( $this->params['sessionId'] );
            if ( $status->isOk() ) {
                $this->storeResultInSession( 'Success',
                    'filename', $this->upload->getLocalFile()->getName() );
            } else {
                $this->storeResultInSession( 'Failure',
                    'errors', $status->getErrorsArray() );
            }
            session_write_close();
        }
    }

    protected function storeResultInSession( $result, $dataKey, $dataValue ) {
        $session =& self::getSessionData( $this->params['sessionKey'] );
        $session['result'] = $result;
        $session[$dataKey] = $dataValue;
    }

    public function initializeSessionData() {
        $session =& self::getSessionData( $this->params['sessionKey'] );
        $$session['result'] = 'Queued';
    }

    public static function &getSessionData( $key ) {
        if ( !isset( $_SESSION[self::SESSION_KEYNAME][$key] ) ) {
            $_SESSION[self::SESSION_KEYNAME][$key] = array();
        }
        return $_SESSION[self::SESSION_KEYNAME][$key];
    }
}