From 631ab97a9c70e32cb5d32a398e366d6d97d25353 Mon Sep 17 00:00:00 2001
From: iamromulan <50184035+iamromulan@users.noreply.github.com>
Date: Mon, 22 Apr 2024 00:15:53 -0400
Subject: [PATCH] Building menus

---
 .../console/menu/sfirewall_settings.sh        | 125 ++++++++++++++++++
 simpleadmin/console/menu/start_menu.sh        |  25 +---
 2 files changed, 131 insertions(+), 19 deletions(-)
 create mode 100644 simpleadmin/console/menu/sfirewall_settings.sh

diff --git a/simpleadmin/console/menu/sfirewall_settings.sh b/simpleadmin/console/menu/sfirewall_settings.sh
new file mode 100644
index 0000000..279e229
--- /dev/null
+++ b/simpleadmin/console/menu/sfirewall_settings.sh
@@ -0,0 +1,125 @@
+#!/bin/bash
+
+SIMPLE_FIREWALL_DIR="/usrdata/simplefirewall"
+SIMPLE_FIREWALL_SCRIPT="$SIMPLE_FIREWALL_DIR/simplefirewall.sh"
+SIMPLE_FIREWALL_SYSTEMD_DIR="$SIMPLE_FIREWALL_DIR/systemd"
+
+# Display Messages in Colors
+display_random_color() {
+    local msg="$1"
+    local colors=(33 34 35 36 37)  # ANSI color codes for yellow, blue, magenta, cyan, white
+    local num_colors=${#colors[@]}
+    local random_color_index=$(($RANDOM % num_colors))  # Pick a random index from the colors array
+    echo -e "\033[${colors[$random_color_index]}m$msg\033[0m"
+}
+
+display_green() {
+    echo -e "\033[0;32m$1\033[0m"
+}
+
+display_red() {
+    echo -e "\033[0;31m$1\033[0m"
+}
+
+set_portblocks() {
+        current_ports_line=$(grep '^PORTS=' "$SIMPLE_FIREWALL_SCRIPT")
+        ports=$(echo "$current_ports_line" | cut -d'=' -f2 | tr -d '()' | tr ' ' '\n' | grep -o '[0-9]\+')
+        echo -e "\e[1;32mCurrent configured ports:\e[0m"
+        echo "$ports" | awk '{print NR") "$0}'
+
+        while true; do
+            echo -e "\e[1;32mEnter a port number to add/remove, or type 'done' or 'exit' to finish:\e[0m"
+            read port
+            if [ "$port" = "done" ] || [ "$port" = "exit" ]; then
+                if [ "$port" = "exit" ]; then
+                    echo -e "\e[1;31mExiting without making changes...\e[0m"
+                    return
+                fi
+                break
+            elif ! echo "$port" | grep -qE '^[0-9]+$'; then
+                echo -e "\e[1;31mInvalid input: Please enter a numeric value.\e[0m"
+            elif echo "$ports" | grep -q "^$port\$"; then
+                ports=$(echo "$ports" | grep -v "^$port\$")
+                echo -e "\e[1;32mPort $port removed.\e[0m"
+            else
+                ports=$(echo "$ports"; echo "$port" | grep -o '[0-9]\+')
+                echo -e "\e[1;32mPort $port added.\e[0m"
+            fi
+        done
+
+        if [ "$port" != "exit" ]; then
+            new_ports_line="PORTS=($(echo "$ports" | tr '\n' ' '))"
+            sed -i "s/$current_ports_line/$new_ports_line/" "$SIMPLE_FIREWALL_SCRIPT"
+        fi
+}
+
+set_ttl(){
+# TTL configuration code
+        ttl_value=$(cat /usrdata/simplefirewall/ttlvalue)
+        if [ "$ttl_value" -eq 0 ]; then
+            echo -e "\e[1;31mTTL is not set.\e[0m"
+        else
+            echo -e "\e[1;32mTTL value is set to $ttl_value.\e[0m"
+        fi
+
+        echo -e "\e[1;31mType 'exit' to cancel.\e[0m"
+        read -p "What do you want the TTL value to be: " new_ttl_value
+        if [ "$new_ttl_value" = "exit" ]; then
+            echo -e "\e[1;31mExiting TTL configuration...\e[0m"
+            return
+        elif ! echo "$new_ttl_value" | grep -qE '^[0-9]+$'; then
+            echo -e "\e[1;31mInvalid input: Please enter a numeric value.\e[0m"
+            return
+        else
+            /usrdata/simplefirewall/ttl-override stop
+	    echo "$new_ttl_value" > /usrdata/simplefirewall/ttlvalue
+     	    /usrdata/simplefirewall/ttl-override start
+            echo -e "\033[0;32mTTL value updated to $new_ttl_value.\033[0m"
+        fi
+}
+
+# function to configure the fetures of simplefirewall
+simple_firewall_menu() {
+    if [ ! -f "$SIMPLE_FIREWALL_SCRIPT" ]; then
+        echo -e "\033[0;31mSimplefirewall is not installed, would you like to install it?\033[0m"
+        echo -e "\033[0;32m1) Yes\033[0m"
+        echo -e "\033[0;31m2) No\033[0m"
+        read -p "Enter your choice (1-2): " install_choice
+
+        case $install_choice in
+            1)
+                install_simple_firewall
+                ;;
+            2)
+                return
+                ;;
+            *)
+                echo -e "\033[0;31mInvalid choice. Please select either 1 or 2.\033[0m"
+                ;;
+        esac
+    fi
+
+    echo -e "\e[1;32mConfigure Simple Firewall:\e[0m"
+    echo -e "\e[38;5;208m1) Configure incoming port block\e[0m"
+    echo -e "\e[38;5;27m2) Configure TTL\e[0m"
+    read -p "Enter your choice (1-2): " menu_choice
+
+    case $menu_choice in
+    1)
+		set_portblocks
+        ;;
+    2)
+        set_ttl
+        ;;
+    *)
+        echo -e "\e[1;31mInvalid choice. Please select either 1 or 2.\e[0m"
+        ;;
+    esac
+
+    systemctl restart simplefirewall
+    echo -e "\e[1;32mFirewall configuration updated.\e[0m"
+}
+
+# Start by checking and installing xml if necessary, then mount filesystem as rw and run the menu
+mount -o remount,rw /
+simple_firewall_menu
\ No newline at end of file
diff --git a/simpleadmin/console/menu/start_menu.sh b/simpleadmin/console/menu/start_menu.sh
index cea218a..0955ebd 100644
--- a/simpleadmin/console/menu/start_menu.sh
+++ b/simpleadmin/console/menu/start_menu.sh
@@ -2,6 +2,7 @@
 
 # Define executable files path
 EXE=/usrdata/root/bin
+MENU_SH=/usrdata/simpleadmin/console/menu
 
 # Display Messages in Colors
 display_random_color() {
@@ -42,7 +43,7 @@ toolkit() {
         read -p "Select an option (1-11): " option
 
         case "$option" in
-            1) $EXE/LAN_settings ;;
+            1) $MEU/LAN_settings ;;
             2) $EXE/simplepasswd ;;
             3) passwd ;;
             4) mc ;;
@@ -78,7 +79,7 @@ apps() {
         read -p "Select an option (1-11): " option
 
         case "$option" in
-            1) $EXE/LAN_settings ;;
+            1) $menu_sh/LAN_settings ;;
             2) $EXE/simplepasswd ;;
             3) passwd ;;
             4) mc ;;
@@ -102,29 +103,15 @@ settings() {
         display_green "1. LAN Settings"
         display_green "2. Change simpleadmin (admin) password"
         display_green "3. Change root password (shell/ssh/console)"
-        display_green "4. Open File Browser/Editor (mc)"
-        display_green "5. View Used/Available space"
-        display_green "6. Open Task Manager/View CPU Load"
-        display_green "7. Run speedtest.net test"
-        display_green "8. Run fast.com test (30Mbps max)"
-        display_green "9. Get and run the Toolkit"
-        display_green "10. Get and run the Development/unstable Toolkit"
-        display_green "11. Exit (Enter Root Shell)"
+        display_green "4. Go back"
         echo
         read -p "Select an option (1-11): " option
 
         case "$option" in
-            1) $EXE/LAN_settings ;;
+            1) $MENU_SH/LAN_settings ;;
             2) $EXE/simplepasswd ;;
             3) passwd ;;
-            4) mc ;;
-            5) dfc ;;
-            6) htop ;;
-            7) $EXE/speedtest ;;
-            8) $EXE/fast ;;
-            9) cd /tmp && wget -O RMxxx_rgmii_toolkit.sh https://raw.githubusercontent.com/iamromulan/quectel-rgmii-toolkit/main/RMxxx_rgmii_toolkit.sh && chmod +x RMxxx_rgmii_toolkit.sh && ./RMxxx_rgmii_toolkit.sh && cd / ;;
-            10) cd /tmp && wget -O RMxxx_rgmii_toolkit.sh https://raw.githubusercontent.com/iamromulan/quectel-rgmii-toolkit/development/RMxxx_rgmii_toolkit.sh && chmod +x RMxxx_rgmii_toolkit.sh && ./RMxxx_rgmii_toolkit.sh && cd / ;;
-            11) break ;;
+            4) break ;;
             *) echo "Invalid option. Please try again." ;;
         esac
     done