org.jetbrains.kotlin.backend.wasm.codegen.NameTable.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler Show documentation

Kotlin Compiler

There is a newer version: 2.0.0

/*
 * Copyright 2010-2019 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

package org.jetbrains.kotlin.backend.wasm.codegen

import org.jetbrains.kotlin.backend.common.ir.isTopLevel
import org.jetbrains.kotlin.ir.IrElement
import org.jetbrains.kotlin.ir.backend.js.utils.*
import org.jetbrains.kotlin.ir.declarations.*
import org.jetbrains.kotlin.ir.util.fqNameWhenAvailable
import org.jetbrains.kotlin.ir.visitors.IrElementVisitorVoid
import org.jetbrains.kotlin.ir.visitors.acceptChildrenVoid

fun  wasmNameTable() = NameTable(sanitizer = ::sanitizeWatIdentifier)

fun generateWatTopLevelNames(packages: List): Map {
    val names = wasmNameTable()

    fun nameTopLevelDecl(declaration: IrDeclarationWithName) {
        val suggestedName = declaration.fqNameWhenAvailable?.toString()
            ?: "fqname???" + declaration.name.asString()
        names.declareFreshName(declaration, suggestedName)
    }

    for (p in packages) {
        p.acceptChildrenVoid(object : IrElementVisitorVoid {
            override fun visitElement(element: IrElement) {
                element.acceptChildrenVoid(this)
            }

            override fun visitSimpleFunction(declaration: IrSimpleFunction) {
                nameTopLevelDecl(declaration)
                super.visitSimpleFunction(declaration)
            }

            override fun visitField(declaration: IrField) {
                if (declaration.isTopLevel)
                    nameTopLevelDecl(declaration)
                super.visitField(declaration)
            }
        })
    }

    return names.names
}

fun sanitizeWatIdentifier(ident: String): String {
    if (ident.isEmpty())
        return "_"
    if (ident.all(::isValidWatIdentifier))
        return ident
    return ident.map { if (isValidWatIdentifier(it)) it else "_" }.joinToString("")
}

// https://webassembly.github.io/spec/core/text/values.html#text-id
fun isValidWatIdentifier(c: Char): Boolean =
    c in '0'..'9' || c in 'A'..'Z' || c in 'a'..'z'
            // TODO: SpiderMonkey js shell can't parse some of the
            //  permitted identifiers: '?', '<'
            // || c in "!#$%&′*+-./:<=>?@\\^_`|~"
            || c in "$.@_"