DEV2-2327 goto def [[WIP]]

build.gradle

@@ -50,7 +50,7 @@ test {
 
 intellij {
     version = '2019.3'
-    type = 'IC'
+    type = 'IU'
     updateSinceUntilBuild = false
 }
 

src/main/java/com/tabnine/inline/InlineCompletionHandler.java

@@ -27,6 +27,8 @@
 import com.tabnine.intellij.completions.CompletionUtils;
 import com.tabnine.prediction.CompletionFacade;
 import com.tabnine.prediction.TabNineCompletion;
+import com.tabnine.psi.LocationLink;
+import com.tabnine.psi.importsResolver.ImportsResolverKt;
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;
@@ -62,6 +64,10 @@ public void retrieveAndShowCompletion(
       @Nullable TabNineCompletion lastShownSuggestion,
       @NotNull String userInput,
       @NotNull CompletionAdjustment completionAdjustment) {
+    long start = System.currentTimeMillis();
+    List<LocationLink> links = ImportsResolverKt.resolveImportsLinksInFile(editor);
+    long end = System.currentTimeMillis();
+    Logger.getInstance(getClass()).warn("took " + (end - start) + "ms - links: " + links);
     Integer tabSize = GraphicsUtilsKt.getTabSize(editor);
 
     ObjectUtils.doIfNotNull(lastFetchInBackgroundTask, task -> task.cancel(false));

src/main/java/com/tabnine/psi/LocationLink.kt

@@ -0,0 +1,9 @@
+package com.tabnine.psi
+
+import java.net.URI
+
+data class Position(val line: Int, val column: Int)
+
+data class Range(val start: Position, val end: Position)
+
+data class LocationLink(val targetUri: URI, val targetRange: Range)

src/main/java/com/tabnine/psi/importsResolver/ImportsResolver.kt

@@ -0,0 +1,69 @@
+package com.tabnine.psi.importsResolver
+
+import com.intellij.openapi.editor.Editor
+import com.intellij.psi.PsiDocumentManager
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiFile
+import com.tabnine.psi.LocationLink
+import com.tabnine.psi.Position
+import com.tabnine.psi.Range
+import java.net.URI
+
+abstract class ImportsResolver {
+    abstract fun potentialElementsPredicate(element: PsiElement): Boolean
+
+    fun resolveImportsLinksInFile(psiFile: PsiFile): List<LocationLink> {
+        return psiFile.children
+            ?.filter { potentialElementsPredicate(it) }
+            ?.flatMap { resolveImportReference(it) } ?: emptyList()
+    }
+}
+
+fun resolveImportsLinksInFile(editor: Editor): List<LocationLink> {
+    val psiFile = editor.project
+        ?.let { PsiDocumentManager.getInstance(it).getPsiFile(editor.document) }
+        ?: return emptyList()
+    val resolver = getImportsResolver(psiFile) ?: return emptyList()
+    return resolver.resolveImportsLinksInFile(psiFile)
+}
+
+private fun resolveImportReference(element: PsiElement): List<LocationLink> {
+    val referee = element.reference?.resolve()
+        ?: return element.children?.flatMap { resolveImportReference(it) } ?: emptyList()
+
+    if (!(referee.isValid && referee.isWritable)) return emptyList()
+    val filePath = referee.containingFile?.virtualFile?.url?.let(URI::create) ?: return emptyList()
+    val range = referee.textRange ?: return emptyList()
+    val text = referee.containingFile?.text ?: return emptyList()
+    val linesAccumulativeLengths = linesAccumulativeLengths(text)
+
+    val startPos = toLogicalPosition(linesAccumulativeLengths, range.startOffset) ?: return emptyList()
+    val endPos = toLogicalPosition(linesAccumulativeLengths, range.endOffset) ?: return emptyList()
+
+    return listOf(LocationLink(filePath, Range(startPos, endPos)))
+}
+
+private fun toLogicalPosition(linesAccumulativeLengths: List<Int>, offset: Int): Position? {
+    val line = linesAccumulativeLengths.indexOfFirst { it >= offset }
+
+    if (line == -1) return null
+    if (line == 0) return Position(0, offset)
+
+    val column = offset - 1 - linesAccumulativeLengths[line - 1]
+    return Position(line, column)
+}
+
+private fun linesAccumulativeLengths(text: String): List<Int> {
+    val linesAccumulativeLengths = mutableListOf<Int>()
+    var index = text.indexOf('\n')
+
+    while (index != -1) {
+        linesAccumulativeLengths.add(index)
+        val nextIndex = text.indexOf('\n', index + 1)
+        index = nextIndex
+    }
+
+    linesAccumulativeLengths.add(text.length)
+
+    return linesAccumulativeLengths
+}

src/main/java/com/tabnine/psi/importsResolver/JavaImportsResolver.kt

@@ -0,0 +1,9 @@
+package com.tabnine.psi.importsResolver
+
+import com.intellij.psi.PsiElement
+
+class JavaImportsResolver : ImportsResolver() {
+    override fun potentialElementsPredicate(element: PsiElement): Boolean {
+        return element.text.contains("import")
+    }
+}

src/main/java/com/tabnine/psi/importsResolver/JavaScriptImportsResolver.kt

@@ -0,0 +1,9 @@
+package com.tabnine.psi.importsResolver
+
+import com.intellij.psi.PsiElement
+
+class JavaScriptImportsResolver : ImportsResolver() {
+    override fun potentialElementsPredicate(element: PsiElement): Boolean {
+        return element.text.contains("import")
+    }
+}

src/main/java/com/tabnine/psi/importsResolver/ResolverFactory.kt

@@ -0,0 +1,19 @@
+package com.tabnine.psi.importsResolver
+
+import com.intellij.lang.Language
+import com.intellij.psi.PsiFile
+
+fun getImportsResolver(psiFile: PsiFile): ImportsResolver? {
+    return when (psiFile.language) {
+        Language.findLanguageByID("JAVA") -> {
+            JavaImportsResolver()
+        }
+        Language.findLanguageByID("TypeScript") -> {
+            TypeScriptImportsResolver()
+        }
+        Language.findLanguageByID("ECMAScript 6") -> {
+            JavaScriptImportsResolver()
+        }
+        else -> null
+    }
+}

src/main/java/com/tabnine/psi/importsResolver/TypeScriptImportsResolver.kt

@@ -0,0 +1,9 @@
+package com.tabnine.psi.importsResolver
+
+import com.intellij.psi.PsiElement
+
+class TypeScriptImportsResolver : ImportsResolver() {
+    override fun potentialElementsPredicate(element: PsiElement): Boolean {
+        return element.text.contains("import")
+    }
+}

src/main/resources/META-INF/plugin.xml

@@ -109,6 +109,10 @@
     <idea-version since-build="193"/>
     <depends>com.intellij.modules.lang</depends>
 
+<!--    <projectListeners>-->
+<!--        <listener topic="com.intellij.openapi.fileEditor.FileEditorManagerListener"-->
+<!--                  class="com.tabnine.general.FilesListener"/>-->
+<!--    </projectListeners>-->
     <extensions defaultExtensionNs="com.intellij">
         <preloadingActivity implementation="com.tabnine.Initializer"/>
         <preloadingActivity implementation="com.tabnine.general.GettingStartedManager"/>