增加 多态关系的模型名映射别名的数组$morphMap

增加 多态关系的模型名映射别名的数组$morphMap

src/model/relation/MorphToMany.php

@@ -24,6 +24,13 @@
 class MorphToMany extends BelongsToMany
 {
 
+    /**
+     * 多态关系的模型名映射别名的数组
+     *
+     * @var array
+     */
+    protected static $morphMap = [];
+
     /**
      * 多态字段名
      * @var string
@@ -58,6 +65,9 @@ public function __construct(Model $parent, string $model, string $middle, string
         $this->morphType  = $morphType;
         $this->inverse    = $inverse;
         $this->morphClass = $inverse ? $model : get_class($parent);
+        if (isset(static::$morphMap[$this->morphClass])) {
+            $this->morphClass = static::$morphMap[$this->morphClass];
+        }
 
         $foreignKey = $inverse ? $morphKey : $localKey;
         $localKey   = $inverse ? $localKey : $morphKey;
@@ -460,4 +470,21 @@ protected function baseQuery(): void
         }
     }
 
+    /**
+     * 设置或获取多态关系的模型名映射别名的数组
+     *
+     * @param  array|null  $map
+     * @param  bool  $merge
+     * @return array
+     */
+    public static function morphMap(array $map = null, $merge = true): array
+    {
+        if (is_array($map)) {
+            static::$morphMap = $merge && static::$morphMap
+                ? $map + static::$morphMap : $map;
+        }
+
+        return static::$morphMap;
+    }
+
 }