Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
57 changes: 57 additions & 0 deletions src/manus_agent/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -1057,6 +1057,7 @@ def _run_variants(argv: list[str]) -> int:
"poc-search",
"changelog",
"blast-radius",
"cluster-variants",
}


Expand Down Expand Up @@ -1935,6 +1936,58 @@ def _run_blast_radius(argv: list[str]) -> int:
return 0


# ---------------------------------------------------------------------------
# cluster-variants subcommand
# ---------------------------------------------------------------------------


def _build_cluster_variants_parser() -> argparse.ArgumentParser:
p = argparse.ArgumentParser(
prog="manus-agent cluster-variants",
description=(
"Group CVEs related to an input CVE across three cluster dimensions:\n"
"same component/vendor, same CWE weakness class, and same\n"
"researcher/disclosure source."
),
add_help=True,
)
p.add_argument("cve_id", metavar="CVE-ID", help="CVE identifier, e.g. CVE-2021-44228")
p.add_argument(
"--output",
choices=["text", "json"],
default="text",
help="Output format (default: text)",
)
return p


def _run_cluster_variants(argv: list[str]) -> int:
import json as _json

parser = _build_cluster_variants_parser()
args = parser.parse_args(argv)
cve_id = args.cve_id.strip().upper()

if not cve_id.startswith("CVE-"):
print("[error] CVE-ID must start with 'CVE-'", file=sys.stderr)
return 1

from manus_agent.tools.cluster_variants import _render_text, cluster_variants

result = cluster_variants(cve_id)

if "error" in result:
print(f"[error] {result['error']}", file=sys.stderr)
return 1

if args.output == "json":
print(_json.dumps(result, indent=2))
else:
print(_render_text(result))

return 0


def _build_run_parser() -> argparse.ArgumentParser:
"""Build the top-level run/interactive parser."""
parser = argparse.ArgumentParser(
Expand Down Expand Up @@ -2269,6 +2322,10 @@ def main() -> None:
idx = argv.index("blast-radius")
sys.exit(_run_blast_radius(argv[idx + 1 :]))

if first_positional == "cluster-variants":
idx = argv.index("cluster-variants")
sys.exit(_run_cluster_variants(argv[idx + 1 :]))

if first_positional == "discover":
idx = argv.index("discover")
discover_args = _build_discover_parser().parse_args(argv[idx + 1 :])
Expand Down
Loading
Loading