#!/usr/bin/env python
#
# led_control.py
#
# Platform-specific LED control functionality for SONiC
#

try:
    from sonic_led.led_control_base import LedControlBase
    import swsssdk
    import threading
    import os
    import logging
    import struct
    import time
    import syslog
    from socket import *
    from select import *
    from minipack.pimutil import PimUtil
except ImportError, e:
    raise ImportError(str(e) + " - required module not found")


class LedControl(LedControlBase):
    """Platform specific LED control class"""
    SONIC_PORT_NAME_PREFIX = "Ethernet"
    
    
    def __init__(self):
        pim=PimUtil()
        pim.init_pim_fpga()
    
    def _port_name_to_index(self, port_name):
        # Strip "Ethernet" off port name
        if not port_name.startswith(self.SONIC_PORT_NAME_PREFIX):
            return -1

        port_idx = int(port_name[len(self.SONIC_PORT_NAME_PREFIX):])
        return port_idx
    
    def _port_state_to_mode(self, port_idx, state):
        if state == "up":
            return 1, 4 #port linkup, led is green
        else:
            return 0, 0 #port linkdown, led is off

    def port_link_state_change(self, portname, state):
        pim=PimUtil()
        port_idx = self._port_name_to_index(portname)
        new_control, led_mode = self._port_state_to_mode(port_idx, state)
        color, control=pim.get_port_led(port_idx)
        
        if color==led_mode:
            if control==new_control:
                return
        
        pim.set_port_led(port_idx, led_mode, new_control)#port linkup, led is green
                                                         #port linkdown, led is off