import json
import jinja2

import legacy_names
import json_index

MAIN_CONTROL_LUT = {
    # 2 Gimbal radios
    "LH": { "str": "Rud", "local": "STR_STICK_NAMES[0]" },
    "LV": { "str": "Ele", "local": "STR_STICK_NAMES[1]" },
    "RV": { "str": "Thr", "local": "STR_STICK_NAMES[2]" },
    "RH": { "str": "Ail", "local": "STR_STICK_NAMES[3]" },
    # Surface radios
    "ST": { "str": "ST", "local": "STR_SURFACE_NAMES[0]" },
    "TH": { "str": "TH", "local": "STR_SURFACE_NAMES[1]" },
}

# MUST be the same order as 'EnumKeys'
KEYS_LUT = [
  "KEY_MENU",
  "KEY_EXIT",
  "KEY_ENTER",

  "KEY_PAGE",
  "KEY_PAGEUP",
  "KEY_PAGEDN",

  "KEY_UP",
  "KEY_DOWN",

  "KEY_LEFT",
  "KEY_RIGHT",

  "KEY_PLUS",
  "KEY_MINUS",

  "KEY_MODEL",
  "KEY_TELE",
  "KEY_SYS",

  "KEY_SHIFT",
  "KEY_BIND",
]

def generate_from_template(json_filename, template_filename, target):

    with open(json_filename) as json_file:
        with open(template_filename) as template_file:

            root_obj = json.load(json_file)

            adc_inputs = root_obj.get('adc_inputs')
            adc_index = json_index.build_adc_index(adc_inputs)
            adc_gpios = json_index.build_adc_gpio_port_index(adc_inputs)

            switches = root_obj.get('switches')
            switch_gpios = json_index.build_switch_gpio_port_index(switches)

            keys = root_obj.get('keys')
            key_gpios = json_index.build_key_gpio_port_index(keys)

            trims = root_obj.get('trims')
            trim_gpios = json_index.build_trim_gpio_port_index(trims)
            
            legacy_inputs = legacy_names.inputs_by_target(target)
            
            env = jinja2.Environment(
                lstrip_blocks=True,
                trim_blocks=True
            )

            template_str = template_file.read()
            template = env.from_string(template_str)

            print(template.render(root_obj,
                                  adc_index=adc_index,
                                  adc_gpios=adc_gpios,
                                  switch_gpios=switch_gpios,
                                  key_gpios=key_gpios,
                                  trim_gpios=trim_gpios,
                                  legacy_inputs=legacy_inputs,
                                  main_labels=MAIN_CONTROL_LUT,
                                  key_index=KEYS_LUT))