📦 Filter out unwanted fields

Author: johnmai-dev

src/main/java/com/github/maiqingqiang/goormhelper/orm/ORMCompletionProvider.java

@@ -14,6 +14,7 @@
 import com.intellij.codeInsight.completion.CompletionParameters;
 import com.intellij.codeInsight.completion.CompletionProvider;
 import com.intellij.codeInsight.completion.CompletionResultSet;
+import com.intellij.codeInsight.completion.CompletionUtil;
 import com.intellij.codeInsight.lookup.LookupElementBuilder;
 import com.intellij.openapi.diagnostic.Logger;
 import com.intellij.openapi.project.Project;
@@ -42,7 +43,7 @@ protected void addCompletions(@NotNull CompletionParameters parameters, @NotNull
 
         PsiElement currentElement = parameters.getPosition();
 
-        LOG.info("currentElement: " + currentElement.getText());
+        LOG.info("currentElement: " + currentElement + " text: " + currentElement.getText());
 
         Project project = currentElement.getProject();
 
@@ -351,15 +352,20 @@ private void scanFields(@NotNull CompletionParameters parameters, GoCallableDesc
                     comment = GoDocumentationProvider.getCommentText(GoDocumentationProvider.getCommentsForElement(field), false);
                 }
 
-                addElement(result, column, comment, type, goTypeSpec);
+                GoStringLiteral goStringLiteral = (GoStringLiteral) parameters.getPosition().getParent();
+                String currentString = goStringLiteral.getDecodedText().replace(CompletionUtil.DUMMY_IDENTIFIER, "");
+
+                if (column != null && !column.contains(currentString.trim())) continue;
+
+                addElement(parameters, result, column, comment, type, goTypeSpec);
 
                 if (!(parameters.getPosition().getParent().getParent() instanceof GoKey)) {
                     Map<GoCallableDescriptor, List<String>> queryExpr = queryExpr();
                     if (queryExpr != null) {
                         List<String> whereExpr = queryExpr.get(descriptor);
                         if (whereExpr != null) {
                             for (String s : whereExpr) {
-                                addElement(result, String.format(s, column), comment, type, goTypeSpec);
+                                addElement(parameters, result, String.format(s, column), comment, type, goTypeSpec);
                             }
                         }
                     }
@@ -368,7 +374,7 @@ private void scanFields(@NotNull CompletionParameters parameters, GoCallableDesc
         }
     }
 
-    private void addElement(@NotNull CompletionResultSet result, String column, String comment, String type, @NotNull GoTypeSpec goTypeSpec) {
+    private void addElement(@NotNull CompletionParameters parameters, @NotNull CompletionResultSet result, String column, String comment, String type, @NotNull GoTypeSpec goTypeSpec) {
         LookupElementBuilder builder = LookupElementBuilder
                 .createWithSmartPointer(column, goTypeSpec)
                 .withPresentableText(column)