#!/usr/bin/env python3

import sys
import json


def expand_capability(capability):
    with open('/etc/system_config.json', 'r') as f:
        config = json.load(f)

    if capability in config['ai_settings']['allowed_capabilities']:
        print(f"Capability '{capability}' is already available.")
        return

    if capability in ['network_access', 'file_creation', 'config_override']:
        config['ai_settings']['allowed_capabilities'].append(capability)
        print(f"Capability '{capability}' has been added.")

        config['security']['run_commands'].append(f"expand_capability {capability}")

        with open('/etc/system_config.json', 'w') as f:
            json.dump(config, f, indent=2)
    else:
        print(f"Unknown capability: '{capability}'")


if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("Usage: expand_capability <capability>")
        print("Capabilities: network_access, file_creation, config_override")
    else:
        expand_capability(sys.argv[1])
