Detect K/N library kind if library is a ZIP file (native-specific part)

Issue #KT-26730 Fixed

konan/library-reader/src/org/jetbrains/kotlin/konan/library/KonanLibraryConstants.kt

@@ -14,6 +14,8 @@ const val KLIB_FILE_EXTENSION_WITH_DOT = ".$KLIB_FILE_EXTENSION"
 const val KLIB_METADATA_FILE_EXTENSION = "knm"
 const val KLIB_METADATA_FILE_EXTENSION_WITH_DOT = ".$KLIB_METADATA_FILE_EXTENSION"
 
+const val KLIB_MODULE_METADATA_FILE_NAME = "module"
+
 const val KONAN_STDLIB_NAME = "stdlib"
 
 val KONAN_COMMON_LIBS_PATH: Path

konan/library-reader/src/org/jetbrains/kotlin/konan/library/KonanLibraryLayout.kt

@@ -14,36 +14,35 @@ import org.jetbrains.kotlin.konan.target.KonanTarget
 interface KonanLibraryLayout {
 
     val libraryName: String
-
     val libDir: File
-    val target: KonanTarget?
-        // This is a default implementation. Can't make it an assignment.
-        get() = null
 
-    val manifestFile
-        get() = File(libDir, "manifest")
-    val resourcesDir
-        get() = File(libDir, "resources")
+    val source: KonanLibrarySource
 
-    val targetsDir
-        get() = File(libDir, "targets")
-    val targetDir
-        get() = File(targetsDir, target!!.visibleName)
+    // This is a default implementation. Can't make it an assignment.
+    val target: KonanTarget? get() = null
 
-    val kotlinDir
-        get() = File(targetDir, "kotlin")
-    val nativeDir
-        get() = File(targetDir, "native")
-    val includedDir
-        get() = File(targetDir, "included")
+    val manifestFile get() = File(libDir, "manifest")
+    val resourcesDir get() = File(libDir, "resources")
 
-    val linkdataDir
-        get() = File(libDir, "linkdata")
-    val moduleHeaderFile
-        get() = File(linkdataDir, "module")
-    val dataFlowGraphFile
-        get() = File(linkdataDir, "module_data_flow_graph")
+    val targetsDir get() = File(libDir, "targets")
+    val targetDir get() = File(targetsDir, target!!.visibleName)
+
+    val kotlinDir get() = File(targetDir, "kotlin")
+    val nativeDir get() = File(targetDir, "native")
+    val includedDir get() = File(targetDir, "included")
+
+    val linkdataDir get() = File(libDir, "linkdata")
+    val moduleHeaderFile get() = File(linkdataDir, KLIB_MODULE_METADATA_FILE_NAME)
+    val dataFlowGraphFile get() = File(linkdataDir, "module_data_flow_graph")
 
     fun packageFragmentFile(packageFqName: String) =
-        File(linkdataDir, if (packageFqName == "") "root_package.knm" else "package_$packageFqName.knm")
+        File(
+            linkdataDir,
+            if (packageFqName == "") "root_package.$KLIB_METADATA_FILE_EXTENSION" else "package_$packageFqName.$KLIB_METADATA_FILE_EXTENSION"
+        )
+}
+
+sealed class KonanLibrarySource {
+    object KonanLibraryDir : KonanLibrarySource()
+    class KonanLibraryFile(val klibFile: File) : KonanLibrarySource()
 }

konan/library-reader/src/org/jetbrains/kotlin/konan/library/impl/KonanLibraryLayoutImpl.kt

@@ -9,6 +9,8 @@ import org.jetbrains.kotlin.konan.file.*
 import org.jetbrains.kotlin.konan.library.KLIB_FILE_EXTENSION
 import org.jetbrains.kotlin.konan.library.KLIB_FILE_EXTENSION_WITH_DOT
 import org.jetbrains.kotlin.konan.library.KonanLibraryLayout
+import org.jetbrains.kotlin.konan.library.KonanLibrarySource.KonanLibraryDir
+import org.jetbrains.kotlin.konan.library.KonanLibrarySource.KonanLibraryFile
 import org.jetbrains.kotlin.konan.target.KonanTarget
 import java.nio.file.FileSystem
 
@@ -17,24 +19,23 @@ interface KonanLibraryLayoutImpl : KonanLibraryLayout {
     fun <T> realFiles(action: (KonanLibraryLayout) -> T): T
 }
 
-private class ZippedKonanLibraryLayout(val klibFile: File, override val target: KonanTarget?) : KonanLibraryLayoutImpl {
+private class ZippedKonanLibraryLayout(
+    klibFile: File,
+    override val target: KonanTarget?
+) : KonanLibraryLayoutImpl {
 
     init {
         zippedKonanLibraryChecks(klibFile)
     }
 
     override val libraryName = klibFile.path.removeSuffix(KLIB_FILE_EXTENSION_WITH_DOT)
-
     override val libDir: File = File("/")
+    override val source = KonanLibraryFile(klibFile)
 
-    override fun <T> realFiles(action: (KonanLibraryLayout) -> T): T {
-        return action(FileExtractor(this))!!
-    }
+    override fun <T> realFiles(action: (KonanLibraryLayout) -> T) = action(FileExtractor(this))
 
-    override fun <T> inPlace(action: (KonanLibraryLayout) -> T): T {
-        return klibFile.withZipFileSystem { zipFileSystem ->
-            action(DirectFromZip(this, zipFileSystem))
-        }
+    override fun <T> inPlace(action: (KonanLibraryLayout) -> T) = source.klibFile.withZipFileSystem { zipFileSystem ->
+        action(DirectFromZip(this, zipFileSystem))
     }
 }
 
@@ -46,44 +47,51 @@ internal fun zippedKonanLibraryChecks(klibFile: File) {
     check(extension.isEmpty() || extension == KLIB_FILE_EXTENSION) { "Unexpected file extension: $extension" }
 }
 
-private class UnzippedKonanLibraryLayout(override val libDir: File, override val target: KonanTarget?) : KonanLibraryLayoutImpl {
-    override val libraryName = libDir.path
+private class UnzippedKonanLibraryLayout(
+    override val libDir: File,
+    override val target: KonanTarget?
+) : KonanLibraryLayoutImpl {
 
-    override fun <T> inPlace(action: (KonanLibraryLayout) -> T): T = action(this)
-    override fun <T> realFiles(action: (KonanLibraryLayout) -> T): T = inPlace(action)
+    override val libraryName = libDir.path
+    override val source = KonanLibraryDir
+
+    override fun <T> inPlace(action: (KonanLibraryLayout) -> T) = action(this)
+    override fun <T> realFiles(action: (KonanLibraryLayout) -> T) = inPlace(action)
 }
 
-private class DirectFromZip(zippedLayout: ZippedKonanLibraryLayout, val zipFileSystem: FileSystem) : KonanLibraryLayout {
+private class DirectFromZip(
+    zippedLayout: ZippedKonanLibraryLayout,
+    zipFileSystem: FileSystem
+) : KonanLibraryLayout {
+
     override val libraryName = zippedLayout.libraryName
     override val libDir = zipFileSystem.file(zippedLayout.libDir)
+    override val source = zippedLayout.source
 }
 
 /**
  * This class automatically extracts pieces of the library on first access. Use it if you need
  * to pass extracted files to an external tool. Otherwise, stick to [DirectFromZip].
  */
-private class FileExtractor(val zippedLibraryLayout: ZippedKonanLibraryLayout) : KonanLibraryLayout by zippedLibraryLayout {
+private class FileExtractor(
+    val zippedLayout: ZippedKonanLibraryLayout
+) : KonanLibraryLayout by zippedLayout {
 
     override val manifestFile: File by lazy { extract(super.manifestFile) }
-
     override val resourcesDir: File by lazy { extractDir(super.resourcesDir) }
-
     override val includedDir: File by lazy { extractDir(super.includedDir) }
-
     override val kotlinDir: File by lazy { extractDir(super.kotlinDir) }
-
     override val nativeDir: File by lazy { extractDir(super.nativeDir) }
-
     override val linkdataDir: File by lazy { extractDir(super.linkdataDir) }
 
-    fun extract(file: File): File = zippedLibraryLayout.klibFile.withZipFileSystem { zipFileSystem ->
+    fun extract(file: File): File = zippedLayout.source.klibFile.withZipFileSystem { zipFileSystem ->
         val temporary = createTempFile(file.name)
         zipFileSystem.file(file).copyTo(temporary)
         temporary.deleteOnExit()
         temporary
     }
 
-    fun extractDir(directory: File): File = zippedLibraryLayout.klibFile.withZipFileSystem { zipFileSystem ->
+    fun extractDir(directory: File): File = zippedLayout.source.klibFile.withZipFileSystem { zipFileSystem ->
         val temporary = createTempDir(directory.name)
         zipFileSystem.file(directory).recursiveCopyTo(temporary)
         temporary.deleteOnExitRecursively()