3 * File backend registration handling.
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
18 * http://www.gnu.org/copyleft/gpl.html
21 * @ingroup FileBackend
22 * @author Aaron Schulz
24 use \MediaWiki\Logger\LoggerFactory
;
27 * Class to handle file backend registration
29 * @ingroup FileBackend
32 class FileBackendGroup
{
33 /** @var FileBackendGroup */
34 protected static $instance = null;
36 /** @var array (name => ('class' => string, 'config' => array, 'instance' => object)) */
37 protected $backends = [];
39 protected function __construct() {
43 * @return FileBackendGroup
45 public static function singleton() {
46 if ( self
::$instance == null ) {
47 self
::$instance = new self();
48 self
::$instance->initFromGlobals();
51 return self
::$instance;
55 * Destroy the singleton instance
57 public static function destroySingleton() {
58 self
::$instance = null;
62 * Register file backends from the global variables
64 protected function initFromGlobals() {
65 global $wgLocalFileRepo, $wgForeignFileRepos, $wgFileBackends, $wgDirectoryMode;
67 // Register explicitly defined backends
68 $this->register( $wgFileBackends, wfConfiguredReadOnlyReason() );
71 // Automatically create b/c backends for file repos...
72 $repos = array_merge( $wgForeignFileRepos, [ $wgLocalFileRepo ] );
73 foreach ( $repos as $info ) {
74 $backendName = $info['backend'];
75 if ( is_object( $backendName ) ||
isset( $this->backends
[$backendName] ) ) {
76 continue; // already defined (or set to the object for some reason)
78 $repoName = $info['name'];
79 // Local vars that used to be FSRepo members...
80 $directory = $info['directory'];
81 $deletedDir = isset( $info['deletedDir'] )
83 : false; // deletion disabled
84 $thumbDir = isset( $info['thumbDir'] )
86 : "{$directory}/thumb";
87 $transcodedDir = isset( $info['transcodedDir'] )
88 ?
$info['transcodedDir']
89 : "{$directory}/transcoded";
90 // Get the FS backend configuration
92 'name' => $backendName,
93 'class' => 'FSFileBackend',
94 'lockManager' => 'fsLockManager',
96 "{$repoName}-public" => "{$directory}",
97 "{$repoName}-thumb" => $thumbDir,
98 "{$repoName}-transcoded" => $transcodedDir,
99 "{$repoName}-deleted" => $deletedDir,
100 "{$repoName}-temp" => "{$directory}/temp"
102 'fileMode' => isset( $info['fileMode'] ) ?
$info['fileMode'] : 0644,
103 'directoryMode' => $wgDirectoryMode,
107 // Register implicitly defined backends
108 $this->register( $autoBackends, wfConfiguredReadOnlyReason() );
112 * Register an array of file backend configurations
114 * @param array $configs
115 * @param string|null $readOnlyReason
116 * @throws InvalidArgumentException
118 protected function register( array $configs, $readOnlyReason = null ) {
119 foreach ( $configs as $config ) {
120 if ( !isset( $config['name'] ) ) {
121 throw new InvalidArgumentException( "Cannot register a backend with no name." );
123 $name = $config['name'];
124 if ( isset( $this->backends
[$name] ) ) {
125 throw new LogicException( "Backend with name `{$name}` already registered." );
126 } elseif ( !isset( $config['class'] ) ) {
127 throw new InvalidArgumentException( "Backend with name `{$name}` has no class." );
129 $class = $config['class'];
131 $config['readOnly'] = !empty( $config['readOnly'] )
132 ?
$config['readOnly']
135 unset( $config['class'] ); // backend won't need this
136 $this->backends
[$name] = [
145 * Get the backend object with a given name
147 * @param string $name
148 * @return FileBackend
149 * @throws InvalidArgumentException
151 public function get( $name ) {
152 if ( !isset( $this->backends
[$name] ) ) {
153 throw new InvalidArgumentException( "No backend defined with the name `$name`." );
155 // Lazy-load the actual backend instance
156 if ( !isset( $this->backends
[$name]['instance'] ) ) {
157 $class = $this->backends
[$name]['class'];
158 $config = $this->backends
[$name]['config'];
160 'wikiId' => wfWikiID(), // e.g. "my_wiki-en_"
161 'mimeCallback' => [ $this, 'guessMimeInternal' ],
162 'obResetFunc' => 'wfResetOutputBuffers',
163 'streamMimeFunc' => [ 'StreamFile', 'contentTypeFromPath' ]
165 $config['lockManager'] =
166 LockManagerGroup
::singleton( $config['wikiId'] )->get( $config['lockManager'] );
167 $config['fileJournal'] = isset( $config['fileJournal'] )
168 ? FileJournal
::factory( $config['fileJournal'], $name )
169 : FileJournal
::factory( [ 'class' => 'NullFileJournal' ], $name );
170 $config['wanCache'] = ObjectCache
::getMainWANInstance();
171 $config['statusWrapper'] = [ 'Status', 'wrap' ];
172 $config['tmpDirectory'] = wfTempDir();
173 $config['logger'] = LoggerFactory
::getInstance( 'FileOperation' );
174 $config['profiler'] = Profiler
::instance();
175 if ( $class === 'FileBackendMultiWrite' ) {
176 foreach ( $config['backends'] as $index => $beConfig ) {
177 if ( isset( $beConfig['template'] ) ) {
178 // Config is just a modified version of a registered backend's.
179 // This should only be used when that config is used only by this backend.
180 $config['backends'][$index] +
= $this->config( $beConfig['template'] );
185 $this->backends
[$name]['instance'] = new $class( $config );
188 return $this->backends
[$name]['instance'];
192 * Get the config array for a backend object with a given name
194 * @param string $name
196 * @throws InvalidArgumentException
198 public function config( $name ) {
199 if ( !isset( $this->backends
[$name] ) ) {
200 throw new InvalidArgumentException( "No backend defined with the name `$name`." );
202 $class = $this->backends
[$name]['class'];
204 return [ 'class' => $class ] +
$this->backends
[$name]['config'];
208 * Get an appropriate backend object from a storage path
210 * @param string $storagePath
211 * @return FileBackend|null Backend or null on failure
213 public function backendFromPath( $storagePath ) {
214 list( $backend, , ) = FileBackend
::splitStoragePath( $storagePath );
215 if ( $backend !== null && isset( $this->backends
[$backend] ) ) {
216 return $this->get( $backend );
223 * @param string $storagePath
224 * @param string|null $content
225 * @param string|null $fsPath
229 public function guessMimeInternal( $storagePath, $content, $fsPath ) {
230 $magic = MimeMagic
::singleton();
231 // Trust the extension of the storage path (caller must validate)
232 $ext = FileBackend
::extensionFromPath( $storagePath );
233 $type = $magic->guessTypesForExtension( $ext );
234 // For files without a valid extension (or one at all), inspect the contents
235 if ( !$type && $fsPath ) {
236 $type = $magic->guessMimeType( $fsPath, false );
237 } elseif ( !$type && strlen( $content ) ) {
238 $tmpFile = TempFSFile
::factory( 'mime_', '', wfTempDir() );
239 file_put_contents( $tmpFile->getPath(), $content );
240 $type = $magic->guessMimeType( $tmpFile->getPath(), false );
242 return $type ?
: 'unknown/unknown';