import html
import json
import math
import random
import re
import sys
import time
from io import BytesIO
from urllib.request import urlopen

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

username = sys.argv[1]
password = sys.argv[2]
percent = 0

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 update_image():
    global img, origin

    wiki_data = s.get('https://www.reddit.com/r/Quebec/wiki/place', stream=True).content.decode()
    js = re.findall(r'''({"origine":.*?})''', html.unescape(wiki_data))[0]
    data = json.loads(js)
    data['url'] = 'https://' + data['url']

    img = Image.open(BytesIO(urlopen(data['url']).read()))
    origin = (data['origine'][0], data['origine'][1])

    print("URL: {}, origine: {}".format(data['url'], data['origine']))

update_image()

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


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)

    try:
        old_color = data["color"] if "color" in data else 0
    except:
        return

    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)
        update_image()
        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 True:
                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)