from tqdm import tqdm
import time


def like(
    self,
    media_id,
    check_media=True,
    container_module="feed_short_url",
    feed_position=0,
    username=None,
    user_id=None,
    hashtag_name=None,
    hashtag_id=None,
    entity_page_name=None,
    entity_page_id=None,
):

    if not self.reached_limit("likes"):
        if self.blocked_actions["likes"]:
            self.logger.warning("YOUR `LIKE` ACTION IS BLOCKED")
            if self.blocked_actions_protection:
                self.logger.warning(
                    "blocked_actions_protection ACTIVE. " "Skipping `like` action."
                )
                return False
        self.delay("like")
        if check_media and not self.check_media(media_id):
            return False
        #
        # TODO: commented out simulation "open link in app"
        #
        # if container_module == "feed_short_url":
        #     if "_" in str(media_id):
        #         media_pk = int(media_id.split("_")[0])
        #     else:
        #         media_pk = int(media_id)
        #     link = self.get_link_from_media_id(media_pk)
        #     self.logger.debug("Opening link {}".format(link))
        #     self.api.open_instagram_link(link)
        #     self.logger.debug("Getting media info...")
        #     self.api.media_info(media_id)
        #
        _r = self.api.like(
            media_id,
            container_module=container_module,
            feed_position=feed_position,
            username=username,
            user_id=user_id,
            hashtag_name=hashtag_name,
            hashtag_id=hashtag_id,
            entity_page_name=entity_page_name,
            entity_page_id=entity_page_id,
        )
        if _r == "feedback_required":
            self.logger.error("`Like` action has been BLOCKED...!!!")
            # if no sleep enabled, default flow,
            # with message of activating blocked actions protection
            if not self.blocked_actions_sleep:
                if self.blocked_actions_protection:
                    self.logger.warning(
                        "Activating blocked actions \
                         protection for `Like` action."
                    )
                    self.blocked_actions["likes"] = True
            else:
                # if the action is sleeping and another block,
                # then enable blocked actions protection
                if self.sleeping_actions["likes"] and self.blocked_actions_protection:
                    self.logger.warning(
                        "This is the second blocked \
                        `Like` action."
                    )
                    self.logger.warning(
                        "Activating blocked actions \
                        protection for `Like` action."
                    )
                    self.sleeping_actions["likes"] = False
                    self.blocked_actions["likes"] = True
                # otherwise (first block or no protection)
                # sleep for specified time
                else:
                    self.logger.info(
                        "`Like` action is going to sleep for \
                        %s seconds."
                        % self.blocked_actions_sleep_delay
                    )
                    self.sleeping_actions["likes"] = True
                    time.sleep(self.blocked_actions_sleep_delay)
            return False
        if _r:
            self.logger.info("Liked media %s." % media_id)
            self.total["likes"] += 1
            # if action just slept and now successful, then no longer sleeping
            if self.blocked_actions_sleep and self.sleeping_actions["likes"]:
                self.logger.info("`Like` action is no longer sleeping.")
                self.sleeping_actions["likes"] = False
            return True
    else:
        self.logger.info("Out of likes for today.")
    return False


def like_comment(self, comment_id):
    if not self.reached_limit("likes"):
        if self.blocked_actions["likes"]:
            self.logger.warning("YOUR `LIKE` ACTION IS BLOCKED")
            if self.blocked_actions_protection:
                from datetime import timedelta

                next_reset = (self.start_time.date() + timedelta(days=1)).strftime(
                    "%Y-%m-%d %H:%M:%S"
                )
                self.logger.warning(
                    (
                        "blocked_actions_protection ACTIVE. Skipping `like` "
                        "action till, at least, {}."
                    ).format(next_reset)
                )
                return False
        self.delay("like")
        _r = self.api.like_comment(comment_id)
        if _r == "feedback_required":
            self.logger.error("`Like` action has been BLOCKED...!!!")
            self.blocked_actions["likes"] = True
            return False
        if _r:
            self.logger.info("Liked comment {}.".format(comment_id))
            self.total["likes"] += 1
            return True
    else:
        self.logger.info("Out of likes for today.")
    return False


def like_media_comments(self, media_id):
    broken_items = []
    media_comments = self.get_media_comments(media_id)
    self.logger.info("Found {} comments".format(len(media_comments)))
    comment_ids = [
        item["pk"]
        for item in media_comments
        if not item.get("has_liked_comment") or not item["has_liked_comment"]
    ]

    if not comment_ids:
        self.logger.info(
            "None comments received: comments not "
            "found or comments have been filtered."
        )
        return broken_items

    self.logger.info("Going to like %d comments." % (len(comment_ids)))

    for comment in tqdm(comment_ids):
        if not self.like_comment(comment):
            self.error_delay()
            broken_items = comment_ids[comment_ids.index(comment) :]
    self.logger.info(
        "DONE: Liked {count} comments.".format(
            count=len(comment_ids) - len(broken_items)
        )
    )
    return broken_items


def like_medias(
    self,
    medias,
    check_media=True,
    container_module="feed_timeline",
    username=None,
    user_id=None,
    hashtag_name=None,
    hashtag_id=None,
    entity_page_name=None,
    entity_page_id=None,
):

    broken_items = []
    if not medias:
        self.logger.info("Nothing to like.")
        return broken_items
    self.logger.info("Going to like %d medias." % (len(medias)))
    feed_position = 0
    for media in tqdm(medias):
        if not self.like(
            media,
            check_media=check_media,
            container_module=container_module,
            feed_position=feed_position,
            username=username,
            user_id=user_id,
            hashtag_name=hashtag_name,
            hashtag_id=hashtag_id,
            entity_page_name=entity_page_name,
            entity_page_id=entity_page_id,
        ):
            self.error_delay()
            broken_items.append(media)
        feed_position += 1
    self.logger.info("DONE: Total liked %d medias." % self.total["likes"])
    return broken_items


def like_timeline(self, amount=None):
    self.logger.info("Liking timeline feed:")
    medias = self.get_timeline_medias()[:amount]
    return self.like_medias(medias, check_media=False)


def like_user(self, user_id, amount=None, filtration=True):
    """ Likes last user_id's medias """
    if filtration:
        if not self.check_user(user_id):
            return False
    self.logger.info("Liking user_%s's feed:" % user_id)
    user_id = self.convert_to_user_id(user_id)
    medias = self.get_user_medias(user_id, filtration=filtration)
    if not medias:
        self.logger.info(
            "None medias received: account is " "closed or medias have been filtered."
        )
        return False
    return self.like_medias(medias[:amount], filtration)


def like_users(self, user_ids, nlikes=None, filtration=True):
    for user_id in user_ids:
        if self.reached_limit("likes"):
            self.logger.info("Out of likes for today.")
            return
        self.like_user(user_id, amount=nlikes, filtration=filtration)


def like_hashtag(self, hashtag, amount=None):
    """ Likes last medias from hashtag """
    self.logger.info("Going to like media with hashtag #%s." % hashtag)
    medias = self.get_total_hashtag_medias(hashtag, amount)
    if self.api.search_tags(hashtag):
        for tag in self.api.last_json["results"]:
            if tag["name"] == hashtag:
                hashtag_id = tag["id"]
                break
        else:
            self.logger.error(
                "Hashtag ID of {} not found within api response".format(hashtag)
            )
            self.logger.debug(
                "Last JSON results: {}".format(self.api.last_json["results"])
            )
            return False
    else:
        self.logger.error("NO INFO FOR HASHTAG: {}".format(hashtag))
        return False
    return self.like_medias(
        medias,
        container_module="feed_contextual_hashtag",
        hashtag_name=hashtag,
        hashtag_id=hashtag_id,
    )


def like_geotag(self, geotag, amount=None):
    # TODO: like medias by geotag
    pass


def like_followers(self, user_id, nlikes=None, nfollows=None):
    self.logger.info("Like followers of: %s." % user_id)
    if self.reached_limit("likes"):
        self.logger.info("Out of likes for today.")
        return
    if not user_id:
        self.logger.info("User not found.")
        return
    follower_ids = self.get_user_followers(user_id, nfollows)
    if not follower_ids:
        self.logger.info("%s not found / closed / has no followers." % user_id)
    else:
        self.like_users(follower_ids[:nfollows], nlikes)


def like_following(self, user_id, nlikes=None, nfollows=None):
    self.logger.info("Like following of: %s." % user_id)
    if self.reached_limit("likes"):
        self.logger.info("Out of likes for today.")
        return
    if not user_id:
        self.logger.info("User not found.")
        return
    following_ids = self.get_user_following(user_id, nfollows)
    if not following_ids:
        self.logger.info("%s not found / closed / has no following." % user_id)
    else:
        self.like_users(following_ids, nlikes)


def like_location_feed(self, place, amount):
    self.logger.info("Searching location: {}".format(place))
    self.api.search_location(place)
    if not self.api.last_json["items"]:
        self.logger.error("{} not found.".format(place))
        return False
    else:
        finded_location = self.api.last_json["items"][0]["location"]["pk"]
        self.api.get_location_feed(finded_location)
        location_feed = self.api.last_json
        if location_feed.get("story"):
            self.logger.info("Liking users from stories...")
            location_to_filter = location_feed["story"]["items"][:amount]
            for i in range(0, len(location_to_filter)):
                user = location_to_filter[i]["user"]["pk"]
                self.like_user(user_id=user, amount=1, filtration=False)
        elif location_feed.get("items"):
            self.logger.info("Liking users from images...")
            max_id = ""
            counter = 0
            while counter < amount:
                location_to_filter = location_feed["items"][:amount]
                medias = self.filter_medias(location_to_filter, filtration=False)
                self.like_medias(medias)
                counter += 1
                if location_feed.get("next_max_id"):
                    max_id = location_feed["next_max_id"]
                else:
                    return False
                self.api.get_location_feed(finded_location, max_id)
                location_feed = self.api.last_json
        else:
            self.logger.error(
                (
                    " '{}' does not seem to have pictures. "
                    "Select a different location."
                ).format(place)
            )
            return False