org.jetbrains.kotlin.resolve.scopes.StaticScopeForKotlinClass.kt Maven / Gradle / Ivy
/*
* 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.resolve.scopes
import org.jetbrains.kotlin.descriptors.ClassDescriptor
import org.jetbrains.kotlin.descriptors.ClassKind
import org.jetbrains.kotlin.descriptors.FunctionDescriptor
import org.jetbrains.kotlin.descriptors.PropertyDescriptor
import org.jetbrains.kotlin.incremental.components.LookupLocation
import org.jetbrains.kotlin.name.Name
import org.jetbrains.kotlin.resolve.DescriptorFactory.*
import org.jetbrains.kotlin.utils.Printer
import java.util.*
// We don't need to track lookups here since this scope used only for introduce special Enum class members
class StaticScopeForKotlinClass(
private val containingClass: ClassDescriptor
) : MemberScopeImpl() {
override fun getContributedClassifier(name: Name, location: LookupLocation) = null // TODO
private val functions: List by lazy {
if (containingClass.kind != ClassKind.ENUM_CLASS) {
listOf()
}
else {
listOf(createEnumValueOfMethod(containingClass), createEnumValuesMethod(containingClass))
}
}
private val properties: List by lazy {
if (containingClass.kind != ClassKind.ENUM_CLASS) {
listOf()
}
else {
listOf(createEnumValuesProperty(containingClass))
}
}
override fun getContributedDescriptors(kindFilter: DescriptorKindFilter,
nameFilter: (Name) -> Boolean) = functions + properties
override fun getContributedVariables(name: Name, location: LookupLocation) = properties.filterTo(ArrayList(1)) { it.name == name }
override fun getContributedFunctions(name: Name, location: LookupLocation) = functions.filterTo(ArrayList(2)) { it.name == name }
override fun printScopeStructure(p: Printer) {
p.println("Static scope for $containingClass")
}
}