增加Macroable trait类

src/helper/Macroable.php

@@ -0,0 +1,66 @@
+<?php
+// +----------------------------------------------------------------------
+// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
+// +----------------------------------------------------------------------
+// | Copyright (c) 2006-2015 http://thinkphp.cn All rights reserved.
+// +----------------------------------------------------------------------
+// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
+// +----------------------------------------------------------------------
+// | Author: yunwuxin <[email protected]>
+// +----------------------------------------------------------------------
+namespace think\helper;
+
+use Closure;
+use think\exception\FuncNotFoundException;
+
+trait Macroable
+{
+    /**
+     * 方法注入.
+     *
+     * @var Closure[]
+     */
+    protected static $macro = [];
+
+    /**
+     * 设置方法注入.
+     *
+     * @param string  $method
+     * @param Closure $closure
+     *
+     * @return void
+     */
+    public static function macro(string $method, Closure $closure)
+    {
+        static::$macro[$method] = $closure;
+    }
+
+    /**
+     * 检查方法是否已经有注入
+     *
+     * @param  string  $name
+     * @return bool
+     */
+    public static function hasMacro(string $method)
+    {
+        return isset(static::$macro[$method]);
+    }
+
+    public function __call($method, $args)
+    {
+        if (!isset(static::$macro[$method])) {
+            throw new FuncNotFoundException('method not exists: ' . static::class . '::' . $method . '()', "{static::class}::{$method}");
+        }
+
+        return call_user_func_array(static::$macro[$method]->bindTo($this, static::class), $args);
+    }
+
+    public static function __callStatic($method, $args)
+    {
+        if (!isset(static::$macro[$method])) {
+            throw new FuncNotFoundException('method not exists: ' . static::class . '::' . $method . '()', "{static::class}::{$method}");
+        }
+
+        return call_user_func_array(static::$macro[$method]->bindTo(null, static::class), $args);        
+    }
+}