"""
SANT_WCH

Micropython library for the SANT_WCH(---)
=======================================================
@dahanzimin From the Mixly Team
"""
_BOT035_ADDRESS     = const(0x13)
_BOT5_TOUCH         = const(0x01)
_BOT035_ADC         = const(0x05)
_BOT035_PWM         = const(0x07)

_BOT035_KB          = const(0x10)
_BOT035_MS          = const(0x14)
_BOT035_STR         = const(0x18)

class BOT035:
    def __init__(self, i2c_bus):
        self._i2c = i2c_bus

    def _wreg(self, reg, val):
        '''Write memory address'''
        self._i2c.writeto_mem(_BOT035_ADDRESS, reg, val.to_bytes(1, 'little'))

    def _rreg(self, reg, nbytes=1):
        '''Read memory address'''
        self._i2c.writeto(_BOT035_ADDRESS, reg.to_bytes(1, 'little'))
        return  self._i2c.readfrom(_BOT035_ADDRESS, nbytes)[0]

    def key_adc(self):
        return (self._rreg(_BOT035_ADC) | self._rreg(_BOT035_ADC + 1) << 8)

    def touch(self, index, value=None):
        index = max(min(index, 1), 0)
        touch = 4095 - (self._rreg(_BOT5_TOUCH + index * 2) | self._rreg(_BOT5_TOUCH + index * 2 + 1) << 8)
        return touch > value if value else  touch

    def touched(self, index, value=600):
        return self.touch(index, value)

    def touch_slide(self):
        values = []
        for i in range(30):
            values.append((self.touch(1) - self._touchs[1]) - (self.touch(0) - self._touchs[0]))
        return round(sorted(values)[15] / 10) 

    def usben(self, index=1, duty=None, freq=None):
        index = max(min(index, 3), 1) - 1
        if duty is not None:
            duty = max(min(duty, 100), 0)
            self._wreg(_BOT035_PWM + index + 2, int(duty))
        if freq is not None:
            freq = max(min(freq, 65535), 10)
            self._wreg(_BOT035_PWM, freq & 0xFF)
            self._wreg(_BOT035_PWM + 1, freq >> 8)
        if freq is None and duty is None:
            return self._rreg(_BOT035_PWM + index + 2), self._rreg(_BOT035_PWM) | self._rreg(_BOT035_PWM + 1) << 8

    def hid_keyboard(self, special=0, general=0, release=True):
        self._buf = bytearray(4)
        self._buf[0] = special
        if type(general) in (tuple, list):
            for i in range(len(general)):
                if i > 2: break
                self._buf[i + 1] = general[i]
        else:
            self._buf[1] = general
        self._i2c.writeto_mem(_BOT035_ADDRESS, _BOT035_KB, self._buf)
        if release:
            time.sleep_ms(10)
            self._i2c.writeto_mem(_BOT035_ADDRESS, _BOT035_KB, bytes(4))

    def hid_keyboard_str(self, string, delay=0):
        for char in str(string):
            self._wreg(_BOT035_STR, ord(char) & 0xFF)
            time.sleep_ms(20 + delay)

    def hid_mouse(self, keys=0, move=(0, 0), wheel=0, release=True):
        self._i2c.writeto_mem(_BOT035_ADDRESS, _BOT035_MS, bytes([keys & 0x0F, move[0] & 0xFF, move[1] & 0xFF, wheel & 0xFF]))
        if release:
            time.sleep_ms(10)
            self._i2c.writeto_mem(_BOT035_ADDRESS, _BOT035_MS, bytes(4))