eee

src/Annotate/DeferRegistry.php

@@ -0,0 +1,127 @@
+<?php
+declare(strict_types=1);
+
+namespace Kiri\Router\Annotate;
+
+class DeferRegistry
+{
+
+	/**
+	 * @var array<string, Defer[]>  "ClassName::method" => Defer[]
+	 */
+	private static array $registry = [];
+
+
+	/**
+	 * @param string $class
+	 * @param string $method
+	 * @param Defer $defer
+	 * @return void
+	 */
+	public static function add(string $class, string $method, Defer $defer): void
+	{
+		$key = self::key($class, $method);
+		self::$registry[$key][] = $defer;
+	}
+
+
+	/**
+	 * @param string $class
+	 * @param string $method
+	 * @return Defer[]
+	 */
+	public static function get(string $class, string $method): array
+	{
+		return self::$registry[self::key($class, $method)] ?? [];
+	}
+
+
+	/**
+	 * @param string $class
+	 * @return bool
+	 */
+	public static function hasAny(string $class): bool
+	{
+		$prefix = $class . '::';
+		foreach (array_keys(self::$registry) as $key) {
+			if (str_starts_with($key, $prefix)) {
+				return true;
+			}
+		}
+		return false;
+	}
+
+
+	/**
+	 * @param string $class
+	 * @return array<string, Defer[]>  method => Defer[]
+	 */
+	public static function getAll(string $class): array
+	{
+		$result = [];
+		$prefix = $class . '::';
+		foreach (self::$registry as $key => $defers) {
+			if (str_starts_with($key, $prefix)) {
+				$method = substr($key, strlen($prefix));
+				$result[$method] = $defers;
+			}
+		}
+		return $result;
+	}
+
+
+	/**
+	 * @param string $class
+	 * @param string $method
+	 * @return void
+	 */
+	public static function execute(string $class, string $method): void
+	{
+		$key = self::key($class, $method);
+		if (!isset(self::$registry[$key])) {
+			return;
+		}
+
+		$defers = self::$registry[$key];
+		unset(self::$registry[$key]);
+
+		foreach ($defers as $defer) {
+			try {
+				$callback = $defer->callback;
+				$params   = $defer->params;
+
+				if (is_array($callback)) {
+					[$cbClass, $cbMethod] = $callback;
+					$instance = \Kiri::getDi()->get($cbClass);
+					call_user_func([$instance, $cbMethod], ...$params);
+				} else {
+					$instance = \Kiri::getDi()->get($callback);
+					call_user_func([$instance, '__invoke'], ...$params);
+				}
+			} catch (\Throwable $throwable) {
+				\Kiri::getLogger()->error('Defer callback failed: ' . $throwable->getMessage());
+			}
+		}
+	}
+
+
+	/**
+	 * @return void
+	 */
+	public static function clear(): void
+	{
+		self::$registry = [];
+	}
+
+
+	/**
+	 * @param string $class
+	 * @param string $method
+	 * @return string
+	 */
+	private static function key(string $class, string $method): string
+	{
+		return $class . '::' . $method;
+	}
+
+}