3 * Local repository that stores files in the local filesystem and registers them
4 * in the wiki's own database.
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License along
17 * with this program; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19 * http://www.gnu.org/copyleft/gpl.html
26 * A repository that stores files in the local filesystem and registers them
27 * in the wiki's own database. This is the most commonly used repository class.
31 class LocalRepo
extends FileRepo
{
33 protected $fileFactory = array( 'LocalFile', 'newFromTitle' );
36 protected $fileFactoryKey = array( 'LocalFile', 'newFromKey' );
39 protected $fileFromRowFactory = array( 'LocalFile', 'newFromRow' );
42 protected $oldFileFromRowFactory = array( 'OldLocalFile', 'newFromRow' );
45 protected $oldFileFactory = array( 'OldLocalFile', 'newFromTitle' );
48 protected $oldFileFactoryKey = array( 'OldLocalFile', 'newFromKey' );
50 function __construct( array $info = null ) {
51 parent
::__construct( $info );
53 $this->hasSha1Storage
= isset( $info['storageLayout'] ) && $info['storageLayout'] === 'sha1';
55 if ( $this->hasSha1Storage() ) {
56 $this->backend
= new FileBackendDBRepoWrapper( array(
57 'backend' => $this->backend
,
58 'repoName' => $this->name
,
59 'dbHandleFactory' => $this->getDBFactory()
66 * @param stdClass $row
69 function newFileFromRow( $row ) {
70 if ( isset( $row->img_name
) ) {
71 return call_user_func( $this->fileFromRowFactory
, $row, $this );
72 } elseif ( isset( $row->oi_name
) ) {
73 return call_user_func( $this->oldFileFromRowFactory
, $row, $this );
75 throw new MWException( __METHOD__
. ': invalid row' );
81 * @param string $archiveName
82 * @return OldLocalFile
84 function newFromArchiveName( $title, $archiveName ) {
85 return OldLocalFile
::newFromArchiveName( $title, $this, $archiveName );
89 * Delete files in the deleted directory if they are not referenced in the
90 * filearchive table. This needs to be done in the repo because it needs to
91 * interleave database locks with file operations, which is potentially a
94 * @param array $storageKeys
96 * @return FileRepoStatus
98 function cleanupDeletedBatch( array $storageKeys ) {
99 if ( $this->hasSha1Storage() ) {
100 wfDebug( __METHOD__
. ": skipped because storage uses sha1 paths\n" );
101 return Status
::newGood();
104 $backend = $this->backend
; // convenience
105 $root = $this->getZonePath( 'deleted' );
106 $dbw = $this->getMasterDB();
107 $status = $this->newGood();
108 $storageKeys = array_unique( $storageKeys );
109 foreach ( $storageKeys as $key ) {
110 $hashPath = $this->getDeletedHashPath( $key );
111 $path = "$root/$hashPath$key";
112 $dbw->startAtomic( __METHOD__
);
113 // Check for usage in deleted/hidden files and preemptively
114 // lock the key to avoid any future use until we are finished.
115 $deleted = $this->deletedFileHasKey( $key, 'lock' );
116 $hidden = $this->hiddenFileHasKey( $key, 'lock' );
117 if ( !$deleted && !$hidden ) { // not in use now
118 wfDebug( __METHOD__
. ": deleting $key\n" );
119 $op = array( 'op' => 'delete', 'src' => $path );
120 if ( !$backend->doOperation( $op )->isOK() ) {
121 $status->error( 'undelete-cleanup-error', $path );
122 $status->failCount++
;
125 wfDebug( __METHOD__
. ": $key still in use\n" );
126 $status->successCount++
;
128 $dbw->endAtomic( __METHOD__
);
135 * Check if a deleted (filearchive) file has this sha1 key
137 * @param string $key File storage key (base-36 sha1 key with file extension)
138 * @param string|null $lock Use "lock" to lock the row via FOR UPDATE
139 * @return bool File with this key is in use
141 protected function deletedFileHasKey( $key, $lock = null ) {
142 $options = ( $lock === 'lock' ) ?
array( 'FOR UPDATE' ) : array();
144 $dbw = $this->getMasterDB();
146 return (bool)$dbw->selectField( 'filearchive', '1',
147 array( 'fa_storage_group' => 'deleted', 'fa_storage_key' => $key ),
153 * Check if a hidden (revision delete) file has this sha1 key
155 * @param string $key File storage key (base-36 sha1 key with file extension)
156 * @param string|null $lock Use "lock" to lock the row via FOR UPDATE
157 * @return bool File with this key is in use
159 protected function hiddenFileHasKey( $key, $lock = null ) {
160 $options = ( $lock === 'lock' ) ?
array( 'FOR UPDATE' ) : array();
162 $sha1 = self
::getHashFromKey( $key );
163 $ext = File
::normalizeExtension( substr( $key, strcspn( $key, '.' ) +
1 ) );
165 $dbw = $this->getMasterDB();
167 return (bool)$dbw->selectField( 'oldimage', '1',
168 array( 'oi_sha1' => $sha1,
169 'oi_archive_name ' . $dbw->buildLike( $dbw->anyString(), ".$ext" ),
170 $dbw->bitAnd( 'oi_deleted', File
::DELETED_FILE
) => File
::DELETED_FILE
),
176 * Gets the SHA1 hash from a storage key
181 public static function getHashFromKey( $key ) {
182 return strtok( $key, '.' );
186 * Checks if there is a redirect named as $title
188 * @param Title $title Title of file
191 function checkRedirect( Title
$title ) {
192 $cache = ObjectCache
::getMainWANInstance();
194 $title = File
::normalizeTitle( $title, 'exception' );
196 $memcKey = $this->getSharedCacheKey( 'image_redirect', md5( $title->getDBkey() ) );
197 if ( $memcKey === false ) {
198 $memcKey = $this->getLocalCacheKey( 'image_redirect', md5( $title->getDBkey() ) );
199 $expiry = 300; // no invalidation, 5 minutes
201 $expiry = 86400; // has invalidation, 1 day
203 $cachedValue = $cache->get( $memcKey );
204 if ( $cachedValue === ' ' ||
$cachedValue === '' ) {
207 } elseif ( strval( $cachedValue ) !== '' && $cachedValue !== ' PURGED' ) {
208 return Title
::newFromText( $cachedValue, NS_FILE
);
209 } // else $cachedValue is false or null: cache miss
211 $id = $this->getArticleID( $title );
213 $cache->set( $memcKey, " ", $expiry );
217 $dbr = $this->getSlaveDB();
218 $row = $dbr->selectRow(
220 array( 'rd_title', 'rd_namespace' ),
221 array( 'rd_from' => $id ),
225 if ( $row && $row->rd_namespace
== NS_FILE
) {
226 $targetTitle = Title
::makeTitle( $row->rd_namespace
, $row->rd_title
);
227 $cache->set( $memcKey, $targetTitle->getDBkey(), $expiry );
231 $cache->set( $memcKey, '', $expiry );
238 * Function link Title::getArticleID().
239 * We can't say Title object, what database it should use, so we duplicate that function here.
241 * @param Title $title
242 * @return bool|int|mixed
244 protected function getArticleID( $title ) {
245 if ( !$title instanceof Title
) {
248 $dbr = $this->getSlaveDB();
249 $id = $dbr->selectField(
253 'page_namespace' => $title->getNamespace(),
254 'page_title' => $title->getDBkey(),
256 __METHOD__
//Function name
262 public function findFiles( array $items, $flags = 0 ) {
263 $finalFiles = array(); // map of (DB key => corresponding File) for matches
265 $searchSet = array(); // map of (normalized DB key => search params)
266 foreach ( $items as $item ) {
267 if ( is_array( $item ) ) {
268 $title = File
::normalizeTitle( $item['title'] );
270 $searchSet[$title->getDBkey()] = $item;
273 $title = File
::normalizeTitle( $item );
275 $searchSet[$title->getDBkey()] = array();
280 $fileMatchesSearch = function ( File
$file, array $search ) {
281 // Note: file name comparison done elsewhere (to handle redirects)
282 $user = ( !empty( $search['private'] ) && $search['private'] instanceof User
)
289 ( empty( $search['time'] ) && !$file->isOld() ) ||
290 ( !empty( $search['time'] ) && $search['time'] === $file->getTimestamp() )
292 ( !empty( $search['private'] ) ||
!$file->isDeleted( File
::DELETED_FILE
) ) &&
293 $file->userCan( File
::DELETED_FILE
, $user )
298 $applyMatchingFiles = function ( ResultWrapper
$res, &$searchSet, &$finalFiles )
299 use ( $that, $fileMatchesSearch, $flags )
302 $info = $that->getInfo();
303 foreach ( $res as $row ) {
304 $file = $that->newFileFromRow( $row );
305 // There must have been a search for this DB key, but this has to handle the
306 // cases were title capitalization is different on the client and repo wikis.
307 $dbKeysLook = array( strtr( $file->getName(), ' ', '_' ) );
308 if ( !empty( $info['initialCapital'] ) ) {
309 // Search keys for "hi.png" and "Hi.png" should use the "Hi.png file"
310 $dbKeysLook[] = $wgContLang->lcfirst( $file->getName() );
312 foreach ( $dbKeysLook as $dbKey ) {
313 if ( isset( $searchSet[$dbKey] )
314 && $fileMatchesSearch( $file, $searchSet[$dbKey] )
316 $finalFiles[$dbKey] = ( $flags & FileRepo
::NAME_AND_TIME_ONLY
)
317 ?
array( 'title' => $dbKey, 'timestamp' => $file->getTimestamp() )
319 unset( $searchSet[$dbKey] );
325 $dbr = $this->getSlaveDB();
329 foreach ( array_keys( $searchSet ) as $dbKey ) {
330 $imgNames[] = $this->getNameFromTitle( File
::normalizeTitle( $dbKey ) );
333 if ( count( $imgNames ) ) {
334 $res = $dbr->select( 'image',
335 LocalFile
::selectFields(), array( 'img_name' => $imgNames ), __METHOD__
);
336 $applyMatchingFiles( $res, $searchSet, $finalFiles );
339 // Query old image table
340 $oiConds = array(); // WHERE clause array for each file
341 foreach ( $searchSet as $dbKey => $search ) {
342 if ( isset( $search['time'] ) ) {
343 $oiConds[] = $dbr->makeList(
345 'oi_name' => $this->getNameFromTitle( File
::normalizeTitle( $dbKey ) ),
346 'oi_timestamp' => $dbr->timestamp( $search['time'] )
353 if ( count( $oiConds ) ) {
354 $res = $dbr->select( 'oldimage',
355 OldLocalFile
::selectFields(), $dbr->makeList( $oiConds, LIST_OR
), __METHOD__
);
356 $applyMatchingFiles( $res, $searchSet, $finalFiles );
359 // Check for redirects...
360 foreach ( $searchSet as $dbKey => $search ) {
361 if ( !empty( $search['ignoreRedirect'] ) ) {
365 $title = File
::normalizeTitle( $dbKey );
366 $redir = $this->checkRedirect( $title ); // hopefully hits memcached
368 if ( $redir && $redir->getNamespace() == NS_FILE
) {
369 $file = $this->newFile( $redir );
370 if ( $file && $fileMatchesSearch( $file, $search ) ) {
371 $file->redirectedFrom( $title->getDBkey() );
372 if ( $flags & FileRepo
::NAME_AND_TIME_ONLY
) {
373 $finalFiles[$dbKey] = array(
374 'title' => $file->getTitle()->getDBkey(),
375 'timestamp' => $file->getTimestamp()
378 $finalFiles[$dbKey] = $file;
388 * Get an array or iterator of file objects for files that have a given
389 * SHA-1 content hash.
391 * @param string $hash A sha1 hash to look for
394 function findBySha1( $hash ) {
395 $dbr = $this->getSlaveDB();
398 LocalFile
::selectFields(),
399 array( 'img_sha1' => $hash ),
401 array( 'ORDER BY' => 'img_name' )
405 foreach ( $res as $row ) {
406 $result[] = $this->newFileFromRow( $row );
414 * Get an array of arrays or iterators of file objects for files that
415 * have the given SHA-1 content hashes.
417 * Overrides generic implementation in FileRepo for performance reason
419 * @param array $hashes An array of hashes
420 * @return array An Array of arrays or iterators of file objects and the hash as key
422 function findBySha1s( array $hashes ) {
423 if ( !count( $hashes ) ) {
424 return array(); //empty parameter
427 $dbr = $this->getSlaveDB();
430 LocalFile
::selectFields(),
431 array( 'img_sha1' => $hashes ),
433 array( 'ORDER BY' => 'img_name' )
437 foreach ( $res as $row ) {
438 $file = $this->newFileFromRow( $row );
439 $result[$file->getSha1()][] = $file;
447 * Return an array of files where the name starts with $prefix.
449 * @param string $prefix The prefix to search for
450 * @param int $limit The maximum amount of files to return
453 public function findFilesByPrefix( $prefix, $limit ) {
454 $selectOptions = array( 'ORDER BY' => 'img_name', 'LIMIT' => intval( $limit ) );
457 $dbr = $this->getSlaveDB();
460 LocalFile
::selectFields(),
461 'img_name ' . $dbr->buildLike( $prefix, $dbr->anyString() ),
466 // Build file objects
468 foreach ( $res as $row ) {
469 $files[] = $this->newFileFromRow( $row );
476 * Get a connection to the slave DB
477 * @return DatabaseBase
479 function getSlaveDB() {
480 return wfGetDB( DB_SLAVE
);
484 * Get a connection to the master DB
485 * @return DatabaseBase
487 function getMasterDB() {
488 return wfGetDB( DB_MASTER
);
492 * Get a callback to get a DB handle given an index (DB_SLAVE/DB_MASTER)
495 protected function getDBFactory() {
496 return function( $index ) {
497 return wfGetDB( $index );
502 * Get a key on the primary cache for this repository.
503 * Returns false if the repository's cache is not accessible at this site.
504 * The parameters are the parts of the key, as for wfMemcKey().
508 function getSharedCacheKey( /*...*/ ) {
509 $args = func_get_args();
511 return call_user_func_array( 'wfMemcKey', $args );
515 * Invalidates image redirect cache related to that image
517 * @param Title $title Title of page
520 function invalidateImageRedirect( Title
$title ) {
521 $cache = ObjectCache
::getMainWANInstance();
523 $memcKey = $this->getSharedCacheKey( 'image_redirect', md5( $title->getDBkey() ) );
525 // Set a temporary value for the cache key, to ensure
526 // that this value stays purged long enough so that
527 // it isn't refreshed with a stale value due to a
529 $cache->delete( $memcKey, 12 );
534 * Return information about the repository.
542 return array_merge( parent
::getInfo(), array(
543 'favicon' => wfExpandUrl( $wgFavicon ),
547 public function store( $srcPath, $dstZone, $dstRel, $flags = 0 ) {
548 return $this->skipWriteOperationIfSha1( __FUNCTION__
, func_get_args() );
551 public function storeBatch( array $triplets, $flags = 0 ) {
552 return $this->skipWriteOperationIfSha1( __FUNCTION__
, func_get_args() );
555 public function cleanupBatch( array $files, $flags = 0 ) {
556 return $this->skipWriteOperationIfSha1( __FUNCTION__
, func_get_args() );
559 public function publish(
564 array $options = array()
566 return $this->skipWriteOperationIfSha1( __FUNCTION__
, func_get_args() );
569 public function publishBatch( array $ntuples, $flags = 0 ) {
570 return $this->skipWriteOperationIfSha1( __FUNCTION__
, func_get_args() );
573 public function delete( $srcRel, $archiveRel ) {
574 return $this->skipWriteOperationIfSha1( __FUNCTION__
, func_get_args() );
577 public function deleteBatch( array $sourceDestPairs ) {
578 return $this->skipWriteOperationIfSha1( __FUNCTION__
, func_get_args() );
582 * Skips the write operation if storage is sha1-based, executes it normally otherwise
584 * @param string $function
586 * @return FileRepoStatus
588 protected function skipWriteOperationIfSha1( $function, array $args ) {
589 $this->assertWritableRepo(); // fail out if read-only
591 if ( $this->hasSha1Storage() ) {
592 wfDebug( __METHOD__
. ": skipped because storage uses sha1 paths\n" );
593 return Status
::newGood();
595 return call_user_func_array( 'parent::' . $function, $args );