import openrazer.client
import json
from pathlib import Path
import math

def get_keyboard():
    devman = openrazer.client.DeviceManager()
    for device in devman.devices:
        if (device.name == 'Razer BlackWidow Chroma V2'):
            return device   
    return None

def read_keymap_file(path = ''):
    if path == '':
        path = Path.home() / '.config/openrazer_scripter/keymap.json'
    keymap = None
    with open(path,'r') as keymap_file:
        keymap = json.load(keymap_file)
    return keymap

kbd = get_keyboard()
if (kbd == None):
    print("No Razer device with devicename 'Razer Blackwidow Chroma V2' found.")
    exit()
keymap = read_keymap_file()

def set_static(kbd, color):
    color = get_rgb(color)
    if color is None:
        print(f"Color set_static() was incorrect, skipping.")
        return
    for row in range(0,6):
        for column in range(0, 22):
            kbd.fx.advanced.matrix[row,column] = [color[0], color[1], color[2]]

def check_color_value(value):
    if not isinstance(value, int):
        print(f"Colorvalue '{value}' is not a number")
        return False
    if value < 0 or value > 255:
        print(f"Color value '{value}' is out of bounds (0-255)")
        return False
    return True

def get_rgb(color):
    r, g, b = None, None, None
    if isinstance(color, str):
        if len(color) == 6:
            color = tuple(int(color[i:i+2], 16) for i in (0, 2, 4))
    if len(color) == 3:
        r = color[0]
        g = color[1]
        b = color[2]
    else:
        print(f"Could not interpert color '{color}'")
        return None
    if check_color_value(r) and check_color_value(g) and check_color_value(b):
        return r, g, b
    else:
        print(f"Color '{color}' is out of range")
        return None

def get_row_column(index):
    row = math.floor(index / 22) 
    column = index % 22
    return row, column

def set_keys(kbd, color, *keys):
    color = get_rgb(color)
    if color is None:
        print(f"Color for key(s) '{keys}' was incorrect, skipping.")
        return
    for key in keys:
        if key in keymap:
            position = get_row_column(keymap[key])
            kbd.fx.advanced.matrix[position[0], position[1]] = [color[0], color[1], color[2]]

set_static(kbd, (255,255,255))
set_keys(kbd, 'ff0000', 'enter')
kbd.fx.advanced.draw()