Disconnect a vpn

Author: Dinesh Kumar

lazy-connect.sh

@@ -63,6 +63,7 @@ lazy-connect - Shell function to fuzzy search an IPSec VPN by name
 -u    - Update lazy-connect
 -r    - Refresh vpn list in ~/.config/lazy-connect
 -h    - Show this help
+-d    - Disconnect a connected VPN
 EOF
 }
 
@@ -105,7 +106,7 @@ function _lazy_connect() {
     esac
   fi
 
-  osascript <<EOF
+  result=$(osascript <<EOF
     on connectVpn(vpnName, password)
       tell application "System Events"
         tell process "SystemUIServer"
@@ -115,7 +116,7 @@ function _lazy_connect() {
             click menu item vpnName of menu 1 of vpnMenu
             delay 1
             keystroke password
-            keystroke return
+            return "true"
           on error errorStr
             if errorStr does not contain "Can’t get menu item" and errorStr does not contain vpnName then
               display dialog errorStr
@@ -127,6 +128,34 @@ function _lazy_connect() {
 
     connectVpn("$vpn_name", "$password")
 EOF
+)
+  [[ $result -eq "true" ]] && echo $vpn_name | sed 's/Connect/Disconnect/g' >> "$_lazy_connect_config_dir/connected_vpns"
+}
+
+function _lazy_disconnect() {
+  vpn_name=$1
+
+  osascript <<EOF
+    on disconnectVpn(vpnName)
+      tell application "System Events"
+        tell process "SystemUIServer"
+          set vpnMenu to (menu bar item 1 of menu bar 1 where description is "VPN")
+          tell vpnMenu to click
+          try
+            click menu item vpnName of menu 1 of vpnMenu
+          on error errorStr
+            if errorStr does not contain "Can’t get menu item" and errorStr does not contain vpnName then
+              display dialog errorStr
+            end if
+          end try
+        end tell
+      end tell
+    end disconnectVpn
+
+    disconnectVpn("$vpn_name")
+    sed "/Connect ${vpn_name}/d" $_lazy_connect_config_dir/vpns
+    sed "/Disconnect ${vpn_name}/d" $_lazy_connect_config_dir/connected_vpns | sort -u
+EOF
 }
 
 function _lazy_connect_update() {
@@ -139,7 +168,7 @@ function lazy-connect() {
   local OPTIND
   mkdir -p $_lazy_connect_config_dir
 
-  while getopts "iruh" opt; do
+  while getopts "iruhd" opt; do
     case $opt in
       h)
         _lazy_connect_usage
@@ -154,6 +183,11 @@ function lazy-connect() {
         _lazy_connect_vpn_refresh
         return 0
         ;;
+      d)
+        vpn_name=$(cat $_lazy_connect_config_dir/connected_vpns | fzf --height=10 --ansi --reverse)
+        _lazy_disconnect "$vpn_name"
+        return 0
+        ;;
       u)
         _lazy_connect_update
         return 0