import logging
import re
import random
import dbhelper
import constants
import helpers
import parser

from bs4 import BeautifulSoup
from selenium.webdriver.chrome.options import Options
from selenium import webdriver
from telegram import Update, ForceReply
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, CallbackContext

def get_chrome_options():
    chrome_options = Options()
    chrome_options.add_argument("--headless")
    chrome_options.add_argument("--no-sandbox")
    chrome_options.add_argument("--disable-dev-shm-usage")
    chrome_prefs = {}
    chrome_options.experimental_options["prefs"] = chrome_prefs
    chrome_prefs["profile.default_content_settings"] = {"images": 2}
    return chrome_options

def find_amazon_link(update: Update, context: CallbackContext) -> None:
    productCode = ""
    extraparams = ""
    try:
        msg = update.message.text
    except AttributeError:
        logging.info(f"Received message has no text")
        return
    start = msg.find("amzn.to")
    if start != -1:
        msg = helpers.unshort_url(msg[start:].split()[0])
    start = msg.find("amzn.eu")
    if start != -1:
        msg = helpers.unshort_url(msg[start:].split()[0])
    start = msg.find(constants.searchURL)
    if start != -1:
        msg = f"https://{msg[start:].split(' ')[0]}"
        m = re.search(r'(?:dp\/[\w]*)|(?:gp\/product\/[\w]*)',msg) # Gets product code in amazon
        m_e = re.search(r'(?:&m=[\w]*)',msg) # Gets vendor and keeps it
        if m != None:
            productCode = m.group(0)
        if m_e != None:
            extraparams = m_e.group(0)

        user, chat, message_id = update.message.from_user, update.message.chat, update.message.message_id
        logging.info(f"Link sent by {user} - {msg}")

        logging.info("Setting up new Chrome Browser")
        driver = webdriver.Chrome(options=get_chrome_options())
        driver.set_window_size(random.randint(1200, 1800), random.randint(600, 1000))
        logging.info("Loading Amazon webpage")
        driver.get(msg)
        logging.info("Scraping information and closing browser")
        soup = BeautifulSoup(driver.page_source, "lxml")
        driver.close()

        logging.info("Getting title...")
        title, price, image = parser.get_title(soup), parser.get_price(soup), parser.get_image(soup)
        if title == "":
            logging.info(f"Title not found, not a valid product")
            return
        logging.info(f"Title found: {title}")

        context.bot.deleteMessage(chat_id=chat['id'], message_id=message_id)
        referurl = helpers.new_refer_url(productCode, extraparams)
        product_id = dbhelper.check_product(referurl, price)
        if not product_id:
            product_id = dbhelper.add_product(user['username'], chat['title'], user['id'], chat['id'], msg, referurl, title, price, image)
            helpers.create_image(product_id, price)
        context.bot.send_photo(chat_id=update.message.chat_id, photo=open(f"/app/data/images/products/{product_id}_composed.png", 'rb'), caption=f"URL enviada por @{user['username']}: \n\n{title} \n\n{referurl}")

def main() -> None:
    dbhelper.setup_db()

    updater = Updater(constants.TELEGRAM_API_KEY)
    dispatcher = updater.dispatcher
    dispatcher.add_handler(MessageHandler(Filters.text & ~Filters.command, find_amazon_link))
    updater.start_polling()
    updater.idle()

if __name__ == '__main__':
    main()