includes/filerepo/LocalRepo.php

   1 <?php
   2 /**
   3  * A repository that stores files in the local filesystem and registers them
   4  * in the wiki's own database. This is the most commonly used repository class.
   5  */
   6 class LocalRepo extends FSRepo {
   7         var $fileFactory = array( 'LocalFile', 'newFromTitle' );
   8         var $oldFileFactory = array( 'OldLocalFile', 'newFromTitle' );
   9
  10         function getSlaveDB() {
  11                 return wfGetDB( DB_SLAVE );
  12         }
  13
  14         function getMasterDB() {
  15                 return wfGetDB( DB_MASTER );
  16         }
  17
  18         function newFileFromRow( $row ) {
  19                 if ( isset( $row->img_name ) ) {
  20                         return LocalFile::newFromRow( $row, $this );
  21                 } elseif ( isset( $row->oi_name ) ) {
  22                         return OldLocalFile::newFromRow( $row, $this );
  23                 } else {
  24                         throw new MWException( __METHOD__.': invalid row' );
  25                 }
  26         }
  27
  28         function newFromArchiveName( $title, $archiveName ) {
  29                 return OldLocalFile::newFromArchiveName( $title, $this, $archiveName );
  30         }
  31
  32         /**
  33          * Delete files in the deleted directory if they are not referenced in the
  34          * filearchive table. This needs to be done in the repo because it needs to
  35          * interleave database locks with file operations, which is potentially a
  36          * remote operation.
  37          * @return FileRepoStatus
  38          */
  39         function cleanupDeletedBatch( $storageKeys ) {
  40                 $root = $this->getZonePath( 'deleted' );
  41                 $dbw = $this->getMasterDB();
  42                 $status = $this->newGood();
  43                 $storageKeys = array_unique($storageKeys);
  44                 foreach ( $storageKeys as $key ) {
  45                         $hashPath = $this->getDeletedHashPath( $key );
  46                         $path = "$root/$hashPath$key";
  47                         $dbw->begin();
  48                         $inuse = $dbw->selectField( 'filearchive', '1',
  49                                 array( 'fa_storage_group' => 'deleted', 'fa_storage_key' => $key ),
  50                                 __METHOD__, array( 'FOR UPDATE' ) );
  51                         if ( !$inuse ) {
  52                                 wfDebug( __METHOD__ . ": deleting $key\n" );
  53                                 if ( !@unlink( $path ) ) {
  54                                         $status->error( 'undelete-cleanup-error', $path );
  55                                         $status->failCount++;
  56                                 }
  57                         } else {
  58                                 wfDebug( __METHOD__ . ": $key still in use\n" );
  59                                 $status->successCount++;
  60                         }
  61                         $dbw->commit();
  62                 }
  63                 return $status;
  64         }
  65
  66         /**
  67          * Function link Title::getArticleID().
  68          * We can't say Title object, what database it should use, so we duplicate that function here.
  69          */
  70         private function getArticleID( $title ) {
  71                 if( !$title instanceof Title ) {
  72                         return 0;
  73                 }
  74                 $dbr = $this->getSlaveDB();
  75                 $id = $dbr->selectField(
  76                         'page', // Table
  77                         'page_id',      //Field
  78                         array(  //Conditions
  79                                 'page_namespace' => $title->getNamespace(),
  80                                 'page_title' => $title->getDbKey(),
  81                         ),
  82                         __METHOD__      //Function name
  83                 );
  84                 return $id;
  85         }
  86
  87         function checkRedirect( $title ) {
  88                 global $wgFileRedirects;
  89                 if( !$wgFileRedirects ) {
  90                         return false;
  91                 }
  92
  93                 if( $title->getNamespace() == NS_MEDIA ) {
  94                         $title = Title::makeTitle( NS_IMAGE, $title->getText() );
  95                 }
  96
  97                 $id = $this->getArticleID( $title );
  98                 if( !$id ) {
  99                         return false;
 100                 }
 101                 $dbr = $this->getSlaveDB();
 102                 $row = $dbr->selectRow(
 103                         'redirect',
 104                         array( 'rd_title', 'rd_namespace' ),
 105                         array( 'rd_from' => $id ),
 106                         __METHOD__
 107                 );
 108                 if( !$row ) {
 109                         return false;
 110                 }
 111                 return Title::makeTitle( $row->rd_namespace, $row->rd_title );
 112         }
 113 }