from __future__ import annotations

import logging

from selenium import webdriver
from selenium.common.exceptions import TimeoutException
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions
from selenium.webdriver.support.wait import WebDriverWait
from webdriver_manager.chrome import ChromeDriverManager


class Instagram:
    """Interface for interacting with Instagram."""

    def __init__(self):
        self.webdriver = None
        self.actiondriver = None

    def setup(self) -> None:
        """Setup connection to Selenium and Instagram."""
        driver = webdriver.Chrome(
            service=Service(
                ChromeDriverManager().install(),
            ),
        )
        self.webdriver = driver
        self.actiondriver = ActionChains(self.webdriver)

    def close(self) -> None:
        self.webdriver.close()

    def get_image_url_list(self, url: str) -> list[str]:
        """Get list of image urls from post."""
        logging.debug('getting %s', url)
        self.webdriver.get(url)
        logging.debug('finished getting url')
        # handle cookie popup
        try:
            logging.debug('waiting for cookie popup to appear')
            # wait forpopup to appear
            elem = WebDriverWait(
                driver=self.webdriver,
                timeout=60,
            ).until(
                expected_conditions.presence_of_element_located(
                    (By.XPATH, "//*[text()='Decline optional cookies']"),
                ),
            )
            logging.debug('located button')
        except TimeoutException:
            logging.error('Timeout before cookie popup appeared.')
        else:
            elem.click()
            logging.debug('clicked button')
        # find image elements with src url matching Instagram's CDN
        img_elem_list = self.webdriver.find_elements(
            by=By.XPATH,
            value=(
                "//*[@role='presentation']"
                "//img[contains(@src,'https://scontent.cdninstagram.com/v/')]"
            ),
        )
        assert len(img_elem_list) > 0
        # extract src url
        img_src_list = [
            img_elem.get_attribute(
                name='src',
            ) for img_elem in img_elem_list
        ]
        # remove src urls with size matching profile images
        img_src_list = [
            img_src for img_src in img_src_list if 's150x150' not in img_src
        ]
        logging.debug('located %s image urls', len(img_src_list))
        return img_src_list