import argparse
import csv
import os

parser = argparse.ArgumentParser(description="Input CSV to .h defines")
parser.add_argument("--csv", required=True, help="Path to Input CSV file")
parser.add_argument("--output", required=True, help="Path to output .h file")
args = parser.parse_args()

def id_enum(name):
  return "INPUT_ACTION_" + name.upper()

# Load CSV
input_ids = []
with open(args.csv, newline="", encoding="utf-8") as f:
  reader = csv.DictReader(f)
  if "id" not in reader.fieldnames:
    raise ValueError("CSV must have an 'id' column")
  for row in reader:
    input_id = row["id"]
    if input_id not in input_ids:
      input_ids.append(input_id)

# Assign enum values
id_values = {input_id: i + 1 for i, input_id in enumerate(input_ids)}

# Build output
out = [
  "#pragma once",
  '#include "dusk.h"',
  "",
  "typedef enum {",
  "  INPUT_ACTION_NULL = 0x0,",
  "",
]
for input_id in input_ids:
  out.append(f"  {id_enum(input_id)} = 0x{id_values[input_id]:x},")
out += [
  "",
  f"  INPUT_ACTION_COUNT = 0x{len(input_ids) + 1:x}",
  "} inputaction_t;",
  "",
  "static const char_t *INPUT_ACTION_IDS[] = {",
]
for input_id in input_ids:
  out.append(f"  [{id_enum(input_id)}] = \"{input_id}\",")
out += [
  "};",
  "",
  "static const char_t *INPUT_ACTION_SCRIPT =",
]
for input_id in input_ids:
  out.append(f"  \"var {id_enum(input_id)} = {id_values[input_id]};\\n\"")
out += [";", ""]

os.makedirs(os.path.dirname(args.output), exist_ok=True)
with open(args.output, "w", encoding="utf-8") as f:
  f.write("\n".join(out))