main.py

from textual.app import App, ComposeResult
from textual.containers import Container, Horizontal, Vertical
from textual.widgets import Header, Footer, Input, Button, DataTable, Static, Label
from textual.worker import Worker
from scanner import Scanner
import asyncio

class IPv4ScannerApp(App):
    CSS = """
    Screen {
        layout: vertical;
    }
    
    .input-container {
        height: auto;
        margin: 1;
        padding: 1;
        border: solid green;
    }
    
    .input-label {
        padding: 1;
    }
    
    #scan-btn {
        margin: 1;
        width: 100%;
    }
    
    DataTable {
        height: 1fr;
        border: solid blue;
    }
    """

    def compose(self) -> ComposeResult:
        yield Header()
        yield Container(
            Vertical(
                Label("Target IPs (comma separated):", classes="input-label"),
                Input(placeholder="e.g. 127.0.0.1, 192.168.1.1", id="ip-input"),
                Label("Target Ports (comma separated or range):", classes="input-label"),
                Input(placeholder="e.g. 80, 443, 8000-8010", value="80", id="port-input"),
                Button("Start Scan", id="scan-btn", variant="primary"),
                classes="input-container"
            )
        )
        yield DataTable()
        yield Footer()

    def on_mount(self) -> None:
        table = self.query_one(DataTable)
        table.add_columns("IP Address", "Port", "Status")
        self.scanner = Scanner()

    async def on_button_pressed(self, event: Button.Pressed) -> None:
        if event.button.id == "scan-btn":
            ip_input = self.query_one("#ip-input", Input).value
            port_input = self.query_one("#port-input", Input).value
            
            if not ip_input:
                self.notify("Please enter IP addresses.", severity="error")
                return

            ips = self.scanner.parse_ips(ip_input)
            ports = self.scanner.parse_ports(port_input)
            
            if not ips:
                self.notify("No valid IPs found.", severity="error")
                return
            if not ports:
                self.notify("No valid ports found.", severity="error")
                return

            self.query_one(DataTable).clear()
            self.run_worker(self.perform_scan(ips, ports), exclusive=True)

    async def perform_scan(self, ips, ports):
        table = self.query_one(DataTable)
        self.notify(f"Scanning {len(ips)} IPs on {len(ports)} ports...")
        
        # We can scan in chunks or all at once. For TUI responsiveness, 
        # let's process them and update table as we go, or just wait for all.
        # To make it update live, we should probably not use the bulk scan_range 
        # but iterate here.
        
        for ip in ips:
            for port in ports:
                # Update status to "Scanning..."
                # table.add_row(ip, str(port), "Scanning...")
                # Actually, let's just await the result
                
                result = await self.scanner.scan_port(ip, port, timeout=0.5)
                # result is (ip, port, is_open)
                status = "[green]OPEN[/green]" if result[2] else "[red]CLOSED[/red]"
                
                # Only show open ports? Or all? Let's show all for now, or maybe just open.
                # Usually scanners show open ports. But user might want to know it checked.
                # Let's show all but maybe sort or filter later.
                
                table.add_row(result[0], str(result[1]), status)
        
        self.notify("Scan complete.")

if __name__ == "__main__":
    app = IPv4ScannerApp()
    app.run()