Fix logic to get module in Azure service linker

Author: Flanker32

PluginsAndFeatures/azure-toolkit-for-intellij/src/com/microsoft/azure/toolkit/intellij/common/AzureArtifactManager.java

@@ -5,7 +5,10 @@
 package com.microsoft.azure.toolkit.intellij.common;
 
 import com.intellij.openapi.externalSystem.model.project.ExternalProjectPojo;
+import com.intellij.openapi.module.Module;
 import com.intellij.openapi.project.Project;
+import com.intellij.openapi.roots.ProjectFileIndex;
+import com.intellij.openapi.vfs.LocalFileSystem;
 import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.packaging.artifacts.Artifact;
 import com.microsoft.azure.toolkit.lib.common.operation.AzureOperation;
@@ -15,16 +18,20 @@
 import org.apache.commons.compress.utils.FileNameUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 import org.jetbrains.idea.maven.model.MavenConstants;
 import org.jetbrains.idea.maven.project.MavenProject;
 import org.jetbrains.idea.maven.project.MavenProjectsManager;
 import org.jetbrains.plugins.gradle.model.ExternalProject;
 import org.jetbrains.plugins.gradle.service.project.data.ExternalProjectDataCache;
 
+import java.io.File;
 import java.util.*;
 import java.util.function.Predicate;
 import java.util.stream.Collectors;
 
+import static com.microsoft.azure.toolkit.intellij.common.AzureArtifactType.File;
+
 public class AzureArtifactManager {
     private static Map<Project, AzureArtifactManager> projectAzureArtifactManagerMap = new HashMap<>();
     private final Project project;
@@ -93,7 +100,7 @@ public AzureArtifact getAzureArtifactById(String artifactId) {
     }
 
     public AzureArtifact getAzureArtifactById(AzureArtifactType azureArtifactType, String artifactId) {
-        return azureArtifactType == AzureArtifactType.File ? AzureArtifact.createFromFile(artifactId) :
+        return azureArtifactType == File ? AzureArtifact.createFromFile(artifactId) :
                getAllSupportedAzureArtifacts().stream().filter(artifact -> StringUtils.equals(getArtifactIdentifier(
                 artifact), artifactId)).findFirst().orElse(null);
     }
@@ -121,6 +128,28 @@ public boolean equalsAzureArtifactIdentifier(AzureArtifact artifact1, AzureArtif
         return StringUtils.equals(getArtifactIdentifier(artifact1), getArtifactIdentifier(artifact2));
     }
 
+    @Nullable
+    @AzureOperation(
+        name = "common|artifact.get_module",
+        type = AzureOperation.Type.TASK
+    )
+    public Module getModuleFromAzureArtifact(AzureArtifact azureArtifact) {
+        if (azureArtifact == null || azureArtifact.getReferencedObject() == null) {
+            return null;
+        }
+        switch (azureArtifact.getType()) {
+            case Gradle:
+                final String gradleModulePath = ((ExternalProjectPojo) azureArtifact.getReferencedObject()).getPath();
+                final VirtualFile gradleVirtualFile = LocalFileSystem.getInstance().findFileByIoFile(new File(gradleModulePath));
+                return ProjectFileIndex.getInstance(project).getModuleForFile(gradleVirtualFile);
+            case Maven:
+                return ProjectFileIndex.getInstance(project).getModuleForFile(((MavenProject) azureArtifact.getReferencedObject()).getFile());
+            default:
+                // IntelliJ artifact is bind to project, can not get the related module, same for File artifact
+                return null;
+        }
+    }
+
     private String getGradleProjectId(ExternalProjectPojo gradleProjectPojo) {
         ExternalProject externalProject = getRelatedExternalProject(gradleProjectPojo);
         return Objects.nonNull(externalProject) ? externalProject.getQName() : null;

PluginsAndFeatures/azure-toolkit-for-intellij/src/com/microsoft/intellij/runner/LinkAzureServiceBeforeRunProvider.java

@@ -15,10 +15,7 @@
 import com.intellij.openapi.actionSystem.DataContext;
 import com.intellij.openapi.diagnostic.Logger;
 import com.intellij.openapi.module.Module;
-import com.intellij.openapi.roots.ProjectFileIndex;
 import com.intellij.openapi.util.Key;
-import com.intellij.openapi.vfs.LocalFileSystem;
-import com.intellij.openapi.vfs.VirtualFile;
 import com.microsoft.azure.toolkit.intellij.common.AzureArtifact;
 import com.microsoft.azure.toolkit.intellij.common.AzureArtifactManager;
 import com.microsoft.azure.toolkit.intellij.webapp.runner.webappconfig.WebAppConfiguration;
@@ -107,8 +104,7 @@ private String getModuleName(@NotNull RunConfiguration runConfiguration) {
             WebAppConfiguration webAppConfiguration = (WebAppConfiguration) runConfiguration;
             final AzureArtifact azureArtifact = AzureArtifactManager.getInstance(runConfiguration.getProject())
                     .getAzureArtifactById(webAppConfiguration.getAzureArtifactType(), webAppConfiguration.getArtifactIdentifier());
-            VirtualFile virtualFile = LocalFileSystem.getInstance().findFileByPath(azureArtifact.getTargetPath());
-            Module module = ProjectFileIndex.getInstance(runConfiguration.getProject()).getModuleForFile(virtualFile, false);
+            final Module module = AzureArtifactManager.getInstance(runConfiguration.getProject()).getModuleFromAzureArtifact(azureArtifact);
             return Objects.nonNull(module) ? module.getName() : StringUtils.EMPTY;
         }
         if (runConfiguration instanceof AbstractRunConfiguration