Added support for "Edit GraphQL fragment" which creates a separate editor window for injected GraphQL (#164)

- Ensured that the in-memory LightVirtualFile in unwrapped to the original file that has a physical path on disk
- Updated find closes config to handle LightVirtualFile as well

Author: jimkyndemeyer

src/main/com/intellij/lang/jsgraphql/ide/project/GraphQLPsiSearchHelper.java

@@ -43,6 +43,7 @@
 import com.intellij.psi.search.UsageSearchContext;
 import com.intellij.psi.search.scope.packageSet.NamedScope;
 import com.intellij.psi.util.PsiTreeUtil;
+import com.intellij.testFramework.LightVirtualFile;
 import org.apache.commons.compress.utils.IOUtils;
 import org.jetbrains.annotations.NotNull;
 
@@ -143,6 +144,12 @@ private static VirtualFile getVirtualFile(PsiFile containingFile) {
         if (virtualFile == null) {
             // in memory PsiFile such as the completion PSI
             virtualFile = containingFile.getOriginalFile().getVirtualFile();
+        } else if (virtualFile instanceof LightVirtualFile) {
+            // in-memory files, e.g. when using "edit GraphQL fragment" on an injected GraphQL tagged template literal
+            VirtualFile originalFile = ((LightVirtualFile) virtualFile).getOriginalFile();
+            if (originalFile != null) {
+                return originalFile;
+            }
         }
         return virtualFile;
     }

src/main/com/intellij/lang/jsgraphql/ide/project/graphqlconfig/GraphQLConfigManager.java

@@ -189,6 +189,13 @@ public VirtualFile getClosestIncludingConfigFile(VirtualFile virtualFile) {
      * Gets the closest .graphqlconfig{.yml,.yaml} file even though it doesn't include the specified file.
      */
     public VirtualFile getClosestConfigFile(VirtualFile virtualFile) {
+        if (virtualFile instanceof LightVirtualFile) {
+            VirtualFile originalFile = ((LightVirtualFile) virtualFile).getOriginalFile();
+            if (originalFile != null) {
+                // need a path on disk to find the closest config, so use the original physical file if one has been set
+                virtualFile = originalFile;
+            }
+        }
         final Set<VirtualFile> contentRoots = getContentRoots(virtualFile);
         VirtualFile directory;
         if (virtualFile.getFileType() == ScratchFileType.INSTANCE) {
@@ -702,19 +709,20 @@ public GraphQLNamedScope getSchemaScope(VirtualFile virtualFile) {
                 if (virtualFileWithPath.get().getFileType() != ScratchFileType.INSTANCE) {
                     if (virtualFileWithPath.get() instanceof LightVirtualFile) {
                         // handle entry files
+                        final LightVirtualFile inMemoryVirtualFile = (LightVirtualFile) virtualFileWithPath.get();
                         configBaseDir = null;
                         for (Map.Entry<VirtualFile, GraphQLConfigData> entry : configPathToConfigurations.entrySet()) {
                             final GraphQLConfigData configData = entry.getValue();
                             GraphQLFile entryFile = getConfigurationEntryFile(configData);
                             boolean found = false;
-                            if (entryFile.getVirtualFile().equals(virtualFileWithPath.get())) {
+                            if (entryFile.getVirtualFile().equals(inMemoryVirtualFile)) {
                                 // the virtual file is an entry file for the specific config base (either the root schema or one of the nested graphql-config project schemas)
                                 configBaseDir = entry.getKey();
                                 found = true;
                             } else if (configData.projects != null) {
                                 for (Map.Entry<String, GraphQLResolvedConfigData> projectEntry : configData.projects.entrySet()) {
                                     entryFile = getConfigurationEntryFile(projectEntry.getValue());
-                                    if (entryFile.getVirtualFile().equals(virtualFileWithPath.get())) {
+                                    if (entryFile.getVirtualFile().equals(inMemoryVirtualFile)) {
                                         configBaseDir = entry.getKey();
                                         found = true;
                                         break;
@@ -726,10 +734,13 @@ public GraphQLNamedScope getSchemaScope(VirtualFile virtualFile) {
                             }
                         }
                         if (configBaseDir == null) {
-                            final PsiFile jsonIntrospectionFile = virtualFileWithPath.get().getUserData(GraphQLSchemaKeys.GRAPHQL_INTROSPECTION_SDL_TO_JSON);
+                            final PsiFile jsonIntrospectionFile = inMemoryVirtualFile.getUserData(GraphQLSchemaKeys.GRAPHQL_INTROSPECTION_SDL_TO_JSON);
                             if (jsonIntrospectionFile != null && jsonIntrospectionFile.getVirtualFile() != null) {
                                 // the file is the SDL derived from a JSON introspection file, so use the JSON file directory to find the associated config
                                 configBaseDir = jsonIntrospectionFile.getVirtualFile().getParent();
+                            } else if(inMemoryVirtualFile.getOriginalFile() != null) {
+                                // editing of injected fragments produce in-memory mapped version of the original file, e.g. editing GraphQL inside a JS as it's own editing window
+                                configBaseDir = inMemoryVirtualFile.getOriginalFile().getParent();
                             }
                         }
                     } else {

src/main/com/intellij/lang/jsgraphql/ide/project/graphqlconfig/GraphQLConfigPackageSet.java

@@ -17,6 +17,7 @@
 import com.intellij.psi.PsiFile;
 import com.intellij.psi.search.scope.packageSet.NamedScopesHolder;
 import com.intellij.psi.search.scope.packageSet.PackageSet;
+import com.intellij.testFramework.LightVirtualFile;
 import org.apache.commons.lang.StringUtils;
 import org.jetbrains.annotations.NotNull;
 
@@ -117,7 +118,17 @@ public boolean includesVirtualFile(@NotNull VirtualFile file) {
             // the file is the in-memory SDL derived from a JSON introspection file, so it's included if the JSON file is set as the schemaPath
             return jsonIntrospectionFile.getVirtualFile().getPath().equals(schemaFilePath);
         }
-        return includesFilePath.computeIfAbsent(file.getPath(), filePath -> {
+
+        String inclusionPath = file.getPath();
+        if (file instanceof LightVirtualFile) {
+            // the light file is potentially derived from a file on disk, so we should use the physical path to check for inclusion
+            final VirtualFile originalFile = ((LightVirtualFile) file).getOriginalFile();
+            if (originalFile != null) {
+                inclusionPath = originalFile.getPath();
+            }
+        }
+
+        return includesFilePath.computeIfAbsent(inclusionPath, filePath -> {
             if (filePath.equals(schemaFilePath)) {
                 // fast-path for always including the schema file if present
                 return true;