Improved registration of Latte functions, providers and macros

Author: tg666

phpstan.neon

@@ -11,3 +11,4 @@ parameters:
 	excludePaths:
 		analyse:
 			- src/Bridge/Latte/AssetMacroSet.php
+			- src/Bridge/Latte/AssetLatte2Extension.php

src/Bridge/Latte/AssetExtension.php

@@ -0,0 +1,28 @@
+<?php
+
+declare(strict_types=1);
+
+namespace SixtyEightPublishers\Asset\Bridge\Latte;
+
+use Symfony\Component\Asset\Packages;
+
+/**
+ * @internal
+ */
+final class AssetFunctionSet
+{
+	private function __construct()
+	{
+	}
+
+	/**
+	 * @return array<string, callable>
+	 */
+	public static function functions(Packages $packages): array
+	{
+		return [
+			'asset' => static fn (string $path, ?string $packageName = NULL): string => $packages->getUrl($path, $packageName),
+			'asset_version' => static fn (string $path, ?string $packageName = NULL): string => $packages->getVersion($path, $packageName),
+		];
+	}
+}

src/Bridge/Latte/AssetFunctionSet.php

@@ -0,0 +1,30 @@
+<?php
+
+declare(strict_types=1);
+
+namespace SixtyEightPublishers\Asset\Bridge\Latte;
+
+use Latte\Engine;
+use Symfony\Component\Asset\Packages;
+
+final class AssetLatte2Extension
+{
+	private function __construct()
+	{
+	}
+
+	public static function extend(Engine $engine, Packages $packages): void
+	{
+		foreach (AssetProviderSet::providers($packages) as $providerName => $provider) {
+			$engine->addProvider($providerName, $provider);
+		}
+
+		foreach (AssetFunctionSet::functions($packages) as $functionName => $functionCallback) {
+			$engine->addFunction($functionName, $functionCallback);
+		}
+
+		$engine->onCompile[] = static function (Engine $engine): void {
+			AssetMacroSet::install($engine->getCompiler());
+		};
+	}
+}

src/Bridge/Latte/AssetLatte2Extension.php

@@ -0,0 +1,41 @@
+<?php
+
+declare(strict_types=1);
+
+namespace SixtyEightPublishers\Asset\Bridge\Latte;
+
+use Latte\Extension;
+use Symfony\Component\Asset\Packages;
+use SixtyEightPublishers\Asset\Bridge\Latte\Node\AssetNode;
+use SixtyEightPublishers\Asset\Bridge\Latte\Node\AssetVersionNode;
+
+final class AssetLatte3Extension extends Extension
+{
+	private Packages $packages;
+
+	public function __construct(Packages $packages)
+	{
+		$this->packages = $packages;
+	}
+
+	/**
+	 * @return array{asset: callable, asset_version: callable}
+	 */
+	public function getTags(): array
+	{
+		return [
+			'asset' => [AssetNode::class, 'create'],
+			'asset_version' => [AssetVersionNode::class, 'create'],
+		];
+	}
+
+	public function getFunctions(): array
+	{
+		return AssetFunctionSet::functions($this->packages);
+	}
+
+	public function getProviders(): array
+	{
+		return AssetProviderSet::providers($this->packages);
+	}
+}

src/Bridge/Latte/AssetLatte3Extension.php

@@ -0,0 +1,27 @@
+<?php
+
+declare(strict_types=1);
+
+namespace SixtyEightPublishers\Asset\Bridge\Latte;
+
+use Symfony\Component\Asset\Packages;
+
+/**
+ * @internal
+ */
+final class AssetProviderSet
+{
+	private function __construct()
+	{
+	}
+
+	/**
+	 * @return array{symfonyPackages: Packages}
+	 */
+	public static function providers(Packages $packages): array
+	{
+		return [
+			'symfonyPackages' => $packages,
+		];
+	}
+}

src/Bridge/Latte/AssetProviderSet.php

@@ -0,0 +1,56 @@
+<?php
+
+declare(strict_types=1);
+
+namespace SixtyEightPublishers\Asset\Bridge\Latte\Node;
+
+use Generator;
+use Latte\Compiler\Tag;
+use Latte\Compiler\PrintContext;
+use Latte\Compiler\Nodes\StatementNode;
+use Latte\Compiler\Nodes\Php\ExpressionNode;
+
+/**
+ * {asset, path [, packageName]}
+ */
+final class AssetNode extends StatementNode
+{
+	public ExpressionNode $path;
+
+	public ?ExpressionNode $packageName = NULL;
+
+	/**
+	 * @throws \Latte\CompileException
+	 */
+	public static function create(Tag $tag): self
+	{
+		$tag->expectArguments();
+		$node = new self;
+		$node->path = $tag->parser->parseUnquotedStringOrExpression();
+
+		if ($tag->parser->stream->tryConsume(',')) {
+			$node->packageName = $tag->parser->parseUnquotedStringOrExpression();
+		}
+
+		return $node;
+	}
+
+	public function print(PrintContext $context): string
+	{
+		return $context->format(
+			'echo %escape($this->global->symfonyPackages->getUrl(%node, %node?)) %line;',
+			$this->path,
+			$this->packageName,
+			$this->position,
+		);
+	}
+
+	public function &getIterator(): Generator
+	{
+		yield $this->path;
+
+		if (NULL !== $this->packageName) {
+			yield $this->packageName;
+		}
+	}
+}

src/Bridge/Latte/Node/AssetNode.php

@@ -0,0 +1,56 @@
+<?php
+
+declare(strict_types=1);
+
+namespace SixtyEightPublishers\Asset\Bridge\Latte\Node;
+
+use Generator;
+use Latte\Compiler\Tag;
+use Latte\Compiler\PrintContext;
+use Latte\Compiler\Nodes\StatementNode;
+use Latte\Compiler\Nodes\Php\ExpressionNode;
+
+/**
+ * {asset_version, path [, packageName]}
+ */
+final class AssetVersionNode extends StatementNode
+{
+	public ExpressionNode $path;
+
+	public ?ExpressionNode $packageName = NULL;
+
+	/**
+	 * @throws \Latte\CompileException
+	 */
+	public static function create(Tag $tag): self
+	{
+		$tag->expectArguments();
+		$node = new self;
+		$node->path = $tag->parser->parseUnquotedStringOrExpression();
+
+		if ($tag->parser->stream->tryConsume(',')) {
+			$node->packageName = $tag->parser->parseUnquotedStringOrExpression();
+		}
+
+		return $node;
+	}
+
+	public function print(PrintContext $context): string
+	{
+		return $context->format(
+			'echo %escape($this->global->symfonyPackages->getVersion(%node, %node?)) %line;',
+			$this->path,
+			$this->packageName,
+			$this->position,
+		);
+	}
+
+	public function &getIterator(): Generator
+	{
+		yield $this->path;
+
+		if (NULL !== $this->packageName) {
+			yield $this->packageName;
+		}
+	}
+}