#! /usr/bin/env python3

import argparse
import json
import os
import re
import sys

# -----------------------------------------------------------------------------

def main():
	arg_parser = argparse.ArgumentParser(
			description = "Generate a picture of the firmware's "
			            + "keyboard layout" )

	arg_parser.add_argument(
			'--ui-info-file',
			required = True )

	args = arg_parser.parse_args(sys.argv[1:])

	args.template_file = './build-scripts/gen-layout--template.svg'

	# normalize paths
	args.ui_info_file = os.path.abspath(args.ui_info_file)

	# do stuff
	doc = ''  # to store the html document we're generating
	template = open(args.template_file).read()
	info = json.loads(open(args.ui_info_file).read())

	matrix_positions = info['mappings']['matrix-positions']
	matrix_layout = info['mappings']['matrix-layout']

	# only consider keycodes, for now
	for layout in matrix_layout:
		template_copy = template
		for (name, (code, press, release)) \
				in zip(matrix_positions, layout):
			template_copy = re.sub( name,
									keycode_to_string.get(code, '[n/a]'),
									template_copy )

		doc += template_copy

	print(doc)

# -----------------------------------------------------------------------------
# -----------------------------------------------------------------------------

keycode_to_string = {
		0x01: "ErrorRollOver",
		0x02: "POSTFail",
		0x03: "ErrorUndefined",
		0x04: "a A",
		0x05: "b B",
		0x06: "c C",
		0x07: "d D",
		0x08: "e E",
		0x09: "f F",
		0x0A: "g G",
		0x0B: "h H",
		0x0C: "i I",
		0x0D: "j J",
		0x0E: "k K",
		0x0F: "l L",
		0x10: "m M",
		0x11: "n N",
		0x12: "o O",
		0x13: "p P",
		0x14: "q Q",
		0x15: "r R",
		0x16: "s S",
		0x17: "t T",
		0x18: "u U",
		0x19: "v V",
		0x1A: "w W",
		0x1B: "x X",
		0x1C: "y Y",
		0x1D: "z Z",
		0x1E: "1 !",
		0x1F: "2 @",
		0x20: "3 #",
		0x21: "4 $",
		0x22: "5 %",
		0x23: "6 ^",
		0x24: "7 &",
		0x25: "8 *",
		0x26: "9 (",
		0x27: "0 )",
		0x28: "Return",
		0x29: "Esc",
		0x2A: "Backspace",
		0x2B: "Tab",
		0x2C: "Space",
		0x2D: "- _",
		0x2E: "= +",
		0x2F: "[ {",
		0x30: "] }",
		0x31: "\ |",
		0x32: "# ~",
		0x33: "; :",
		0x34: "\' \"",
		0x35: "` ~",
		0x36: ", <",
		0x37: ". >",
		0x38: "/ ?",
		0x39: "CapsLock",
		0x3A: "F1",
		0x3B: "F2",
		0x3C: "F3",
		0x3D: "F4",
		0x3E: "F5",
		0x3F: "F6",
		0x40: "F7",
		0x41: "F8",
		0x42: "F9",
		0x43: "F10",
		0x44: "F11",
		0x45: "F12",
		0x46: "PrintScreen",
		0x47: "ScrollLock",
		0x48: "Pause",
		0x49: "Insert",
		0x4A: "Home",
		0x4B: "PageUp",
		0x4C: "Delete",
		0x4D: "End",
		0x4E: "PageDown",
		0x4F: "\u2192",  # right arrow
		0x50: "\u2190",  # left arrow
		0x51: "\u2193",  # down arrow
		0x52: "\u2191",  # up arrow

		0x53: "NumLock Clear",
		0x54: "/",
		0x55: "*",
		0x56: "-",
		0x57: "+",
		0x58: "Enter(kp)",
		0x59: "1 End",
		0x5A: "2 \u2193",  # down arrow
		0x5B: "3 PageDown",
		0x5C: "4 \u2190",  # left arrow
		0x5D: "5",
		0x5E: "6 \u2192",  # right arrow
		0x5F: "7 Home",
		0x60: "8 \u2191",  # up arrow
		0x61: "9 PageUp",
		0x62: "0 Insert",
		0x63: ". Del",

		0x64: "\ |",
		0x65: "Application",
		0x66: "Power",

		0x67: "=",

		0x68: "F13",
		0x69: "F14",
		0x6A: "F15",
		0x6B: "F16",
		0x6C: "F17",
		0x6D: "F18",
		0x6E: "F19",
		0x6F: "F20",
		0x70: "F21",
		0x71: "F22",
		0x72: "F23",
		0x73: "F24",
		0x74: "Exec",
		0x75: "Help",
		0x76: "Menu",
		0x77: "Select",
		0x78: "Stop",
		0x79: "Again",
		0x7A: "Undo",
		0x7B: "Cut",
		0x7C: "Copy",
		0x7D: "Paste",
		0x7E: "Find",
		0x7F: "Mute",
		0x80: "VolumeUp",
		0x81: "VolumeDown",
		0x82: "LockingCapsLock",
		0x83: "LockingNumLock",
		0x84: "LockingScrollLock",

		0x85: ",",
		0x86: "=",

		0x87: "International1",
		0x88: "International2",
		0x89: "International3",
		0x8A: "International4",
		0x8B: "International5",
		0x8C: "International6",
		0x8D: "International7",
		0x8E: "International8",
		0x8F: "International9",
		0x90: "LANG1",
		0x91: "LANG2",
		0x92: "LANG3",
		0x93: "LANG4",
		0x94: "LANG5",
		0x95: "LANG6",
		0x96: "LANG7",
		0x97: "LANG8",
		0x98: "LANG9",
		0x99: "AlternateErase",
		0x9A: "SysReq_Attention",
		0x9B: "Cancel",
		0x9C: "Clear",
		0x9D: "Prior",
		0x9E: "Return",
		0x9F: "Separator",
		0xA0: "Out",
		0xA1: "Oper",
		0xA2: "Clear_Again",
		0xA3: "CrSel_Props",
		0xA4: "ExSel",

		0xB0: "00",
		0xB1: "000",

		0xB2: "Thousands_Sep",
		0xB3: "Decimal_Sep",
		0xB4: "Currency_Unit",
		0xB5: "Currency_Subunit",

		0xB6: "(",
		0xB7: ")",
		0xB8: "{",
		0xB9: "}",

		0xBA: "Tab",
		0xBB: "Backspace",
		0xBC: "A",
		0xBD: "B",
		0xBE: "C",
		0xBF: "D",
		0xC0: "E",
		0xC1: "F",
		0xC2: "XOR",
		0xC3: "^",
		0xC4: "%",
		0xC5: "<",
		0xC6: ">",
		0xC7: "&",
		0xC8: "&&",
		0xC9: "|",
		0xCA: "||",
		0xCB: ":",
		0xCC: "#",
		0xCD: "Space",
		0xCE: "@",
		0xCF: "!",
		0xD0: "Mem_Store",
		0xD1: "Mem_Recall",
		0xD2: "Mem_Clear",
		0xD3: "Mem_+",
		0xD4: "Mem_-",
		0xD5: "Mem_*",
		0xD6: "Mem_/",
		0xD7: "+-",
		0xD8: "Clear",
		0xD9: "ClearEntry",
		0xDA: "Binary",
		0xDB: "Octal",
		0xDC: "Decimal",
		0xDD: "Hexadecimal",

		0xE0: "LeftControl",
		0xE1: "LeftShift",
		0xE2: "LeftAlt",
		0xE3: "LeftGUI",
		0xE4: "RightControl",
		0xE5: "RightShift",
		0xE6: "RightAlt",
		0xE7: "RightGUI",
		}

# -----------------------------------------------------------------------------
# -----------------------------------------------------------------------------

if __name__ == '__main__':
	main()