diff --git a/src/Nette/WebLoader.php b/src/Nette/WebLoader.php index 6888af8..1e7e6a4 100755 --- a/src/Nette/WebLoader.php +++ b/src/Nette/WebLoader.php @@ -9,6 +9,7 @@ use WebLoader\Compiler; use WebLoader\Enum\RenderMode; use WebLoader\File; +use WebLoader\FileCollection; /** * Web loader @@ -78,7 +79,16 @@ protected function getUrl(File $file): string */ public function render(): void { - $file = $this->compiler->generate(); + $hasArgs = func_num_args() > 0; + + if ($hasArgs) { + $backup = $this->compiler->getFileCollection(); + $newFiles = new FileCollection($backup->getRoot()); + $newFiles->addFiles(func_get_args()); + $this->compiler->setFileCollection($newFiles); + } + + $file = $this->compiler->generate(); if ($file === null) { return; @@ -91,6 +101,10 @@ public function render(): void }; echo $output, PHP_EOL; + + if ($hasArgs) { + $this->compiler->setFileCollection($backup); + } }