map entity class with orm.yml file as linemarker #309

Author: Haehnchen

src/fr/adrienbrault/idea/symfony2plugin/Symfony2Icons.java

@@ -13,6 +13,7 @@ public class Symfony2Icons {
     public static final Icon SYMFONY = IconLoader.getIcon("icons/symfony.png");
     public static final Icon SYMFONY_TOOL_WINDOW = IconLoader.getIcon("icons/symfony_tool_window.png");
     public static final Icon DOCTRINE = IconLoader.getIcon("icons/doctrine.png");
+    public static final Icon DOCTRINE_LINE_MARKER = IconLoader.getIcon("icons/doctrine_line_marker.png");
     public static final Icon MONGODB = IconLoader.getIcon("icons/mongodb.png");
     public static final Icon ROUTE = IconLoader.getIcon("icons/route.png");
     public static final Icon ROUTE_WEAK = IconLoader.getIcon("icons/route_weak.png");

src/fr/adrienbrault/idea/symfony2plugin/config/ServiceLineMarkerProvider.java

@@ -6,16 +6,19 @@
 import com.intellij.codeInsight.navigation.NavigationGutterIconBuilder;
 import com.intellij.openapi.project.Project;
 import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiFile;
 import com.intellij.psi.util.PsiTreeUtil;
 import com.jetbrains.php.lang.psi.elements.*;
 import fr.adrienbrault.idea.symfony2plugin.Settings;
 import fr.adrienbrault.idea.symfony2plugin.Symfony2Icons;
 import fr.adrienbrault.idea.symfony2plugin.Symfony2InterfacesUtil;
 import fr.adrienbrault.idea.symfony2plugin.Symfony2ProjectComponent;
 import fr.adrienbrault.idea.symfony2plugin.dic.XmlServiceParser;
+import fr.adrienbrault.idea.symfony2plugin.doctrine.EntityHelper;
 import fr.adrienbrault.idea.symfony2plugin.form.util.FormUtil;
 import fr.adrienbrault.idea.symfony2plugin.stubs.ServiceIndexUtil;
 import fr.adrienbrault.idea.symfony2plugin.util.PhpElementsUtil;
+import fr.adrienbrault.idea.symfony2plugin.util.dict.DoctrineModel;
 import fr.adrienbrault.idea.symfony2plugin.util.service.ServiceXmlParserFactory;
 import org.apache.commons.lang.StringUtils;
 import org.jetbrains.annotations.NotNull;
@@ -55,6 +58,7 @@ public void collectSlowLineMarkers(@NotNull List<PsiElement> psiElements, @NotNu
 
             if(PhpElementsUtil.getClassNamePattern().accepts(psiElement)) {
                 this.classNameMarker(psiElement, results);
+                this.entityClassMarker(psiElement, results);
             }
 
             if(phpHighlightServices) {
@@ -121,6 +125,38 @@ private void classNameMarker(PsiElement psiElement, Collection<? super RelatedIt
 
     }
 
+    private void entityClassMarker(PsiElement psiElement, Collection<? super RelatedItemLineMarkerInfo> result) {
+
+        PsiElement phpClassContext = psiElement.getContext();
+        if(!(phpClassContext instanceof PhpClass)) {
+            return;
+        }
+
+        String originFqn = ((PhpClass) phpClassContext).getPresentableFQN();
+        if(originFqn == null || !originFqn.toLowerCase().contains("entity")) {
+            return;
+        }
+
+        for(DoctrineModel doctrineModel: EntityHelper.getModelClasses(psiElement.getProject())) {
+            PhpClass phpClass = doctrineModel.getPhpClass();
+            if(phpClass != null) {
+                String presentableFQN = phpClass.getPresentableFQN();
+
+                if(presentableFQN != null && presentableFQN.equals(originFqn)) {
+                    PsiFile psiFile = EntityHelper.getModelConfigFile(phpClass);
+                    if(psiFile != null) {
+                        NavigationGutterIconBuilder<PsiElement> builder = NavigationGutterIconBuilder.create(Symfony2Icons.DOCTRINE_LINE_MARKER).
+                            setTarget(psiFile).
+                            setTooltipText("Navigate to model");
+
+                        result.add(builder.createLineMarkerInfo(psiElement));
+                    }
+                }
+            }
+        }
+
+    }
+
     private void formNameMarker(PsiElement psiElement, Collection<? super RelatedItemLineMarkerInfo> result) {
 
         if(!(psiElement instanceof StringLiteralExpression)) {