#!/usr/bin/env python3
"""
JSON Key Validator - Verifies JSON files contain expected keys.
Usage: python json_key_validator.py <json_file> <key1,key2,...>
Exit 0: all keys present
Exit 1: missing keys
Exit 2: file error
"""
import sys
import json


def validate_json_keys(filepath: str, expected_keys: list) -> tuple[bool, list]:
    """Validate that a JSON file contains all expected keys."""
    with open(filepath, 'r') as f:
        data = json.load(f)
    
    if not isinstance(data, dict):
        return False, ["JSON root is not an object"]
    
    missing = [k for k in expected_keys if k not in data]
    return len(missing) == 0, missing


def main():
    if len(sys.argv) != 3:
        print("Usage: python json_key_validator.py <json_file> <key1,key2,...>")
        sys.exit(2)
    
    filepath = sys.argv[1]
    expected_keys = [k.strip() for k in sys.argv[2].split(',')]
    
    try:
        valid, missing = validate_json_keys(filepath, expected_keys)
        if valid:
            print(f"PASS: All {len(expected_keys)} keys present in {filepath}")
            sys.exit(0)
        else:
            print(f"FAIL: Missing keys in {filepath}: {missing}")
            sys.exit(1)
    except FileNotFoundError:
        print(f"ERROR: File not found: {filepath}")
        sys.exit(2)
    except json.JSONDecodeError as e:
        print(f"ERROR: Invalid JSON: {e}")
        sys.exit(2)


if __name__ == "__main__":
    main()