96 lines
3.0 KiB
Python
96 lines
3.0 KiB
Python
from __future__ import annotations
|
|
from datetime import datetime
|
|
from html import escape
|
|
from pathlib import Path
|
|
from typing import Dict, List
|
|
|
|
from jinja2 import Environment, FileSystemLoader, select_autoescape
|
|
|
|
from utils.models import HostResult
|
|
|
|
def fmt_ports(ports: List[int]) -> str:
|
|
if not ports:
|
|
return "none"
|
|
return ", ".join(str(p) for p in sorted(set(int(x) for x in ports)))
|
|
|
|
def badge(text: str, bg: str, fg: str = "#ffffff") -> str:
|
|
return (
|
|
f'<span style="display:inline-block;padding:2px 6px;border-radius:12px;'
|
|
f'font-size:12px;line-height:16px;background:{bg};color:{fg};'
|
|
f'font-family:Segoe UI,Arial,sans-serif">{escape(text)}</span>'
|
|
)
|
|
|
|
def pill_state(state: str) -> str:
|
|
s = (state or "").lower()
|
|
if s == "open":
|
|
return badge("open", "#16a34a")
|
|
if s in ("open|filtered",):
|
|
return badge("open|filtered", "#0ea5e9")
|
|
if s == "filtered":
|
|
return badge("filtered", "#f59e0b", "#111111")
|
|
if s == "closed":
|
|
return badge("closed", "#ef4444")
|
|
return badge(s, "#6b7280")
|
|
|
|
def pill_proto(proto: str) -> str:
|
|
return badge((proto or "").lower(), "#334155")
|
|
|
|
def _env(templates_dir: Path) -> Environment:
|
|
env = Environment(
|
|
loader=FileSystemLoader(str(templates_dir)),
|
|
autoescape=select_autoescape(["html", "xml"]),
|
|
trim_blocks=True,
|
|
lstrip_blocks=True,
|
|
)
|
|
env.globals.update(badge=badge, pill_state=pill_state, pill_proto=pill_proto, fmt_ports=fmt_ports)
|
|
return env
|
|
|
|
def render_html_report_jinja(
|
|
reports: Dict[str, Dict[str, List[int]]],
|
|
host_results: List[HostResult],
|
|
templates_dir: Path,
|
|
template_name: str = "report.html.j2",
|
|
title: str = "Port Compliance Report",
|
|
only_issues: bool = False,
|
|
) -> str:
|
|
env = _env(templates_dir)
|
|
template = env.get_template(template_name)
|
|
|
|
total_hosts = len(reports)
|
|
hosts_with_issues = [ip for ip, r in reports.items() if any(r.values())]
|
|
ok_hosts = total_hosts - len(hosts_with_issues)
|
|
|
|
# No filtering — we'll show all, but only_issues changes template behavior.
|
|
by_ip = {hr.address: hr for hr in host_results}
|
|
for hr in by_ip.values():
|
|
if hr and hr.ports:
|
|
hr.ports.sort(key=lambda p: (p.protocol, p.port))
|
|
|
|
html = template.render(
|
|
title=title,
|
|
generated=datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
|
total_hosts=total_hosts,
|
|
ok_hosts=ok_hosts,
|
|
hosts_with_issues=hosts_with_issues,
|
|
reports=reports,
|
|
host_results_by_ip=by_ip,
|
|
only_issues=only_issues,
|
|
)
|
|
return html
|
|
|
|
|
|
def write_html_report_jinja(
|
|
reports: Dict[str, Dict[str, List[int]]],
|
|
host_results: List[HostResult],
|
|
out_path: Path,
|
|
templates_dir: Path = Path("templates"),
|
|
template_name: str = "report.html.j2",
|
|
title: str = "Port Compliance Report",
|
|
only_issues: bool = False,
|
|
) -> Path:
|
|
html = render_html_report_jinja(
|
|
reports, host_results, templates_dir, template_name, title, only_issues
|
|
)
|
|
out_path.write_text(html, encoding="utf-8")
|
|
return out_path
|