from . import command, photo_fn, always, sticker_reply_fn
from sh import convert as conv_unbaked, composite as comp_unbaked
from schema import StickerPack
import telegram


# disable _tty_out to compact the AppArmor profile :)
convert = conv_unbaked.bake(_tty_out=False)
composite = comp_unbaked.bake(_tty_out=False)

collect_uid_pack_map = {}

@command('list packs')
def list_packs(bea, bot, update):
    output = []
    packs = [pack for pack in StickerPack.select()]
    packs.sort(key=lambda pack: len(pack.title))
    for pack in packs:
        output.append(f"<a href='https://telegram.me/addstickers/{pack.shortname}'>{pack.title}</a>: {pack.shortname.split('_by_tfckbot')[0]}")
    bea.reply('\n'.join(output), parse_mode='HTML')

@command('newstickerset (\S+) (.+)', pass_groups=True)
def create_set(bea, bot, update, groups):
    """ Create a new sticker set. Arguments: short name, title. """
    shortname, title = groups
    shortname += "_by_tfckbot" # fuck de bot api maar whatever
    user_id = update.effective_user.id

    res = True
    try:
        res = bot.create_new_sticker_set(user_id, shortname, title)
    except telegram.TelegramError:
        update.message.reply_text("Unable to create pack.. are the arguments in the right order?")
        res = False

    if res:
        StickerPack.create(uid=user_id, shortname=shortname, title=title)
        update.message.reply_text("Created new pack by that name!")


@command('collect (\S+)', pass_groups=True)
def collect_stickers(bea, bot, update, groups):
    """ Start collecting stickers for a pack. This causes Bea to interpret every sent image as a new sticker, and the emoji in its caption as the corresponding emoji. """
    name = groups[0]
    name += "_by_tfckbot"

    pack = StickerPack.get_or_none(StickerPack.shortname == name)
    if not pack:
        return update.message.reply_text("I don't know of a pack by that name..")

    id = update.effective_user.id

    if not bea.try_register_bypass(id, collection_helper):
        return update.message.reply_text("Registering bypass function failed - are you already bypassing something?")
    else:
        collect_uid_pack_map[id] = (name, 0)
        return update.message.reply_text("Collecting... (end with 'Bea: end collect')")

@command('delsticker', additional_match_fn=sticker_reply_fn)
def del_sticker(bea, bot, update, sticker):
    """ Delete a sticker from a pack. Send this in reply to a sticker. """
    pack = StickerPack.get_or_none(StickerPack.shortname == sticker.set_name)
    if not pack:
        return update.message.reply_text("I don't manage the pack that sticker is from.")

    try:
        if bot.delete_sticker_from_set(sticker.file_id):
            update.message.reply_text("Successfully deleted!")
    except telegram.TelegramError:
        update.message.reply_text("Unable to delete sticker.")

@command('addsticker (\S+) (\S+)', pass_groups=True, additional_match_fn=photo_fn)
def add_sticker(bea, bot, update, photo, groups):
    """ Add a sticker with the given emoji to pack. Arguments: emoji, pack shortname. """
    emoji, name = groups
    name += "_by_tfckbot"

    pack = StickerPack.get_or_none(StickerPack.shortname == name)
    if not pack:
        return update.message.reply_text("I don't know of a pack by that name..")

    photo.get_file().download('newsticker.png')
    convert('newsticker.png', '-resize', '512x512', 'newstickerout.png')
    f = open('newstickerout.png', 'rb')

    id = pack.uid
    try:
        if bot.add_sticker_to_set(id, name, emoji, f):
            update.message.reply_text('Sticker added.')
    except telegram.TelegramError:
        update.message.reply_text("Something went wrong. Are your arguments in the right order?")

@command('(end collect)?', pass_groups=True, bypass_fn=always)
def collection_helper(bea, bot, update, groups):
    id = update.effective_user.id
    if groups and (groups[0] == 'end collect'):
        if not bea.unregister_bypass(id, collection_helper):
            return update.message.reply_text("Unable to unregister bypass, did you try to end collecting stickers when I wasn't doing that?")
        else:
            _, added_stickers = collect_uid_pack_map[id]
            del collect_uid_pack_map[id]
            return update.message.reply_text(f"Success. Added {added_stickers} stickers.")

    try:
        photo = photo_fn(update)
    except ValueError:
        return

    caption = update.effective_message.caption
    if not caption:
        return

    photo.get_file().download('newsticker.png')
    convert('newsticker.png', '-resize', '512x512', 'newstickerout.png')
    f = open('newstickerout.png', 'rb')

    name, added_stickers = collect_uid_pack_map[id]
    emoji = caption

    try:
        if bot.add_sticker_to_set(id, name, emoji, f):
            collect_uid_pack_map[id] = (name, added_stickers + 1)
    except telegram.TelegramError:
        update.message.reply_text("Something went wrong.. are those emoji in the caption?")

@command('steal', additional_match_fn=sticker_reply_fn)
def steal_sticker(bea, bot, update, sticker):
    """ Steal a sticker, adding it to Beapack. """
    pack = StickerPack.get(StickerPack.shortname == 'bpack_by_tfckbot')
    sticker.get_file().download('sticker.webp')
    convert('sticker.webp','sticker.png')
    if bot.add_sticker_to_set(pack.uid, pack.shortname, sticker.emoji, open('sticker.png', 'rb')):
        update.message.reply_to_message.reply_sticker(sticker="CAACAgQAAxkBAAIO_V6NAnjnxSxVaoOgSRN2GyA_yec6AAJ3AgAC6DSzAqmfA-HmNynaGAQ", quote=True)


def init(bea, _config):
    return [list_packs, del_sticker, collect_stickers, add_sticker, create_set, steal_sticker]