modify

2021-07-27 00:35:50 +08:00
parent 170de0b822
commit b89554b6c6
6 changed files with 1285 additions and 1328 deletions
@@ -5,6 +5,7 @@ namespace Annotation;


 use Exception;
+use HttpServer\Http\Context;
 use ReflectionException;
 use ReflectionProperty;
 use Snowflake\Core\Str;
@@ -18,75 +19,89 @@ use Snowflake\Snowflake;
 {


-	/**
-	 * Inject constructor.
-	 * @param string $className
-	 * @param array $args
-	 */
-	public function __construct(private string $className, private array $args = [])
-	{
-	}
+    /**
+     * Inject constructor.
+     * @param string $className
+     * @param array $args
+     */
+    public function __construct(private string $className, public bool $withContext = false, private array $args = [])
+    {
+    }


-	/**
-	 * @param mixed $class
-	 * @param mixed|null $method
-	 * @return bool
-	 * @throws ReflectionException
-	 * @throws Exception
-	 */
-	public function execute(mixed $class, mixed $method = null): bool
-	{
-		$injectValue = $this->parseInjectValue();
-		if (!($method = $this->getProperty($class, $method))) {
-			return false;
-		}
-		/** @var ReflectionProperty $class */
-		if ($method->isPrivate() || $method->isProtected()) {
-			$method = 'set' . ucfirst(Str::convertUnderline($method->getName()));
-			if (!method_exists($class, $method)) {
-				return false;
-			}
-			$class->$method($injectValue);
-		} else {
-			$class->{$method->getName()} = $injectValue;
-		}
-		return true;
-	}
+    /**
+     * @param mixed $class
+     * @param mixed|null $method
+     * @return bool
+     * @throws ReflectionException
+     * @throws Exception
+     */
+    public function execute(mixed $class, mixed $method = null): bool
+    {
+        if (!($method = $this->getProperty($class, $method))) {
+            return false;
+        }
+        /** @var ReflectionProperty $class */
+        $injectValue = $this->parseInjectValue();
+        if ($method->isPrivate() || $method->isProtected()) {
+            $this->setter($class, $method, $injectValue);
+        } else {
+            $class->{$method->getName()} = $injectValue;
+        }
+        return true;
+    }


-	/**
-	 * @param $class
-	 * @param $method
-	 * @return ReflectionProperty|bool
-	 */
-	private function getProperty($class, $method): ReflectionProperty|bool
-	{
-		if ($method instanceof ReflectionProperty) {
-			return $method;
-		}
-		if (is_object($class)) $class = $class::class;
-		$method = Snowflake::getDi()->getClassProperty($class, $method);
-		if (!$method) {
-			return false;
-		}
-		return $method;
-	}
+    /**
+     * @param $class
+     * @param $method
+     * @param $injectValue
+     */
+    private function setter($class, $method, $injectValue)
+    {
+        $method = 'set' . ucfirst(Str::convertUnderline($method->getName()));
+        if (!method_exists($class, $method)) {
+            return;
+        }
+        $class->$method($injectValue);
+    }


-	/**
-	 * @return mixed
-	 * @throws Exception
-	 */
-	private function parseInjectValue(): mixed
-	{
-		if (class_exists($this->className)) {
-			return Snowflake::getDi()->get($this->className, $this->args);
-		} else if (Snowflake::app()->has($this->className)) {
-			return Snowflake::app()->get($this->className);
-		} else {
-			return $this->className;
-		}
-	}
+    /**
+     * @param $class
+     * @param $method
+     * @return ReflectionProperty|bool
+     */
+    private function getProperty($class, $method): ReflectionProperty|bool
+    {
+        if ($method instanceof ReflectionProperty && !$method->isStatic()) {
+            return $method;
+        }
+        if (is_object($class)) $class = $class::class;
+        $method = Snowflake::getDi()->getClassProperty($class, $method);
+        if (!$method || $method->isStatic()) {
+            return false;
+        }
+        return $method;
+    }
+
+
+    /**
+     * @return mixed
+     * @throws Exception
+     */
+    private function parseInjectValue(): mixed
+    {
+        if ($this->withContext) {
+            return Context::getContext($this->className);
+        }
+        if (class_exists($this->className)) {
+            return Snowflake::getDi()->get($this->className, $this->args);
+        } else if (Snowflake::app()->has($this->className)) {
+            return Snowflake::app()->get($this->className);
+        } else {
+            return $this->className;
+        }
+    }

 }