main.py

import sys
from core.scanner import run_nmap_scan, run_nmap_scan_no_ping, run_nmap_network_scan
from core.parser import parse_nmap_xml, has_open_ports
from core.analyzer import analyze_with_ollama
from core.reporter import build_report, build_network_report, save_report
from core.utils import ensure_directories, timestamp


def is_network_target(target: str) -> bool:
    return "/" in target


def main() -> None:
    if len(sys.argv) < 2:
        print("Usage: python3 main.py <target> [mode]")
        print("Examples:")
        print("  python3 main.py 192.168.20.103")
        print("  python3 main.py 192.168.20.103 dual")
        print("  python3 main.py 192.168.20.103 offensive")
        print("  python3 main.py 192.168.20.103 defensive")
        print("  python3 main.py 192.168.20.0/24")
        sys.exit(1)

    target = sys.argv[1]
    mode = sys.argv[2] if len(sys.argv) > 2 else "dual"

    ensure_directories()
    ts = timestamp()

    safe_target = target.replace("/", "_")
    report_path = f"reports/{safe_target}_{ts}.md"

    if is_network_target(target):
        scan_base = f"scans/{safe_target}_{ts}"
        print(f"[+] Network overview mode: {target}")
        xml_path, scan_mode = run_nmap_network_scan(target, scan_base)
        print(f"[+] Scan saved: {xml_path}")

        print("[+] Parsing XML")
        parsed = parse_nmap_xml(str(xml_path))

        print("[+] Generating network overview report")
        report_text = build_network_report(target, parsed, scan_mode)
        final_report = save_report(report_text, report_path)

        print(f"[+] Done. Report saved to: {final_report}")
        return

    print(f"[+] Mode: {mode}")

    scan_base = f"scans/{safe_target}_{ts}"
    print(f"[+] Starting scan on {target}")
    xml_path, scan_mode = run_nmap_scan(target, scan_base)
    print(f"[+] Scan saved: {xml_path}")

    print("[+] Parsing XML")
    parsed = parse_nmap_xml(str(xml_path))

    if not has_open_ports(parsed):
        print("[!] No open ports found. Retrying with -Pn...")
        scan_base_retry = f"scans/{safe_target}_{ts}_pn"
        xml_path, scan_mode = run_nmap_scan_no_ping(target, scan_base_retry)
        print(f"[+] Retry scan saved: {xml_path}")
        parsed = parse_nmap_xml(str(xml_path))

    print("[+] Analyzing with local LLM")
    analysis = analyze_with_ollama(parsed, mode=mode)

    print("[+] Generating report")
    report_text = build_report(target, parsed, analysis, scan_mode)
    final_report = save_report(report_text, report_path)

    print(f"[+] Done. Report saved to: {final_report}")


if __name__ == "__main__":
    main()