From 878b13b2fa87f82652a0411070bdd6dd9971a3fe Mon Sep 17 00:00:00 2001
From: Joe Groocock <frebib@gmail.com>
Date: Thu, 16 Nov 2017 00:30:45 +0000
Subject: [PATCH] Customise sudo plugin to fit my aliases

---
 .config/zsh/plugins/sudo.zsh | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/.config/zsh/plugins/sudo.zsh b/.config/zsh/plugins/sudo.zsh
index e1b2317..8f3d680 100644
--- a/.config/zsh/plugins/sudo.zsh
+++ b/.config/zsh/plugins/sudo.zsh
@@ -4,11 +4,13 @@ sudo-command-line() {
     [[ -z $BUFFER ]] && zle up-history
     if [[ $BUFFER == sudo\ * ]]; then
         LBUFFER="${LBUFFER#sudo }"
-    elif [[ $BUFFER == $EDITOR\ * ]]; then
+    elif [[ $BUFFER == $EDITOR\ * ]] || [[ $BUFFER == vi\ * ]]; then
         LBUFFER="${LBUFFER#$EDITOR }"
-        LBUFFER="sudoedit $LBUFFER"
-    elif [[ $BUFFER == sudoedit\ * ]]; then
+        LBUFFER="${LBUFFER#vi }"
+        LBUFFER="svi $LBUFFER"
+    elif [[ $BUFFER == sudoedit\ * ]] || [[ $BUFFER == svi\ * ]]; then
         LBUFFER="${LBUFFER#sudoedit }"
+        LBUFFER="${LBUFFER#svi }"
         LBUFFER="$EDITOR $LBUFFER"
     else
         LBUFFER="sudo $LBUFFER"