import sys
import random
import time
from pprint import pprint
import telepot
import urllib.request
import time
from PIL import Image
import os
import calendar
import logging
logging.basicConfig(format='%(asctime)s %(message)s')

taulu_im = Image.open("taulu.png")
golden_im = Image.open("golden.png")
img_size = (640, 640)
GOLDEN_CHANCE = 2 #%
START_TIME = calendar.timegm(time.gmtime())

taulu_im.thumbnail(img_size, Image.ANTIALIAS)
golden_im.thumbnail(img_size, Image.ANTIALIAS)

def run_taulu(chat_id, user_id):

	photos = bot.getUserProfilePhotos(user_id)

	if photos["total_count"] == 0:
		logging.error("No pictures found, sending message...")
		bot.sendMessage(chat_id, "Vitun taulu, hommaa kuva!")
		return

	logging.warning("Looking up picture...")
	pic = photos["photos"][0][-1]
	file_id = pic["file_id"]

	if not os.path.exists("photo/"):
		logging.warning("No 'photo' directory in working directory, creating one...")
		os.makedirs("photo/")

	# download to a file with user_id in the name for duplicate recognition
	filepath = "photo/" + str(abs(user_id)) + ".jpg"
	finalpath = "photo/" + str(abs(user_id)) + "final.png"

	# every 100th image has golden borders :)
	golden_number= random.randrange(1, 100)
	logging.warning("Random gold number was: {}".format(golden_number))
	is_golden = (golden_number < GOLDEN_CHANCE)

	if os.path.exists(finalpath) and not is_golden:
		logging.warning("User profile already found. Skipping download and edit...")
	else:
		logging.warning("Downloading picture...")
		bot.downloadFile(file_id, filepath)
		# time.sleep(2)

		logging.warning("Opening image for editing...")
		im = Image.open(filepath)
		im.thumbnail(img_size, Image.ANTIALIAS)

		if is_golden:
			logging.warning("Golden taulu!")
			im.paste(golden_im, (0, 0), golden_im)
			bot.sendMessage(chat_id, "Onneksi olkoon! Kultainen taulu ilmestyy vain kerran tuhannessa vuodessa!")
		else:
			im.paste(taulu_im, (0, 0), taulu_im)

		logging.warning("Saving image...")
		im.save(finalpath)

	f = open(finalpath, "rb")
	logging.warning("Sending photo {} to chat {}...".format(finalpath, chat_id))
	bot.sendPhoto(chat_id, f)

	logging.warning("Image sent succesfully.")
	return

# Getting the token from command-line is better than embedding it in code,
# because tokens are supposed to be kept secret.
TOKEN = sys.argv[1]

bot = telepot.Bot(TOKEN)
logging.warning("Connected with token {}".format(TOKEN))

last_user_by_chat_id = {}
done_list = []

def is_taulu_command(text):
	return "taulu" in text.strip().lower()

def handle(msg):

	chat_id = msg["chat"]["id"]
	user_id = msg["from"]["id"]
	user_name = msg["from"]["first_name"]
	msg_text = msg["text"]
	logging.warning("New message! chat_id: {}, user_name: {}, msg_text: {}".format(chat_id, user_name, msg_text))

	if is_taulu_command(msg_text):
		logging.warning("Taulu command found!")
		attempt, ATTEMPTS = 0, 3
		while attempt < ATTEMPTS:
			try:
				taulu_id = last_user_by_chat_id[chat_id]
				logging.warning("Running taulu for user id {}".format(taulu_id))
				run_taulu(chat_id, taulu_id)
				break
			except KeyError:
				logging.error("No chat id found in chat dict. Maybe no previous messages in the chat?")
				break
			except:
				logging.error("Error occured during process. Retrying... (attempt {})".format(attempt))
				attempt += 1

	last_user_by_chat_id[chat_id] = user_id

bot.notifyOnMessage(handle)
logging.warning("Listening...")

# Keep the program running.
while 1:
	time.sleep(5)