"""
Basically what I want this program to dynamically update the colors based on the desktop background
so it will grab colors from the desktop background by finding the most promident colors and then set
the system colors to these colors. I believe there are two major components involved in making this work

i. The part of the program that analyzes an image and determines the most prominent colors
ii. The part of the program that goes into all the configuration files and updates their color schemes
"""
import sys
import numpy as np
import PIL
import cv2 as cv
from sklearn.cluster import KMeans
from paths import Paths
def grabColors(img_path: str, num_colors: int) -> list():
    """
    Takes in an image, and Number of colors, then returns a list of those colors.
    The list of colors will contain the most prominent colors present in the image.
    img_path - the path where your image lives (IE, /home/chandler/Pictures/moss.png)
    num_colors - the number of colors you need back from the image
    """
    img = cv.imread(img_path)
    img = cv.cvtColor(img, cv.COLOR_BGR2RGB)
    
    # scale image down by factor of 10 to decrease computation time
    dim = (int(len(img[0])/10), int(len(img)/10))
    img = cv.resize(img, dim, interpolation= cv.INTER_AREA)
    clt = KMeans(n_clusters=num_colors, n_init='auto')
    clt.fit(img.reshape(-1, 3))
    return clt.cluster_centers_

def rgbToHex(input_values):
    """
    Takes in a list of RBG color values and returns a list of those same colors as hex values
    """
    hex_list=[]
    for color in input_values:
        red = int(color[0])
        green = int(color[1])
        blue = int(color[2])
        hex_list.append('#{:02x}{:02x}{:02x}'.format(red, green, blue))
    return hex_list

def compColors(color_list):
    """
    given a list of colors, generate complimentary colors to contrast the prominent colors.
    return a list of these colors.
    """
    compliments = []
    for color in color_list:
        curr_hex = color[1:] # slice off the # from the hex code
        rgb = (curr_hex[0:2], curr_hex[2:4], curr_hex[4:6])
        comp = ['%02X' % (255 - int(a, 16)) for a in rgb]
        compliments.append('#' + ''.join(comp))
    return compliments
def updatei3Theme(config_path: str, img_path: str, colors: list, compliments: list):
    # read in the config file
    data = ''
    with open(config_path, 'r') as file:
        data = file.readlines()
    for i, line in enumerate(data):
        # update colors
        if "set $bgcolor" in line:
            data[i] = 'set $bgcolor ' + colors[0] + '\n'
        if "set $in-bgcolor" in line:
            data[i] = 'set $in-bgcolor ' + colors[1] + '\n'
        if "set $text" in line:
            data[i] = 'set $text ' + compliments[0] + '\n'
        if "set $indicator" in line:
            data[i] = 'set $indicator ' + colors[2] + '\n'
        if "set $in-text" in line:
            data[i] = 'set $in-text ' + compliments[1] + '\n'
        #update background image
        if "exec_always feh" in line:
            data[i] = 'exec_always feh --bg-fil ' + img_path + '\n'
    with open(config_path, 'w') as file:
        file.writelines(data)
def updatePolybarTheme(config_path: str, colors: list, compliments: list):
    data = ''
    with open(config_path, 'r') as file:
        data = file.readlines()
    for i,line in enumerate(data):
        #update colors
        if "background =" in line and i == 19:
            data[i] = 'background = ' + colors[0] + '\n'
        if "background-alt =" in line and i == 20:
            data[i] = 'background-alt = ' + colors[1] + '\n'
        if "foreground =" in line and i == 21:
            data[i] = 'foreground = ' + compliments[1] + '\n'
        if "primary =" in line and i == 22:
            data[i] = 'primary = ' + compliments[2] + '\n'
        if "secondary =" in line and i == 23:
            data[i] = 'secondary = ' + compliments[3] + '\n'
        if "disabled =" in line and i == 25:
            data[i] = 'disabled = ' + colors[2] + '\n'
    with open(config_path, 'w') as file:
        file.writelines(data)

def main():
    img_path = sys.argv[1]
    popularColors = grabColors(img_path, 5)
    hex_colors = rgbToHex(popularColors)
    hex_compliments = compColors(hex_colors)
    updatei3Theme(Paths['i3'], img_path, hex_colors, hex_compliments)
    updatePolybarTheme(Paths['polybar'], hex_colors, hex_compliments)
    print("Theme changed successfully, please reload i3")


main()