Merge pull request #3 from cedricziel/object-manager

Add ObjectManagerTypeResolver

Author: cedricziel

META-INF/plugin.xml

@@ -27,6 +27,7 @@
   <extensions defaultExtensionNs="com.intellij">
     <!-- Add your extensions here -->
     <php.typeProvider2 implementation="com.cedricziel.idea.typo3.provider.GeneralUtilityTypeProvider" />
+    <php.typeProvider2 implementation="com.cedricziel.idea.typo3.provider.ObjectManagerTypeProvider" />
   </extensions>
 
   <extensions defaultExtensionNs="com.jetbrains.php">

README.md

@@ -3,6 +3,7 @@
 ## Features
 
 * TypeProvider for `GeneralUtility::makeInstance`
+* TypeProvider for `ObjectManager::get`
 
 # License
 

src/com/cedricziel/idea/typo3/provider/ObjectManagerTypeProvider.java

@@ -0,0 +1,47 @@
+package com.cedricziel.idea.typo3.provider;
+
+import com.cedricziel.idea.typo3.psi.PhpElementsUtil;
+import com.intellij.openapi.project.DumbService;
+import com.intellij.psi.PsiElement;
+import com.jetbrains.php.PhpIndex;
+import com.jetbrains.php.lang.psi.elements.MethodReference;
+import com.jetbrains.php.lang.psi.elements.PhpReference;
+import org.jetbrains.annotations.Nullable;
+
+/**
+ * TypeProvider for `ObjectManager::get`
+ */
+public class ObjectManagerTypeProvider extends AbstractServiceLocatorTypeProvider {
+
+    @Override
+    public char getKey() {
+        return '\u0215';
+    }
+
+    @Nullable
+    @Override
+    public String getType(PsiElement psiElement) {
+        if (DumbService.getInstance(psiElement.getProject()).isDumb()) {
+            return null;
+        }
+
+        if (!(psiElement instanceof MethodReference) || !PhpElementsUtil.isMethodWithFirstStringOrFieldReference(psiElement, "get")) {
+            return null;
+        }
+
+        MethodReference methodReference = (MethodReference) psiElement;
+        if (methodReference.getParameters().length == 0) {
+            return null;
+        }
+
+        PsiElement firstParam = methodReference.getParameters()[0];
+        if (firstParam instanceof PhpReference) {
+            PhpReference ref = (PhpReference) firstParam;
+            if (ref.getText().toLowerCase().contains("::class")) {
+                return methodReference.getSignature() + "%" + ref.getSignature();
+            }
+        }
+
+        return null;
+    }
+}