small update to gui

Author: sitzmaa

frontend/gui.py

@@ -1,32 +1,86 @@
 import sys
 import socket
 import argparse
-from PyQt6.QtWidgets import QApplication, QMainWindow, QPushButton, QLabel
-from PyQt6.QtCore import QThread, pyqtSignal
+from PyQt6.QtWidgets import (QApplication, QMainWindow, QPushButton, QLabel, 
+                             QGraphicsScene, QGraphicsView, QGraphicsEllipseItem, QGraphicsTextItem, QVBoxLayout, QWidget, QLineEdit)
+from PyQt6.QtCore import Qt, QThread, pyqtSignal
+from PyQt6.QtGui import QBrush, QColor
+
+class NetworkNode(QGraphicsEllipseItem):
+    def __init__(self, x, y, ip, details):
+        super().__init__(0, 0, 50, 50)
+        self.setBrush(QBrush(QColor(0, 0, 255)))
+        self.setFlag(QGraphicsEllipseItem.GraphicsItemFlag.ItemIsSelectable)
+        self.setPos(x, y)
+        self.ip = ip
+        self.details = details
+
+        self.text = QGraphicsTextItem(ip, self)
+        self.text.setDefaultTextColor(Qt.GlobalColor.white)
+        self.text.setPos(10, 10)
+
+    def mousePressEvent(self, event):
+        print(f"Clicked on node: {self.ip}\nDetails: {self.details}")
+        super().mousePressEvent(event)
 
 class NetworkVisualizerApp(QMainWindow):
     def __init__(self, port):
         super().__init__()
         self.port = port
         self.setWindowTitle("NetMap - Network Visualizer")
-        self.setGeometry(100, 100, 800, 600)
+        self.setGeometry(100, 100, 1000, 700)
+
+        self.central_widget = QWidget()
+        self.setCentralWidget(self.central_widget)
+        self.layout = QVBoxLayout(self.central_widget)
 
         self.label = QLabel("Network Map", self)
-        self.label.move(350, 50)
+        self.layout.addWidget(self.label)
+
+        # Graphics View for Network Map
+        self.scene = QGraphicsScene()
+        self.view = QGraphicsView(self.scene)
+        self.layout.addWidget(self.view)
 
-        self.btn = QPushButton("Scan Network", self)
-        self.btn.move(350, 100)
-        self.btn.clicked.connect(self.scan_network)
+        # Command Input Box
+        self.command_input = QLineEdit(self)
+        self.command_input.setPlaceholderText("Enter command...")
+        self.layout.addWidget(self.command_input)
+        self.command_input.returnPressed.connect(self.send_command)
 
+        # Buttons
+        self.scan_btn = QPushButton("Scan Network", self)
+        self.scan_btn.clicked.connect(self.scan_network)
+        self.layout.addWidget(self.scan_btn)
+
+        # Backend Thread
         self.backend_thread = BackendThread(self.port)
-        self.backend_thread.new_data.connect(self.update_label)
+        self.backend_thread.new_data.connect(self.update_network)
         self.backend_thread.start()
 
     def scan_network(self):
-        self.backend_thread.send_command("ping")
+        self.backend_thread.send_command("scan")
+
+    def send_command(self):
+        command = self.command_input.text().strip()
+        if command:
+            self.backend_thread.send_command(command)
+            self.command_input.clear()
 
-    def update_label(self, data):
-        self.label.setText(f"Server Response: {data}")
+    def update_network(self, data):
+        self.scene.clear()
+        nodes = data.split(";")
+        x, y = 50, 50
+        for node_info in nodes:
+            parts = node_info.split(",")
+            if len(parts) >= 2:
+                ip, details = parts[0], parts[1]
+                node = NetworkNode(x, y, ip, details)
+                self.scene.addItem(node)
+                x += 100
+                if x > 800:
+                    x = 50
+                    y += 100
 
 class BackendThread(QThread):
     new_data = pyqtSignal(str)
@@ -36,15 +90,14 @@ def __init__(self, port):
         self.port = port
         self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         self.socket.connect(("localhost", self.port))
-        # Send 'True' to indicate that it's a GUI client
         self.socket.sendall(b"1\n")
-    
+
     def run(self):
         while True:
             data = self.socket.recv(1024).decode()
             if data:
                 self.new_data.emit(data)
-    
+
     def send_command(self, command):
         self.socket.sendall(command.encode() + b"\n")
 
@@ -59,4 +112,4 @@ def run():
     sys.exit(app.exec())
 
 if __name__ == "__main__":
-    run()
+    run()