Woops, fix bug with last commit when the number of rows <= 2.
[lhc/web/wiklou.git] / includes / SpecialUpload.php
index 46d61c8..9ef7bd4 100644 (file)
@@ -19,6 +19,21 @@ function wfSpecialUpload() {
  * @addtogroup SpecialPage
  */
 class UploadForm {
+       const SUCCESS = 0;
+       const BEFORE_PROCESSING = 1;
+       const LARGE_FILE_SERVER = 2;
+       const EMPTY_FILE = 3;
+       const MIN_LENGHT_PARTNAME = 4;
+       const ILLEGAL_FILENAME = 5;
+       const PROTECTED_PAGE = 6;
+       const OVERWRITE_EXISTING_FILE = 7;
+       const FILETYPE_MISSING = 8;
+       const FILETYPE_BADTYPE = 9;
+       const VERIFICATION_ERROR = 10;
+       const UPLOAD_VERIFICATION_ERROR = 11;
+       const UPLOAD_WARNING = 12;
+       const INTERNAL_ERROR = 13;
+
        /**#@+
         * @access private
         */
@@ -253,34 +268,114 @@ class UploadForm {
                $this->cleanupTempFile();
        }
 
-       /* -------------------------------------------------------------- */
+       /**
+        * Do the upload
+        * Checks are made in SpecialUpload::execute()
+        *
+        * @access private
+        */
+       function processUpload(){
+               global $wgUser, $wgOut, $wgFileExtensions;
+               $details = null;
+               $value = null;
+               $value = $this->internalProcessUpload( $details );
+               
+               switch($value) {
+                       case self::SUCCESS:
+                               $wgOut->redirect( $this->mLocalFile->getTitle()->getFullURL() );
+                               break;
+
+                       case self::BEFORE_PROCESSING:
+                               break;
+
+                       case self::LARGE_FILE_SERVER:
+                               $this->mainUploadForm( wfMsgHtml( 'largefileserver' ) );
+                               break;
+
+                       case self::EMPTY_FILE:
+                               $this->mainUploadForm( wfMsgHtml( 'emptyfile' ) );
+                               break;
+
+                       case self::MIN_LENGHT_PARTNAME:
+                               $this->mainUploadForm( wfMsgHtml( 'minlength1' ) );
+                               break;
+
+                       case self::ILLEGAL_FILENAME:
+                               $filtered = $details['filtered'];
+                               $this->uploadError( wfMsgWikiHtml( 'illegalfilename', htmlspecialchars( $filtered ) ) );
+                               break;
+
+                       case self::PROTECTED_PAGE:
+                               $this->uploadError( wfMsgWikiHtml( 'protectedpage' ) );
+                               break;
+
+                       case self::OVERWRITE_EXISTING_FILE:
+                               $errorText = $details['overwrite'];
+                               $overwrite = new WikiError( $wgOut->parse( $errorText ) );
+                               $this->uploadError( $overwrite->toString() );
+                               break;
+
+                       case self::FILETYPE_MISSING:
+                               $this->uploadError( wfMsgExt( 'filetype-missing', array ( 'parseinline' ) ) );
+                               break;
+
+                       case self::FILETYPE_BADTYPE:
+                               $finalExt = $details['finalExt'];
+                               $this->uploadError( wfMsgExt( 'filetype-badtype', array ( 'parseinline' ), htmlspecialchars( $finalExt ), implode ( ', ', $wgFileExtensions ) ) );
+                               break;
+
+                       case self::VERIFICATION_ERROR:
+                               $veri = $details['veri'];
+                               $this->uploadError( $veri->toString() );
+                               break;
+
+                       case self::UPLOAD_VERIFICATION_ERROR:
+                               $error = $details['error'];
+                               $this->uploadError( $error );
+                               break;
+
+                       case self::UPLOAD_WARNING:
+                               $warning = $details['warning'];
+                               $this->uploadWarning( $warning );
+                               break;
+
+                       case self::INTERNAL_ERROR:
+                               $internal = $details['internal'];
+                               $this->showError( $internal );
+                               break;
+
+                       default:
+                               throw new MWException( __METHOD__ . ": Unknown value `{$value}`" );
+               }
+       }
 
        /**
         * Really do the upload
         * Checks are made in SpecialUpload::execute()
+        *
+        * @param array $resultDetails contains result-specific dict of additional values
+        *
         * @access private
         */
-       function processUpload() {
-               global $wgUser, $wgOut;
+       function internalProcessUpload( &$resultDetails ) {
+               global $wgUser;
 
                if( !wfRunHooks( 'UploadForm:BeforeProcessing', array( &$this ) ) )
                {
                        wfDebug( "Hook 'UploadForm:BeforeProcessing' broke processing the file." );
-                       return false;
+                       return self::BEFORE_PROCESSING;
                }
 
                /* Check for PHP error if any, requires php 4.2 or newer */
                if( $this->mCurlError == 1/*UPLOAD_ERR_INI_SIZE*/ ) {
-                       $this->mainUploadForm( wfMsgHtml( 'largefileserver' ) );
-                       return;
+                       return self::LARGE_FILE_SERVER;
                }
 
                /**
                 * If there was no filename or a zero size given, give up quick.
                 */
                if( trim( $this->mSrcName ) == '' || empty( $this->mFileSize ) ) {
-                       $this->mainUploadForm( wfMsgHtml( 'emptyfile' ) );
-                       return;
+                       return self::EMPTY_FILE;
                }
 
                # Chop off any directories in the given filename
@@ -311,8 +406,7 @@ class UploadForm {
                }
 
                if( strlen( $partname ) < 1 ) {
-                       $this->mainUploadForm( wfMsgHtml( 'minlength1' ) );
-                       return;
+                       return self::MIN_LENGHT_PARTNAME;
                }
 
                /**
@@ -322,8 +416,8 @@ class UploadForm {
                $filtered = preg_replace ( "/[^".Title::legalChars()."]|:/", '-', $filtered );
                $nt = Title::makeTitleSafe( NS_IMAGE, $filtered );
                if( is_null( $nt ) ) {
-                       $this->uploadError( wfMsgWikiHtml( 'illegalfilename', htmlspecialchars( $filtered ) ) );
-                       return;
+                       $resultDetails = array( 'filtered' => $filtered );
+                       return self::ILLEGAL_FILENAME;
                }
                $this->mLocalFile = wfLocalFile( $nt );
                $this->mDestName = $this->mLocalFile->getName();
@@ -333,26 +427,27 @@ class UploadForm {
                 * to modify it by uploading a new revision.
                 */
                if( !$nt->userCan( 'edit' ) ) {
-                       return $this->uploadError( wfMsgWikiHtml( 'protectedpage' ) );
+                       return self::PROTECTED_PAGE;
                }
 
                /**
                 * In some cases we may forbid overwriting of existing files.
                 */
                $overwrite = $this->checkOverwrite( $this->mDestName );
-               if( WikiError::isError( $overwrite ) ) {
-                       return $this->uploadError( $overwrite->toString() );
+               if( $overwrite !== true ) {
+                       $resultDetails = array( 'overwrite' => $overwrite );
+                       return self::OVERWRITE_EXISTING_FILE;
                }
 
                /* Don't allow users to override the blacklist (check file extension) */
                global $wgStrictFileExtensions;
                global $wgFileExtensions, $wgFileBlacklist;
                if ($finalExt == '') {
-                       return $this->uploadError( wfMsgExt( 'filetype-missing', array ( 'parseinline' ) ) );
+                       return self::FILETYPE_MISSING;
                } elseif ( $this->checkFileExtensionList( $ext, $wgFileBlacklist ) ||
                                ($wgStrictFileExtensions && !$this->checkFileExtension( $finalExt, $wgFileExtensions ) ) ) {
-                       return $this->uploadError( wfMsgExt( 'filetype-badtype', array ( 'parseinline' ), 
-                               htmlspecialchars( $finalExt ), implode ( ', ', $wgFileExtensions ) ) );
+                       $resultDetails = array( 'finalExt' => $finalExt );
+                       return self::FILETYPE_BADTYPE;
                }
 
                /**
@@ -366,7 +461,8 @@ class UploadForm {
                        $veri = $this->verify( $this->mTempPath, $finalExt );
 
                        if( $veri !== true ) { //it's a wiki error...
-                               return $this->uploadError( $veri->toString() );
+                               $resultDetails = array( 'veri' => $veri );
+                               return self::VERIFICATION_ERROR;
                        }
 
                        /**
@@ -375,7 +471,8 @@ class UploadForm {
                        $error = '';
                        if( !wfRunHooks( 'UploadVerification',
                                        array( $this->mDestName, $this->mTempPath, &$error ) ) ) {
-                               return $this->uploadError( $error );
+                               $resultDetails = array( 'error' => $error );
+                               return self::UPLOAD_VERIFICATION_ERROR;
                        }
                }
 
@@ -397,7 +494,7 @@ class UploadForm {
                        global $wgCheckFileExtensions;
                        if ( $wgCheckFileExtensions ) {
                                if ( ! $this->checkFileExtension( $finalExt, $wgFileExtensions ) ) {
-                                       $warning .= '<li>'.wfMsgExt( 'filetype-badtype', array ( 'parseinline' ), 
+                                       $warning .= '<li>'.wfMsgExt( 'filetype-badtype', array ( 'parseinline' ),
                                                htmlspecialchars( $finalExt ), implode ( ', ', $wgFileExtensions ) ).'</li>';
                                }
                        }
@@ -421,7 +518,8 @@ class UploadForm {
                                 * Stash the file in a temporary location; the user can choose
                                 * to let it through and we'll complete the upload then.
                                 */
-                               return $this->uploadWarning( $warning );
+                               $resultDetails = array( 'warning' => $warning );
+                               return self::UPLOAD_WARNING;
                        }
                }
 
@@ -432,19 +530,20 @@ class UploadForm {
                $pageText = self::getInitialPageText( $this->mComment, $this->mLicense,
                        $this->mCopyrightStatus, $this->mCopyrightSource );
 
-               $status = $this->mLocalFile->upload( $this->mTempPath, $this->mComment, $pageText, 
+               $status = $this->mLocalFile->upload( $this->mTempPath, $this->mComment, $pageText,
                        File::DELETE_SOURCE, $this->mFileProps );
                if ( !$status->isGood() ) {
-                       $this->showError( $status->getWikiText() );
+                       $resultDetails = array( 'internal' => $status->getWikiText() );
+                       return self::INTERNAL_ERROR;
                } else {
                        if ( $this->mWatchthis ) {
                                global $wgUser;
                                $wgUser->addWatch( $this->mLocalFile->getTitle() );
                        }
                        // Success, redirect to description page
-                       $wgOut->redirect( $this->mLocalFile->getTitle()->getFullURL() );
                        $img = null; // @todo: added to avoid passing a ref to null - should this be defined somewhere?
                        wfRunHooks( 'UploadComplete', array( &$img ) );
+                       return self::SUCCESS;
                }
        }
 
@@ -459,7 +558,6 @@ class UploadForm {
                // Check for uppercase extension. We allow these filenames but check if an image
                // with lowercase extension exists already
                $warning = '';
-               wfDebugLog( 'borko', 'wtf' );
                
                if( strpos( $file->getName(), '.' ) == false ) {
                        $partname = $file->getName();
@@ -467,7 +565,6 @@ class UploadForm {
                } else {
                        list( $partname, $rawExtension ) = explode( '.', $file->getName(), 2 );
                }
-               wfDebugLog( 'borko', sprintf( "%s - %s - %s", $partname, $rawExtension, $file->getExtension() ) );
                $sk = $wgUser->getSkin();
 
                if ( $rawExtension != $file->getExtension() ) {
@@ -478,14 +575,8 @@ class UploadForm {
                        // Check for another file using the normalized form...
                        $nt_lc = Title::newFromText( $partname . '.' . $file->getExtension() );
                        $file_lc = wfLocalFile( $nt_lc );
-                       if( $file_lc ) {
-                               wfDebugLog( 'borko', 'whee: ' . $file_lc->getName() );
-                       } else {
-                               wfDebugLog( 'borko', 'no lc match');
-                       }
                } else {
                        $file_lc = false;
-                       wfDebugLog( 'borko', 'extensions ok');
                }
 
                if( $file->exists() ) {
@@ -504,7 +595,6 @@ class UploadForm {
                        $warning .= '<li>' . wfMsgExt( 'fileexists', 'parseline', $dlink ) . '</li>' . $dlink2;
 
                } elseif ( $file_lc && $file_lc->exists() ) {
-                       wfDebugLog( 'borko', 'whee: ' . $file_lc->getName() );
                        # Check if image with lowercase extension exists.
                        # It's not forbidden but in 99% it makes no sense to upload the same filename with uppercase extension
                        $dlink = $sk->makeKnownLinkObj( $nt_lc );
@@ -551,9 +641,19 @@ class UploadForm {
                                        substr( $partname , 0, strpos( $partname , '-' ) +1 ) ) . '</li>';
                        }
                }
-               if ( $file->wasDeleted() ) {
+
+               $filenamePrefixBlacklist = self::getFilenamePrefixBlacklist();
+               # Do the match
+               foreach( $filenamePrefixBlacklist as $prefix ) {
+                       if ( substr( $partname, 0, strlen( $prefix ) ) == $prefix ) {
+                               $warning .= '<li>' . wfMsgExt( 'filename-bad-prefix', 'parseinline', $prefix ) . '</li>';
+                               break;
+                       }
+               }
+
+               if ( $file->wasDeleted() && !$file->exists() ) {
                        # If the file existed before and was deleted, warn the user of this
-                       # Don't bother doing so if the image exists now, however
+                       # Don't bother doing so if the file exists now, however
                        $ltitle = SpecialPage::getTitleFor( 'Log' );
                        $llink = $sk->makeKnownLinkObj( $ltitle, wfMsgHtml( 'deletionlog' ), 
                                'type=delete&page=' . $file->getTitle()->getPrefixedUrl() );
@@ -562,6 +662,12 @@ class UploadForm {
                return $warning;
        }
 
+       /**
+        * Get a list of warnings
+        *
+        * @param string local filename, e.g. 'file exists', 'non-descriptive filename'
+        * @return array list of warning messages
+        */
        static function ajaxGetExistsWarning( $filename ) {
                $file = wfFindFile( $filename );
                if( !$file ) {
@@ -598,6 +704,33 @@ class UploadForm {
                return $output->getText();
        }
 
+       /**
+        * Get a list of blacklisted filename prefixes from [[MediaWiki:filename-prefix-blacklist]]
+        *
+        * @return array list of prefixes
+        */
+       public static function getFilenamePrefixBlacklist() {
+               $blacklist = array();
+               $message = wfMsgForContent( 'filename-prefix-blacklist' );
+               if( $message && !( wfEmptyMsg( 'filename-prefix-blacklist', $message ) || $message == '-' ) ) {
+                       $lines = explode( "\n", $message );
+                       foreach( $lines as $line ) {
+                               // Remove comment lines
+                               $comment = substr( trim( $line ), 0, 1 );
+                               if ( $comment == '#' || $comment == '' ) {
+                                       continue;
+                               }
+                               // Remove additional comments after a prefix
+                               $comment = strpos( $line, '#' );
+                               if ( $comment > 0 ) {
+                                       $line = substr( $line, 0, $comment-1 );
+                               }
+                               $blacklist[] = trim( $line );
+                       }
+               }
+               return $blacklist;
+       }
+
        /**
         * Stash a file in a temporary directory for later processing
         * after the user has confirmed it.
@@ -761,10 +894,10 @@ class UploadForm {
 
                $useAjaxDestCheck = $wgUseAjax && $wgAjaxUploadDestCheck;
                $useAjaxLicensePreview = $wgUseAjax && $wgAjaxLicensePreview;
-               
+
                $adc = wfBoolToStr( $useAjaxDestCheck );
                $alp = wfBoolToStr( $useAjaxLicensePreview );
-               
+
                $wgOut->addScript( "<script type=\"text/javascript\">
 wgAjaxUploadDestCheck = {$adc};
 wgAjaxLicensePreview = {$alp};
@@ -777,20 +910,26 @@ wgAjaxLicensePreview = {$alp};
                        wfDebug( "Hook 'UploadForm:initial' broke output of the upload form" );
                        return false;
                }
-               
-               if( $this->mDesiredDestName && $wgUser->isAllowed( 'deletedhistory' ) ) {
+
+               if( $this->mDesiredDestName ) {
                        $title = Title::makeTitleSafe( NS_IMAGE, $this->mDesiredDestName );
-                       if( $title instanceof Title && ( $count = $title->isDeleted() ) > 0 ) {
+                       // Show a subtitle link to deleted revisions (to sysops et al only)
+                       if( $title instanceof Title && ( $count = $title->isDeleted() ) > 0 && $wgUser->isAllowed( 'deletedhistory' ) ) {
                                $link = wfMsgExt(
                                        $wgUser->isAllowed( 'delete' ) ? 'thisisdeleted' : 'viewdeleted',
                                        array( 'parse', 'replaceafter' ),
                                        $wgUser->getSkin()->makeKnownLinkObj(
                                                SpecialPage::getTitleFor( 'Undelete', $title->getPrefixedText() ),
-                                               wfMsgHtml( 'restorelink', $count )
+                                               wfMsgExt( 'restorelink', array( 'parsemag', 'escape' ), $count )
                                        )
                                );
                                $wgOut->addHtml( "<div id=\"contentSub2\">{$link}</div>" );
-                       }                               
+                       }
+
+                       // Show the relevant lines from deletion log (for still deleted files only)
+                       if( $title instanceof Title && $title->isDeleted() > 0 && !$title->exists() ) {
+                               $this->showDeletionLog( $wgOut, $title->getPrefixedText() );
+                       }
                }
 
                $cols = intval($wgUser->getOption( 'cols' ));
@@ -1362,7 +1501,7 @@ EOT
 
                if( $error ) {
                        $errorText = wfMsg( $error, wfEscapeWikiText( $img->getName() ) );
-                       return new WikiError( $wgOut->parse( $errorText ) );
+                       return $errorText;
                }
 
                // Rockin', go ahead and upload
@@ -1429,4 +1568,29 @@ EOT
                }
                return $pageText;
        }
+
+       /**
+        * If there are rows in the deletion log for this file, show them,
+        * along with a nice little note for the user
+        *
+        * @param OutputPage $out
+        * @param string filename
+        */
+       private function showDeletionLog( $out, $filename ) {
+               $reader = new LogReader(
+                       new FauxRequest(
+                               array(
+                                       'page' => $filename,
+                                       'type' => 'delete',
+                                       )
+                       )
+               );
+               if( $reader->hasRows() ) {
+                       $out->addHtml( '<div id="mw-upload-deleted-warn">' );
+                       $out->addWikiText( wfMsg( 'upload-wasdeleted' ) );
+                       $viewer = new LogViewer( $reader );
+                       $viewer->showList( $out );
+                       $out->addHtml( '</div>' );
+               }
+       }
 }