Rewrite all services, cleate new cli interface, fix more bugs, add root check

Author: trapplus

app/interfaces/api/api.py

@@ -0,0 +1,118 @@
+from app.interfaces.cli.menu import Menu
+from app.services.bbr import BBRService
+from app.services.docker import DockerService
+from app.services.fail2ban import Fail2BanService
+from app.services.ufw import UFWService
+
+
+def run_interactive_script():
+    menu = Menu()
+
+    while True:
+        menu.update_status()
+        menu.display()
+
+        choice = input()
+
+        if choice == "0":
+            break
+        elif choice == "1":
+            bbr_menu()
+        elif choice == "2":
+            docker_menu()
+        elif choice == "3":
+            fail2ban_menu()
+        elif choice == "4":
+            ufw_menu()
+
+
+def bbr_menu():
+    while True:
+        print("\n1 - Enable BBR")
+        print("2 - Disable BBR")
+        print("0 - Назад")
+
+        choice = input("Выбор: ")
+
+        if choice == "0":
+            break
+
+        bbr = BBRService.create()
+
+        if choice == "1":
+            bbr.enable()
+            print("BBR включен")
+            input("Press Enter...")
+        elif choice == "2":
+            bbr.disable()
+            print("BBR выключен")
+            input("Press Enter...")
+
+
+def docker_menu():
+    while True:
+        print("\n1 - Install Docker")
+        print("2 - Uninstall Docker")
+        print("0 - Назад")
+
+        choice = input("Выбор: ")
+
+        if choice == "0":
+            break
+
+        docker = DockerService.create()
+
+        if choice == "1":
+            docker.install()
+            print("Docker установлен")
+            input("Press Enter...")
+        elif choice == "2":
+            docker.uninstall()
+            print("Docker удален")
+            input("Press Enter...")
+
+
+def fail2ban_menu():
+    while True:
+        print("\n1 - Install Fail2Ban")
+        print("2 - Uninstall Fail2Ban")
+        print("0 - Назад")
+
+        choice = input("Выбор: ")
+
+        if choice == "0":
+            break
+
+        fail2ban = Fail2BanService.create()
+
+        if choice == "1":
+            fail2ban.install()
+            print("Fail2Ban установлен")
+            input("Press Enter...")
+        elif choice == "2":
+            fail2ban.uninstall()
+            print("Fail2Ban удален")
+            input("Press Enter...")
+
+
+def ufw_menu():
+    while True:
+        print("\n1 - Install UFW")
+        print("2 - Uninstall UFW")
+        print("0 - Назад")
+
+        choice = input("Выбор: ")
+
+        if choice == "0":
+            break
+
+        ufw = UFWService.create()
+
+        if choice == "1":
+            ufw.install()
+            print("UFW установлен")
+            input("Press Enter...")
+        elif choice == "2":
+            ufw.uninstall()
+            print("UFW удален")
+            input("Press Enter...")

app/interfaces/cli/cli.py

@@ -0,0 +1,90 @@
+import platform
+from subprocess import run
+
+from app.utils import sysinfo_utils
+
+
+class Menu:
+    def __init__(self):
+        self.distro = sysinfo_utils.detect_distro()
+        self.kernel = platform.release()
+        self.hostname = platform.node()
+
+        # Service statuses
+        self.bbr_status = "Unknown"
+        self.docker_status = "Unknown"
+        self.fail2ban_status = "Unknown"
+        self.ufw_status = "Unknown"
+
+        self.menu_text = ""
+        self.update_status()
+
+    def update_status(self):
+        """Update all service statuses"""
+        self.bbr_status = self._check_bbr()
+        self.docker_status = self._check_service("docker")
+        self.fail2ban_status = self._check_service("fail2ban")
+        self.ufw_status = self._check_ufw()
+        self._build_menu_text()
+
+    def _check_bbr(self):
+        """Check BBR status"""
+        try:
+            result = run(
+                ["sysctl", "net.ipv4.tcp_congestion_control"],
+                capture_output=True,
+                text=True,
+            )
+            if "bbr" in result.stdout:
+                return "Enabled"
+            return "Disabled"
+        except Exception:
+            return "Unknown"
+
+    def _check_service(self, service_name):
+        """Check service status (systemd)"""
+        try:
+            result = run(
+                ["systemctl", "is-active", service_name], capture_output=True, text=True
+            )
+            status = result.stdout.strip()
+            if status == "active":
+                return "Running"
+            return "Stopped"
+        except Exception:
+            return "Not Installed"
+
+    def _check_ufw(self):
+        """Check UFW status"""
+        try:
+            result = run(["ufw", "status"], capture_output=True, text=True)
+            if "Status: active" in result.stdout:
+                return "Active"
+            return "Inactive"
+        except Exception:
+            return "Not Installed"
+
+    def _build_menu_text(self):
+        """Build menu text"""
+        self.menu_text = f"""
+╔════════════════════════════════════════════════════════════╗
+║          DevOps Automation Scripts v1.0.4                  ║
+╚════════════════════════════════════════════════════════════╝
+
+System: {self.distro} | Kernel: {self.kernel} | Host: {self.hostname}
+Services: BBR [{self.bbr_status}] | Docker [{self.docker_status}] | Fail2Ban [{self.fail2ban_status}] | UFW [{self.ufw_status}]
+
+────────────────────────────────────────────────────────────
+
+Выберите скрипт:
+0 - Выход
+1 - BBR
+2 - Docker
+3 - Fail2Ban
+4 - UFW
+
+Выбор: """
+
+    def display(self):
+        """Display menu"""
+        print(self.menu_text, end="")

app/interfaces/cli/menu.py

@@ -1,14 +1,43 @@
+from abc import ABC, abstractmethod
+
 from app.utils import sysinfo_utils
 
-__os: str | None = sysinfo_utils.detect_distro().lower()
 
-if __os == "debian":
-    from app.services.distro.debian.bbr import BBRService
-elif __os == "arch":
-    from app.services.distro.arch.bbr import BBRService
-elif __os == "openwrt":
-    from app.services.distro.wrt.bbr import BBRService
-else:
-    raise OSError("Unsupported distro!")
+class BBRService(ABC):
+    """Base class for BBR congestion control management"""
+
+    def __init__(self):
+        self.path_to_sysctl_config = "/etc/sysctl.conf"
+        self.bbr_config = """
+net.core.default_qdisc=fq_codel
+net.ipv4.tcp_congestion_control=bbr
+"""
+
+    @abstractmethod
+    def enable(self):
+        """Enable BBR"""
+        pass
+
+    @abstractmethod
+    def disable(self):
+        """Disable BBR"""
+        pass
+
+    @staticmethod
+    def create():
+        """Factory method to create instance based on OS"""
+        __os: str | None = sysinfo_utils.detect_distro().lower()
+
+        if __os == "debian":
+            from app.services.distro.debian.bbr import DebianBBRService
+
+            return DebianBBRService()
+        if __os == "arch":
+            from app.services.distro.arch.bbr import ArchBBRService
+
+            return ArchBBRService()
+        if __os == "openwrt":
+            from app.services.distro.wrt.bbr import WrtBBRService
 
-bbrservice = BBRService()
+            return WrtBBRService()
+        raise OSError(f"Unsupported distro: {__os}")