ZipUtils.kt

import java.nio.file.*
import java.util.zip.ZipEntry
import java.util.zip.ZipOutputStream
import kotlin.io.path.*

@OptIn(ExperimentalPathApi::class)
fun ZipOutputStream.addFolder(entryName: String, folder: Path) {
    for (path in folder.walk()) {
        val relativized = folder.relativize(path)
        val prefix = if (entryName.isNotEmpty()) {
            "$entryName/"
        } else ""
        addEntry(prefix + relativized.joinToString(separator = "/") { it.name }) {
            path.inputStream().use {
                it.copyTo(this)
            }
        }
    }
}

fun ZipOutputStream.addFile(entryName: String, byteArray: ByteArray) {
    addEntry(entryName) { write(byteArray) }
}

fun ZipOutputStream.addEntry(entryName: String, operation: () -> Unit) {
    val zipEntry = ZipEntry(entryName)
    try {
        putNextEntry(zipEntry)
        operation()
    }
    catch (ex: Exception) {
        fileLogger().error("Failed to add file to stream", ex)
    }
    finally {
        closeEntry()
    }
}