from .. import loader, utils
from telethon.tl.types import DocumentAttributeFilename
from requests import get, post
from PIL import Image
from io import BytesIO

@loader.tds
class QRtoolsMod(loader.Module):
	"""Generator and reader of QR codes"""
	strings = {"name": "QR tool's"}
	@loader.owner
	async def makeqrcmd(self, message):
		""".makeqr <text or reply>"""
		text = utils.get_args_raw(message)
		reply = await message.get_reply_message()
		file = False
		if not text or text.lower() == ".file":
			if text and text == ".file":
				file = True
			if not reply or not reply.message:
				await message.edit("<b>Нет текста для кодирования!</b>")
				return
			text = reply.raw_text
		else:
			if text.startswith(".file"):
				file = True
				text = text[5:].strip()
		url = "https://api.qrserver.com/v1/create-qr-code/?data={}&size=512x512&charset-source=UTF-8&charset-target=UTF-8&ecc=L&color=0-0-0&bgcolor=255-255-255&margin=1&qzone=1&format=png"
		r = get(url.format(text), stream=True)
		qrcode = BytesIO() 
		qrcode.name = "qr.png" if file else "qr.webp"
		Image.open(BytesIO(r.content)).save(qrcode)
		qrcode.seek(0)
		await message.delete()
		await message.client.send_file(message.to_id, qrcode, reply_to=reply, force_document=file)
		
	@loader.owner
	async def readqrcmd(self, message):
		""".readqr <qrcode or reply to qrcode>"""
		ok = await check(message)
		if not ok:
			reply = await message.get_reply_message()
			ok = await check(reply)
			if not ok:
				text = "<b>Это не изображение!</b>" if reply else "<b>Нечего не передано!</b>"
				await message.edit(text)
				return
		file = BytesIO()
		file.name = "qr.png"
		data = await message.client.download_file(ok)
		Image.open(BytesIO(data)).save(file)
		url = "https://api.qrserver.com/v1/read-qr-code/?outputformat=json"
		resp = post(url, files={"file": file.getvalue()})
		text = resp.json()[0]["symbol"][0]["data"]
		if not text:
			text = "<b>Невозможно распознать или QR пуст!<b>"
		await utils.answer(message, text)
		
async def check(msg):
	if msg and msg.media:
		if msg.photo:
			ok = msg.photo
		elif msg.document:
			if DocumentAttributeFilename(file_name='AnimatedSticker.tgs') in msg.media.document.attributes:
				return False
			if msg.gif or msg.video or msg.audio or msg.voice:
				return False
			ok = msg.media.document
		else:
			return False
	else:
		return False
	if not ok or ok is None:
		return False
	else:
		return ok