From dddc7bfedb813ffde4c8d632f5416cfa8c09dda5 Mon Sep 17 00:00:00 2001
From: Thomas Goldbrunner <thomas.goldbrunner@tum.de>
Date: Thu, 27 Jul 2017 12:53:49 +0200
Subject: [PATCH] Enhancements for remove-go

- add completion for remove-go
- remove-go can now handle the "tip" version
---
 plugins/gimme/README.md        |  2 +-
 plugins/gimme/gimme.plugin.zsh | 25 ++++++++++++++++++++++++-
 2 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/plugins/gimme/README.md b/plugins/gimme/README.md
index 095a36b9..0d5eae98 100644
--- a/plugins/gimme/README.md
+++ b/plugins/gimme/README.md
@@ -11,7 +11,7 @@ Provides following functions/aliases:
     If called without argument, the 'stable' version will be loaded.
   - ```remove-go```: Remove the go version specified by the first argument from
     ~/.gimme/*
-  - Completions for ```gimme``` and ```load-go```
+  - Completions for ```gimme```, ```load-go``` and ```remove-go```
 
 ```gimme``` is assumed to be in ```~/bin```, so make
 sure to add this folder to the $PATH variable.
diff --git a/plugins/gimme/gimme.plugin.zsh b/plugins/gimme/gimme.plugin.zsh
index 3b387e43..97c493f5 100644
--- a/plugins/gimme/gimme.plugin.zsh
+++ b/plugins/gimme/gimme.plugin.zsh
@@ -39,7 +39,16 @@ remove-go() {
     echo "Usage: remove-go GO_VERSION"
     return 0
   else
-    if [ "$1" = "stable" ] ; then
+    if [ "$1" = "tip" ] ; then
+      if ! [ -e ~/.gimme/versions/go ] ; then
+        echo "go version tip is not installed"
+        return 0
+      fi
+      rm -rf ~/.gimme/versions/go || return 1
+      rm ~/.gimme/envs/gotip.env || return 1
+      rm ~/.gimme/envs/go.git.* || return 1
+      return 0
+    elif [ "$1" = "stable" ] ; then
       if ! [ -e ~/.gimme/versions/stable ] ; then
         echo "go version stable is not installed"
         return 0
@@ -116,5 +125,19 @@ __load-go_completion() {
   _values 'go_versions' ${go_versions[@]}
 }
 
+__remove-go_completion() {
+  local context state state_descr line
+  typeset -a installed_versions
+  if [ "$(ls ~/.gimme/versions | grep stable)" ] ; then
+    installed_versions+=('stable')
+  fi
+  if [ -d ~/.gimme/versions/go ] ; then
+    installed_versions+=('tip')
+  fi
+  installed_versions+=($(ls ~/.gimme/versions | egrep -oZ '[0-9].[0-9].[0-9]'))
+  _values -w 'installed_versions' ${installed_versions[@]}
+}
+
 compdef __gimme_completion gimme
 compdef __load-go_completion load-go
+compdef __remove-go_completion remove-go