import json
import os

def split_json_file(file_path, output_dir, batch_size):
    # Ensure output directory exists
    os.makedirs(output_dir, exist_ok=True)

    # Load the main JSON file
    with open(file_path, 'r', encoding='utf-8') as f:
        data = json.load(f)
    
    # Check if it's the expected structure
    if 'rules' not in data:
        print("Invalid JSON format: 'rules' key not found.")
        return

    # Get the list of rules
    rules = data['rules']
    total_rules = len(rules)
    
    # Split rules into batches
    for i in range(0, total_rules, batch_size):
        # Slice the list of rules for the current batch
        batch_rules = rules[i:i + batch_size]
        batch_data = {"rules": batch_rules}

        # Define the output file path
        batch_file_name = f"batch_{i // batch_size + 1}.json"
        batch_file_path = os.path.join(output_dir, batch_file_name)

        # Write the batch to a new JSON file
        with open(batch_file_path, 'w', encoding='utf-8') as f:
            json.dump(batch_data, f, indent=4)

        print(f"Created batch file: {batch_file_path} with {len(batch_rules)} rules")

# Usage example
file_path = 'automation-rules.json'  # Path to the large JSON file
output_dir = 'output_batches'  # Directory to save the smaller files
batch_size = 200  # Number of rules per batch

split_json_file(file_path, output_dir, batch_size)