org.jetbrains.kotlin.js.inline.util.collectUtils.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler Show documentation
the Kotlin compiler
There is a newer version: 2.0.0
/*
 * 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.*

public fun collectFunctionReferencesInside(scope: JsNode): List =
    collectReferencesInside(scope) filter { it.staticRef is JsFunction }

public fun collectReferencesInside(scope: JsNode): List {
    return with(ReferenceNameCollector()) {
        accept(scope)
        references
    }
}

public fun collectLocalNames(function: JsFunction): List {
    val functionScope = function.getScope()

    return with(NameCollector(functionScope)) {
        accept(function.getBody())
        names.values().toList()
    }
}

public fun collectJsProperties(scope: JsNode): IdentityHashMap {
    val collector = PropertyCollector()
    collector.accept(scope)
    return collector.properties
}

public 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
}

public fun collectInstances(klass: Class, scope: JsNode): List {
    return with(InstanceCollector(klass, visitNestedDeclarations = false)) {
        accept(scope)
        collected
    }
}