#!/usr/bin/env python3
#
# Copyright (C) 2022 Intel Corporation.
#
# SPDX-License-Identifier: BSD-3-Clause
#

import sys
import acrn_config_utilities, lib.error
from acrn_config_utilities import get_node
from collections import defaultdict

# The COM1 was used for console vUART, so we alloc io_port frome COM2~COM4
val = lambda:list(range(0x9000, 0x9100, 8))
vm_port_list = defaultdict(val)

def create_s5_vuart_connection(allocation_etree, service_vm_name, service_vm_port, user_vm_name, user_vm_port):
    vuart_connections_node = get_node(f"/acrn-config/hv/vuart_connections", allocation_etree)
    if vuart_connections_node is None:
        vuart_connections_node = acrn_config_utilities.append_node("/acrn-config/hv/vuart_connections", None, allocation_etree)

    connection_name = service_vm_name + "_"  + user_vm_name

    vuart_connection_node = acrn_config_utilities.append_node(f"./vuart_connection", None, vuart_connections_node)
    acrn_config_utilities.append_node(f"./name", connection_name, vuart_connection_node)
    acrn_config_utilities.append_node(f"./type", "legacy", vuart_connection_node)

    service_vm_endpoint = acrn_config_utilities.append_node(f"./endpoint", None, vuart_connection_node)
    acrn_config_utilities.append_node(f"./vm_name", service_vm_name, service_vm_endpoint)
    acrn_config_utilities.append_node(f"./io_port", service_vm_port, service_vm_endpoint)

    user_vm_endpoint = acrn_config_utilities.append_node(f"./endpoint", None, vuart_connection_node)
    acrn_config_utilities.append_node(f"./vm_name", user_vm_name, user_vm_endpoint)
    acrn_config_utilities.append_node(f"./io_port", user_vm_port, user_vm_endpoint)

def get_console_vuart_port(scenario_etree, vm_name):
    port = get_node(f"//vm[name = '{vm_name}']/console_vuart/text()", scenario_etree)

    if port == "COM Port 1":
        port = "0x3F8U"
    elif port == "COM Port 2":
        port = "0x2F8U"
    elif port == "COM Port 3":
        port = "0x3E8U"
    elif port == "COM Port 4":
        port = "0x2E8U"
    else:
        port = "0x0U"

    return port

def alloc_free_port(scenario_etree, load_order, vm_name):
    port_list = scenario_etree.xpath(f"//endpoint[vm_name = '{vm_name}']/io_port/text()")
    if load_order == "SERVICE_VM":
        vm_id = int(get_node(f"//vm[load_order = 'SERVICE_VM']/@id", scenario_etree))
    else:
        vm_id = int(get_node(f"//vm[name = '{vm_name}']/@id", scenario_etree))
    console_port = get_console_vuart_port(scenario_etree, vm_name)
    if console_port is not None:
        port_list.append(console_port.replace("U", ""))
    
    tmp_list = []
    for port in port_list:
        tmp_list.append(int(port, 16))


    global vm_port_list
    vm_port_list[vm_id] = list(set(vm_port_list[vm_id]) - set(tmp_list))
    vm_port_list[vm_id].sort()
    port = hex(vm_port_list[vm_id][0])
    vm_port_list[vm_id].remove(vm_port_list[vm_id][0])
    return str(port).upper()

def alloc_vuart_connection_info(board_etree, scenario_etree, allocation_etree):
    user_vm_list = scenario_etree.xpath(f"//vm[load_order != 'SERVICE_VM']")
    service_vm_id = get_node(f"//vm[load_order = 'SERVICE_VM']/@id", scenario_etree)
    service_vm_name = get_node(f"//vm[load_order = 'SERVICE_VM']/name/text()", scenario_etree)

    if (service_vm_id is None) or (service_vm_name is None):
        return

    for index,vm_node in enumerate(user_vm_list):
        vm_id = get_node("./@id", vm_node)
        load_order = get_node("./load_order/text()", vm_node)
        user_vm_name = get_node(f"./name/text()", vm_node)
        service_vm_port = alloc_free_port(scenario_etree, "SERVICE_VM", user_vm_name)
        user_vm_port = alloc_free_port(scenario_etree, load_order, user_vm_name)

        create_s5_vuart_connection(allocation_etree, service_vm_name, service_vm_port, user_vm_name, user_vm_port)

def fn(board_etree, scenario_etree, allocation_etree):
    alloc_vuart_connection_info(board_etree, scenario_etree, allocation_etree)