org.jetbrains.jet.j2k.ast.Imports.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2010-2014 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.jet.j2k.ast

import com.intellij.psi.PsiImportStatementBase
import org.jetbrains.jet.j2k.*
import com.intellij.psi.PsiImportList
import org.jetbrains.jet.lang.resolve.name.FqName
import org.jetbrains.jet.lang.resolve.java.mapping.JavaToKotlinClassMap
import org.jetbrains.jet.lang.resolve.name.isValidJavaFqName


class Import(val name: String) : Element {
    override fun toKotlin() = "import " + name
}

class ImportList(val imports: List) : Element {
    val filteredImports = imports.filter {
        !it.name.isEmpty() && it.name !in NOT_NULL_ANNOTATIONS
    }.filter {
        // If name is invalid, like with star imports, don't try to filter
        if (!isValidJavaFqName(it.name))
            true
        else {
            // If imported class has a kotlin analog, drop the import
            val kotlinAnalogsForClass = JavaToKotlinClassMap.getInstance().mapPlatformClass(FqName(it.name))
            kotlinAnalogsForClass.isEmpty()
        }
    }

    override fun isEmpty(): Boolean {
        return filteredImports.isEmpty()
    }

    override fun toKotlin() = filteredImports.toKotlin("\n")
}

fun Converter.convertImportList(importList: PsiImportList): ImportList =
        ImportList(importList.getAllImportStatements() map { convertImport(it) })

fun Converter.convertImport(i: PsiImportStatementBase): Import {
    val reference = i.getImportReference()
    if (reference != null) {
        return Import(quoteKeywords(reference.getQualifiedName()!!) + if (i.isOnDemand()) ".*" else "")
    }
    return Import("")
}