#!/usr/bin/env python3
import yaml
from telethon import TelegramClient, events, sync
import logging
from telethon import functions, types, errors
from pathlib import Path

mimes={
    "image/webp": "webp",
    "application/x-tgsticker": "tgs",
    "video/webm": "webm"
    }
w = Path(".")
wn = w / "env"
wn.mkdir(exist_ok=True)
wd = wn/"download"
wd.mkdir(exist_ok=True)
wdi = wd/"inline"
wdi.mkdir(exist_ok=True)
logging.basicConfig(filename=str(wn/"bot.log"),level=logging.DEBUG)
log = logging.getLogger("gentoobot")
dllog=logging.getLogger("gentoobot.download")

def download_sticker(client, document):
    stickerdir, setid = get_sticker_setid(document)
    if stickerdir is None:
        return
    dldir = wd / stickerdir
    dldir.mkdir(exist_ok=True)
    dlpath = dldir / (str(document.id)+"."+mimes.get(document.mime_type, "")) 
    if not  dlpath.exists() or dlpath.stat().st_size != document.size:
        dllog.info("downloading %s to %s", document.id, str(dlpath))
        client.download_file(document, dlpath)
    else:
        dllog.info("document %s already downloaded at %s", document.id, str(dlpath))

def get_sticker_setid(document):
    stickerset = None
    for a in document.attributes:
        if hasattr(a, "stickerset"):
            stickerset = a.stickerset
    if stickerset is None:
        dllog.info("document %s is not a sticker", document.id)
        return None, None
    if isinstance(stickerset, types.InputStickerSetID):
        log.info("document %s is a normal sticker", document.id)
        return str(stickerset.id), stickerset
    if isinstance(stickerset, types.InputStickerSetEmpty):
        dllog.info("document %s is an inline sticker", document.id)
        return "inline", stickerset

def fetch_dialogs(client):
    sticker_archives=list()
    for dialog in client.iter_dialogs():
        log.debug(dialog)
        if dialog.entity.id in cfg["stickers"]["sources"]["dialogs"]:
            log.info(dialog.stringify())
            yield dialog

def main(client):
    stickerset_seen=set()
    for sticker_archive in fetch_dialogs(client):
        for msg in client.iter_messages(sticker_archive, limit=None):
            log.debug(msg)
            if not hasattr(msg, "media"):
                log.debug("message %i has no media", msg.id)
                continue
            if not hasattr(msg.media, "document"):
                log.debug("message %i has no documents", msg.id)
                continue
            try:
                dldir, setid = get_sticker_setid(msg.media.document)
                if dldir == "inline":
                    download_sticker(client, msg.media.document)
                elif dldir is not None:
                    download_sticker(client, msg.media.document)
                    if setid.id not in stickerset_seen:
                        log.debug("preparing to download whole stickerset %s as %s", setid, dldir)
                        stickerset_seen.add(setid.id)
                        try:
                            for doc in client(functions.messages.GetStickerSetRequest(stickerset=setid,hash=0)).documents:
                                try: 
                                    download_sticker(client, doc)
                                except Exception as e:
                                    log.critical("oops: %s", exc_info=e)
                        except errors.rpcerrorlist.StickersetInvalidError:
                            log.warning("sadly, stickerset %s no longer exists", setid.id)
            except Exception as e:
                log.error("somethin wrong happened during checking message: %s", msg.stringify(), exc_info=e)

    #  for doc in stickers_inline:
        #  try:
            #  download_sticker(client, doc)
        #  except Exception as e:
            #  log.critical("ouchie: %s", exc_info=e)

    #  for stickerset in stickers_sets.values():
        #  try:
            #  for doc in client(functions.messages.GetStickerSetRequest(stickerset=stickerset,hash=0)).documents:
                #  try:
                    #  download_sticker(client, doc)
                #  except Exception as e:
                    #  log.critical("oops: %s", exc_info=e)
        #  except errors.rpcerrorlist.StickersetInvalidError:
            #  log.warning("sadly, stickerset %s no longer exists", stickerset.id)
        #  except Exception as e:
            #  log.critical("omg: %s", exc_info=e)


####################


log.debug("opening %s", repr("config/bot.yaml"))
with open("config/bot.yaml") as cfgstream:
    cfg = yaml.safe_load(cfgstream)
    log.debug(cfg)
    app = cfg["apps"][0]
log.debug("starting client with id %s and hash %s", app['id'], app['hash'])
client = TelegramClient('env/gentoo_session', app['id'], app['hash'])
client.start()
log.debug(client.get_me().stringify())
stickers_sets=dict()
stickers_favorites=set()
stickers_inline=list()
sticker_archives=list()
try:
    main(client)
except KeyboardInterrupt:
    pass
client.disconnect()