Updated frkn tool

tools/frkn.sh

@@ -12,15 +12,38 @@ function connect() {
     echo
 }
 
+function in_array() {
+    local find=$1
+    shift
+    for e in "$@"; do
+        [[ "$e" == "$find" ]] && return 0
+    done
+    return 1
+}
+
 country="$1"
-countries=(uk ru nl nl2 ch)
+countries=()
 current=$(sudo wg show | head -n 1 | awk '{print $2}')
 
-if [ -z "$country" ]; then
+for file in /etc/wireguard/*.conf; do
-    IFS= read -rp "Выбери страну (${countries[*]}): " country
+    filename=${file/\/etc\/wireguard\/frkn-}
+    code=${filename/.conf/}
+    countries+=($code)
+done
+
+correct=-1
+if [ -z "$country" ] ; then
+    while [ $correct -lt 0 ]; do
+        read -rp "Выбери страну (${countries[*]}): " country
+        if in_array "$country" ${countries[@]}; then
+            correct=1
+        else
+            echo "Неверный код страны!"
+        fi
+    done
 fi
 
-case $country in
+case "$country" in
     "down" )
         if [ -n "$current" ]; then
             disconnect "$current"