Complete main packages in run configuration

Author: zolotov

src/com/goide/codeInsight/imports/GoImportPackageQuickFix.java

@@ -180,7 +180,7 @@ public String fun(@NotNull GoFile file) {
               return null;
             }
           }
-          if (!GoPsiImplUtil.canBeAutoImported(file, module)) {
+          if (!GoPsiImplUtil.canBeAutoImported(file, false, module)) {
             return null;
           }
           String importPath = file.getImportPath(vendoringEnabled);

src/com/goide/completion/GoAutoImportCompletionContributor.java

@@ -320,7 +320,7 @@ public boolean process(@NotNull GoNamedElement element) {
     @NotNull
     private Boolean cachedAllowed(@NotNull GoNamedElement element, @Nullable Boolean existingValue) {
       if (existingValue != null) return existingValue;
-      return GoPsiImplUtil.canBeAutoImported(element.getContainingFile(), myModule);
+      return GoPsiImplUtil.canBeAutoImported(element.getContainingFile(), false, myModule);
     }
 
     @NotNull

src/com/goide/completion/GoImportPathsCompletionProvider.java

@@ -58,14 +58,15 @@ protected void addCompletions(@NotNull CompletionParameters parameters, Processi
 
     Module module = ModuleUtilCore.findModuleForPsiElement(parameters.getOriginalFile());
     if (module != null) {
-      addCompletions(result, module, parameters.getOriginalFile(), GoUtil.goPathResolveScope(module, parameters.getOriginalFile()));
+      addCompletions(result, module, parameters.getOriginalFile(), GoUtil.goPathResolveScope(module, parameters.getOriginalFile()), false);
     }
   }
 
   public static void addCompletions(@NotNull CompletionResultSet result,
                                     @NotNull Module module,
                                     @Nullable PsiElement context,
-                                    @NotNull GlobalSearchScope scope) {
+                                    @NotNull GlobalSearchScope scope,
+                                    boolean allowMain) {
     Project project = module.getProject();
     boolean vendoringEnabled = GoVendoringUtil.isVendoringEnabled(module);
     String contextImportPath = GoCompletionUtil.getContextImportPath(context, vendoringEnabled);
@@ -80,7 +81,7 @@ public static void addCompletions(@NotNull CompletionResultSet result,
       if (directory == null) continue;
 
       GoFile goFile = (GoFile)psiFile;
-      if (!GoPsiImplUtil.canBeAutoImported(goFile, module)) continue;
+      if (!GoPsiImplUtil.canBeAutoImported(goFile, allowMain, module)) continue;
 
       String importPath = goFile.getImportPath(vendoringEnabled);
       if (StringUtil.isNotEmpty(importPath) && !excludedSettings.isExcluded(importPath) 

src/com/goide/psi/impl/GoPsiImplUtil.java

@@ -1289,8 +1289,8 @@ public static GoTypeSpec getTypeSpecSafe(@NotNull GoType type) {
     return ObjectUtils.tryCast(parent, GoTypeSpec.class);
   }
 
-  public static boolean canBeAutoImported(@NotNull GoFile file, @Nullable Module module) {
-    if (isBuiltinFile(file) || StringUtil.equals(file.getCanonicalPackageName(), GoConstants.MAIN)) {
+  public static boolean canBeAutoImported(@NotNull GoFile file, boolean allowMain, @Nullable Module module) {
+    if (isBuiltinFile(file) || !allowMain && StringUtil.equals(file.getCanonicalPackageName(), GoConstants.MAIN)) {
       return false;
     }
     if (!GoUtil.matchedForModuleBuildTarget(file, module) || GoUtil.isExcludedFile(file)) {

src/com/goide/runconfig/testing/ui/GoPackageFieldCompletionProvider.java

@@ -20,6 +20,7 @@
 import com.goide.util.GoUtil;
 import com.intellij.codeInsight.completion.CompletionResultSet;
 import com.intellij.openapi.module.Module;
+import com.intellij.psi.search.GlobalSearchScope;
 import com.intellij.util.Producer;
 import com.intellij.util.TextFieldCompletionProvider;
 import org.jetbrains.annotations.NotNull;
@@ -38,7 +39,8 @@ protected void addCompletionVariants(@NotNull String text,
                                        @NotNull CompletionResultSet result) {
     Module module = myModuleProducer.produce();
     if (module != null) {
-      GoImportPathsCompletionProvider.addCompletions(result, module, null, GoUtil.moduleScopeWithoutLibraries(module.getProject(), module));
+      GlobalSearchScope scope = GoUtil.moduleScopeWithoutLibraries(module.getProject(), module);
+      GoImportPathsCompletionProvider.addCompletions(result, module, null, scope, true);
     }
   }
 }