Fixes titan-data#81

Signed-off-by: Derek Smart <[email protected]>

src/main/kotlin/io/titandata/titan/clients/Docker.kt

@@ -110,6 +110,11 @@ class Docker(private val executor: CommandExecutor) {
         return executor.exec(listOf("docker", "container", "rm", containerId))
     }
 
+    fun removeTitanImages(version: String): String {
+        val imageId = executor.exec(listOf("docker", "images", "titan:$version", "--format", "{{.ID}}")).trim()
+        return executor.exec(listOf("docker", "rmi", imageId, "-f"))
+    }
+
     fun run(image: String, entry: String, arguments: List<String>): String {
         val mutArgs = mutableListOf("docker", "run")
         mutArgs.addAll(arguments)

src/main/kotlin/io/titandata/titan/providers/Local.kt

@@ -137,7 +137,7 @@ class Local: Provider {
     }
 
     override fun uninstall(force: Boolean) {
-        val uninstallCommand = Uninstall(::exit, ::remove,  commandExecutor, docker)
+        val uninstallCommand = Uninstall(titanServerVersion, ::exit, ::remove,  commandExecutor, docker)
         return uninstallCommand.uninstall(force)
     }
 

src/main/kotlin/io/titandata/titan/providers/local/Uninstall.kt

@@ -10,6 +10,7 @@ import io.titandata.titan.exceptions.CommandException
 import io.titandata.titan.utils.CommandExecutor
 
 class Uninstall (
+    private val titanServerVersion: String,
     private val exit: (message: String, code: Int) -> Unit,
     private val remove: (container: String, force: Boolean) -> Unit,
     private val commandExecutor: CommandExecutor = CommandExecutor(),
@@ -35,6 +36,7 @@ class Uninstall (
         try {
             docker.removeVolume("titan-data")
         } catch (e: CommandException) {}
+        docker.removeTitanImages(titanServerVersion)
         println("Uninstalled titan infrastructure")
     }
 }

src/test/kotlin/io/titandata/titan/providers/local/UninstallTest.kt

@@ -7,7 +7,7 @@ import org.junit.Test
 class UninstallTest {
     private fun exit(message: String, code: Int) {}
     private fun remove(container: String, force: Boolean) {}
-    private val command = Uninstall(::exit, ::remove)
+    private val command = Uninstall("version", ::exit, ::remove)
 
     @Test
     fun `can instantiate`(){