import math
import sys
import time
import random

import requests
from PIL import Image
from requests.adapters import HTTPAdapter

img = Image.open(sys.argv[1]) # First arg is filename
origin = (int(sys.argv[2]), int(sys.argv[3])) # Top left pixel coordinate (x, y)
username = sys.argv[4] # Your reddit username
password = sys.argv[5] # Your reddit password
percent = 0


def find_palette(point):
    rgb_code_dictionary = {
        (255, 255, 255): 0,
        (228, 228, 228): 1,
        (136, 136, 136): 2,
        (34, 34, 34): 3,
        (255, 167, 209): 4,
        (229, 0, 0): 5,
        (229, 149, 0): 6,
        (160, 106, 66): 7,
        (229, 217, 0): 8,
        (148, 224, 68): 9,
        (2, 190, 1): 10,
        (0, 211, 211): 11,
        (0, 131, 199): 12,
        (0, 0, 234): 13,
        (207, 110, 228): 14,
        (130, 0, 128): 15
    }

    def distance(c1, c2):
        (r1, g1, b1) = c1
        (r2, g2, b2) = c2
        return math.sqrt((r1 - r2) ** 2 + (g1 - g2) ** 2 + (b1 - b2) ** 2)

    colors = list(rgb_code_dictionary.keys())
    closest_colors = sorted(colors, key=lambda color: distance(color, point))
    closest_color = closest_colors[0]
    code = rgb_code_dictionary[closest_color]
    return code


s = requests.Session()
s.mount('https://www.reddit.com', HTTPAdapter(max_retries=5))
s.headers["User-Agent"] = "PlacePlacer"
r = s.post("https://www.reddit.com/api/login/{}".format(username),
           data={"user": username, "passwd": password, "api_type": "json"})
s.headers['x-modhash'] = r.json()["json"]["data"]["modhash"]


def place_pixel(ax, ay, new_color):
    message = "Probing absolute pixel {},{}".format(ax, ay)

    while True:
        r = s.get("http://reddit.com/api/place/pixel.json?x={}&y={}".format(ax, ay), timeout=5)
        if r.status_code == 200:
            data = r.json()
            break
        else:
            print("ERROR: ", r, r.text)
        time.sleep(5)

    old_color = data["color"] if "color" in data else 0
    if old_color == new_color:
        print("{}: skipping, color #{} set by {}".format(message, new_color, data[
            "user_name"] if "user_name" in data else "<nobody>"))
        time.sleep(.25)
    else:
        print("{}: Placing color #{}".format(message, new_color, ax, ay))
        r = s.post("https://www.reddit.com/api/place/draw.json",
                   data={"x": str(ax), "y": str(ay), "color": str(new_color)})

        secs = float(r.json()["wait_seconds"])
        if "error" not in r.json():
            message = "Placed color, waiting {} seconds. {}% complete."
        else:
            message = "Cooldown already active - waiting {} seconds. {}% complete."
        waitTime = int(secs) + 2
        while(waitTime > 0):
            m = message.format(waitTime, percent)
            time.sleep(1)
            waitTime -= 1
            if waitTime > 0:
                print(m, end="              \r")
            else:
                print(m)

        if "error" in r.json():
            place_pixel(ax, ay, new_color)

# From: http://stackoverflow.com/questions/27337784/how-do-i-shuffle-a-multidimensional-list-in-python
def shuffle2d(arr2d, rand=random):
    """Shuffes entries of 2-d array arr2d, preserving shape."""
    reshape = []
    data = []
    iend = 0
    for row in arr2d:
        data.extend(row)
        istart, iend = iend, iend+len(row)
        reshape.append((istart, iend))
    rand.shuffle(data)
    return [data[istart:iend] for (istart,iend) in reshape]

while True:
    print("starting image placement for img height: {}, width: {}".format(img.height, img.width))
    arr2d = shuffle2d([[[i,j] for i in range(img.width)] for j in range(img.height)])
    total = img.width * img.height
    checked = 0
    for y in range(img.width ):
        for x in range(img.height ):
            xx = arr2d[x][y]
            pixel = img.getpixel((xx[0], xx[1]))

            if pixel[3] > 0:
                pal = find_palette((pixel[0], pixel[1], pixel[2]))

                ax = xx[0] + origin[0]
                ay = xx[1] + origin[1]

                place_pixel(ax, ay, pal)
                checked += 1
                percent = round((checked/total) * 100, 2)
    message = "All pixels placed, sleeping {}s..."
    waitTime = 60
    while(waitTime > 0):
        m = message.format(waitTime)
        time.sleep(1)
        waitTime -= 1
        if waitTime > 0:
            print(m, end="              \r")
        else:
            print(m)