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"]

wiki_info = None

def update_image():
    global img, origin, wiki_info

    text = s.get('https://www.reddit.com/r/Quebec/wiki/place')
    while (text.status_code != 200):
            s.get('https://www.reddit.com/r/Quebec/wiki/place')

    wiki_data = text.content.decode()
    js = re.findall(r'''({"origine":.*?})''', html.unescape(wiki_data))[0]
    data = json.loads(js)
    if data == wiki_info:
        return False
    else:
        wiki_info = data
    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']))

    return True

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)
        if update_image():
            return False
        if "error" in r.json():
            place_pixel(ax, ay, new_color)
    return True

# 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]

class RestartTheLoop(Exception):
    pass

while True:
    try:
        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 x in range(img.width ):
            for y in range(img.height ):
                xx = arr2d[x][y]
                pixel = img.getpixel((xx[0], xx[1]))

                if pixel[3] > 0:  # Check if the pixel should be transparent
                    pal = find_palette((pixel[0], pixel[1], pixel[2]))

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

                    if not place_pixel(ax, ay, pal):
                        raise RestartTheLoop()
                    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)
    except RestartTheLoop:
        pass