From fa6caf4c226d89dce7d920237d27b4bed37fc588 Mon Sep 17 00:00:00 2001
From: schizza <lukyn@lukyns.com>
Date: Sun, 28 Apr 2024 02:35:04 +0200
Subject: [PATCH] Script fixed.

* Duplicity rule test
* color loging
---
 .../sws12500/iptables_redirect.sh             | 121 +++++++++++-------
 1 file changed, 72 insertions(+), 49 deletions(-)

diff --git a/custom_components/sws12500/iptables_redirect.sh b/custom_components/sws12500/iptables_redirect.sh
index ed34e25..3729cf3 100755
--- a/custom_components/sws12500/iptables_redirect.sh
+++ b/custom_components/sws12500/iptables_redirect.sh
@@ -1,62 +1,85 @@
-#!/bin/zsh
+#!/bin/bash
 
 # Script pro přesměrování portu pro stanici SWS12500
 
-STATION_IP = 192.168.2.95
-HA = 192.168.2.219
-SRC_PORT = 80
-DST_PORT = 8123
+# set -e
 
-INSTALL_IPTABLES = 0
-APK_MISSING = 0
+STATION_IP=192.168.2.95
+HA=192.168.2.219
+SRC_PORT=80
+DST_PORT=8123
 
-echo "Spoštím iptables pro 80 -> 8123 přesměrování"
+INSTALL_IPTABLES=0
+APK_MISSING=0
 
-# Máme nainstalované iptables?
+RED_COLOR='\033[0;31m'
+GREEN_COLOR='\033[0;32m'
+GREEN_YELLOW='\033[1;33m'
+NO_COLOR='\033[0m'
 
-echo -n "Kontrola zda jsou dostupné iptable ... "
-IPTABLES='$(type -p "iptables")'
-if ! [ -f "$IPTABLES" ]; then
-	echo "chybí"
-	INSTALL_IPTABLES = 1
-else
-	echo "OK"
-fi
 
-# Máme apk?
-echo -n "Kontrola zda je dostupný apk  ..."
-APK='$(type -p "apk")'
-if ! [ -f "$APK" ]; then
-	echo "chybí"
-	APK_MISSING = 1
-else
-	echo "OK"
-fi
+function info () { echo -e "${GREEN_COLOR}INFO: $1${NO_COLOR}";}
+function warn () { echo -e "${GREEN_YELLOW}WARN: $1${NO_COLOR}";}
+function error () { echo -e "${RED_COLOR}ERROR: $1${NO_COLOR}"; if [ "$2" != "false" ]; then exit 1;fi; }
 
-if [ APK_MISSING == 1 -a INSTALL_IPTABLES == 1 ]
-	echo "Nelze nakonfigurovat IP Tables. iptables chybí a chybí i instalační aplikace apk!!"
-	exit 1
-fi
-
-if [ INSTALL_IPTABLES == 1 -a APK_MISSING == 0]
-	runinstall=(apk add iptables)
-        echo -n "Spouštím instalaci iptables ... ${runinstall[@]} ... "
-        ${runinstall[@]}
-        EXIT_STATUS=$?
-	if [ $EXIT_STATUS -ne 0 ]
-                echo "Instalace iptables se nezdařila!"
-                exit $EXIT_STATUS
-	fi
-        runiptables=(iptables -t nat -I PREROUTING --src $STATION_IP --dst $HA -p tcp --dport $SRC_PORT -j REDIRECT --to-ports $DST_PORT)
-        echo -n "Spouštím iptables ... ${runiptables[@]} ..."
-        ${runiptables[@]}
-        EXIT_STATUS=$?
-        if [ $EXIT_STATUS -ne 0 ]
-                echo "Přidní pravidla do iptables se nezdařilo!"
-                exit $EXIT_STATUS
+function check () {
+	echo -n "Kontrola dostupnosti $1 ... "
+	if [ -z "$(command -v "$1")" ]; then
+		error "'$1' není nainstalován." $2
+		return 1
 	fi
+	info "OK."
+	return 0
+}
+
+
+echo 
+echo "**************************************************************"
+echo "*                                                            *"
+echo -e "*     ${GREEN_YELLOW}Spouštím iptables přesměrování pro port $SRC_PORT -> $DST_PORT ${NO_COLOR}    *"
+echo "*                                                            *"
+echo "**************************************************************"
+echo
+
+# Máme nainstalované iptables a apk?
+
+check "iptables" false
+INSTALL_IPTABLES=$?
+
+check "apk" false
+
+APK_MISSING=$?
+
+
+if [ $APK_MISSING -eq 1 ] && [ $INSTALL_IPTABLES -eq 1 ]; then
+        error "Nelze nakonfigurovat IP Tables. \n'iptables' chybí a chybí i instalační aplikace 'apk'!!\n"
 fi
-echo "iptables jsou nastaveny na přesměrování portu $SRC_PORT -> $DST_PORT pro stanici na IP: $STATION_IP"
-exit
 
+if [ $INSTALL_IPTABLES -eq 1 ] && [ $APK_MISSING -eq 0 ]; then
+        declare -a RUNINSTALL=(apk add iptables)
+        echo -n "Spouštím instalaci iptables ... ${RUNINSTAll[@]} ... "
+		${RUNINSTALL[@]}
+        EXIT_STATUS=$?
+        if [ $EXIT_STATUS -ne 0 ]; then
+				warn "Chybový kód instalace: $EXIT_STATUS"
+                error "Instalace iptables se nezdařila!"
+        else
+            info "'iptables' úspěšně nainstalovány."
+        fi
+fi
+declare -a RULE=(PREROUTING -t nat -s $STATION_IP -d $HA -p tcp -m tcp --dport $SRC_PORT -j REDIRECT --to-ports $DST_PORT)
+echo -ne "Spouštím iptables ... "
+$(iptables -C ${RULE[@]} 2>/dev/null)
+if [ $? -eq 0 ]; then
+    warn "Pravidlo je již v iptables zapsáno."
+else
+    $(iptables -I ${RULE[@]})
+fi
+EXIT_STATUS=$?
+if [ $EXIT_STATUS -ne 0 ]; then
+	warn "Chybový kód iptables: ${EXIT_STATUS} "
+	error "Přidání pravidla do iptables se nezdařilo!"
+fi
+
+info "iptables jsou nastaveny na přesměrování portu $SRC_PORT -> $DST_PORT pro stanici na IP: $STATION_IP"