#!/usr/bin/env python
import os
import sys
import json
import toml
import signal
from PIL import Image
from pathlib import Path
from random import randint
from subprocess import run

import colors

def dmenu(input_, prompt):
    result = os.popen(f'echo "{input_}" | dmenu -p "{prompt}"').read().strip("\n")

    return result

def gen_pylogo(top_color, bottom_color):
    home = str(Path.home())
    run(['convert', f'{home}/.config/qtile/topsnake.png', '+level-colors', f"gray,'{top_color}'", '-transparent', 'gray', '/tmp/topsnake.png'])

    run(['convert', f'{home}/.config/qtile/bottomsnake.png', '+level-colors', f"gray,'{bottom_color}'", '-transparent', 'gray', '/tmp/bottomsnake.png'])

    run(['convert', '-size', '600x600', '-composite', '/tmp/bottomsnake.png', '/tmp/topsnake.png', '/tmp/python.png'])

    #run(['convert', '-size', '610x610', '-composite', f'{home}/.config/qtile/border.png', '/tmp/python.png', '/tmp/python.png'])

    run(['rm', '-r', '/tmp/topsnake.png', '/tmp/bottomsnake.png'])

def preset(wall, mode):
    global home
    presets = os.popen(f"ls {home}/Pictures/Wallpapers/presets").read().strip('\n').split('\n')
    wall = wall[:-3]
    wall += 'json'

    if mode == 'find':
        return wall in presets

    elif mode == 'set':
        wall = presets.index(wall)
        return presets[wall]

def new(name, folder):
    os.system(f"ln -s {home}/Pictures/Wallpapers/wallpapers/{name} {home}/Pictures/Wallpapers/wallpaperdirs/{folder}/{name}")

def pywal(wall, loc):
    global home

    if wall == 'archstylized.png':
        nitro(wall, loc)

        os.system(f'wal --theme {home}/.config/qtile/colors/defaultcolors.json')

        return 0


    else:
        pass

    os.system(f'wal -i {loc}/{wall}')

def nitro(wall, loc):
    global home
    os.system(f'nitrogen --set-scaled {loc}/{wall}')

def random(walls):
    wallcount = len(walls) - 1
    ranint = randint(0, wallcount)
    ranwall = walls[ranint]
    
    return ranwall

if __name__ == '__main__':
    #Set dir variables
    home = str(Path.home())
    stdwalls = f'{home}/Pictures/Wallpapers/wallpapers'
    hallo = f'{home}/Pictures/Wallpapers/halloweenwallpapers'
    xmas = f'{home}/Pictures/Wallpapers/xmaswallpapers'

    if len(sys.argv) >= 2:
        if sys.argv[1] == 'none':
            wallpapers = os.popen(f'ls {stdwalls}').read().strip('\n').split('\n')
            wall = random(wallpapers)

            if preset(wall, 'find'):
                os.system(f'wal --theme {home}/Pictures/Wallpapers/presets/{preset(wall, "set")}')
                run([f'{home}/.local/bin/pywalfox', 'update'])
        
            else:

                pywal(wall, stdwalls)

            with open(f"{home}/.cache/wal/colors.json", 'r') as f:
                data = json.load(f)

                top_color = data["colors"].get('color8')

                bottom_color = data["colors"].get('color6')

            gen_pylogo(top_color, bottom_color)

        elif sys.argv[1] == 'halloween':
            curr_dir = hallo

            wallpapers = os.popen(f'ls {curr_dir}').read().strip('\n').split('\n')
    
            walls = random(wallpapers)

            nitro(walls, curr_dir)

            colors.setColor(home, f'{home}/.config/qtile/colors', 'halloweencolors.json', norestart = True)

            run([f'{home}/.local/bin/pywalfox', 'update'])

            sys.exit(0)

        elif sys.argv[1] == 'xmas':
            curr_dir = xmas

            wallpapers = os.popen(f'ls {curr_dir}').read().strip('\n').split('\n')
    
            walls = random(wallpapers)

            nitro(walls, curr_dir)

            colors.setColor(home, f'{home}/.config/qtile/colors', 'xmascolors.json', norestart = True)

            run([f'{home}/.local/bin/pywalfox', 'update'])

            sys.exit(0)

        elif sys.argv[1] == 'new':
            new(sys.argv[2], sys.argv[3])

        elif '-i' == sys.argv[1]:
            if sys.argv[2][0] == '/':
                indx = sys.argv[2].rindex('/')
                curr_dir  = sys.argv[2][:indx]
                wall = sys.argv[2][indx + 1:]

            else:
                curr_dir = os.getcwd()
                wall = sys.argv[2]

            pywal(wall, curr_dir)

            run([f'{home}/.local/bin/pywalfox', 'update'])

            with open(f"{home}/.cache/wal/colors.json", 'r') as f:
                data = json.load(f)

                top_color = data["colors"].get('color8')

                bottom_color = data["colors"].get('color6')

            gen_pylogo(top_color, bottom_color)

            try:
                #subprocess.check_call(['qtile', 'shell', '-c', 'restart()'])
                #run(['kill', '-SIGUSR1', os.popen('pidof qtile').read().strip('\n')])
                #run(['qtile', 'shell', '-c', 'restart()'])
                run([f'{home}/git/qtile/bin/qtile', 'shell', '-c', 'reload_config()'])

            except:
                os.system('killall dwm')

        elif 'a' in sys.argv[1]:
            curr_dir = sys.argv[1] + 'wallpapers'

            wallpapers = os.popen(f'ls {home}/Pictures/Wallpapers/{curr_dir}').read().strip('\n').split('\n')
            nitro(random(wallpapers), curr_dir)

        sys.exit(0)

    else:
        pass

    #Inital prompt
    walldirs = "All\n"
    walldirs += os.popen(f"ls {home}/Pictures/Wallpapers/wallpaperdirs/").read().strip("\n")
    #wallpaperchoice = os.popen(f'echo "{walldirs}" | dmenu -p "Wallpaper:"').read().strip("\n")
    wallpaperchoice = dmenu(walldirs, "Wallpaper:")

    #Closes on ESC
    if wallpaperchoice == '':
        sys.exit(0)
    
    #Sets to other dirs
    elif wallpaperchoice == "halloween":
        #wallpaperchoice = os.popen(f'ls {hallo}/ | dmenu -p "Wallpaper:"').read().strip('\n')
        wallpaperchoice = dmenu(os.popen(f"ls {hallo}").read().strip('\n'), "Wallpaper:")        

        if wallpaperchoice == '':
            sys.exit(0)
    
        curr_dir = hallo

    elif wallpaperchoice == "xmas":
        #wallpaperchoice = os.popen(f'ls {xmas}/ | dmenu -p "Wallpaper:"').read().strip('\n')
        wallpaperchoice = dmenu(os.popen(f"ls {xmas}").read().strip('\n'), "Wallpaper:")        
        
        if wallpaperchoice == '':
            sys.exit(0)
    
        curr_dir = xmas

    elif wallpaperchoice == "All":
        #wallpaperchoice = os.popen(f'ls {stdwalls}/ | dmenu -p "Wallpaper:"').read().strip('\n')
        wallpaperchoice = dmenu(os.popen(f"ls {stdwalls}").read().strip('\n'), "Wallpaper:")        
        
        if wallpaperchoice == '':
            sys.exit(0)
    
        curr_dir = stdwalls

    else:
        #wallpaperchoice = os.popen(f'ls {home}/Pictures/Wallpapers/wallpaperdirs/{wallpaperchoice} | dmenu -p "Wallpaper:"').read().strip('\n')
        wallpaperchoice = dmenu(os.popen(f"ls {home}/Pictures/Wallpapers/wallpaperdirs/{wallpaperchoice}").read().strip('\n'), "Wallpaper:")        

        if wallpaperchoice == '':
            sys.exit(0)
    
        curr_dir = stdwalls

    #Split to list 
    wallargs = wallpaperchoice.split(' ')
    
    #Handle random selection
    if 'random' in wallargs:
        wallpapers = os.popen(f'ls {home}/Pictures/Wallpapers/{curr_dir}').read().strip('\n').split('\n')
    
        wallargs[0] = random(wallpapers)

    #Set wallpaper with nitrogen
    if ':nitro' in wallargs:
        nitro(wallargs[0], curr_dir)
        sys.exit(0)

    elif preset(wallargs[0], 'find') and ':auto' not in wallargs:
        os.system(f'wal --theme {home}/Pictures/Wallpapers/presets/{preset(wallargs[0], "set")}')

        run([f'{home}/.local/bin/pywalfox', 'update'])

    #Set wallpaper with pywal
    else:
        pywal(wallargs[0], curr_dir)
        run([f'{home}/.local/bin/pywalfox', 'update'])

    with open(f"{home}/.cache/wal/colors.json", 'r') as f:
        data = json.load(f)

        top_color = data["colors"].get('color8')

        bottom_color = data["colors"].get('color6')

    gen_pylogo(top_color, bottom_color)

    try:
        print('restarting qtile')
        #subprocess.check_call(['qtile', 'shell', '-c', 'restart()'])
        #run(['qtile', 'shell', '-c', 'restart()'])
        #run(['kill', '-SIGUSR1', os.popen('pidof qtile').read().strip('\n')])
        run([f'{home}/git/qtile/bin/qtile', 'shell', '-c', 'reload_config()'])
        #os.system('qtile shell -c "restart()"')

    except:
        print('killing dwm')
        os.system('killall dwm')
        #pass