main.py

import sys
import json
import os

from CodeZap.driver.parser import build_ir


def main():

    """
    Entry point for the CLI.

    Workflow:
    1. Read file path from command-line arguments
    2. Validate file existence (with fallback inside CodeZap/)
    3. Build IR from the source file
    4. Convert IR to clean JSON
    5. Output result:
       - print to stdout (default)
       - or write to file if output path is provided

    Usage:
        py -m CodeZap.main <file.py> [output.json]
    """

    # Step 1: Validate CLI arguments

    if len(sys.argv) < 2:
        print("Usage: py -m CodeZap.main <file.py> [output.json]")
        sys.exit(1)

    filepath = os.path.abspath(sys.argv[1])

    # Step 2: Check file existence

    if not os.path.exists(filepath):

        alt_path = os.path.join("CodeZap", filepath)

        if os.path.exists(alt_path):
            filepath = alt_path
        else:
            print(f"Error: file '{filepath}' not found")
            sys.exit(1)

    

    # Step 3: Build IR

    root = build_ir(filepath)

    # Step 4: Convert IR to clean JSON

    json_output = root.to_clean_dict()

    # Step 5: Output result

    output_file = sys.argv[2] if len(sys.argv) >= 3 else None

    if output_file:
        if os.path.exists(output_file):
            print(f"Error: '{output_file}' already exists. Use a different name.")
            sys.exit(1)

        with open(output_file, "w", encoding="utf-8") as f:
            json.dump(json_output, f, indent=2)
    else:
        print(json.dumps(json_output, indent=2))


if __name__ == "__main__":
    main()