"""
Module for RaSCSI device management utility methods
"""

from flask_babel import _


def get_valid_scsi_ids(devices, reserved_ids):
    """
    Takes a list of (dict)s devices, and list of (int)s reserved_ids.
    Returns:
    - (list) of (int)s valid_ids, which are the SCSI ids that are not reserved
    - (int) recommended_id, which is the id that the Web UI should default to recommend
    """
    occupied_ids = []
    for device in devices:
        occupied_ids.append(device["id"])

    unoccupied_ids = [i for i in list(range(8)) if i not in reserved_ids + occupied_ids]
    unoccupied_ids.sort()
    valid_ids = [i for i in list(range(8)) if i not in reserved_ids]
    valid_ids.sort(reverse=True)

    if unoccupied_ids:
        recommended_id = unoccupied_ids[-1]
    else:
        recommended_id = occupied_ids.pop(0)

    return valid_ids, recommended_id


def sort_and_format_devices(devices):
    """
    Takes a (list) of (dict)s devices and returns a (list) of (dict)s.
    Sorts by SCSI ID acending (0 to 7).
    For SCSI IDs where no device is attached, inject a (dict) with placeholder text.
    """
    occupied_ids = []
    for device in devices:
        occupied_ids.append(device["id"])

    formatted_devices = devices

    # Add padding devices and sort the list
    for i in range(8):
        if i not in occupied_ids:
            formatted_devices.append({"id": i, "device_type": "-", \
                    "status": "-", "file": "-", "product": "-"})
    # Sort list of devices by id
    formatted_devices.sort(key=lambda dic: str(dic["id"]))

    return formatted_devices


def extend_device_names(device_types):
    """
    Takes a (list) of (str) device_types with the four letter device acronyms
    Returns a (dict) of device_type:device_name mappings of localized device names
    """
    mapped_device_types = {}
    for device_type in device_types:
        if device_type is "SAHD":
            device_name = _("SASI Hard Drive")
        elif device_type is "SCHD":
            device_name = _("SCSI Hard Drive")
        elif device_type is "SCRM":
            device_name = _("Removable Drive")
        elif device_type is "SCMO":
            device_name = _("Magneto-Optical Drive")
        elif device_type is "SCCD":
            device_name = _("CD-ROM Drive")
        elif device_type is "SCBR":
            device_name = _("X68000 Host Bridge")
        elif device_type is "SCDP":
            device_name = _("DaynaPORT SCSI/Link")
        else:
            device_name = _("Unknown Device")
        mapped_device_types[device_type] = device_name

    return mapped_device_types