mirror of
https://github.com/Rarebuffalo/securelens-backend.git
synced 2026-06-19 07:00:30 +00:00
added cli installer ]
This commit is contained in:
rarebuffalo
208
cli/securelens/output/__init__.py
Normal file
208
cli/securelens/output/__init__.py
Normal file
@@ -0,0 +1,208 @@
|
||||
"""
|
||||
Terminal Renderer
|
||||
=================
|
||||
All Rich-based output for the CLI — banners, progress, tables, panels.
|
||||
"""
|
||||
|
||||
import json
|
||||
from datetime import datetime
|
||||
from typing import Optional
|
||||
|
||||
from rich.console import Console
|
||||
from rich.panel import Panel
|
||||
from rich.progress import Progress, SpinnerColumn, BarColumn, TextColumn, TimeElapsedColumn
|
||||
from rich.table import Table
|
||||
from rich.text import Text
|
||||
from rich import box
|
||||
from rich.columns import Columns
|
||||
from rich.rule import Rule
|
||||
from rich.syntax import Syntax
|
||||
from rich.live import Live
|
||||
from rich.padding import Padding
|
||||
|
||||
console = Console()
|
||||
|
||||
# ── Severity colours ──────────────────────────────────────────────────────────
|
||||
SEVERITY_COLOR = {
|
||||
"Critical": "bold red",
|
||||
"High": "bold orange1",
|
||||
"Warning": "bold yellow",
|
||||
"Medium": "bold yellow",
|
||||
"Info": "bold blue",
|
||||
"Low": "bold cyan",
|
||||
}
|
||||
|
||||
GRADE_COLOR = {
|
||||
"A": "bold green",
|
||||
"B": "bold cyan",
|
||||
"C": "bold yellow",
|
||||
"D": "bold orange1",
|
||||
"F": "bold red",
|
||||
}
|
||||
|
||||
|
||||
def print_banner() -> None:
|
||||
banner = Text()
|
||||
banner.append("\n")
|
||||
banner.append(" ███████╗███████╗ ██████╗██╗ ██╗██████╗ ███████╗██╗ ███████╗███╗ ██╗███████╗\n", style="bold cyan")
|
||||
banner.append(" ██╔════╝██╔════╝██╔════╝██║ ██║██╔══██╗██╔════╝██║ ██╔════╝████╗ ██║██╔════╝\n", style="bold cyan")
|
||||
banner.append(" ███████╗█████╗ ██║ ██║ ██║██████╔╝█████╗ ██║ █████╗ ██╔██╗ ██║███████╗\n", style="bold blue")
|
||||
banner.append(" ╚════██║██╔══╝ ██║ ██║ ██║██╔══██╗██╔══╝ ██║ ██╔══╝ ██║╚██╗██║╚════██║\n", style="bold blue")
|
||||
banner.append(" ███████║███████╗╚██████╗╚██████╔╝██║ ██║███████╗███████╗███████╗██║ ╚████║███████║\n", style="bold magenta")
|
||||
banner.append(" ╚══════╝╚══════╝ ╚═════╝ ╚═════╝ ╚═╝ ╚═╝╚══════╝╚══════╝╚══════╝╚═╝ ╚═══╝╚══════╝\n", style="bold magenta")
|
||||
banner.append(" AI Security Agent v2.0.0\n", style="dim")
|
||||
console.print(banner)
|
||||
|
||||
|
||||
def print_scan_header(target: str, model: str) -> None:
|
||||
console.print(f" [bold]🔍 Target:[/bold] [cyan]{target}[/cyan]")
|
||||
console.print(f" [bold]🧠 Model:[/bold] [dim]{model}[/dim]")
|
||||
console.print()
|
||||
|
||||
|
||||
def make_progress() -> Progress:
|
||||
return Progress(
|
||||
SpinnerColumn(),
|
||||
TextColumn("[bold blue]{task.description}"),
|
||||
BarColumn(bar_width=30),
|
||||
TextColumn("[progress.percentage]{task.percentage:>3.0f}%"),
|
||||
TextColumn("• [dim]{task.fields[detail]}[/dim]"),
|
||||
TimeElapsedColumn(),
|
||||
console=console,
|
||||
transient=False,
|
||||
)
|
||||
|
||||
|
||||
def print_code_scan_report(result) -> None:
|
||||
"""Render a full local code scan report."""
|
||||
console.print()
|
||||
console.rule("[bold white] SECURITY REPORT [/bold white]", style="bright_black")
|
||||
console.print()
|
||||
|
||||
# Score panel
|
||||
grade_color = GRADE_COLOR.get(result.grade, "white")
|
||||
score_text = Text()
|
||||
score_text.append(f" {result.score}/100", style=f"bold {grade_color}")
|
||||
score_text.append(" Grade: ", style="dim")
|
||||
score_text.append(result.grade, style=grade_color)
|
||||
score_text.append(f" • {len(result.vulnerabilities)} issue(s) found", style="dim")
|
||||
score_text.append(f" • {len(result.files_triaged)} file(s) scanned", style="dim")
|
||||
console.print(Panel(score_text, title="[bold]Overall Score[/bold]", border_style="bright_black"))
|
||||
console.print()
|
||||
|
||||
if not result.vulnerabilities:
|
||||
console.print(" [bold green]✓ No vulnerabilities found![/bold green]")
|
||||
console.print()
|
||||
else:
|
||||
_print_vulnerability_table(result.vulnerabilities)
|
||||
|
||||
# AI Summary
|
||||
if result.ai_summary:
|
||||
console.print(Panel(
|
||||
result.ai_summary,
|
||||
title="[bold cyan]🤖 AI Security Summary[/bold cyan]",
|
||||
border_style="cyan",
|
||||
padding=(1, 2),
|
||||
))
|
||||
console.print()
|
||||
|
||||
|
||||
def print_web_scan_report(result) -> None:
|
||||
"""Render a full web scan report."""
|
||||
console.print()
|
||||
console.rule("[bold white] WEB SECURITY REPORT [/bold white]", style="bright_black")
|
||||
console.print()
|
||||
|
||||
if not result.reachable:
|
||||
console.print(" [bold red]✗ Could not reach the target URL[/bold red]")
|
||||
return
|
||||
|
||||
grade_color = GRADE_COLOR.get(result.grade, "white")
|
||||
score_text = Text()
|
||||
score_text.append(f" {result.score}/100", style=f"bold {grade_color}")
|
||||
score_text.append(" Grade: ", style="dim")
|
||||
score_text.append(result.grade, style=grade_color)
|
||||
if result.ssl_expiry_days is not None:
|
||||
score_text.append(f" • SSL expires in {result.ssl_expiry_days} days", style="dim")
|
||||
console.print(Panel(score_text, title="[bold]Overall Score[/bold]", border_style="bright_black"))
|
||||
console.print()
|
||||
|
||||
if result.exposed_paths:
|
||||
console.print(f" [bold red]⚠ Exposed sensitive paths:[/bold red] {', '.join(result.exposed_paths)}")
|
||||
console.print()
|
||||
|
||||
if not result.issues:
|
||||
console.print(" [bold green]✓ No issues found![/bold green]")
|
||||
else:
|
||||
_print_web_issue_table(result.issues)
|
||||
|
||||
if result.ai_summary:
|
||||
console.print(Panel(
|
||||
result.ai_summary,
|
||||
title="[bold cyan]🤖 AI Security Summary[/bold cyan]",
|
||||
border_style="cyan",
|
||||
padding=(1, 2),
|
||||
))
|
||||
console.print()
|
||||
|
||||
|
||||
def _print_vulnerability_table(vulns) -> None:
|
||||
"""Render grouped vulnerability table by severity."""
|
||||
severity_order = ["Critical", "High", "Medium", "Low"]
|
||||
grouped: dict = {s: [] for s in severity_order}
|
||||
for v in vulns:
|
||||
sev = v.severity if v.severity in grouped else "Low"
|
||||
grouped[sev].append(v)
|
||||
|
||||
for sev in severity_order:
|
||||
items = grouped[sev]
|
||||
if not items:
|
||||
continue
|
||||
color = SEVERITY_COLOR.get(sev, "white")
|
||||
console.print(f" [{color}]▶ {sev.upper()} ({len(items)})[/{color}]")
|
||||
for v in items:
|
||||
loc = f"[dim]{v.file_path}"
|
||||
if v.line_number:
|
||||
loc += f":{v.line_number}"
|
||||
loc += "[/dim]"
|
||||
console.print(f" [bold]{v.issue}[/bold] {loc}")
|
||||
console.print(f" [dim]{v.explanation}[/dim]")
|
||||
console.print(f" [green]Fix:[/green] [dim]{v.suggested_fix}[/dim]")
|
||||
console.print()
|
||||
|
||||
|
||||
def _print_web_issue_table(issues) -> None:
|
||||
"""Render web scan issues grouped by layer."""
|
||||
layers: dict = {}
|
||||
for issue in issues:
|
||||
layers.setdefault(issue.layer, []).append(issue)
|
||||
|
||||
for layer, items in layers.items():
|
||||
console.print(f" [bold bright_black]── {layer} ──[/bold bright_black]")
|
||||
for item in items:
|
||||
color = SEVERITY_COLOR.get(item.severity, "white")
|
||||
console.print(f" [{color}]●[/{color}] [bold]{item.issue}[/bold]")
|
||||
console.print(f" [green]Fix:[/green] [dim]{item.fix}[/dim]")
|
||||
console.print()
|
||||
|
||||
|
||||
def print_repl_prompt() -> None:
|
||||
console.print("\n[bold cyan]💬 Ask a follow-up[/bold cyan] [dim](or press Ctrl+C to exit)[/dim]")
|
||||
|
||||
|
||||
def print_ai_response(text: str) -> None:
|
||||
console.print()
|
||||
console.print(Panel(text, border_style="dim", padding=(0, 1)))
|
||||
console.print()
|
||||
|
||||
|
||||
def print_error(msg: str) -> None:
|
||||
console.print(f"\n [bold red]✗ {msg}[/bold red]\n")
|
||||
|
||||
|
||||
def print_success(msg: str) -> None:
|
||||
console.print(f"\n [bold green]✓ {msg}[/bold green]\n")
|
||||
|
||||
|
||||
def print_info(msg: str) -> None:
|
||||
console.print(f" [dim]{msg}[/dim]")
|
||||
148
cli/securelens/output/exporters.py
Normal file
148
cli/securelens/output/exporters.py
Normal file
@@ -0,0 +1,148 @@
|
||||
"""
|
||||
Export Formatters
|
||||
=================
|
||||
Serializes scan results to JSON and Markdown.
|
||||
"""
|
||||
|
||||
import json
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
|
||||
|
||||
# ── JSON ──────────────────────────────────────────────────────────────────────
|
||||
|
||||
def to_json(result, target_type: str = "code") -> str:
|
||||
"""Serialize a scan result to a JSON string."""
|
||||
if target_type == "code":
|
||||
data = {
|
||||
"scan_type": "code",
|
||||
"target": result.target,
|
||||
"timestamp": datetime.now().isoformat(),
|
||||
"score": result.score,
|
||||
"grade": result.grade,
|
||||
"files_scanned": result.files_triaged,
|
||||
"total_issues": len(result.vulnerabilities),
|
||||
"vulnerabilities": [
|
||||
{
|
||||
"file": v.file_path,
|
||||
"line": v.line_number,
|
||||
"severity": v.severity,
|
||||
"issue": v.issue,
|
||||
"explanation": v.explanation,
|
||||
"fix": v.suggested_fix,
|
||||
}
|
||||
for v in result.vulnerabilities
|
||||
],
|
||||
"ai_summary": result.ai_summary,
|
||||
}
|
||||
else: # web
|
||||
data = {
|
||||
"scan_type": "web",
|
||||
"target": result.url,
|
||||
"timestamp": datetime.now().isoformat(),
|
||||
"score": result.score,
|
||||
"grade": result.grade,
|
||||
"ssl_expiry_days": result.ssl_expiry_days,
|
||||
"exposed_paths": result.exposed_paths,
|
||||
"total_issues": len(result.issues),
|
||||
"issues": [
|
||||
{
|
||||
"layer": i.layer,
|
||||
"severity": i.severity,
|
||||
"issue": i.issue,
|
||||
"fix": i.fix,
|
||||
}
|
||||
for i in result.issues
|
||||
],
|
||||
"ai_summary": result.ai_summary,
|
||||
}
|
||||
return json.dumps(data, indent=2)
|
||||
|
||||
|
||||
def save_json(result, target_type: str = "code") -> Path:
|
||||
ts = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||
path = Path(f"securelens-report-{ts}.json")
|
||||
path.write_text(to_json(result, target_type))
|
||||
return path
|
||||
|
||||
|
||||
# ── Markdown ──────────────────────────────────────────────────────────────────
|
||||
|
||||
def to_markdown(result, target_type: str = "code") -> str:
|
||||
ts = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||
lines = []
|
||||
|
||||
if target_type == "code":
|
||||
lines.append(f"# SecureLens AI — Code Security Report\n")
|
||||
lines.append(f"**Target:** `{result.target}` ")
|
||||
lines.append(f"**Score:** {result.score}/100 **Grade:** {result.grade} ")
|
||||
lines.append(f"**Files Scanned:** {len(result.files_triaged)} ")
|
||||
lines.append(f"**Issues Found:** {len(result.vulnerabilities)} ")
|
||||
lines.append(f"**Generated:** {ts}\n")
|
||||
|
||||
if result.ai_summary:
|
||||
lines.append("## AI Summary\n")
|
||||
lines.append(result.ai_summary)
|
||||
lines.append("\n")
|
||||
|
||||
severity_order = ["Critical", "High", "Medium", "Low"]
|
||||
grouped: dict = {s: [] for s in severity_order}
|
||||
for v in result.vulnerabilities:
|
||||
sev = v.severity if v.severity in grouped else "Low"
|
||||
grouped[sev].append(v)
|
||||
|
||||
for sev in severity_order:
|
||||
items = grouped[sev]
|
||||
if not items:
|
||||
continue
|
||||
lines.append(f"## {sev} ({len(items)})\n")
|
||||
for v in items:
|
||||
loc = v.file_path
|
||||
if v.line_number:
|
||||
loc += f":{v.line_number}"
|
||||
lines.append(f"### `{v.issue}`")
|
||||
lines.append(f"**File:** `{loc}` ")
|
||||
lines.append(f"**Risk:** {v.explanation} ")
|
||||
lines.append(f"**Fix:** {v.suggested_fix}\n")
|
||||
|
||||
lines.append("## Files Scanned\n")
|
||||
for f in result.files_triaged:
|
||||
lines.append(f"- `{f}`")
|
||||
|
||||
else: # web
|
||||
lines.append(f"# SecureLens AI — Web Security Report\n")
|
||||
lines.append(f"**Target:** {result.url} ")
|
||||
lines.append(f"**Score:** {result.score}/100 **Grade:** {result.grade} ")
|
||||
lines.append(f"**Issues Found:** {len(result.issues)} ")
|
||||
if result.ssl_expiry_days is not None:
|
||||
lines.append(f"**SSL Expires In:** {result.ssl_expiry_days} days ")
|
||||
lines.append(f"**Generated:** {ts}\n")
|
||||
|
||||
if result.ai_summary:
|
||||
lines.append("## AI Summary\n")
|
||||
lines.append(result.ai_summary)
|
||||
lines.append("\n")
|
||||
|
||||
if result.exposed_paths:
|
||||
lines.append("## Exposed Paths\n")
|
||||
for p in result.exposed_paths:
|
||||
lines.append(f"- `{p}`")
|
||||
lines.append("")
|
||||
|
||||
layers: dict = {}
|
||||
for issue in result.issues:
|
||||
layers.setdefault(issue.layer, []).append(issue)
|
||||
for layer, items in layers.items():
|
||||
lines.append(f"## {layer}\n")
|
||||
for item in items:
|
||||
lines.append(f"**[{item.severity}]** {item.issue} ")
|
||||
lines.append(f"*Fix:* {item.fix}\n")
|
||||
|
||||
return "\n".join(lines)
|
||||
|
||||
|
||||
def save_markdown(result, target_type: str = "code") -> Path:
|
||||
ts = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||
path = Path(f"securelens-report-{ts}.md")
|
||||
path.write_text(to_markdown(result, target_type))
|
||||
return path
|
||||
Reference in New Issue
Block a user