fix extended model morph map by checking parent class map

packages/core/src/Base/ModelManifest.php

@@ -143,7 +143,7 @@ public function guessModelClass(string $modelContract): string
         return 'Lunar\\Models\\'.$shortName;
     }
 
-    public function isLunarModel(BaseModel $model): bool
+    public function isLunarModel(string|BaseModel $model): bool
     {
         $class = (new \ReflectionClass($model));
 

packages/core/src/Base/Traits/HasModelExtending.php

@@ -3,6 +3,7 @@
 namespace Lunar\Base\Traits;
 
 use Illuminate\Database\Eloquent\Builder;
+use Illuminate\Database\Eloquent\Relations\Relation;
 use Illuminate\Support\Arr;
 use Lunar\Facades\ModelManifest;
 
@@ -46,6 +47,23 @@ public static function modelClass(): string
         return ModelManifest::get($contractClass) ?? static::class;
     }
 
+    public function getMorphClass(): string
+    {
+        $morphMap = Relation::morphMap();
+
+        if ($customModelMorphMap = array_search(static::modelClass(), $morphMap, true)) {
+            return $customModelMorphMap;
+        }
+
+        $parentClass = get_parent_class(static::class);
+
+        if (ModelManifest::isLunarModel($parentClass) && $lunarModelMorphMap = array_search($parentClass, $morphMap, true)) {
+            return $lunarModelMorphMap;
+        }
+
+        return parent::getMorphClass();
+    }
+
     public static function isLunarInstance(): bool
     {
         return static::class == static::modelClass();

packages/core/src/Facades/ModelManifest.php

@@ -16,7 +16,7 @@
  * @method static string|null get(string $interfaceClass)
  * @method static string guessContractClass(string $modelClass)
  * @method static string guessModelClass(string $modelContract)
- * @method static bool isLunarModel(BaseModel $model)
+ * @method static bool isLunarModel(string|BaseModel $model)
  * @method static string getTable(BaseModel $model)
  * @method static void morphMap()
  * @method static string getMorphMapKey(string $className)

tests/core/Unit/Base/Traits/HasModelExtendingTest.php

@@ -45,3 +45,16 @@ function () {
     expect($newStaticMethod)->toBeInstanceOf(Collection::class);
     expect($newStaticMethod)->toHaveCount(3);
 });
+
+test('morph map is correct when models are extended', function () {
+    \Lunar\Facades\ModelManifest::replace(
+        \Lunar\Models\Contracts\Product::class,
+        \Lunar\Tests\Core\Stubs\Models\CustomProduct::class
+    );
+
+   expect((new \Lunar\Tests\Core\Stubs\Models\CustomProduct)->getMorphClass())
+       ->toBe('product')
+       ->and((new Product)->getMorphClass())
+       ->toBe('product');
+});
+