Merge pull request #3 from jaikdean/entity-manager-interface

Stub EntityManagerInterface and ObjectManager
psalm · Jan 3, 2019 · 8156747 · 8156747
2 parents 30a00e6 + f8b8bfe
commit 8156747
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 10 deletions.
diff --git a/Plugin.php b/Plugin.php
@@ -11,8 +11,9 @@ class Plugin implements PluginEntryPointInterface
     public function __invoke(RegistrationInterface $psalm, ?SimpleXMLElement $config = null)
     {
         $psalm->addStubFile(__DIR__ . '/' . 'stubs/Collections.php');
-        $psalm->addStubFile(__DIR__ . '/' . 'stubs/EntityManager.php');
+        $psalm->addStubFile(__DIR__ . '/' . 'stubs/EntityManagerInterface.php');
         $psalm->addStubFile(__DIR__ . '/' . 'stubs/EntityRepository.php');
+        $psalm->addStubFile(__DIR__ . '/' . 'stubs/ObjectManager.php');
         $psalm->addStubFile(__DIR__ . '/' . 'stubs/Paginator.php');
     }
 }
diff --git a/stubs/EntityManager.php → stubs/EntityManagerInterface.php b/stubs/EntityManager.php → stubs/EntityManagerInterface.php
@@ -1,7 +1,7 @@
 <?php
 namespace Doctrine\ORM;
 
-class EntityManager implements EntityManagerInterface
+class EntityManagerInterface
 {
     /**
      * @template T
@@ -11,14 +11,6 @@ class EntityManager implements EntityManagerInterface
      */
     public function getRepository(string $entityName) {}
 
-    /**
-     * @template T
-     * @param class-string $className
-     * @template-typeof T $className
-     * @return Mapping\ClassMetadata<T>
-     */
-    public function getClassMetadata(string $className) {}
-
     /**
      * @template T
      * @param class-string $entityName

diff --git a/stubs/ObjectManager.php b/stubs/ObjectManager.php
@@ -0,0 +1,29 @@
+<?php
+namespace Doctrine\ORM;
+
+class ObjectManager
+{
+    /**
+     * @template T
+     * @param class-string $className
+     * @template-typeof T $className
+     * @return ObjectRepository<T>
+     */
+    public function getRepository(string $className) {}
+
+    /**
+     * @template T
+     * @param class-string $className
+     * @template-typeof T $className
+     * @return Mapping\ClassMetadata<T>
+     */
+    public function getClassMetadata(string $className) {}
+
+    /**
+     * @template T
+     * @param class-string $className
+     * @template-typeof T $className
+     * @return ?T
+     */
+    public function find(string $className, $id) {}
+}