From 4547825e0336011fc79e5f0a281e79d36c5416c7 Mon Sep 17 00:00:00 2001
From: shockrahwow <alejandros714@protonmail.com>
Date: Fri, 30 Aug 2019 12:52:43 -0700
Subject: [PATCH] fixed some """"issues"""", really just made behavior follow
 expectations

---
 wifi | 76 +++++++++++++++++++++++++++++++++++++++---------------------
 1 file changed, 49 insertions(+), 27 deletions(-)

diff --git a/wifi b/wifi
index 73c8a23..3911e04 100755
--- a/wifi
+++ b/wifi
@@ -5,16 +5,38 @@
 cfg_loc="$HOME/.config/wifi-configs/"
 iface="wlp1s0"
 
-debug_kill='on'
-__write_config() {
+# wpa_supplicant and dhclient are only exposed to root by default
+if [ "`id -u`" -ne 0 ]
+then
+	echo 'Must be ran as root'
+	exit 1
+fi
+# Only activating this for testing the menu tbh
+debug_kill=
+
+__write_config_personal() {
 printf "network={
 	ssid=\"$2\"
 	psk=\"$3\"
-}\n" # > "$cfg_loc/$1"
+}\n" > "$cfg_loc/$1"
 }
 
+__write_config_enterprise() {
+printf "network={
+	ssid=\"$1\"
+	scan_ssid=1
+	key_mgmt=WPA-EAP
+	identity=\"$2\"
+	password=\"$3\"
+	eap=PEAP
+	phase1=\"peaplabel=0\"
+	phase2=\"auth=MSCHAPV2\"
+}" > "$cfg_loc/$4"
+}
+
+
 __kill_old() {
-	if [ ! -z "$debug_kill" ]
+	if [ -z "$debug_kill" ]
 	then
 		kill `pgrep wpa`
 		dhclient -r
@@ -23,56 +45,56 @@ __kill_old() {
 	fi
 }
 
-_reconnect_config() {
-	__kill_old
-	if [ ! -z "$debug_kill" ]
+__connect_config() {
+	if [ -z "$debug_kill" ]
 	then
 		__kill_old
+		wpa_supplicant -B -i wlp1s0 -c /etc/wpa_supplicant.conf -D wext
+		dhclient wlp1s0
 	else
 		echo 'No restart'
 	fi
 }
 
 remove_old_config() {
-	name="`echo '' | dmenu -i -p 'Name of config to remove(This will disconnect you)'`"
+	name="`echo '' | dmenu -i -p 'Name of config to remove'`"
 	rm -f "$cfg_loc/$name"
-	__kill_old
+}
+
+reconnect_old_config() {
+	name="`ls $cfg_loc | dmenu -i -p 'Choose config to connect with'`"
+	if [ ! -z "$name" ] 
+	then
+		__kill_old
+		__connect_config
+	fi
 }
 
 create_new_config() {
 	name="`echo '' | dmenu -i -p 'Name of new config'`"
 	type="`printf "WPA2 Personal\nWPA2 Enterprise" | dmenu -i -p 'Connection Type'`"
+	mkdir -p $cfg_loc
 
 	case $type in
 		*Personal)
 			ssid="`echo '' | dmenu -p 'SSID'`"
 			psk="`echo '' | dmenu -p 'Passkey'`"
-			__write_config "$HOME/.config//$name" "$ssid" "$psk"
-			con="`printf "Yes\nNo" | dmenu -i -p 'Connect with new config?'`"
-			if [ "$con" = "Yes" ]
-			then
-				_reconnect_config $con
-			fi
+			__write_config_personal "$cfg_loc/$name" "$ssid" "$psk"
 			;;
 		*Enterprise)
-			echo 'Not yet implemented' | dmenu -p 'SSID'
-			exit
+			ssid="`echo '' | dmenu -p 'SSID'`"
+			identity="`echo '' | dmenu -p 'Identity'`"
+			psk="`echo '' | dmenu -p 'Passkey'`"
+			__write_config_enterprise $ssid $identity $psk $name
 			;;
 	esac
 }
 
-reconnect_old_config() {
-	name="`ls $cfg_loc | dmenu -i -p 'Choose config to connect with'`"
-	if [ -f $name ] 
-	then
-		rm -f "$cfg_loc/$name"
-		__kill_old
-	fi
-}
 
-option=`printf "Remove\nReconnect\nNew\n" | dmenu -i`
+option=`printf "Remove\nConnect\nNew\nDiconnect\n" | dmenu -i`
 case $option in 
 	New) create_new_config;;
 	Remove) remove_old_config;;
-	Reconnect) reconnect_old_config;;
+	Connect) reconnect_old_config;;
+	Disconnect) __kill_old;;
 esac