"""
Generate a colorscheme using imagemagick.
"""
import os
import re
import shutil
import subprocess
import sys

from .settings import CACHE_DIR, COLOR_COUNT
from . import util


def imagemagick(color_count, img):
    """Call Imagemagick to generate a scheme."""
    if shutil.which("magick"):
        magick_command = ["magick", "convert"]

    elif shutil.which("convert"):
        magick_command = ["convert"]

    else:
        print("error: imagemagick not found, exiting...\n"
              "error: wal requires imagemagick to function.")
        sys.exit(1)

    colors = subprocess.run([*magick_command, img,
                             "-resize", "25%", "+dither",
                             "-colors", str(color_count),
                             "-unique-colors", "txt:-"],
                            stdout=subprocess.PIPE)

    return colors.stdout.splitlines()


def gen_colors(img, color_count):
    """Format the output from imagemagick into a list
       of hex colors."""
    raw_colors = imagemagick(color_count, img)

    index = 0
    while len(raw_colors) - 1 < color_count:
        index += 1
        raw_colors = imagemagick(color_count + index, img)

        print("colors: Imagemagick couldn't generate a", color_count,
              "color palette, trying a larger palette size",
              color_count + index)

        if index > 20:
            print("colors: Imagemagick couldn't generate a suitable scheme",
                  "for the image. Exiting...")
            sys.exit(1)

    # Remove the first element because it isn't a color code.
    del raw_colors[0]

    return [re.search("#.{6}", str(col)).group(0) for col in raw_colors]


def sort_colors(img, colors):
    """Sort the generated colors and store them in a dict that
       we will later save in json format."""
    raw_colors = colors[:1] + colors[8:] + colors[8:-1]

    # Darken the background color if it's too light.
    # The value can be a letter or an int so we treat the
    # entire test as strings.
    if raw_colors[0][1] != "0":
        raw_colors[0] = util.darken_color(raw_colors[0], 0.25)

    # Create a comment color from the background.
    raw_colors[8] = util.lighten_color(raw_colors[0], 0.40)

    colors = {}
    colors["wallpaper"] = img
    colors["special"] = {}
    colors["colors"] = {}

    colors["special"]["background"] = raw_colors[0]
    colors["special"]["foreground"] = raw_colors[15]
    colors["special"]["cursor"] = raw_colors[15]

    for index, color in enumerate(raw_colors):
        colors["colors"]["color%s" % index] = color

    return colors


def get(img, cache_dir=CACHE_DIR,
        color_count=COLOR_COUNT, notify=False):
    """Get the colorscheme."""
    # _home_dylan_img_jpg.json
    cache_file = img.replace("/", "_").replace("\\", "_").replace(".", "_")
    cache_file = os.path.join(cache_dir, "schemes", cache_file + ".json")

    if os.path.isfile(cache_file):
        colors = util.read_file_json(cache_file)
        print("colors: Found cached colorscheme.")

    else:
        util.msg("wal: Generating a colorscheme...", notify)

        colors = gen_colors(img, color_count)
        colors = sort_colors(img, colors)

        util.save_file_json(colors, cache_file)
        util.msg("wal: Generation complete.", notify)

    return colors


def file(input_file):
    """Import colorscheme from json file."""
    data = util.read_file_json(input_file)

    if "wallpaper" not in data:
        data["wallpaper"] = "None"

    return data

@@ -1908,260 +1908,262 @@
 lor 
-if it's too light.%0A    # The value can be a letter or an int so we treat the%0A    # entire test as strings.%0A    if raw_colors%5B0%5D%5B1%5D != %220%22:%0A        raw_colors%5B0%5D = util.darken_color(raw_colors%5B0%5D, 0.25)%0A%0A    # Create a comment color from the background.
+slightly.%0A    if raw_colors%5B0%5D%5B1%5D != %220%22:%0A        raw_colors%5B0%5D = util.darken_color(raw_colors%5B0%5D, 0.25)%0A%0A    # Manually adjust colors.%0A    raw_colors%5B7%5D = util.lighten_color(raw_colors%5B7%5D, 0.25)%0A    raw_colors%5B8%5D = util.darken_color(raw_colors%5B7%5D, 0.30)
 %0A   
@@ -2166,33 +2166,34 @@
 %0A    raw_colors%5B
-8
+15
 %5D = util.lighten
@@ -2214,16 +2214,17 @@
 ors%5B
-0
+15
 %5D, 0.
-40
+25
 )%0A%0A 

Improve contrast and brightness.