132 lines
3.3 KiB
PHP
132 lines
3.3 KiB
PHP
<?php
|
|
|
|
/**
|
|
* This file is part of the Nette Framework (https://nette.org)
|
|
* Copyright (c) 2004 David Grudl (https://davidgrudl.com)
|
|
*/
|
|
|
|
namespace Nette\DI;
|
|
|
|
use Nette;
|
|
|
|
|
|
/**
|
|
* DI container loader.
|
|
*/
|
|
class ContainerLoader
|
|
{
|
|
use Nette\SmartObject;
|
|
|
|
/** @var bool */
|
|
private $autoRebuild = false;
|
|
|
|
/** @var string */
|
|
private $tempDirectory;
|
|
|
|
|
|
public function __construct($tempDirectory, $autoRebuild = false)
|
|
{
|
|
$this->tempDirectory = $tempDirectory;
|
|
$this->autoRebuild = $autoRebuild;
|
|
}
|
|
|
|
|
|
/**
|
|
* @param callable function (Nette\DI\Compiler $compiler): string|null
|
|
* @param mixed
|
|
* @return string
|
|
*/
|
|
public function load($generator, $key = null)
|
|
{
|
|
if (!is_callable($generator)) { // back compatiblity
|
|
trigger_error(__METHOD__ . ': order of arguments has been swapped.', E_USER_DEPRECATED);
|
|
list($generator, $key) = [$key, $generator];
|
|
}
|
|
$class = $this->getClassName($key);
|
|
if (!class_exists($class, false)) {
|
|
$this->loadFile($class, $generator);
|
|
}
|
|
return $class;
|
|
}
|
|
|
|
|
|
/**
|
|
* @return string
|
|
*/
|
|
public function getClassName($key)
|
|
{
|
|
return 'Container_' . substr(md5(serialize($key)), 0, 10);
|
|
}
|
|
|
|
|
|
/**
|
|
* @return void
|
|
*/
|
|
private function loadFile($class, $generator)
|
|
{
|
|
$file = "$this->tempDirectory/$class.php";
|
|
if (!$this->isExpired($file) && (@include $file) !== false) { // @ file may not exist
|
|
return;
|
|
}
|
|
|
|
Nette\Utils\FileSystem::createDir($this->tempDirectory);
|
|
|
|
$handle = @fopen("$file.lock", 'c+'); // @ is escalated to exception
|
|
if (!$handle) {
|
|
throw new Nette\IOException("Unable to create file '$file.lock'. " . error_get_last()['message']);
|
|
} elseif (!@flock($handle, LOCK_EX)) { // @ is escalated to exception
|
|
throw new Nette\IOException("Unable to acquire exclusive lock on '$file.lock'. " . error_get_last()['message']);
|
|
}
|
|
|
|
if (!is_file($file) || $this->isExpired($file, $updatedMeta)) {
|
|
if (isset($updatedMeta)) {
|
|
$toWrite["$file.meta"] = $updatedMeta;
|
|
} else {
|
|
list($toWrite[$file], $toWrite["$file.meta"]) = $this->generate($class, $generator);
|
|
}
|
|
|
|
foreach ($toWrite as $name => $content) {
|
|
if (file_put_contents("$name.tmp", $content) !== strlen($content) || !rename("$name.tmp", $name)) {
|
|
@unlink("$name.tmp"); // @ - file may not exist
|
|
throw new Nette\IOException("Unable to create file '$name'.");
|
|
} elseif (function_exists('opcache_invalidate')) {
|
|
@opcache_invalidate($name, true); // @ can be restricted
|
|
}
|
|
}
|
|
}
|
|
|
|
if ((@include $file) === false) { // @ - error escalated to exception
|
|
throw new Nette\IOException("Unable to include '$file'.");
|
|
}
|
|
flock($handle, LOCK_UN);
|
|
}
|
|
|
|
|
|
private function isExpired($file, &$updatedMeta = null)
|
|
{
|
|
if ($this->autoRebuild) {
|
|
$meta = @unserialize((string) file_get_contents("$file.meta")); // @ - file may not exist
|
|
$orig = $meta[2];
|
|
return empty($meta[0])
|
|
|| DependencyChecker::isExpired(...$meta)
|
|
|| ($orig !== $meta[2] && $updatedMeta = serialize($meta));
|
|
}
|
|
return false;
|
|
}
|
|
|
|
|
|
/**
|
|
* @return array of (code, file[])
|
|
*/
|
|
protected function generate($class, $generator)
|
|
{
|
|
$compiler = new Compiler;
|
|
$compiler->setClassName($class);
|
|
$code = call_user_func_array($generator, [&$compiler]) ?: $compiler->compile();
|
|
return [
|
|
"<?php\n$code",
|
|
serialize($compiler->exportDependencies()),
|
|
];
|
|
}
|
|
}
|