1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
<?php
/**
* Refresh file headers from metadata.
*
* Usage: php refreshFileHeaders.php
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
* http://www.gnu.org/copyleft/gpl.html
*
* @file
* @author Aaron Schulz
* @ingroup Maintenance
*/
require_once( __DIR__ . '/Maintenance.php' );
/**
* Maintenance script to refresh file headers from metadata
*
* @ingroup Maintenance
*/
class RefreshFileHeaders extends Maintenance {
function __construct() {
parent::__construct();
$this->mDescription = 'Script to update file HTTP headers';
$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->setBatchSize( 200 );
}
public function execute() {
$repo = RepoGroup::singleton()->getLocalRepo();
$start = str_replace( ' ', '_', $this->getOption( 'start', '' ) ); // page on img_name
$end = str_replace( ' ', '_', $this->getOption( 'end', '' ) ); // page on img_name
$count = 0;
$dbr = wfGetDB( DB_SLAVE );
do {
$conds = array( "img_name > {$dbr->addQuotes( $start )}" );
if ( strlen( $end ) ) {
$conds[] = "img_name <= {$dbr->addQuotes( $end )}";
}
$res = $dbr->select( 'image', '*', $conds,
__METHOD__, array( 'LIMIT' => $this->mBatchSize, 'ORDER BY' => 'img_name ASC' ) );
foreach ( $res as $row ) {
$file = $repo->newFileFromRow( $row );
$headers = $file->getStreamHeaders();
if ( count( $headers ) ) {
$this->updateFileHeaders( $file, $headers );
}
// Do all of the older file versions...
foreach ( $file->getHistory() as $oldFile ) {
$headers = $oldFile->getStreamHeaders();
if ( count( $headers ) ) {
$this->updateFileHeaders( $oldFile, $headers );
}
}
if ( $this->hasOption( 'verbose' ) ) {
$this->output( "Updated headers for file '{$row->img_name}'.\n" );
}
++$count;
$start = $row->img_name; // advance
}
} while ( $res->numRows() > 0 );
$this->output( "Done. Updated headers for $count file(s).\n" );
}
protected function updateFileHeaders( File $file, array $headers ) {
$status = $file->getRepo()->getBackend()->describe( array(
'src' => $file->getPath(), 'headers' => $headers
) );
if ( !$status->isGood() ) {
$this->error( "Encountered error: " . print_r( $status, true ) );
}
}
}
$maintClass = 'RefreshFileHeaders';
require_once( RUN_MAINTENANCE_IF_MAIN );
|
