139 lines
3.5 KiB
PHP
139 lines
3.5 KiB
PHP
|
<?php
|
||
|
|
/**
|
||
|
|
* Autoloader Class.
|
||
|
|
*
|
||
|
|
* A basic PSR-4 autoloader for theme developers.
|
||
|
|
*
|
||
|
|
* @author WPTRT <themes@wordpress.org>
|
||
|
|
* @copyright 2019 WPTRT
|
||
|
|
* @license https://www.gnu.org/licenses/gpl-2.0.html GPL-2.0-or-later
|
||
|
|
* @link https://github.com/WPTRT/autoload
|
||
|
|
*/
|
||
|
|
|
||
|
|
namespace WPTRT\Autoload;
|
||
|
|
|
||
|
|
class Loader
|
||
|
|
{
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Array of loaders.
|
||
|
|
*
|
||
|
|
* @since 1.0.0
|
||
|
|
* @access protected
|
||
|
|
* @var array
|
||
|
|
*/
|
||
|
|
protected $loaders = [];
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Adds a new prefix and path to load.
|
||
|
|
*
|
||
|
|
* @since 1.0.0
|
||
|
|
* @access public
|
||
|
|
* @param string $prefix Namespace prefix.
|
||
|
|
* @param array|string $paths Absolute path(s) where to look for classes.
|
||
|
|
* @return void
|
||
|
|
*/
|
||
|
|
public function add($prefix, $paths)
|
||
|
|
{
|
||
|
|
foreach ((array) $paths as $path) {
|
||
|
|
$this->loaders[ $prefix ][] = $path;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Removes a loader by prefix or prefix + path.
|
||
|
|
*
|
||
|
|
* @since 1.0.0
|
||
|
|
* @access public
|
||
|
|
* @param string $prefix Namespace prefix.
|
||
|
|
* @param string $path Absolute path.
|
||
|
|
* @return void
|
||
|
|
*/
|
||
|
|
public function remove($prefix, $path = '')
|
||
|
|
{
|
||
|
|
|
||
|
|
// Remove specific loader if both the prefix and path are provided.
|
||
|
|
if ($path) {
|
||
|
|
if ($this->has($prefix, $path)) {
|
||
|
|
$key = array_search($path, $this->loaders[ $prefix ], true);
|
||
|
|
unset($this->loaders[ $prefix ][ $key ]);
|
||
|
|
}
|
||
|
|
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Remove all loaders for a prefix if no path is provided.
|
||
|
|
if ($this->has($prefix)) {
|
||
|
|
unset($this->loaders[ $prefix ]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Checks if a loader is already added.
|
||
|
|
*
|
||
|
|
* @since 1.0.0
|
||
|
|
* @access public
|
||
|
|
* @param string $prefix Namespace prefix.
|
||
|
|
* @param string $path Absolute path.
|
||
|
|
* @return bool
|
||
|
|
*/
|
||
|
|
public function has($prefix, $path = '')
|
||
|
|
{
|
||
|
|
if ($path) {
|
||
|
|
return isset($this->loaders[ $prefix ]) && in_array($path, $this->loaders[ $prefix ], true);
|
||
|
|
}
|
||
|
|
|
||
|
|
return isset($this->loaders[ $prefix ]);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Registers all loaders.
|
||
|
|
*
|
||
|
|
* @since 1.0.0
|
||
|
|
* @access public
|
||
|
|
* @return void
|
||
|
|
*/
|
||
|
|
public function register()
|
||
|
|
{
|
||
|
|
if ($this->loaders) {
|
||
|
|
spl_autoload_register(function ($class) {
|
||
|
|
$this->load($class);
|
||
|
|
}, true, true);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Loads a class if it's within the given namespace.
|
||
|
|
*
|
||
|
|
* @since 1.0.0
|
||
|
|
* @access protected
|
||
|
|
* @param string $class Fully-qualified class name.
|
||
|
|
* @return void
|
||
|
|
*/
|
||
|
|
protected function load($class)
|
||
|
|
{
|
||
|
|
foreach ($this->loaders as $prefix => $paths) {
|
||
|
|
|
||
|
|
// Continue if the class is not in our namespace.
|
||
|
|
if (0 !== strpos($class, $prefix)) {
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Build a class filename to append to the path.
|
||
|
|
$suffix = ltrim(str_replace($prefix, '', $class), '\\');
|
||
|
|
$suffix = DIRECTORY_SEPARATOR . str_replace('\\', DIRECTORY_SEPARATOR, $suffix) . '.php';
|
||
|
|
|
||
|
|
// Loop through the paths to see if we can find the file
|
||
|
|
// for the class.
|
||
|
|
foreach ($paths as $path) {
|
||
|
|
|
||
|
|
// Load the class file if it exists and return.
|
||
|
|
if (file_exists($file = realpath($path) . $suffix)) {
|
||
|
|
include $file;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|