import argparse
import os
import csv
from tools.util.type import detectType, stringToCType, typeToCType

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 csvIdToEnumName(inputId):
  return "INPUT_ACTION_" + inputId.upper()

# Load up CSV file.
outHeader = "#pragma once\n"
outHeader += '#include "dusk.h"\n\n'

with open(args.csv, newline="", encoding="utf-8") as csvfile:
  reader = csv.DictReader(csvfile)

  # CSV must have id column.
  if "id" not in reader.fieldnames:
    raise Exception("CSV file must have 'id' column")
  
  # For each ID
  inputIds = []
  for row in reader:
    inputId = row["id"]
    if inputId not in inputIds:
      inputIds.append(inputId)

  # For each ID, create enum entry.
  outHeader += "typedef enum {\n"
  outHeader += "  INPUT_ACTION_NULL = 0,\n\n"
  for inputId in inputIds:
    outHeader += f"  {csvIdToEnumName(inputId)},\n"
  outHeader += f"\n  INPUT_ACTION_COUNT\n"
  outHeader += "} inputaction_t;\n\n"

  # Write IDs to char array.
  outHeader += f"static const char_t* INPUT_ACTION_IDS[] = {{\n"
  for inputId in inputIds:
    outHeader += f"  [{csvIdToEnumName(inputId)}] = \"{inputId}\",\n"
  outHeader += f"}};\n\n"

# Write to output file.
os.makedirs(os.path.dirname(args.output), exist_ok=True)
with open(args.output, "w", encoding="utf-8") as outFile:
  outFile.write(outHeader)