salvacybersec/strix
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: improve completion message display for scan results and user interruptions

Browse Source
This commit is contained in:
Ahmed Allam
2025-11-01 03:02:47 +02:00
parent b9f8ee3f67
commit cf1d43706a
1 changed files with 32 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
strix/interface/main.py
View File

@@ -329,14 +329,25 @@ def display_completion_message(args: argparse.Namespace, results_path: Path) ->
console = Console() console = Console()
tracer = get_global_tracer() tracer = get_global_tracer()
scan_completed = False
if tracer and tracer.scan_results:
scan_completed = tracer.scan_results.get("scan_completed", False)
has_vulnerabilities = tracer and len(tracer.vulnerability_reports) > 0
completion_text = Text() completion_text = Text()
completion_text.append("🦉 ", style="bold white") if scan_completed:
completion_text.append("AGENT FINISHED", style="bold green") completion_text.append("🦉 ", style="bold white")
completion_text.append(" • ", style="dim white") completion_text.append("AGENT FINISHED", style="bold green")
completion_text.append("Penetration test completed", style="white") completion_text.append(" • ", style="dim white")
completion_text.append("Penetration test completed", style="white")
else:
completion_text.append("🦉 ", style="bold white")
completion_text.append("SESSION ENDED", style="bold yellow")
completion_text.append(" • ", style="dim white")
completion_text.append("Penetration test interrupted by user", style="white")
stats_text = build_stats_text(tracer) stats_text = build_stats_text(tracer)
llm_stats_text = build_llm_stats_text(tracer) llm_stats_text = build_llm_stats_text(tracer)
target_text = Text() target_text = Text()
@@ -352,39 +363,29 @@ def display_completion_message(args: argparse.Namespace, results_path: Path) ->
if i < len(args.targets_info) - 1: if i < len(args.targets_info) - 1:
target_text.append("\n") target_text.append("\n")
results_text = Text() panel_parts = [completion_text, "\n\n", target_text]
results_text.append("📊 Results Saved To: ", style="bold cyan")
results_text.append(str(results_path), style="bold yellow")
if stats_text.plain: if stats_text.plain:
if llm_stats_text.plain: panel_parts.extend(["\n", stats_text])
panel_content = Text.assemble(
completion_text, if llm_stats_text.plain:
"\n\n", panel_parts.extend(["\n", llm_stats_text])
target_text,
"\n", if scan_completed or has_vulnerabilities:
stats_text, results_text = Text()
"\n", results_text.append("📊 Results Saved To: ", style="bold cyan")
llm_stats_text, results_text.append(str(results_path), style="bold yellow")
"\n\n", panel_parts.extend(["\n\n", results_text])
results_text,
) panel_content = Text.assemble(*panel_parts)
else:
panel_content = Text.assemble( border_style = "green" if scan_completed else "yellow"
completion_text, "\n\n", target_text, "\n", stats_text, "\n\n", results_text
)
elif llm_stats_text.plain:
panel_content = Text.assemble(
completion_text, "\n\n", target_text, "\n", llm_stats_text, "\n\n", results_text
)
else:
panel_content = Text.assemble(completion_text, "\n\n", target_text, "\n\n", results_text)
panel = Panel( panel = Panel(
panel_content, panel_content,
title="[bold green]🛡️ STRIX CYBERSECURITY AGENT", title="[bold green]🛡️ STRIX CYBERSECURITY AGENT",
title_align="center", title_align="center",
border_style="green", border_style=border_style,
padding=(1, 2), padding=(1, 2),
) )