#!/usr/bin/env python3
"""
Generate a series of state files with progressively more items.

Usage: ./generate_state_files.py <count> <output_dir>

Creates: output_dir/state_1.json, state_2.json, ..., state_N.json
"""

import json
import os
import sys

def generate_state(n):
    return {
        "colors": [f"color_{i}" for i in range(1, n + 1)],
        "numbers": [f"number_{i}" for i in range(1, n + 1)],
        "animals": [f"animal_{i}" for i in range(1, n + 1)],
    }

def main():
    if len(sys.argv) != 3:
        print("Usage: generate_state_files.py <count> <output_dir>", file=sys.stderr)
        sys.exit(1)

    count = int(sys.argv[1])
    output_dir = sys.argv[2]

    os.makedirs(output_dir, exist_ok=True)

    for i in range(1, count + 1):
        state = generate_state(i)
        path = os.path.join(output_dir, f"state_{i}.json")
        with open(path, "w") as f:
            json.dump(state, f)
        print(f"Created {path}")

if __name__ == "__main__":
    main()