main_scan.py

import sys
import os
import argparse
import signal

# Ensure we can import app
sys.path.append(os.getcwd())

from app import create_app, db
from app.exploration.orchestrator.main import ExplorationOrchestrator

app = create_app()
orchestrator = None

def signal_handler(sig, frame):
    global orchestrator
    if orchestrator:
        print("\n[*] Interrupt received. Initiating graceful shutdown...")
        orchestrator.shutdown_flag.set()
    else:
        sys.exit(0)

def run_scan(url, enable_brute=False, scan_id=None, output_dir=None):
    global orchestrator
    with app.app_context():
        # Ensure FS isolation works by creating tables if missing
        db.create_all()
        orchestrator = ExplorationOrchestrator(app=app)
        try:
            orchestrator.start_scan(seed_url=url, clear_previous_data=True, enable_brute=enable_brute, scan_id=scan_id, output_dir=output_dir)
        finally:
            orchestrator.export_artifacts()

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="PentesterPro Autonomous Scanner")
    parser.add_argument("url", nargs="?", default="https://secladder.com", help="Target URL to scan")
    parser.add_argument("--brute", action="store_true", help="Enable brute force directory enumeration (requires ffuf)")
    parser.add_argument("--scan-id", help="Unique ID for this scan run (for batch processing)", default=None)
    parser.add_argument("--output-dir", help="Directory to save artifacts", default=None)
    args = parser.parse_args()
    
    signal.signal(signal.SIGINT, signal_handler)
    signal.signal(signal.SIGTERM, signal_handler)
    
    print(f"[*] Starting Autonomous Scan for: {args.url}")
    if args.brute:
        print("[*] Brute force directory enumeration: ENABLED")
    run_scan(args.url, enable_brute=args.brute, scan_id=args.scan_id, output_dir=args.output_dir)