Merge pull request #78 from cedricziel/extension-name-index

Implement ExtensionNameStubIndex

Author: cedricziel

src/main/java/com/cedricziel/idea/typo3/index/ExtensionIndex.java

@@ -0,0 +1,80 @@
+package com.cedricziel.idea.typo3.index;
+
+import com.cedricziel.idea.typo3.index.externalizer.ObjectStreamDataExternalizer;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.util.indexing.*;
+import com.intellij.util.io.DataExternalizer;
+import com.intellij.util.io.EnumeratorStringDescriptor;
+import com.intellij.util.io.KeyDescriptor;
+import gnu.trove.THashMap;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.Map;
+
+public class ExtensionNameStubIndex extends FileBasedIndexExtension<String, String> {
+
+    private static ObjectStreamDataExternalizer<String> EXTERNALIZER = new ObjectStreamDataExternalizer<>();
+
+    public static final ID<String, String> KEY = ID.create("com.cedricziel.idea.typo3.index.extension_name");
+
+    private final KeyDescriptor<String> myKeyDescriptor = new EnumeratorStringDescriptor();
+
+    @NotNull
+    @Override
+    public ID<String, String> getName() {
+        return KEY;
+    }
+
+    @NotNull
+    @Override
+    public DataIndexer<String, String, FileContent> getIndexer() {
+        return virtualFile -> {
+            final Map<String, String> items = new THashMap<>();
+
+            VirtualFile file = virtualFile.getFile();
+            VirtualFile parentDirectory = file.getParent();
+            if (parentDirectory != null) {
+                VirtualFile ancestorDirectory = parentDirectory.getParent();
+                if (ancestorDirectory != null && ancestorDirectory.isDirectory()) {
+                    String ancestorDirectoryName = ancestorDirectory.getName();
+                    if (ancestorDirectoryName.equals("sysext") || ancestorDirectoryName.equals("ext")) {
+                        String path = file.getPath();
+                        items.put(parentDirectory.getName(), path);
+                    }
+
+                    // handle the case where the extension name can *not* be inferred from the directory name
+                }
+            }
+
+            return items;
+        };
+    }
+
+    @NotNull
+    @Override
+    public KeyDescriptor<String> getKeyDescriptor() {
+        return myKeyDescriptor;
+    }
+
+    @NotNull
+    @Override
+    public DataExternalizer<String> getValueExternalizer() {
+        return EXTERNALIZER;
+    }
+
+    @Override
+    public int getVersion() {
+        return 0;
+    }
+
+    @NotNull
+    @Override
+    public FileBasedIndex.InputFilter getInputFilter() {
+        return virtualFile -> virtualFile.getName().equals("ext_emconf.php");
+    }
+
+    @Override
+    public boolean dependsOnFileContent() {
+        return true;
+    }
+}

src/main/java/com/cedricziel/idea/typo3/index/ExtensionNameStubIndex.java

@@ -0,0 +1,41 @@
+package com.cedricziel.idea.typo3.index.externalizer;
+
+import com.intellij.util.io.DataExternalizer;
+import org.jetbrains.annotations.NotNull;
+
+import java.io.*;
+
+/**
+ * @author Daniel Espendiller <daniel@espendiller.net>
+ */
+public class ObjectStreamDataExternalizer<T extends Serializable> implements DataExternalizer<T> {
+
+    @Override
+    public void save(@NotNull DataOutput out, T value) throws IOException {
+        ByteArrayOutputStream stream = new ByteArrayOutputStream();
+        ObjectOutput output = new ObjectOutputStream(stream);
+
+        output.writeObject(value);
+
+        out.writeInt(stream.size());
+        out.write(stream.toByteArray());
+    }
+
+    @Override
+    public T read(@NotNull DataInput in) throws IOException {
+        int bufferSize = in.readInt();
+        byte[] buffer = new byte[bufferSize];
+        in.readFully(buffer, 0, bufferSize);
+
+        ByteArrayInputStream stream = new ByteArrayInputStream(buffer);
+        ObjectInput input  = new ObjectInputStream(stream);
+
+        T object = null;
+        try {
+            object = (T) input.readObject();
+        } catch (ClassNotFoundException | ClassCastException ignored) {
+        }
+
+        return object;
+    }
+}

src/main/java/com/cedricziel/idea/typo3/index/externalizer/ObjectStreamDataExternalizer.java

@@ -266,6 +266,7 @@ It is a great inspiration for possible solutions and parts of the code.</p>
     <extensions defaultExtensionNs="com.intellij">
         <!-- indexes -->
         <fileBasedIndex implementation="com.cedricziel.idea.typo3.index.CoreServiceMapStubIndex"/>
+        <fileBasedIndex implementation="com.cedricziel.idea.typo3.index.ExtensionNameStubIndex"/>
 
         <!-- completion -->
         <completion.contributor language="PHP"