org.jetbrains.kotlin.js.inline.util.collectUtils.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.js.inline.util
import com.google.dart.compiler.backend.js.ast.*
import com.google.dart.compiler.backend.js.ast.metadata.staticRef
import java.util.IdentityHashMap
import org.jetbrains.kotlin.js.inline.util.collectors.ReferenceNameCollector
import org.jetbrains.kotlin.js.inline.util.collectors.NameCollector
import org.jetbrains.kotlin.js.inline.util.collectors.InstanceCollector
import org.jetbrains.kotlin.js.inline.util.collectors.PropertyCollector
import org.jetbrains.kotlin.js.translate.expression.*
fun collectFunctionReferencesInside(scope: JsNode): List =
collectReferencesInside(scope).filter { it.staticRef is JsFunction }
fun collectReferencesInside(scope: JsNode): List {
return with(ReferenceNameCollector()) {
accept(scope)
references
}
}
fun collectLocalNames(function: JsFunction): List {
val functionScope = function.scope
return with(NameCollector(functionScope)) {
accept(function.body)
names.values.toList()
}
}
fun collectJsProperties(scope: JsNode): IdentityHashMap {
val collector = PropertyCollector()
collector.accept(scope)
return collector.properties
}
fun collectNamedFunctions(scope: JsNode): IdentityHashMap {
val namedFunctions = IdentityHashMap()
for ((name, value) in collectJsProperties(scope)) {
val function: JsFunction? = when (value) {
is JsFunction -> value
else -> InlineMetadata.decompose(value)?.function
}
if (function != null) {
namedFunctions[name] = function
}
}
return namedFunctions
}
fun collectInstances(klass: Class, scope: JsNode): List {
return with(InstanceCollector(klass, visitNestedDeclarations = false)) {
accept(scope)
collected
}
}