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 = []
  inputIdValues = {}
  for row in reader:
    inputId = row["id"]
    if inputId not in inputIds:
      inputIds.append(inputId)

  # For each ID, create enum entry.
  count = 0
  outHeader += "typedef enum {\n"
  outHeader += f"  INPUT_ACTION_NULL = {count},\n\n"
  count += 1
  for inputId in inputIds:
    inputIdValues[inputId] = count
    outHeader += f"  {csvIdToEnumName(inputId)} = {count},\n"
    count += 1
  outHeader += f"\n  INPUT_ACTION_COUNT = {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"

  # Lua Script
  outHeader += f"static const char_t *INPUT_ACTION_SCRIPT = \n"
  for inputId in inputIds:
    # Reference the enum 
    outHeader += f"  \"{csvIdToEnumName(inputId)} = {inputIdValues[inputId]}\\n\"\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)