#!/usr/bin/env python
# coding: utf-8

# Copyright (c) 2020 Huawei Technologies Co., Ltd.
# oec-hardware is licensed under the Mulan PSL v2.
# You can use this software according to the terms and conditions of the Mulan PSL v2.
# You may obtain a copy of Mulan PSL v2 at:
#     http://license.coscl.org.cn/MulanPSL2
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR
# PURPOSE.
# See the Mulan PSL v2 for more details.
# Create: 2020-04-01

from .command import Command, CertCommandError


def filter_char(string):
    """
    fileter char
    :param string:
    :return:
    """
    ascii_blacklist = map(chr, range(9) + range(11, 13) + range(14, 32))
    filtered = u''
    start = 0
    for i in range(len(string)):
        c = string[i]
        if c in ascii_blacklist or (type(string) != unicode and ord(c) >= 128):
            if start < i:
                filtered += string[start:i]
            start = i + 1
    filtered += string[start:]
    return filtered


class CertDevice(object):
    """
    Certified device
    """
    def __init__(self):
        self.devices = None

    def get_devices(self):
        """
        get devices information
        :return:
        """
        self.devices = list()
        try:
            pipe = Command("udevadm info --export-db")
            pipe.start()
            properties = dict()
            while True:
                line = pipe.readline()
                if line:
                    if line == "\n":
                        if len(properties) > 0:
                            device = Device(properties)
                            if device.path != "":
                                self.devices.append(device)
                            properties = dict()
                    else:
                        prop = line.split(":", 1)
                        if len(prop) == 2:
                            tp = prop[0].strip(r'\ \'\n')
                            attribute = prop[1].strip(r'\ \'\n')
                            if tp == "E":
                                keyvalue = attribute.split("=", 1)
                                if len(keyvalue) == 2:
                                    properties[keyvalue[0]] = keyvalue[1]
                            elif tp == "P":
                                properties["INFO"] = attribute
                else:
                    break
        except CertCommandError as e:
            print("Warning: get devices fail")
            print(e)
        self.devices.sort(key=lambda k: k.path)
        return self.devices


class Device(object):
    """
    get device properties
    """
    def __init__(self, properties=None):
        self.path = ""
        if properties:
            self.properties = properties
            self.path = properties["DEVPATH"]
        else:
            self.properties = dict()

    def get_property(self, prop):
        """
        get properties
        :param prop:
        :return:
        """
        try:
            return self.properties[prop]
        except KeyError:
            return ""

    def get_name(self):
        """
        get property value
        :return:
        """
        if "INTERFACE" in self.properties.keys():
            return self.properties["INTERFACE"]
        elif "DEVNAME" in self.properties.keys():
            return self.properties["DEVNAME"].split("/")[-1]
        elif self.path:
            return self.path.split("/")[-1]
        else:
            return ""