from pygameControls.controlsbase import ControlsBase
from pydualsense import *

BATTERY_STATE = {
    "0": "Discharging",
    "1": "Charging",
    "2": "Full",
    "11": "Not charging",
    "15": "Error",
    "10": "Temp or voltage out of range"
    }

class DualSenseController(ControlsBase):
    def __init__(self, joy, connection_type):
        self.device = pydualsense()
        self.device.init()
        self.name = self.device.device.get_product_string()
        self.guid = self.device.get_guid()
        self.connection_type = connection_type
        self.powerlevel = self.device.battery.Level
        self.batterystate = BATTERY_STATE[str(self.device.battery.State)]
        self.set_player_id(PlayerID.PLAYER_1)
        self.numaxis: int = joy.get_numaxes()
        self.axis: list = [joy.get_axis(a) for a in range(self.numaxis)]
        self.numhats: int = joy.get_numhats()
        self.hats: list = [joy.get_hat(h) for h in range(self.numhats)]
        self.numbuttons: int = joy.get_numbuttons()
        self.buttons: list = [joy.get_button(b) for b in range(self.numbuttons)]
        self.mapping = {
            "l1 button": 4,
            "r1 button": 5,
            "cross button": 0,
            "triangle button": 2,
            "circle button": 1,
            "square button": 3,
            "left stick button": 11,
            "right stick button": 12,
            "connect button": 8,
            "list button": 9,
            "logo button": 10
            }
        print(f"{self.name} connected")
        print(f"Power level: {self.powerlevel}")
        print(f"Battery state: {self.batterystate}")
    
    def close(self):
        self.device.close()
        
    def handle_input(self, event):
        pass
    
    def set_led(self, red: int, green: int, blue: int):
        self.device.light.setColorI(red, green, blue)

    def set_player_id(self, playerid: PlayerID):
        self.device.light.setPlayerID(playerid)
        
    def left(self):
        pass
    
    def right(self):
        pass
    
    def up(self):
        pass
    
    def down(self):
        pass
    
    def pause(self):
        pass
    
    def rumble(self, left, right):
        if not left in range(256) or not right in range(256):
            raise ValueError("left and right values must be in the range 0 - 255")
        self.device.setLeftMotor(left)
        self.device.setRightMotor(right)
    
    def stop_rumble(self):
        self.device.setLeftMotor(0)
        self.device.setRightMotor(0)
        
    @property
    def name(self) -> str:
        return self._name
    
    @name.setter
    def name(self, name: str) -> None:
        self._name = name
    
    @property
    def powerlevel(self) -> str:
        return self._powerlevel
    
    @powerlevel.setter
    def powerlevel(self, lvl: str) -> None:
        self._powerlevel = lvl
    
    @property
    def batterystate(self) -> int:
        return self._batterystate
    
    @batterystate.setter
    def batterystate(self, state) -> None:
        self._batterystate = state