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() {
                # 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
                $status = $this->upload->fetchFile();
                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];
        }
}