Fazendaaa/AnilistBot

// tslint:disable: no-submodule-imports
import { aboutExtra, countdownExtra, counterExtra, guideExtra, handleMediaExtra, handleUserExtra, menuExtra, recommendationExtra,
startExtra } from 'extra';
import { IBotContext, LanguageRequest, ListFilterRequest, ListRequest, MenuRequest, NotifyRequests, TimeRequest, UserRequest
} from 'telegraf-bot-typings';
import Scene from 'telegraf/scenes/base';
import { IUserTTFInfo } from '.';
import { handleMediaMoreExtra } from '../extra/media';
import { handleMediaMore } from '../parse/media';
import { handleCountdown, handleCounter, handleMedia, handleUser } from '../parse/menu';

export const menuScene = new Scene('Menu');

menuScene.leave(async ({ i18n, updateType, message, replyWithMarkdown }: IBotContext) => {
    if ('callback_query' !== updateType && undefined !== message.text && i18n.t('menu') !== message.text) {
        return replyWithMarkdown(i18n.t('leavingMenu'));
    }
});

menuScene.enter(async ({ i18n, replyWithMarkdown }: IBotContext) => {
    await replyWithMarkdown(i18n.t('menuGreetings'), startExtra(i18n));

    return replyWithMarkdown(i18n.t('menuOptions'), menuExtra(i18n));
});

menuScene.on('callback_query', async (ctx: IBotContext) => {
    const { i18n, from, scene, answerCbQuery, callbackQuery, editMessageText, replyWithMarkdown } = ctx;
    const { id } = from;
    const translation = i18n;
    const data = callbackQuery.data.split('/');
    const kind = <MenuRequest> data[0];
    const user = <IUserTTFInfo> scene.state;

    await answerCbQuery(i18n.t('loading'));

    if ('ABOUT' === kind) {
        return editMessageText(i18n.t('aboutOptions'), aboutExtra());
    } if ('MENU' === kind) {
        return editMessageText(i18n.t('menuOptions'), menuExtra(translation));
    } if ('GUIDE' === kind) {
        return editMessageText(i18n.t('guideOptions'), guideExtra(translation));
    } if ('RECOMMENDATION' === kind) {
        return editMessageText(i18n.t('recommendationOptions'), recommendationExtra());
    } if ('COUNTDOWN' === kind) {
        return editMessageText(i18n.t('countdownOptions', { anime: await handleCountdown({
            id,
            user,
            translation
        })}), countdownExtra());
    } if ('COUNTER' === kind) {
        return editMessageText(await handleCounter({ id, user, translation }), counterExtra());
    } if ('USER' === kind) {
        const request = (2 <= data.length) ? <UserRequest | TimeRequest> data[1] : null;
        const value = (3 === data.length) ? <LanguageRequest | NotifyRequests | number> data[2] : null;

        return editMessageText(await handleUser({
            id,
            value,
            request,
            translation
        }), handleUserExtra({ value, request, translation }));
    } if ('MEDIA' === kind) {
        const list = <ListRequest> data[1];
        const filter = <ListFilterRequest> data[2];

        return editMessageText(await handleMedia({
            id,
            list,
            user,
            filter,
            translation
        }), await handleMediaExtra({ user, list, filter, translation }));
    } if ('LOCATION' === kind) {
        return scene.enter('Location');
    } if ('MORE' === kind) {
        const request = <'ANIME' | 'MANGA'> data[1];
        const content = parseInt(data[2], 10);

        return replyWithMarkdown(await handleMediaMore({
            content,
            request,
            translation
        }), handleMediaMoreExtra({ id: content, request, translation }));
    }

    return scene.leave();
});