import random
import string
import requests


def get_image(boards, nsfw=False):
    """
    get_image(boards, nsfw=False)

    Returns a URL to an image on reddit from a random board in the boards list
    as long as it is hosted on one of the domains in the domains list
    """

    if isinstance(boards, list):
        boards = random.choice(boards)

    domains = [
        "cdn.awwni.m",
        "gfycat.com",
        "i.imgur.com",
        "i.redd.it",
        "i.reddituploads.com",
        "i3.kym-cdn.com",
        "imgur.com",
        "media.giphy.com",
    ]
    request_string = "https://reddit.com/r/{}/random.json".format(boards)

    if not nsfw:
        # Append this header to the request. Tells the API to only return SFW results
        request_string += "?obey_over18=true"

    # Spoof our user agent with each request so we dont get rate limited
    random_user_agent = "".join(
        random.SystemRandom().choice(string.ascii_uppercase + string.digits)
        for _ in range(5)
    )

    response = requests.get(
        request_string, headers={"User-agent": random_user_agent}
    ).json()[0]["data"]["children"][0]

    if response["data"]["domain"] not in domains:
        # If we dont find an approved domain, re-try the request
        return get_image(boards, nsfw=nsfw)
    image_url = response["data"]["url"].replace("http://", "https://")

    return image_url