import aiohttp
import asyncio
from bs4 import BeautifulSoup as bs4

session = aiohttp.ClientSession()


class ThumbNailer(object):

    async def get_img(self, url):
            async with session.get(url, timeout=30) as response:
                assert response.status == 200
                html = await response.read()
                return await ThumbNailer.__parse_result(html)

    async def __parse_result(html):
        try:
            soup = bs4(html, 'html.parser')
            img = soup.find("meta", property="twitter:image")["content"]
            return img
        except Exception as e:
            raise e

    async def identify(self, identifier, uri):
        if "youtube" in uri:
            thumbnail_url = f"https://img.youtube.com/vi/{identifier}/0.jpg"
            return thumbnail_url
        elif "soundcloud" in uri:
            thumbnail_url = await ThumbNailer.get_img(ThumbNailer, url=uri)
            return thumbnail_url
        else:
            return None