Major changes with new sem versioning.

Author: sureshg

build.gradle.kts

@@ -1,22 +1,24 @@
 import org.gradle.api.tasks.wrapper.Wrapper
 import org.gradle.api.tasks.wrapper.Wrapper.DistributionType
 import org.gradle.api.tasks.wrapper.Wrapper.DistributionType.ALL
-import org.jetbrains.kotlin.gradle.dsl.Coroutines.ENABLE
 import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
 import us.kirchmeier.capsule.manifest.CapsuleManifest
 import us.kirchmeier.capsule.spec.ReallyExecutableSpec
 import us.kirchmeier.capsule.task.*
 import org.gradle.jvm.tasks.Jar
+import java.time.ZonedDateTime
+import java.time.format.DateTimeFormatter
 import java.util.jar.Attributes.Name.*
 
-
 buildscript {
     var javaVersion: JavaVersion by extra
     var kotlinVersion: String by extra
     var kotlinEAPRepo: String by extra
+    var wrapperVersion: String by extra
 
     javaVersion = JavaVersion.VERSION_1_8
     kotlinVersion = "1.1.2-eap-44"
+    wrapperVersion = "3.5-20170331195952+0000"
     kotlinEAPRepo = "https://dl.bintray.com/kotlin/kotlin-eap-1.1"
 
     repositories {
@@ -29,6 +31,7 @@ val appAuthor by project
 val javaVersion: JavaVersion by extra
 val kotlinVersion: String by extra
 val kotlinEAPRepo: String by extra
+val wrapperVersion: String by extra
 printHeader()
 
 plugins {
@@ -58,14 +61,7 @@ java {
  */
 application {
     applicationName = rootProject.name
-    mainClassName = "io.sureshg.InstallCertsKt"
-}
-
-/**
- * Enable coroutines.
- */
-kotlin {
-    experimental.coroutines = ENABLE
+    mainClassName = "io.sureshg.MainKt"
 }
 
 /**
@@ -92,7 +88,8 @@ dependencies {
  */
 tasks.withType<Jar> {
     manifest {
-        attributes(mapOf("Author" to appAuthor,
+        attributes(mapOf("Built-By" to appAuthor,
+                "Built-Date" to buildDateTime,
                 IMPLEMENTATION_VERSION.toString() to appVersion,
                 IMPLEMENTATION_TITLE.toString() to application().applicationName,
                 MAIN_CLASS.toString() to application().mainClassName))
@@ -114,7 +111,6 @@ task<FatCapsule>("makeExecutable") {
         applicationVersion = version
         jvmArgs = listOf("-client")
         minJavaVersion = minJavaVer
-        //args = listOf("$@")
     }
     description = "Create $archiveName executable."
     dependsOn("clean")
@@ -128,9 +124,12 @@ task<FatCapsule>("makeExecutable") {
  * Generate Gradle Script Kotlin wrapper.
  */
 task<Wrapper>("wrapper") {
-    description = "Generate Gradle Script Kotlin wrapper v0.8"
+    description = "Generate Gradle Script Kotlin wrapper v$wrapperVersion"
     distributionType = ALL
-    distributionUrl = getGskURL("3.5-20170331195952+0000")
+    distributionUrl = getGskURL(wrapperVersion)
+    doFirst {
+        println(description)
+    }
 }
 
 
@@ -149,4 +148,6 @@ fun printHeader() {
     println()
 }
 
+val buildDateTime get() = ZonedDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd hh:mm:ss a z"))
+
 fun getGskURL(version: String, type: DistributionType = ALL) = "https://repo.gradle.org/gradle/dist-snapshots/gradle-script-kotlin-$version-${type.name.toLowerCase()}.zip"

gradle.properties

@@ -3,4 +3,4 @@ org.gradle.script.lang.kotlin.accessors.auto=true
 
 # Project custom properties
 appAuthor=Suresh
-appVersion=1.0
+appVersion=1.0.0

gradle/wrapper/gradle-wrapper.jar

@@ -1,4 +1,4 @@
-#Mon Apr 10 02:06:42 PDT 2017
+#Thu Apr 13 16:08:47 PDT 2017
 distributionBase=GRADLE_USER_HOME
 distributionPath=wrapper/dists
 zipStoreBase=GRADLE_USER_HOME

gradle/wrapper/gradle-wrapper.properties

@@ -0,0 +1,29 @@
+package io.sureshg
+
+import io.airlift.airline.SingleCommand
+import io.sureshg.cmd.Install
+import io.sureshg.extn.*
+
+/**
+ * Main method that runs [InstallCerts]
+ *
+ * @author  Suresh
+ */
+fun main(args: Array<String>) {
+    val cmd = SingleCommand.singleCommand(Install::class.java)
+    var install: Install? = null
+    try {
+        install = cmd.parse(*args)
+        if (install.helpOption.showHelpIfRequested()) {
+            return
+        }
+        install.run()
+    } catch (e: Throwable) {
+        println("""|${e.message?.err}
+                   |See ${"'installcerts --help'".bold}
+                    """.trimMargin())
+        install?.let {
+            if (it.verbose) e.printStackTrace()
+        }
+    }
+}

src/main/kotlin/io/sureshg/InstallCerts.kt

@@ -4,10 +4,10 @@ import io.airlift.airline.Arguments
 import io.airlift.airline.Command
 import io.airlift.airline.HelpOption
 import io.airlift.airline.Option
-import io.sureshg.extn.AnsiColor.*
-import io.sureshg.extn.color
-import io.sureshg.extn.sux
+import io.sureshg.crypto.InstallCerts
+import io.sureshg.extn.*
 import java.net.URL
+import java.util.jar.Attributes.Name.*
 import javax.inject.Inject
 
 /**
@@ -27,6 +27,9 @@ class Install {
     @Option(name = arrayOf("-p"), description = "Trust store password. Default is 'changeit'")
     var storePasswd = "changeit"
 
+    @Option(name = arrayOf("-a", "--all"), description = "Install all certs")
+    var all = false
+
     @Option(name = arrayOf("-v", "--verbose"), description = "Verbose mode")
     var verbose = false
 
@@ -50,20 +53,34 @@ class Install {
     /**
      * Tool version.
      */
-    val version by lazy {
-      Install::class.java.`package`.implementationVersion
+    val buildInfo by lazy {
+        Install::class.jarManifest?.let {
+            val attr = it.mainAttributes
+            BuildInfo(attr.getValue("Built-By"),
+                    attr.getValue("Built-Date"),
+                    attr.getValue(IMPLEMENTATION_VERSION))
+        } ?: BuildInfo()
     }
 
     /**
      * Executes the command
      */
     fun run() {
-        if (uri.isEmpty()) throw IllegalArgumentException("Server URL can't be empty!")
-        if(showVersion) {
-            println("".sux)
-            System.exit(0)
+        when {
+            showVersion -> {
+                val version = """|InstallCerts version: ${buildInfo.version ?: "N/A"}
+                                 |Build Date: ${buildInfo.date ?: "N/A"}
+                              """.trimMargin()
+                println(version.bold.cyan)
+                System.exit(0)
+            }
+            uri.isEmpty() -> throw IllegalArgumentException("Server URL can't be empty!")
         }
+        InstallCerts.exec(this)
     }
+}
 
-    override fun toString() = "Install(uri=$uri, hostPort=$hostPort)"
-}
+/**
+ * Build info class
+ */
+data class BuildInfo(val by: String? = null, val date: String? = null, val version: String? = null)

src/main/kotlin/io/sureshg/Main.kt

@@ -0,0 +1,73 @@
+package io.sureshg.crypto
+
+import io.sureshg.cmd.Install
+import io.sureshg.extn.*
+import java.io.File
+import javax.net.ssl.SSLException
+import javax.net.ssl.SSLSocket
+
+/**
+ * Creates a PKCS12 TrustStore by retrieving server's certificates
+ * with JDK trusted certificates.
+ *
+ * @author Suresh
+ */
+object InstallCerts {
+
+    /**
+     * Executes the action.
+     */
+    fun exec(args: Install) {
+        val (host, port) = args.hostPort
+        val storePasswd = args.storePasswd
+
+        val keystoreFile = File(host.replace(".", "_").plus(".p12"))
+
+        if (keystoreFile.isFile) {
+            val uin: String? = System.console()?.readLine(" $keystoreFile PKCS12 file exists. Do you want to overwrite it (y/n)? ".warn)
+            if (uin?.toLowerCase() != "y") {
+                println("Existing...".red)
+                System.exit(-1)
+            }
+        }
+
+        println("Loading default ca truststore...".cyan)
+        val keyStore = CACertsKeyStore
+        val tm = keyStore.defaultTrustManager.saving()
+        val sslFactory = getSSLSockFactory("TLS", trustManagers = arrayOf(tm))
+
+        println("Opening connection to $host:$port...".cyan)
+        val socket = sslFactory.createSocket(host, port) as SSLSocket
+
+        try {
+            with(socket) {
+                soTimeout = 5_000
+                startHandshake()
+                close()
+            }
+            println(" No error, certificate is already trusted using jre ca certs!".sux)
+            System.exit(0)
+        } catch(e: SSLException) {
+            e.printStackTrace()
+        }
+
+        if (tm.chain.isEmpty()) {
+            println("Could not obtain server certificate chain".err)
+            System.exit(-1)
+        } else {
+
+            tm.chain.forEach {
+                println(it.info().fg256((RAND.nextDouble() * 255).toInt()))
+            }
+
+        }
+
+    }
+
+
+}
+
+
+
+
+