blob: fce8549f0781bafdc7da2301b84d048286757445 [file] [log] [blame]
avm9996399bb77c2020-01-27 03:15:08 +01001<?php
2
3/*
4 * This file is part of Composer.
5 *
6 * (c) Nils Adermann <naderman@naderman.de>
7 * Jordi Boggiano <j.boggiano@seld.be>
8 *
9 * For the full copyright and license information, please view the LICENSE
10 * file that was distributed with this source code.
11 */
12
13namespace Composer\Autoload;
14
15/**
16 * ClassLoader implements a PSR-0, PSR-4 and classmap class loader.
17 *
18 * $loader = new \Composer\Autoload\ClassLoader();
19 *
20 * // register classes with namespaces
21 * $loader->add('Symfony\Component', __DIR__.'/component');
22 * $loader->add('Symfony', __DIR__.'/framework');
23 *
24 * // activate the autoloader
25 * $loader->register();
26 *
27 * // to enable searching the include path (eg. for PEAR packages)
28 * $loader->setUseIncludePath(true);
29 *
30 * In this example, if you try to use a class in the Symfony\Component
31 * namespace or one of its children (Symfony\Component\Console for instance),
32 * the autoloader will first look for the class under the component/
33 * directory, and it will then fallback to the framework/ directory if not
34 * found before giving up.
35 *
36 * This class is loosely based on the Symfony UniversalClassLoader.
37 *
38 * @author Fabien Potencier <fabien@symfony.com>
39 * @author Jordi Boggiano <j.boggiano@seld.be>
40 * @see http://www.php-fig.org/psr/psr-0/
41 * @see http://www.php-fig.org/psr/psr-4/
42 */
43class ClassLoader
44{
45 // PSR-4
46 private $prefixLengthsPsr4 = array();
47 private $prefixDirsPsr4 = array();
48 private $fallbackDirsPsr4 = array();
49
50 // PSR-0
51 private $prefixesPsr0 = array();
52 private $fallbackDirsPsr0 = array();
53
54 private $useIncludePath = false;
55 private $classMap = array();
56 private $classMapAuthoritative = false;
57 private $missingClasses = array();
58 private $apcuPrefix;
59
60 public function getPrefixes()
61 {
62 if (!empty($this->prefixesPsr0)) {
63 return call_user_func_array('array_merge', $this->prefixesPsr0);
64 }
65
66 return array();
67 }
68
69 public function getPrefixesPsr4()
70 {
71 return $this->prefixDirsPsr4;
72 }
73
74 public function getFallbackDirs()
75 {
76 return $this->fallbackDirsPsr0;
77 }
78
79 public function getFallbackDirsPsr4()
80 {
81 return $this->fallbackDirsPsr4;
82 }
83
84 public function getClassMap()
85 {
86 return $this->classMap;
87 }
88
89 /**
90 * @param array $classMap Class to filename map
91 */
92 public function addClassMap(array $classMap)
93 {
94 if ($this->classMap) {
95 $this->classMap = array_merge($this->classMap, $classMap);
96 } else {
97 $this->classMap = $classMap;
98 }
99 }
100
101 /**
102 * Registers a set of PSR-0 directories for a given prefix, either
103 * appending or prepending to the ones previously set for this prefix.
104 *
105 * @param string $prefix The prefix
106 * @param array|string $paths The PSR-0 root directories
107 * @param bool $prepend Whether to prepend the directories
108 */
109 public function add($prefix, $paths, $prepend = false)
110 {
111 if (!$prefix) {
112 if ($prepend) {
113 $this->fallbackDirsPsr0 = array_merge(
114 (array) $paths,
115 $this->fallbackDirsPsr0
116 );
117 } else {
118 $this->fallbackDirsPsr0 = array_merge(
119 $this->fallbackDirsPsr0,
120 (array) $paths
121 );
122 }
123
124 return;
125 }
126
127 $first = $prefix[0];
128 if (!isset($this->prefixesPsr0[$first][$prefix])) {
129 $this->prefixesPsr0[$first][$prefix] = (array) $paths;
130
131 return;
132 }
133 if ($prepend) {
134 $this->prefixesPsr0[$first][$prefix] = array_merge(
135 (array) $paths,
136 $this->prefixesPsr0[$first][$prefix]
137 );
138 } else {
139 $this->prefixesPsr0[$first][$prefix] = array_merge(
140 $this->prefixesPsr0[$first][$prefix],
141 (array) $paths
142 );
143 }
144 }
145
146 /**
147 * Registers a set of PSR-4 directories for a given namespace, either
148 * appending or prepending to the ones previously set for this namespace.
149 *
150 * @param string $prefix The prefix/namespace, with trailing '\\'
151 * @param array|string $paths The PSR-4 base directories
152 * @param bool $prepend Whether to prepend the directories
153 *
154 * @throws \InvalidArgumentException
155 */
156 public function addPsr4($prefix, $paths, $prepend = false)
157 {
158 if (!$prefix) {
159 // Register directories for the root namespace.
160 if ($prepend) {
161 $this->fallbackDirsPsr4 = array_merge(
162 (array) $paths,
163 $this->fallbackDirsPsr4
164 );
165 } else {
166 $this->fallbackDirsPsr4 = array_merge(
167 $this->fallbackDirsPsr4,
168 (array) $paths
169 );
170 }
171 } elseif (!isset($this->prefixDirsPsr4[$prefix])) {
172 // Register directories for a new namespace.
173 $length = strlen($prefix);
174 if ('\\' !== $prefix[$length - 1]) {
175 throw new \InvalidArgumentException("A non-empty PSR-4 prefix must end with a namespace separator.");
176 }
177 $this->prefixLengthsPsr4[$prefix[0]][$prefix] = $length;
178 $this->prefixDirsPsr4[$prefix] = (array) $paths;
179 } elseif ($prepend) {
180 // Prepend directories for an already registered namespace.
181 $this->prefixDirsPsr4[$prefix] = array_merge(
182 (array) $paths,
183 $this->prefixDirsPsr4[$prefix]
184 );
185 } else {
186 // Append directories for an already registered namespace.
187 $this->prefixDirsPsr4[$prefix] = array_merge(
188 $this->prefixDirsPsr4[$prefix],
189 (array) $paths
190 );
191 }
192 }
193
194 /**
195 * Registers a set of PSR-0 directories for a given prefix,
196 * replacing any others previously set for this prefix.
197 *
198 * @param string $prefix The prefix
199 * @param array|string $paths The PSR-0 base directories
200 */
201 public function set($prefix, $paths)
202 {
203 if (!$prefix) {
204 $this->fallbackDirsPsr0 = (array) $paths;
205 } else {
206 $this->prefixesPsr0[$prefix[0]][$prefix] = (array) $paths;
207 }
208 }
209
210 /**
211 * Registers a set of PSR-4 directories for a given namespace,
212 * replacing any others previously set for this namespace.
213 *
214 * @param string $prefix The prefix/namespace, with trailing '\\'
215 * @param array|string $paths The PSR-4 base directories
216 *
217 * @throws \InvalidArgumentException
218 */
219 public function setPsr4($prefix, $paths)
220 {
221 if (!$prefix) {
222 $this->fallbackDirsPsr4 = (array) $paths;
223 } else {
224 $length = strlen($prefix);
225 if ('\\' !== $prefix[$length - 1]) {
226 throw new \InvalidArgumentException("A non-empty PSR-4 prefix must end with a namespace separator.");
227 }
228 $this->prefixLengthsPsr4[$prefix[0]][$prefix] = $length;
229 $this->prefixDirsPsr4[$prefix] = (array) $paths;
230 }
231 }
232
233 /**
234 * Turns on searching the include path for class files.
235 *
236 * @param bool $useIncludePath
237 */
238 public function setUseIncludePath($useIncludePath)
239 {
240 $this->useIncludePath = $useIncludePath;
241 }
242
243 /**
244 * Can be used to check if the autoloader uses the include path to check
245 * for classes.
246 *
247 * @return bool
248 */
249 public function getUseIncludePath()
250 {
251 return $this->useIncludePath;
252 }
253
254 /**
255 * Turns off searching the prefix and fallback directories for classes
256 * that have not been registered with the class map.
257 *
258 * @param bool $classMapAuthoritative
259 */
260 public function setClassMapAuthoritative($classMapAuthoritative)
261 {
262 $this->classMapAuthoritative = $classMapAuthoritative;
263 }
264
265 /**
266 * Should class lookup fail if not found in the current class map?
267 *
268 * @return bool
269 */
270 public function isClassMapAuthoritative()
271 {
272 return $this->classMapAuthoritative;
273 }
274
275 /**
276 * APCu prefix to use to cache found/not-found classes, if the extension is enabled.
277 *
278 * @param string|null $apcuPrefix
279 */
280 public function setApcuPrefix($apcuPrefix)
281 {
282 $this->apcuPrefix = function_exists('apcu_fetch') && filter_var(ini_get('apc.enabled'), FILTER_VALIDATE_BOOLEAN) ? $apcuPrefix : null;
283 }
284
285 /**
286 * The APCu prefix in use, or null if APCu caching is not enabled.
287 *
288 * @return string|null
289 */
290 public function getApcuPrefix()
291 {
292 return $this->apcuPrefix;
293 }
294
295 /**
296 * Registers this instance as an autoloader.
297 *
298 * @param bool $prepend Whether to prepend the autoloader or not
299 */
300 public function register($prepend = false)
301 {
302 spl_autoload_register(array($this, 'loadClass'), true, $prepend);
303 }
304
305 /**
306 * Unregisters this instance as an autoloader.
307 */
308 public function unregister()
309 {
310 spl_autoload_unregister(array($this, 'loadClass'));
311 }
312
313 /**
314 * Loads the given class or interface.
315 *
316 * @param string $class The name of the class
317 * @return bool|null True if loaded, null otherwise
318 */
319 public function loadClass($class)
320 {
321 if ($file = $this->findFile($class)) {
322 includeFile($file);
323
324 return true;
325 }
326 }
327
328 /**
329 * Finds the path to the file where the class is defined.
330 *
331 * @param string $class The name of the class
332 *
333 * @return string|false The path if found, false otherwise
334 */
335 public function findFile($class)
336 {
337 // class map lookup
338 if (isset($this->classMap[$class])) {
339 return $this->classMap[$class];
340 }
341 if ($this->classMapAuthoritative || isset($this->missingClasses[$class])) {
342 return false;
343 }
344 if (null !== $this->apcuPrefix) {
345 $file = apcu_fetch($this->apcuPrefix.$class, $hit);
346 if ($hit) {
347 return $file;
348 }
349 }
350
351 $file = $this->findFileWithExtension($class, '.php');
352
353 // Search for Hack files if we are running on HHVM
354 if (false === $file && defined('HHVM_VERSION')) {
355 $file = $this->findFileWithExtension($class, '.hh');
356 }
357
358 if (null !== $this->apcuPrefix) {
359 apcu_add($this->apcuPrefix.$class, $file);
360 }
361
362 if (false === $file) {
363 // Remember that this class does not exist.
364 $this->missingClasses[$class] = true;
365 }
366
367 return $file;
368 }
369
370 private function findFileWithExtension($class, $ext)
371 {
372 // PSR-4 lookup
373 $logicalPathPsr4 = strtr($class, '\\', DIRECTORY_SEPARATOR) . $ext;
374
375 $first = $class[0];
376 if (isset($this->prefixLengthsPsr4[$first])) {
377 $subPath = $class;
378 while (false !== $lastPos = strrpos($subPath, '\\')) {
379 $subPath = substr($subPath, 0, $lastPos);
380 $search = $subPath . '\\';
381 if (isset($this->prefixDirsPsr4[$search])) {
382 $pathEnd = DIRECTORY_SEPARATOR . substr($logicalPathPsr4, $lastPos + 1);
383 foreach ($this->prefixDirsPsr4[$search] as $dir) {
384 if (file_exists($file = $dir . $pathEnd)) {
385 return $file;
386 }
387 }
388 }
389 }
390 }
391
392 // PSR-4 fallback dirs
393 foreach ($this->fallbackDirsPsr4 as $dir) {
394 if (file_exists($file = $dir . DIRECTORY_SEPARATOR . $logicalPathPsr4)) {
395 return $file;
396 }
397 }
398
399 // PSR-0 lookup
400 if (false !== $pos = strrpos($class, '\\')) {
401 // namespaced class name
402 $logicalPathPsr0 = substr($logicalPathPsr4, 0, $pos + 1)
403 . strtr(substr($logicalPathPsr4, $pos + 1), '_', DIRECTORY_SEPARATOR);
404 } else {
405 // PEAR-like class name
406 $logicalPathPsr0 = strtr($class, '_', DIRECTORY_SEPARATOR) . $ext;
407 }
408
409 if (isset($this->prefixesPsr0[$first])) {
410 foreach ($this->prefixesPsr0[$first] as $prefix => $dirs) {
411 if (0 === strpos($class, $prefix)) {
412 foreach ($dirs as $dir) {
413 if (file_exists($file = $dir . DIRECTORY_SEPARATOR . $logicalPathPsr0)) {
414 return $file;
415 }
416 }
417 }
418 }
419 }
420
421 // PSR-0 fallback dirs
422 foreach ($this->fallbackDirsPsr0 as $dir) {
423 if (file_exists($file = $dir . DIRECTORY_SEPARATOR . $logicalPathPsr0)) {
424 return $file;
425 }
426 }
427
428 // PSR-0 include paths.
429 if ($this->useIncludePath && $file = stream_resolve_include_path($logicalPathPsr0)) {
430 return $file;
431 }
432
433 return false;
434 }
435}
436
437/**
438 * Scope isolated include.
439 *
440 * Prevents access to $this/self from included files.
441 */
442function includeFile($file)
443{
444 include $file;
445}