main.py

# -*- coding: utf-8 -*-
"""Main module"""

import uvicorn
from telegram import BotCommand
from telegram.ext import (
    CallbackQueryHandler,
    CommandHandler,
    Dispatcher,
    Filters,
    MessageHandler,
    Updater,
)

from module.callback_handlers import (
    exit_handler,
    informative_callback,
    localization_handler,
    md_handler,
    none_handler,
    submenu_handler,
)
from module.commands.aulario import (
    aulario,
    calendar_handler,
    month_handler,
    subjects_arrow_handler,
    subjects_handler,
)
from module.commands.drive_contribute import drive_contribute
from module.commands.esami import esami, esami_handler, esami_input_insegnamento
from module.commands.gdrive import drive, drive_handler
from module.commands.help import help_cmd
from module.commands.lezioni import lezioni, lezioni_handler, lezioni_input_insegnamento
from module.commands.professori import prof
from module.commands.regolamento_didattico import (
    cdl_handler,
    regolamentodidattico,
    regolamentodidattico_handler,
    send_regolamento,
)
from module.commands.reminder import (
    reminder,
    reminder_annullato_handler,
    reminder_appello_handler,
    reminder_confermato_handler,
    reminder_del_button,
    reminder_del_handler,
    reminder_input_insegnamento,
    reminder_new_handler,
    reminder_prof_handler,
    reminder_sessione_handler,
)
from module.commands.report import report
from module.commands.start import start
from module.commands.stats import stats, stats_tot
from module.data.vars import TEXT_IDS
from module.debug import error_handler, log_message
from module.easter_egg_func import (
    bladrim,
    lei_che_ne_pensa_signorina,
    prof_sticker,
    santino,
    smonta_portoni,
    uni_bandita,
)
from module.gitlab import git, gitlab_handler
from module.job_updater import check_exam_reminders, updater_lep
from module.shared import config_map
from module.utils.multi_lang_utils import get_regex_multi_lang, load_translations
from webapp.app import app


def add_commands(up: Updater) -> None:
    """Adds the list of commands with their description to the bot

    Args:
        up (Updater): supplyed Updater
    """
    commands = [
        BotCommand("start", "messaggio di benvenuto"),
        BotCommand("help ", "help"),
        BotCommand("gruppi", "link alla lista dei gruppi telegram delle materie"),
        BotCommand("esami", "cerca informazioni sugli esami"),
        BotCommand("reminder", "imposta un reminder per la prenotazione ad un appello"),
        BotCommand("lezioni", "cerca informazioni sulle lezioni"),
        BotCommand("prof", "cerca informazioni sui professori"),
        BotCommand("aulario", "cerca informazioni sull'aulario"),
        BotCommand("ufficioersu", "locazione e orari sede ufficio ersu"),
        BotCommand("ersu", "locazione e orari sede ersu"),
        BotCommand("sdidattica", "locazione e orari segreteria didattica"),
        BotCommand("studenti", "locazione e orari segreteria studenti"),
        BotCommand("cus", "locazione e orari del CUS"),
        BotCommand("cea", "CEA"),
        BotCommand("urp", "URP"),
        BotCommand("mensa", "orari e menù della mensa"),
        BotCommand("biblioteca", "orari della biblioteca"),
        BotCommand("aule_studio", "tutta la descrizione qui"),
        BotCommand("drive", "accedi alla cartella Drive"),
        BotCommand(
            "drive_contribute",
            "ottieni i permessi per caricare materiale sulla cartella Drive",
        ),
        BotCommand("git", "accedi al materiale didattico su GitLab"),
        BotCommand("gitlab", "accedi al materiale didattico su GitLab"),
        BotCommand("rappresentanti", "lista rappresentanti"),
        BotCommand("rappresentanti_dmi", "lista rappresentanti dmi"),
        BotCommand("rappresentanti_matematica", "lista rappresentanti matematica"),
        BotCommand("rappresentanti_informatica", "lista rappresentanti informatica"),
        BotCommand("report", "segnala un problema"),
        BotCommand("contributors", "sviluppatori del bot"),
        BotCommand("chatid", "mostra la chat id di questa chat"),
        BotCommand("cloud", "risorse didattiche in cloud"),
        BotCommand("regolamentodidattico", "lista dei regolamenti didattici"),
        BotCommand("ricevimenti", "lista orari ricevimenti dei professori"),
        BotCommand(
            "syllabus", "lista e informazioni pe rogni materia di ogni corso di laurea"
        ),
        BotCommand("trasporto_urbano_unict", "link orari Metro Shuttle FCE"),
        BotCommand("dmi_3d", "mappa in 3D del DMI"),
        BotCommand("mercatino", "bot per la vendita e l'acquisto di libri"),
        BotCommand("faq", "mostra le domande chieste più frequentemente"),
    ]
    up.bot.set_my_commands(commands=commands)


# pylint: disable=too-many-statements
def add_handlers(dp: Dispatcher) -> None:
    """Adds all the handlers the bot will react to

    Args:
        dp: supplied Dispatcher
    """
    dp.add_error_handler(error_handler)
    dp.add_handler(MessageHandler(Filters.all, log_message), 1)

    # Easter Egg
    dp.add_handler(CommandHandler('smonta_portoni', smonta_portoni))
    dp.add_handler(CommandHandler('santino', santino))
    dp.add_handler(CommandHandler('prof_sticker', prof_sticker))
    dp.add_handler(MessageHandler(Filters.regex('/lezioni cazzeggio'), bladrim))
    dp.add_handler(CommandHandler('leiCheNePensaSignorina', lei_che_ne_pensa_signorina))
    dp.add_handler(CommandHandler('universita_bandita', uni_bandita))

    # Informative command
    dp.add_handler(CommandHandler('sdidattica', informative_callback))
    dp.add_handler(CommandHandler('studenti', informative_callback))
    dp.add_handler(CommandHandler('cea', informative_callback))
    dp.add_handler(CommandHandler('ersu', informative_callback))
    dp.add_handler(CommandHandler('ufficioersu', informative_callback))
    dp.add_handler(CommandHandler('urp', informative_callback))
    dp.add_handler(CommandHandler('biblioteca', informative_callback))
    dp.add_handler(CommandHandler('aule_studio', informative_callback))
    dp.add_handler(CommandHandler('gruppi', informative_callback))
    dp.add_handler(CommandHandler('cus', informative_callback))
    dp.add_handler(CommandHandler('ricevimenti', informative_callback))
    dp.add_handler(CommandHandler('syllabus', informative_callback))
    dp.add_handler(CommandHandler('trasporto_urbano_unict', informative_callback))
    dp.add_handler(CommandHandler('dmi_3d', informative_callback))
    dp.add_handler(CommandHandler('faq', informative_callback))

    dp.add_handler(CommandHandler('lezioni', lezioni))
    dp.add_handler(CommandHandler('esami', esami))
    dp.add_handler(CommandHandler('reminder', reminder))

    dp.add_handler(CommandHandler('prof', prof))

    dp.add_handler(CommandHandler('aulario', aulario))
    dp.add_handler(
        MessageHandler(
            Filters.regex(get_regex_multi_lang(TEXT_IDS.AULARIO_KEYBOARD_TEXT_ID)),
            aulario,
        )
    )
    dp.add_handler(CommandHandler('help', help_cmd))
    dp.add_handler(
        MessageHandler(
            Filters.regex(get_regex_multi_lang(TEXT_IDS.HELP_KEYBOARD_TEXT_ID)),
            help_cmd,
        )
    )
    dp.add_handler(CommandHandler('contributors', informative_callback))
    dp.add_handler(CommandHandler('mercatino', informative_callback))
    dp.add_handler(
        MessageHandler(
            Filters.regex(
                get_regex_multi_lang(TEXT_IDS.MERCATINO_LIBRI_KEYBOARD_TEXT_ID)
            ),
            informative_callback,
        )
    )

    dp.add_handler(CommandHandler('rappresentanti', informative_callback))
    dp.add_handler(CommandHandler('rappresentanti_dmi', informative_callback))
    dp.add_handler(CommandHandler('rappresentanti_informatica', informative_callback))
    dp.add_handler(CommandHandler('rappresentanti_matematica', informative_callback))
    dp.add_handler(CommandHandler('report', report))
    dp.add_handler(
        CommandHandler('chatid', lambda u, c: u.message.reply_text(u.message.chat_id))
    )
    dp.add_handler(CommandHandler('start', start))
    dp.add_handler(CommandHandler('cloud', informative_callback))
    dp.add_handler(
        MessageHandler(
            Filters.regex(get_regex_multi_lang(TEXT_IDS.CLOUD_KEYBOARD_TEXT_ID)),
            informative_callback,
        )
    )
    dp.add_handler(
        MessageHandler(
            Filters.regex(get_regex_multi_lang(TEXT_IDS.REPORT_TO_KEYBOARD_TEXT_ID)),
            informative_callback,
        )
    )

    # generic buttons
    dp.add_handler(CallbackQueryHandler(exit_handler, pattern='^(exit_cmd)'))
    dp.add_handler(CallbackQueryHandler(submenu_handler, pattern='sm_.*'))
    dp.add_handler(CallbackQueryHandler(md_handler, pattern='md_.*'))
    dp.add_handler(
        CallbackQueryHandler(localization_handler, pattern='localization_.*')
    )
    dp.add_handler(CallbackQueryHandler(none_handler, pattern='NONE'))

    # aulario and calendar
    dp.add_handler(CallbackQueryHandler(calendar_handler, pattern='cal_.*'))
    dp.add_handler(CallbackQueryHandler(month_handler, pattern='m_[np]_.*'))
    dp.add_handler(CallbackQueryHandler(subjects_handler, pattern='sb_.*'))
    dp.add_handler(CallbackQueryHandler(subjects_arrow_handler, pattern='pg_.*'))

    # drive & gitlab buttons
    dp.add_handler(CallbackQueryHandler(drive_handler, pattern=r'^drive_file_.*'))
    dp.add_handler(CallbackQueryHandler(gitlab_handler, pattern='git_.*'))

    # regolamento didattico
    dp.add_handler(CommandHandler('regolamentodidattico', regolamentodidattico))
    dp.add_handler(
        CallbackQueryHandler(regolamentodidattico_handler, pattern=r'^reg_button_.*')
    )
    dp.add_handler(CallbackQueryHandler(cdl_handler, pattern=r'^cdl_button_.*'))
    dp.add_handler(
        CallbackQueryHandler(send_regolamento, pattern=r'^Regolamento Didattico.*')
    )

    # esami
    # regex accetta [/ins: nome] oppure [/Ins: nome], per agevolare chi usa il cellulare
    dp.add_handler(CallbackQueryHandler(esami_handler, pattern='esami_button_.*'))
    dp.add_handler(
        MessageHandler(Filters.regex(r"^(?!=<[/])[Ii]ns:\s+"), esami_input_insegnamento)
    )

    # reminder
    # regex accetta [/ins: nome] oppure [\Ins:nome], per agevolare chi usa il cellulare
    dp.add_handler(CallbackQueryHandler(reminder_prof_handler, pattern='^rem_prof_'))
    dp.add_handler(CallbackQueryHandler(reminder_new_handler, pattern='rem_add'))
    dp.add_handler(CallbackQueryHandler(reminder_del_button, pattern='rem_del$'))
    dp.add_handler(CallbackQueryHandler(reminder_del_handler, pattern='^rem_delete_*'))
    dp.add_handler(
        CallbackQueryHandler(reminder_sessione_handler, pattern='^rem_sess_')
    )
    dp.add_handler(
        CallbackQueryHandler(reminder_appello_handler, pattern='^rem_appello_')
    )
    dp.add_handler(
        CallbackQueryHandler(reminder_annullato_handler, pattern='^rem_conf_no')
    )
    dp.add_handler(
        CallbackQueryHandler(reminder_confermato_handler, pattern='^rem_conf_yes')
    )
    dp.add_handler(
        MessageHandler(
            Filters.regex(r"^(?!=<[/])[Ii]ns:\s+"), reminder_input_insegnamento
        ),
        2,
    )

    # lezioni
    dp.add_handler(CallbackQueryHandler(lezioni_handler, pattern='lezioni_button_*'))
    dp.add_handler(
        MessageHandler(
            Filters.regex(r"^(?!=<[/])[Nn]ome:\s+"), lezioni_input_insegnamento
        )
    )

    # drive and gitlab commands
    if config_map['debug']['disable_drive'] == 0:
        dp.add_handler(CommandHandler('drive', drive))
        dp.add_handler(CommandHandler('drive_contribute', drive_contribute))

    if config_map['debug']['disable_gitlab'] == 0:
        dp.add_handler(CommandHandler('git', git))
        dp.add_handler(CommandHandler('gitlab', git))

    # stats command
    if config_map['debug']['disable_db'] == 0:
        dp.add_handler(CommandHandler('stats', stats))
        dp.add_handler(CommandHandler('stats_tot', stats_tot))


def add_jobs(dp: Dispatcher) -> None:
    """Schedule the jobs in the JobQueue

    Args:
        dp: supplyed Dispatcher
    """
    dp.job_queue.run_repeating(
        updater_lep, interval=86400, first=1
    )  # job_updater_lep (24h)

    dp.job_queue.run_repeating(
        check_exam_reminders, interval=86400, first=100
    )  # ogni 24h ma first=100 per evitare overlapping con le DB queries di updater_lep


def main() -> None:
    """Main function"""
    load_translations()
    updater = Updater(
        config_map['token'],
        request_kwargs={'read_timeout': 20, 'connect_timeout': 20},
        use_context=True,
    )
    add_commands(updater)
    add_handlers(updater.dispatcher)
    add_jobs(updater.dispatcher)

    updater.start_polling()
    uvicorn.run(app, host="0.0.0.0", port=8000)
    updater.idle()


if __name__ == '__main__':
    main()