]> scripts.mit.edu Git - autoinstallsdev/mediawiki.git/blobdiff - includes/filerepo/FileRepo.php
MediaWiki 1.15.1
[autoinstallsdev/mediawiki.git] / includes / filerepo / FileRepo.php
index cf6d65c2cec04bb682de7688f18c0885ce5a9a21..c9d34377c33266cf35b4e911bc0f1653a6fbe056 100644 (file)
@@ -3,30 +3,36 @@
 /**
  * Base class for file repositories
  * Do not instantiate, use a derived class.
+ * @ingroup FileRepo
  */
 abstract class FileRepo {
        const DELETE_SOURCE = 1;
+       const FIND_PRIVATE = 1;
+       const FIND_IGNORE_REDIRECT = 2;
        const OVERWRITE = 2;
        const OVERWRITE_SAME = 4;
 
        var $thumbScriptUrl, $transformVia404;
        var $descBaseUrl, $scriptDirUrl, $articleUrl, $fetchDescription, $initialCapital;
        var $pathDisclosureProtection = 'paranoid';
+       var $descriptionCacheExpiry, $apiThumbCacheExpiry, $hashLevels;
 
-       /** 
+       /**
         * Factory functions for creating new files
         * Override these in the base class
         */
        var $fileFactory = false, $oldFileFactory = false;
+       var $fileFactoryKey = false, $oldFileFactoryKey = false;
 
        function __construct( $info ) {
                // Required settings
                $this->name = $info['name'];
-               
+
                // Optional settings
                $this->initialCapital = true; // by default
-               foreach ( array( 'descBaseUrl', 'scriptDirUrl', 'articleUrl', 'fetchDescription', 
-                       'thumbScriptUrl', 'initialCapital', 'pathDisclosureProtection' ) as $var ) 
+               foreach ( array( 'descBaseUrl', 'scriptDirUrl', 'articleUrl', 'fetchDescription',
+                       'thumbScriptUrl', 'initialCapital', 'pathDisclosureProtection', 
+                       'descriptionCacheExpiry', 'apiThumbCacheExpiry', 'hashLevels' ) as $var )
                {
                        if ( isset( $info[$var] ) ) {
                                $this->$var = $info[$var];
@@ -45,15 +51,15 @@ abstract class FileRepo {
        /**
         * Create a new File object from the local repository
         * @param mixed $title Title object or string
-        * @param mixed $time Time at which the image is supposed to have existed. 
-        *                    If this is specified, the returned object will be an 
+        * @param mixed $time Time at which the image was uploaded.
+        *                    If this is specified, the returned object will be an
         *                    instance of the repository's old file class instead of
-        *                    a current file. Repositories not supporting version 
+        *                    a current file. Repositories not supporting version
         *                    control should return false if this parameter is set.
         */
        function newFile( $title, $time = false ) {
                if ( !($title instanceof Title) ) {
-                       $title = Title::makeTitleSafe( NS_IMAGE, $title );
+                       $title = Title::makeTitleSafe( NS_FILE, $title );
                        if ( !is_object( $title ) ) {
                                return null;
                        }
@@ -70,26 +76,123 @@ abstract class FileRepo {
        }
 
        /**
-        * Find an instance of the named file that existed at the specified time
-        * Returns false if the file did not exist. Repositories not supporting 
+        * Find an instance of the named file created at the specified time
+        * Returns false if the file does not exist. Repositories not supporting
         * version control should return false if the time is specified.
         *
+        * @param mixed $title Title object or string
         * @param mixed $time 14-character timestamp, or false for the current version
         */
-       function findFile( $title, $time = false ) {
+       function findFile( $title, $time = false, $flags = 0 ) {
+               if ( !($title instanceof Title) ) {
+                       $title = Title::makeTitleSafe( NS_FILE, $title );
+                       if ( !is_object( $title ) ) {
+                               return false;
+                       }
+               }
                # First try the current version of the file to see if it precedes the timestamp
                $img = $this->newFile( $title );
                if ( !$img ) {
                        return false;
                }
-               if ( $img->exists() && ( !$time || $img->getTimestamp() <= $time ) ) {
+               if ( $img->exists() && ( !$time || $img->getTimestamp() == $time ) ) {
                        return $img;
                }
                # Now try an old version of the file
-               $img = $this->newFile( $title, $time );
-               if ( $img->exists() ) {
+               if ( $time !== false ) {
+                       $img = $this->newFile( $title, $time );
+                       if ( $img && $img->exists() ) {
+                               if ( !$img->isDeleted(File::DELETED_FILE) ) {
+                                       return $img;
+                               } else if ( ($flags & FileRepo::FIND_PRIVATE) && $img->userCan(File::DELETED_FILE) ) {
+                                       return $img;
+                               }
+                       }
+               }
+                               
+               # Now try redirects
+               if ( $flags & FileRepo::FIND_IGNORE_REDIRECT ) {
+                       return false;
+               }
+               $redir = $this->checkRedirect( $title );                
+               if( $redir && $redir->getNamespace() == NS_FILE) {
+                       $img = $this->newFile( $redir );
+                       if( !$img ) {
+                               return false;
+                       }
+                       if( $img->exists() ) {
+                               $img->redirectedFrom( $title->getDBkey() );
+                               return $img;
+                       }
+               }
+               return false;
+       }
+       
+       /*
+        * Find many files at once. 
+        * @param array $titles, an array of titles
+        * @todo Think of a good way to optionally pass timestamps to this function.
+        */
+       function findFiles( $titles ) {
+               $result = array();
+               foreach ( $titles as $index => $title ) {
+                       $file = $this->findFile( $title );
+                       if ( $file )
+                               $result[$file->getTitle()->getDBkey()] = $file;
+               }
+               return $result;
+       }
+       
+       /**
+        * Create a new File object from the local repository
+        * @param mixed $sha1 SHA-1 key
+        * @param mixed $time Time at which the image was uploaded.
+        *                    If this is specified, the returned object will be an
+        *                    instance of the repository's old file class instead of
+        *                    a current file. Repositories not supporting version
+        *                    control should return false if this parameter is set.
+        */
+       function newFileFromKey( $sha1, $time = false ) {
+               if ( $time ) {
+                       if ( $this->oldFileFactoryKey ) {
+                               return call_user_func( $this->oldFileFactoryKey, $sha1, $this, $time );
+                       } else {
+                               return false;
+                       }
+               } else {
+                       return call_user_func( $this->fileFactoryKey, $sha1, $this );
+               }
+       }
+       
+       /**
+        * Find an instance of the file with this key, created at the specified time
+        * Returns false if the file does not exist. Repositories not supporting
+        * version control should return false if the time is specified.
+        *
+        * @param string $sha1 string
+        * @param mixed $time 14-character timestamp, or false for the current version
+        */
+       function findFileFromKey( $sha1, $time = false, $flags = 0 ) {
+               # First try the current version of the file to see if it precedes the timestamp
+               $img = $this->newFileFromKey( $sha1 );
+               if ( !$img ) {
+                       return false;
+               }
+               if ( $img->exists() && ( !$time || $img->getTimestamp() == $time ) ) {
                        return $img;
                }
+               # Now try an old version of the file
+               if ( $time !== false ) {
+                       $img = $this->newFileFromKey( $sha1, $time );
+                       if ( $img->exists() ) {
+                               if ( !$img->isDeleted(File::DELETED_FILE) ) {
+                                       return $img;
+                               } else if ( ($flags & FileRepo::FIND_PRIVATE) && $img->userCan(File::DELETED_FILE) ) {
+                                       return $img;
+                               }
+                       }
+               }
+               return false;
        }
 
        /**
@@ -135,6 +238,14 @@ abstract class FileRepo {
                        return $path;
                }
        }
+       
+       /**
+        * Get a relative path including trailing slash, e.g. f/fa/
+        * If the repo is not hashed, returns an empty string
+        */
+       function getHashPath( $name ) {
+               return self::getHashPathForLevel( $name, $this->hashLevels );
+       }
 
        /**
         * Get the name of this repository, as specified by $info['name]' to the constructor
@@ -143,58 +254,62 @@ abstract class FileRepo {
                return $this->name;
        }
 
-       /**
-        * Get the file description page base URL, or false if there isn't one.
-        * @private
-        */
-       function getDescBaseUrl() {
-               if ( is_null( $this->descBaseUrl ) ) {
-                       if ( !is_null( $this->articleUrl ) ) {
-                               $this->descBaseUrl = str_replace( '$1', 
-                                       wfUrlencode( Namespace::getCanonicalName( NS_IMAGE ) ) . ':', $this->articleUrl );
-                       } elseif ( !is_null( $this->scriptDirUrl ) ) {
-                               $this->descBaseUrl = $this->scriptDirUrl . '/index.php?title=' . 
-                                       wfUrlencode( Namespace::getCanonicalName( NS_IMAGE ) ) . ':';
-                       } else {
-                               $this->descBaseUrl = false;
-                       }
-               }
-               return $this->descBaseUrl;
-       }
-
        /**
         * Get the URL of an image description page. May return false if it is
-        * unknown or not applicable. In general this should only be called by the 
-        * File class, since it may return invalid results for certain kinds of 
+        * unknown or not applicable. In general this should only be called by the
+        * File class, since it may return invalid results for certain kinds of
         * repositories. Use File::getDescriptionUrl() in user code.
         *
         * In particular, it uses the article paths as specified to the repository
         * constructor, whereas local repositories use the local Title functions.
         */
        function getDescriptionUrl( $name ) {
-               $base = $this->getDescBaseUrl();
-               if ( $base ) {
-                       return $base . wfUrlencode( $name );
-               } else {
-                       return false;
+               $encName = wfUrlencode( $name );
+               if ( !is_null( $this->descBaseUrl ) ) {
+                       # "http://example.com/wiki/Image:"
+                       return $this->descBaseUrl . $encName;
                }
+               if ( !is_null( $this->articleUrl ) ) {
+                       # "http://example.com/wiki/$1"
+                       #
+                       # We use "Image:" as the canonical namespace for
+                       # compatibility across all MediaWiki versions.
+                       return str_replace( '$1',
+                               "Image:$encName", $this->articleUrl );
+               }
+               if ( !is_null( $this->scriptDirUrl ) ) {
+                       # "http://example.com/w"
+                       #
+                       # We use "Image:" as the canonical namespace for
+                       # compatibility across all MediaWiki versions,
+                       # and just sort of hope index.php is right. ;)
+                       return $this->scriptDirUrl .
+                               "/index.php?title=Image:$encName";
+               }
+               return false;
        }
 
        /**
-        * Get the URL of the content-only fragment of the description page. For 
-        * MediaWiki this means action=render. This should only be called by the 
-        * repository's file class, since it may return invalid results. User code 
+        * Get the URL of the content-only fragment of the description page. For
+        * MediaWiki this means action=render. This should only be called by the
+        * repository's file class, since it may return invalid results. User code
         * should use File::getDescriptionText().
+        * @param string $name Name of image to fetch
+        * @param string $lang Language to fetch it in, if any.
         */
-       function getDescriptionRenderUrl( $name ) {
+       function getDescriptionRenderUrl( $name, $lang = null ) {
+               $query = 'action=render';
+               if ( !is_null( $lang ) ) {
+                       $query .= '&uselang=' . $lang;
+               }
                if ( isset( $this->scriptDirUrl ) ) {
-                       return $this->scriptDirUrl . '/index.php?title=' . 
-                               wfUrlencode( Namespace::getCanonicalName( NS_IMAGE ) . ':' . $name ) .
-                               '&action=render';
+                       return $this->scriptDirUrl . '/index.php?title=' .
+                               wfUrlencode( 'Image:' . $name ) .
+                               "&$query";
                } else {
-                       $descBase = $this->getDescBaseUrl();
-                       if ( $descBase ) {
-                               return wfAppendQuery( $descBase . wfUrlencode( $name ), 'action=render' );
+                       $descUrl = $this->getDescriptionUrl( $name );
+                       if ( $descUrl ) {
+                               return wfAppendQuery( $descUrl, $query );
                        } else {
                                return false;
                        }
@@ -210,7 +325,7 @@ abstract class FileRepo {
         * @param integer $flags Bitwise combination of the following flags:
         *     self::DELETE_SOURCE     Delete the source file after upload
         *     self::OVERWRITE         Overwrite an existing destination file instead of failing
-        *     self::OVERWRITE_SAME    Overwrite the file if the destination exists and has the 
+        *     self::OVERWRITE_SAME    Overwrite the file if the destination exists and has the
         *                             same contents as the source
         * @return FileRepoStatus
         */
@@ -234,7 +349,7 @@ abstract class FileRepo {
         * Pick a random name in the temp zone and store a file to it.
         * Returns a FileRepoStatus object with the URL in the value.
         *
-        * @param string $originalName The base name of the file as specified 
+        * @param string $originalName The base name of the file as specified
         *     by the user. The file extension will be maintained.
         * @param string $srcPath The current location of the file.
         */
@@ -255,7 +370,7 @@ abstract class FileRepo {
         * virtual URL, into this repository at the specified destination location.
         *
         * Returns a FileRepoStatus object. On success, the value contains "new" or
-        * "archived", to indicate whether the file was new with that name. 
+        * "archived", to indicate whether the file was new with that name.
         *
         * @param string $srcPath The source path or URL
         * @param string $dstRel The destination relative path
@@ -288,16 +403,16 @@ abstract class FileRepo {
        /**
         * Move a group of files to the deletion archive.
         *
-        * If no valid deletion archive is configured, this may either delete the 
+        * If no valid deletion archive is configured, this may either delete the
         * file or throw an exception, depending on the preference of the repository.
         *
         * The overwrite policy is determined by the repository -- currently FSRepo
-        * assumes a naming scheme in the deleted zone based on content hash, as 
+        * assumes a naming scheme in the deleted zone based on content hash, as
         * opposed to the public zone which is assumed to be unique.
         *
-        * @param array $sourceDestPairs Array of source/destination pairs. Each element 
+        * @param array $sourceDestPairs Array of source/destination pairs. Each element
         *        is a two-element array containing the source file path relative to the
-        *        public root in the first element, and the archive file path relative 
+        *        public root in the first element, and the archive file path relative
         *        to the deleted zone root in the second element.
         * @return FileRepoStatus
         */
@@ -305,10 +420,10 @@ abstract class FileRepo {
 
        /**
         * Move a file to the deletion archive.
-        * If no valid deletion archive exists, this may either delete the file 
+        * If no valid deletion archive exists, this may either delete the file
         * or throw an exception, depending on the preference of the repository
         * @param mixed $srcRel Relative path for the file to be deleted
-        * @param mixed $archiveRel Relative path for the archive location. 
+        * @param mixed $archiveRel Relative path for the archive location.
         *        Relative to a private archive directory.
         * @return WikiError object (wikitext-formatted), or true for success
         */
@@ -400,5 +515,58 @@ abstract class FileRepo {
         * STUB
         */
        function cleanupDeletedBatch( $storageKeys ) {}
-}
 
+       /**
+        * Checks if there is a redirect named as $title. If there is, return the
+        * title object. If not, return false.
+        * STUB
+        *
+        * @param Title $title Title of image
+        */
+       function checkRedirect( $title ) {
+               return false;
+       }
+
+       /**
+        * Invalidates image redirect cache related to that image
+        *
+        * @param Title $title Title of image
+        */     
+       function invalidateImageRedirect( $title ) {
+               global $wgMemc;
+               $memcKey = $this->getMemcKey( "image_redirect:" . md5( $title->getPrefixedDBkey() ) );
+               $wgMemc->delete( $memcKey );
+       }
+       
+       function findBySha1( $hash ) {
+               return array();
+       }
+       
+       /**
+        * Get the human-readable name of the repo. 
+        * @return string
+        */
+       public function getDisplayName() {
+               // We don't name our own repo, return nothing
+               if ( $this->name == 'local' ) {
+                       return null;
+               }
+               $repoName = wfMsg( 'shared-repo-name-' . $this->name );
+               if ( !wfEmptyMsg( 'shared-repo-name-' . $this->name, $repoName ) ) {
+                       return $repoName;
+               }
+               return wfMsg( 'shared-repo' ); 
+       }
+       
+       function getSlaveDB() {
+               return wfGetDB( DB_SLAVE );
+       }
+
+       function getMasterDB() {
+               return wfGetDB( DB_MASTER );
+       }
+       
+       function getMemcKey( $key ) {
+               return wfWikiID( $this->getSlaveDB() ) . ":{$key}";
+       }
+}