use custom insert handler inside bundle resources file completion for #185

Author: Haehnchen

src/fr/adrienbrault/idea/symfony2plugin/util/SymfonyBundleFileCompletionProvider.java

@@ -12,6 +12,7 @@
 import com.jetbrains.php.PhpIndex;
 import fr.adrienbrault.idea.symfony2plugin.Symfony2ProjectComponent;
 import fr.adrienbrault.idea.symfony2plugin.util.dict.BundleFile;
+import fr.adrienbrault.idea.symfony2plugin.util.dict.ResourceFileInsertHandler;
 import fr.adrienbrault.idea.symfony2plugin.util.dict.SymfonyBundle;
 import fr.adrienbrault.idea.symfony2plugin.util.dict.SymfonyBundleFileLookupElement;
 import org.jetbrains.annotations.NotNull;
@@ -50,7 +51,7 @@ protected void addCompletions(@NotNull CompletionParameters completionParameters
         }
 
         for(BundleFile bundleFile : bundleFiles) {
-            completionResultSet.addElement(new SymfonyBundleFileLookupElement(bundleFile));
+            completionResultSet.addElement(new SymfonyBundleFileLookupElement(bundleFile, ResourceFileInsertHandler.getInstance()));
         }
 
     }

src/fr/adrienbrault/idea/symfony2plugin/util/dict/ResourceFileInsertHandler.java

@@ -0,0 +1,43 @@
+package fr.adrienbrault.idea.symfony2plugin.util.dict;
+
+import com.intellij.codeInsight.completion.InsertHandler;
+import com.intellij.codeInsight.completion.InsertionContext;
+import com.intellij.codeInsight.lookup.LookupElement;
+import com.intellij.openapi.editor.Editor;
+import com.intellij.psi.PsiElement;
+import com.jetbrains.php.completion.insert.PhpInsertHandlerUtil;
+import com.jetbrains.php.lang.psi.elements.PhpClass;
+import fr.adrienbrault.idea.symfony2plugin.util.PsiElementUtils;
+import org.jetbrains.annotations.NotNull;
+
+
+public class ResourceFileInsertHandler implements InsertHandler<LookupElement> {
+
+    private static final ResourceFileInsertHandler instance = new ResourceFileInsertHandler();
+
+    public void handleInsert(@NotNull InsertionContext context, @NotNull LookupElement lookupElement) {
+
+        // resource: "@AsseticBundle/Resources/config/filters/cssimport.xml"
+        // if no @ is before cursor add one
+        if(!isStringBeforeCaret(context.getEditor(), context, "@")) {
+            context.getDocument().insertString(context.getStartOffset(), "@");
+        }
+
+    }
+
+    public static boolean isStringBeforeCaret(@NotNull Editor editor, InsertionContext context, @NotNull String string) {
+
+        String fileText = editor.getDocument().getText();
+        if (fileText.length() < string.length()) {
+            return false;
+        }
+
+        return fileText.substring(context.getStartOffset() - string.length(), context.getStartOffset()).equals(string);
+    }
+
+
+    public static ResourceFileInsertHandler getInstance(){
+        return instance;
+    }
+
+}

src/fr/adrienbrault/idea/symfony2plugin/util/dict/SymfonyBundleFileLookupElement.java

@@ -1,5 +1,7 @@
 package fr.adrienbrault.idea.symfony2plugin.util.dict;
 
+import com.intellij.codeInsight.completion.InsertHandler;
+import com.intellij.codeInsight.completion.InsertionContext;
 import com.intellij.codeInsight.lookup.LookupElement;
 import com.intellij.codeInsight.lookup.LookupElementPresentation;
 import com.intellij.util.IconUtil;
@@ -11,16 +13,42 @@
 public class SymfonyBundleFileLookupElement extends LookupElement {
 
     private BundleFile bundleFile;
+    private InsertHandler<LookupElement> insertHandler = null;
 
     public SymfonyBundleFileLookupElement(BundleFile bundleFile) {
         this.bundleFile = bundleFile;
     }
 
+    public SymfonyBundleFileLookupElement(BundleFile bundleFile, InsertHandler<LookupElement> insertHandler) {
+        this(bundleFile);
+        this.insertHandler = insertHandler;
+    }
+
     @NotNull
     @Override
     public String getLookupString() {
         String shortcutName = this.bundleFile.getShortcutPath();
-        return shortcutName != null ? shortcutName : "";
+        if(shortcutName == null) {
+            return "";
+        }
+
+        // we strip any control char, so only use the pathname
+        if(shortcutName.startsWith("@")) {
+            shortcutName = shortcutName.substring(1);
+        }
+
+        return shortcutName;
+    }
+
+    @Override
+    public void handleInsert(InsertionContext context) {
+
+        if(this.insertHandler != null) {
+            this.insertHandler.handleInsert(context, this);
+            return;
+        }
+
+        super.handleInsert(context);
     }
 
     public void renderElement(LookupElementPresentation presentation) {