# -*- coding: utf-8 -*-

# Module author: Official Repo, @GovnoCodules

import io
import json

import requests

from .. import loader, utils


@loader.tds
class SearchMod(loader.Module):
    """Searcher module"""

    strings = {
        "name": "Search",
        "search": "⚪⚪⚪\n⚪❓⚪\n⚪⚪⚪",
        "no_reply": "<b>Reply to image or sticker!</b>",
        "ya_result": '<a href="{}"><b>🔴⚪🔴|See</b>\n<b>⚪🔴⚪|Search</b>\n<b>⚪🔴⚪|Results</b></a>',
        "error": "<b>Something went wrong...</b>",
    }

    @loader.owner
    async def yarscmd(self, message):
        """.yars <repy to image>"""
        reply = await message.get_reply_message()
        data = await check_media(message, reply)
        if not data:
            await utils.answer(message, self.strings("no_reply", message))
            return
        await utils.answer(message, self.strings("search", message))
        searchUrl = "https://yandex.ru/images/search"
        files = {"upfile": ("blob", data, "image/jpeg")}
        params = {
            "rpt": "imageview",
            "format": "json",
            "request": '{"blocks":[{"block":"b-page_type_search-by-image__link"}]}',
        }
        response = requests.post(searchUrl, params=params, files=files)
        if response.ok:
            query_string = json.loads(response.content)["blocks"][0]["params"]["url"]
            link = searchUrl + "?" + query_string
            text = self.strings("ya_result", message).format(link)
            await utils.answer(message, text)
        else:
            await utils.answer(message, self.strings("error", message))


async def check_media(message, reply):
    if not reply or not reply.media:
        return None
    if reply.photo:
        data = reply.photo
    elif reply.document:
        if reply.gif or reply.video or reply.audio or reply.voice:
            return None
        data = reply.media.document
    else:
        return None
    if not data or data is None:
        return None
    data = await message.client.download_file(data, bytes)
    return io.BytesIO(data)