From a9db9bb28ab0817f6a0d4bb28bf0a15b425af4a9 Mon Sep 17 00:00:00 2001
From: Dmitry Zhuravlev <dmitry.anatolyevich.zhuravlev@gmail.com>
Date: Wed, 12 Apr 2017 15:50:53 +0300
Subject: [PATCH] * package all Kobalt sources in zip distribution. relates to
 #237

---
 kobalt/src/Build.kt | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/kobalt/src/Build.kt b/kobalt/src/Build.kt
index 6f15035a..f335067c 100644
--- a/kobalt/src/Build.kt
+++ b/kobalt/src/Build.kt
@@ -17,6 +17,8 @@ import java.io.File
 import java.nio.file.Files
 import java.nio.file.Paths
 import java.nio.file.StandardCopyOption
+import java.util.zip.ZipEntry
+import java.util.zip.ZipOutputStream
 
 val bs = buildScript {
     repos("http://dl.bintray.com/cbeust/maven")
@@ -220,6 +222,12 @@ val kobaltApp = project(kobaltPluginApi, wrapper) {
             (0 .. files.size - 1 step 3).forEach { i ->
                 include(from(files[i]), to(files[i + 1]), files[i + 2])
             }
+
+            val currentDir = Paths.get(".").toAbsolutePath().normalize().toString()
+            zipFolders("$currentDir/$buildDirectory/libs/all-sources/$projectName-$version-sources.jar",
+                    "$currentDir/$directory/src/main/kotlin",
+                    "$currentDir/${kobaltPluginApi.directory}/src/main/kotlin")
+            include(from("$buildDirectory/libs/all-sources"), to("$dir/kobalt/wrapper"), "$projectName-$version-sources.jar")
         }
     }
 
@@ -244,6 +252,28 @@ val kobaltApp = project(kobaltPluginApi, wrapper) {
     }
 }
 
+fun zipFolders(zipFilePath: String, vararg foldersPath: String) {
+    val zip = Paths.get(zipFilePath)
+    Files.deleteIfExists(zip)
+    Files.createDirectories(zip.parent)
+    val zipPath = Files.createFile(zip)
+    ZipOutputStream(Files.newOutputStream(zipPath)).use {
+        foldersPath.map {Paths.get(it)}.forEach { folderPath ->
+            Files.walk(folderPath)
+                    .filter { path -> !Files.isDirectory(path) }
+                    .forEach { path ->
+                        val zipEntry = ZipEntry(folderPath.relativize(path).toString())
+                        try {
+                            it.putNextEntry(zipEntry)
+                            Files.copy(path, it)
+                            it.closeEntry()
+                        } catch (e: Exception) {
+                        }
+                    }
+        }
+    }
+}
+
 fun readVersion() : String {
     val localFile =
             listOf("src/main/resources/kobalt.properties",