* http://www.gnu.org/copyleft/gpl.html
*
* @file
- * @author Aaron Schulz
* @ingroup Maintenance
*/
$this->addOption( 'verbose', 'Output information about each file.', false, false, 'v' );
$this->addOption( 'start', 'Name of file to start with', false, true );
$this->addOption( 'end', 'Name of file to end with', false, true );
+ $this->addOption( 'media_type', 'Media type to filter for', false, true );
+ $this->addOption( 'major_mime', 'Major mime type to filter for', false, true );
+ $this->addOption( 'minor_mime', 'Minor mime type to filter for', false, true );
+ $this->addOption(
+ 'refreshContentType',
+ 'Set true to refresh file content type from mime data in db',
+ false,
+ false
+ );
$this->setBatchSize( 200 );
}
$repo = RepoGroup::singleton()->getLocalRepo();
$start = str_replace( ' ', '_', $this->getOption( 'start', '' ) ); // page on img_name
$end = str_replace( ' ', '_', $this->getOption( 'end', '' ) ); // page on img_name
+ // filter by img_media_type
+ $media_type = str_replace( ' ', '_', $this->getOption( 'media_type', '' ) );
+ // filter by img_major_mime
+ $major_mime = str_replace( ' ', '_', $this->getOption( 'major_mime', '' ) );
+ // filter by img_minor_mime
+ $minor_mime = str_replace( ' ', '_', $this->getOption( 'minor_mime', '' ) );
$count = 0;
$dbr = $this->getDB( DB_REPLICA );
+
+ $fileQuery = LocalFile::getQueryInfo();
+
do {
$conds = [ "img_name > {$dbr->addQuotes( $start )}" ];
+
if ( strlen( $end ) ) {
$conds[] = "img_name <= {$dbr->addQuotes( $end )}";
}
- $res = $dbr->select( 'image', '*', $conds,
- __METHOD__, [ 'LIMIT' => $this->mBatchSize, 'ORDER BY' => 'img_name ASC' ] );
+
+ if ( strlen( $media_type ) ) {
+ $conds[] = "img_media_type = {$dbr->addQuotes( $media_type )}";
+ }
+
+ if ( strlen( $major_mime ) ) {
+ $conds[] = "img_major_mime = {$dbr->addQuotes( $major_mime )}";
+ }
+
+ if ( strlen( $minor_mime ) ) {
+ $conds[] = "img_minor_mime = {$dbr->addQuotes( $minor_mime )}";
+ }
+
+ $res = $dbr->select( $fileQuery['tables'],
+ $fileQuery['fields'],
+ $conds,
+ __METHOD__,
+ [
+ 'LIMIT' => $this->getBatchSize(),
+ 'ORDER BY' => 'img_name ASC'
+ ],
+ $fileQuery['joins']
+ );
if ( $res->numRows() > 0 ) {
$row1 = $res->current();
$res->rewind();
}
+ $backendOperations = [];
+
foreach ( $res as $row ) {
$file = $repo->newFileFromRow( $row );
$headers = $file->getContentHeaders();
+ if ( $this->getOption( 'refreshContentType', false ) ) {
+ $headers['Content-Type'] = $row->img_major_mime . '/' . $row->img_minor_mime;
+ }
+
if ( count( $headers ) ) {
- $this->updateFileHeaders( $file, $headers );
+ $backendOperations[] = [
+ 'op' => 'describe', 'src' => $file->getPath(), 'headers' => $headers
+ ];
}
+
// Do all of the older file versions...
foreach ( $file->getHistory() as $oldFile ) {
$headers = $oldFile->getContentHeaders();
if ( count( $headers ) ) {
- $this->updateFileHeaders( $oldFile, $headers );
+ $backendOperations[] = [
+ 'op' => 'describe', 'src' => $oldFile->getPath(), 'headers' => $headers
+ ];
}
}
+
if ( $this->hasOption( 'verbose' ) ) {
- $this->output( "Updated headers for file '{$row->img_name}'.\n" );
+ $this->output( "Queued headers update for file '{$row->img_name}'.\n" );
}
- ++$count;
+
$start = $row->img_name; // advance
}
- } while ( $res->numRows() === $this->mBatchSize );
+
+ $backendOperationsCount = count( $backendOperations );
+ $count += $backendOperationsCount;
+
+ $this->output( "Updating headers for {$backendOperationsCount} file(s).\n" );
+ $this->updateFileHeaders( $repo, $backendOperations );
+ } while ( $res->numRows() === $this->getBatchSize() );
$this->output( "Done. Updated headers for $count file(s).\n" );
}
- protected function updateFileHeaders( File $file, array $headers ) {
- $status = $file->getRepo()->getBackend()->describe( [
- 'src' => $file->getPath(), 'headers' => $headers
- ] );
+ protected function updateFileHeaders( $repo, $backendOperations ) {
+ $status = $repo->getBackend()->doQuickOperations( $backendOperations );
+
if ( !$status->isGood() ) {
$this->error( "Encountered error: " . print_r( $status, true ) );
}