From ebfc89e114fd536e66019bae5a15cb3d30467013 Mon Sep 17 00:00:00 2001
From: Yaroslav Pankratyev <yaroslav.pankratyev@jetbrains.com>
Date: Thu, 9 Nov 2017 20:12:37 +0700
Subject: [PATCH] IDEA-115118 DevKitUseScopeEnlarger to provide custom scope
 for extensions

---
 plugins/devkit/src/DevKitUseScopeEnlarger.java | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/plugins/devkit/src/DevKitUseScopeEnlarger.java b/plugins/devkit/src/DevKitUseScopeEnlarger.java
index 82757dee5b23..2a48e8d230d7 100644
--- a/plugins/devkit/src/DevKitUseScopeEnlarger.java
+++ b/plugins/devkit/src/DevKitUseScopeEnlarger.java
@@ -29,15 +29,13 @@ import com.intellij.util.xml.DomElement;
 import com.intellij.util.xml.DomService;
 import com.intellij.util.xml.DomTarget;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.idea.devkit.dom.Extension;
 import org.jetbrains.idea.devkit.dom.ExtensionPoint;
 import org.jetbrains.idea.devkit.dom.IdeaPlugin;
 import org.jetbrains.idea.devkit.util.PsiUtil;
 
 import java.util.Collection;
 
-/**
- * @author peter
- */
 public class DevKitUseScopeEnlarger extends UseScopeEnlarger {
 
   @Override
@@ -46,7 +44,7 @@ public class DevKitUseScopeEnlarger extends UseScopeEnlarger {
       PomTarget target = ((PomTargetPsiElement)element).getTarget();
       if (target instanceof DomTarget) {
         DomElement domElement = ((DomTarget)target).getDomElement();
-        if (domElement instanceof ExtensionPoint) {
+        if (domElement instanceof ExtensionPoint || domElement instanceof Extension) {
           return createAllPluginDescriptorFilesSearchScope(element);
         }
       }