New J2K: Add basic import list mapping
This commit is contained in:
committed by
Ilya Kirillov
parent
213aa9a86d
commit
df5c2f90c1
@@ -19,7 +19,12 @@ import org.jetbrains.kotlin.j2k.tree.visitors.JKVisitorVoid
|
|||||||
class ImportStatementConversion : RecursiveApplicableConversionBase() {
|
class ImportStatementConversion : RecursiveApplicableConversionBase() {
|
||||||
override fun applyToElement(element: JKTreeElement): JKTreeElement {
|
override fun applyToElement(element: JKTreeElement): JKTreeElement {
|
||||||
if (element !is JKFile) return recurse(element)
|
if (element !is JKFile) return recurse(element)
|
||||||
|
element.importList = element.importList.map {
|
||||||
|
mapOf("java.util.Collection" to "kotlin.collections.Collection")[it.name.value]?.let {
|
||||||
|
JKImportStatementImpl(JKNameIdentifierImpl(it))
|
||||||
|
} ?: it
|
||||||
|
}
|
||||||
|
val packageName = element.packageDeclaration.packageName.value
|
||||||
for (import in element.declarationList.collectImports()) {
|
for (import in element.declarationList.collectImports()) {
|
||||||
if (!element.importList.containsImport(import) && importIsInPacakge(import, packageName)) {
|
if (!element.importList.containsImport(import) && importIsInPacakge(import, packageName)) {
|
||||||
element.importList += JKImportStatementImpl(JKNameIdentifierImpl(import))
|
element.importList += JKImportStatementImpl(JKNameIdentifierImpl(import))
|
||||||
|
|||||||
Reference in New Issue
Block a user