1 files changed, 23 insertions, 3 deletions
diff --git a/dactyl.sh b/dactyl.sh
index bf6259c..c0bc41e 100755
--- a/dactyl.sh
+++ b/dactyl.sh
@@ -9,8 +9,10 @@ set -e
 
 container=""
 shellContainer=DM-shell
-configContainer=DM-config
 modelContainer=DM-model
+configContainer=DM-config
+containers=("$shellContainer" "$configContainer" "$modelContainer")
+
 imageName=dactyl-keyboard
 srcBind="$(pwd)/src:/app/src"
 thingsBind="$(pwd)/things:/app/things"
@@ -289,11 +291,15 @@ function handleRebuildImage() {
   buildImage
 }
 
+function removeImage() {
+  inform "Removing docker image: $imageName..."
+  docker image rm $imageName
+}
+
 function handleRemoveImage() {
   warn "This will remove docker image: $imageName"
   confirmContinue "Would you like to continue?"
-  inform "Removing docker image: $imageName..."
-  docker image rm $imageName
+  removeImage
 }
 
 function handleInspectImage() {
@@ -551,6 +557,20 @@ function promptStartShellSession() {
 }
 
 ################################
+# Uninstaller
+################################
+
+function handleUninstall() {
+  for currentContainer in "${containers[@]}"; do
+    container="$currentContainer"
+    removeContainer
+  done
+
+  removeImage
+  exit
+}
+
+################################
 # Main Menu Logic
 ################################