import random
import requests

import get_from_reddit


def get_from_danbooru(boards):
    """
    get_from_danbooru(boards)

    returns a URL to an image on danbooru
    """

    request = requests.get(
        'https://danbooru.donmai.us/posts/random.json?tags=rating%3Aexplicit'
    ).json()

    # List of tags we dont want images from
    undesired_tags = [
        'bestiality',
        'pee',
        'futa',
        'futanari',
        'yaoi'
    ]
    # If any undesired tags show up in the request, try again
    if not set(request['tag_string'].split()).isdisjoint(undesired_tags):
        return get_from_danbooru(boards)

    # Return the large firl url if available
    if 'large_file_url' in request.keys():
        if request['large_file_url'].startswith('/data/'):
            return "https://danbooru.donmai.us{}".format(request['large_file_url'])
        return request['large_file_url']

    if 'file_url' in request.keys():
        if request['file_url'].startswith('/data/'):
            return "https://danbooru.donmai.us{}".format(request['file_url'])
        return request['file_url']

    return get_from_danbooru(boards)


def get_lewd(channel_name):
    if 'nsfw' in channel_name:
        boards = [
            '2Booty',
            'ahegao',
            'bakunyuu_hentai',
            'ecchi',
            'hentai',
            'rule34',
            'WesternHentai',
            'doujinshi',
            'hentai_gif',
            'muchihentai',
            'chiisaihentai',
            'dekaihentai',
            'WaifusGonewild',
            'thighdeology'
        ]

        # This is really bad practice but pass boards to get_from_reddit AND danbooru so it doesnt error
        return random.choice([get_from_reddit.get_image, get_from_danbooru])(boards=boards)
    return 'You can only use this command in NSFW channels'