run_analysis.py

import argparse
import importlib
import json
import sys
from pathlib import Path


def load_analyzer(module_path):
    """
    Dynamically load analyzer module
    Example: core.password_policy_analyzer
    """
    try:
        module = importlib.import_module(module_path)
        if not hasattr(module, "run"):
            raise AttributeError("Analyzer must expose a run() function")
        return module
    except Exception as e:
        print(f"[ERROR] Failed to load analyzer: {e}")
        sys.exit(1)


def main():
    parser = argparse.ArgumentParser(description="ADShield Generic Analysis Runner")

    parser.add_argument(
        "--analyzer",
        required=True,
        help="Analyzer module path (e.g. core.password_policy_analyzer)"
    )

    parser.add_argument(
        "--input",
        required=True,
        nargs="+",
        help="Input JSON file path(s); some analyzers may require multiple files"
    )

    parser.add_argument(
        "--output",
        default=None,
        help="Optional output file path (JSON)"
    )

    args = parser.parse_args()

    analyzer = load_analyzer(args.analyzer)

    # support one or more input files (some analyzers, e.g. stale_objects, need two)
    input_paths = [Path(p) for p in args.input]
    for p in input_paths:
        if not p.exists():
            print(f"[ERROR] Input file not found: {p}")
            sys.exit(1)

    # forward paths as separate arguments to the analyzer
    report = analyzer.run(*[str(p) for p in input_paths])

    if args.output:
        output_path = Path(args.output)
        output_path.parent.mkdir(parents=True, exist_ok=True)
        with open(output_path, "w", encoding="utf-8") as f:
            json.dump(report, f, indent=4)
        print(f"[+] Report written to {output_path}")
    else:
        print(json.dumps(report, indent=4))


if __name__ == "__main__":
    main()