Add API to record lookups and introduce KotlinLookupLocation to store the processing JetElement

compiler/frontend/src/org/jetbrains/kotlin/incremental/KotlinLookupLocation.kt

@@ -0,0 +1,22 @@
+/*
+ * Copyright 2010-2015 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.kotlin.incremental
+
+import org.jetbrains.kotlin.incremental.components.LookupLocation
+import org.jetbrains.kotlin.psi.JetElement
+
+class KotlinLookupLocation(val element: JetElement) : LookupLocation

compiler/frontend/src/org/jetbrains/kotlin/incremental/utils.kt

@@ -0,0 +1,64 @@
+/*
+ * Copyright 2010-2015 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.kotlin.incremental
+
+import org.jetbrains.kotlin.descriptors.ClassifierDescriptor
+import org.jetbrains.kotlin.descriptors.PackageFragmentDescriptor
+import org.jetbrains.kotlin.diagnostics.DiagnosticUtils.getLineAndColumnInPsiFile
+import org.jetbrains.kotlin.incremental.components.LookupLocation
+import org.jetbrains.kotlin.incremental.components.LookupTracker
+import org.jetbrains.kotlin.incremental.components.ScopeKind
+import org.jetbrains.kotlin.name.Name
+import org.jetbrains.kotlin.psi.doNotAnalyze
+import org.jetbrains.kotlin.resolve.DescriptorToSourceUtils
+import org.jetbrains.kotlin.resolve.descriptorUtil.fqNameUnsafe
+import org.jetbrains.kotlin.resolve.scopes.JetScope
+
+
+public fun LookupTracker.record(from: LookupLocation, inScope: JetScope, name: Name) {
+    if (this == LookupTracker.DO_NOTHING || from == LookupLocation.NO_LOCATION) return
+
+    if (from !is KotlinLookupLocation) throw AssertionError("Unexpected location type: ${from.javaClass}")
+
+    val containingJetFile = from.element.getContainingJetFile()
+
+    // TODO: provide additionally info for fake elements/files
+    if (containingJetFile.doNotAnalyze != null) return
+
+    val containingFilePath = containingJetFile.virtualFile.path
+    val lookupLocationSuffix =
+            when {
+                verbose ->
+                    getLineAndColumnInPsiFile(containingJetFile, from.element.textRange).let {
+                        ":" + it.line + ":" + it.column
+                    }
+                else -> ""
+            }
+    val lookupLocation = "$containingFilePath$lookupLocationSuffix"
+
+    val scopeContainingDeclaration = inScope.getContainingDeclaration()
+    val scopeFilePath = DescriptorToSourceUtils.getContainingFile(scopeContainingDeclaration)?.virtualFile?.path
+
+    val scopeKind =
+            when (scopeContainingDeclaration) {
+                is ClassifierDescriptor -> ScopeKind.CLASSIFIER
+                is PackageFragmentDescriptor -> ScopeKind.PACKAGE
+                else -> throw AssertionError("Unexpected containing declaration type: ${scopeContainingDeclaration.javaClass}")
+            }
+
+    record(lookupLocation, scopeContainingDeclaration.fqNameUnsafe.asString(), scopeFilePath, scopeKind, name.asString())
+}

core/descriptors/src/org/jetbrains/kotlin/incremental/components/LookupTracker.kt

@@ -17,7 +17,19 @@
 package org.jetbrains.kotlin.incremental.components
 
 public interface LookupTracker {
+    val verbose: Boolean
+        get() = false
+
+    fun record(lookupLocation: String, scopeFqName: String, scopeContainingFile: String?, scopeKind: ScopeKind, name: String)
+
     companion object {
-        val DO_NOTHING = object : LookupTracker {}
+        val DO_NOTHING = object : LookupTracker {
+            override fun record(lookupLocation: String, scopeFqName: String, scopeContainingFile: String?, scopeKind: ScopeKind, name: String) {}
+        }
     }
 }
+
+public enum class ScopeKind {
+    PACKAGE,
+    CLASSIFIER
+}