app.py

import os
import logging
from telegram.ext import (
    Application,
    CallbackQueryHandler,
    CommandHandler,
    ConversationHandler,
    InlineQueryHandler,
    filters,
    MessageHandler,
)
from add_cred_callbacks import (
    # variable states
    CRED_NAME,
    CRED_USERNAME,
    CRED_PASSWORD,
    CRED_URI,
    CRED_TAGS,
    CRED_NOTE,
    CRED_CONFIRMATION,
    ADD_MORE,
    # callback functions
    start,
    cancel,
    gen_pass,
    silent_cancel,
    get_pass_callback,
    add_cred_callback,
    add_cred_name,
    add_cred_username,
    add_cred_password,
    add_cred_uri,
    add_cred_tags,
    add_cred_note,
    add_cred_confirm,
    add_more_confirm,
    inline_generator,
    unknown,
)
from dashboard_callbacks import (
    # chosen_cred,
    done_query,
    get_cred,
    inline_credential,
    dash,
    fav,
)
from delete_callbacks import (
    del_cred,
    do_del_cred,
    return_del_cred,
)
from edit_callbacks import (
    # states
    CRED_NAME_EDIT,
    CRED_USERNAME_EDIT,
    CRED_PASSWORD_EDIT,
    CRED_URI_EDIT,
    CRED_TAGS_EDIT,
    CRED_NOTE_EDIT,
    # callbacks
    edit_cred,
    return_edit_cred,
    edit_cred_name,
    do_edit_cred_name,
    edit_cred_username,
    do_edit_cred_username,
    edit_cred_password,
    do_edit_cred_password,
    edit_cred_uri,
    do_edit_cred_uri,
    edit_cred_tags,
    do_edit_cred_tags,
    edit_cred_note,
    do_edit_cred_note,
)


voltpass_token = str(os.getenv("VOLTPASS_TOKEN"))


logging.basicConfig(
    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
    level=logging.INFO,
)
# for monitoring activities
logger = logging.getLogger(__name__)


def main() -> None:
    application = Application.builder().token(voltpass_token).build()

    # Conversation handlers
    conv_add_cred = ConversationHandler(
        entry_points=[
            CallbackQueryHandler(
                add_cred_callback,
                pattern="^add_cred$",
            ),
            CommandHandler("add_cred", add_cred_callback),
        ],
        states={
            CRED_NAME: [
                MessageHandler(filters.TEXT & ~filters.COMMAND, add_cred_name),
            ],
            CRED_USERNAME: [
                MessageHandler(
                    (filters.TEXT | filters.Entity("email")) & ~filters.COMMAND,
                    add_cred_username,
                ),
            ],
            CRED_PASSWORD: [
                MessageHandler(filters.TEXT & ~filters.COMMAND, add_cred_password),
            ],
            CRED_URI: [
                MessageHandler(
                    filters.TEXT & ~filters.COMMAND & filters.Entity("url"),
                    add_cred_uri,
                ),
            ],
            CRED_TAGS: [
                MessageHandler(
                    filters.TEXT & ~filters.COMMAND,
                    add_cred_tags,
                ),
                CallbackQueryHandler(
                    add_cred_tags,
                    pattern="^skip_tags$",
                ),
            ],
            CRED_NOTE: [
                MessageHandler(filters.TEXT & ~filters.COMMAND, add_cred_note),
                CallbackQueryHandler(
                    add_cred_note,
                    pattern="^skip_note$",
                ),
            ],
            CRED_CONFIRMATION: [
                CallbackQueryHandler(
                    add_cred_confirm,
                    pattern="^(confirm_cred|cancel_cred)$",
                ),
            ],
            ADD_MORE: [
                CallbackQueryHandler(
                    add_more_confirm,
                    pattern="^(add_cred_more|add_cred_done)$",
                ),
            ],
        },
        fallbacks=[
            MessageHandler(filters.COMMAND, silent_cancel),
            CallbackQueryHandler(
                silent_cancel,
                pattern="^dash$",
            ),
        ],
        allow_reentry=True,
        per_chat=False,
    )

    edit_name_conv_handler = ConversationHandler(
        entry_points=[
            CallbackQueryHandler(
                edit_cred_name,
                pattern="^edit_name_[0-9a-fA-F]{24}$",
            ),
        ],
        states={
            CRED_NAME_EDIT: [
                MessageHandler(filters.TEXT & ~filters.COMMAND, do_edit_cred_name),
            ],
        },
        fallbacks=[
            MessageHandler(filters.COMMAND, silent_cancel),
            CallbackQueryHandler(
                silent_cancel,
                pattern="^dash$",
            ),
        ],
        allow_reentry=True,
        per_chat=False,
    )

    edit_username_conv_handler = ConversationHandler(
        entry_points=[
            CallbackQueryHandler(
                edit_cred_username,
                pattern="^edit_username_[0-9a-fA-F]{24}$",
            ),
        ],
        states={
            CRED_USERNAME_EDIT: [
                MessageHandler(
                    (filters.TEXT | filters.Entity("email")) & ~filters.COMMAND,
                    do_edit_cred_username,
                ),
            ],
        },
        fallbacks=[
            MessageHandler(filters.COMMAND, silent_cancel),
            CallbackQueryHandler(
                silent_cancel,
                pattern="^dash$",
            ),
        ],
        allow_reentry=True,
        per_chat=False,
    )

    edit_password_conv_handler = ConversationHandler(
        entry_points=[
            CallbackQueryHandler(
                edit_cred_password,
                pattern="^edit_password_[0-9a-fA-F]{24}$",
            ),
        ],
        states={
            CRED_PASSWORD_EDIT: [
                MessageHandler(filters.TEXT & ~filters.COMMAND, do_edit_cred_password),
            ],
        },
        fallbacks=[
            MessageHandler(filters.COMMAND, silent_cancel),
            CallbackQueryHandler(
                silent_cancel,
                pattern="^dash$",
            ),
        ],
        allow_reentry=True,
        per_chat=False,
    )

    edit_uri_conv_handler = ConversationHandler(
        entry_points=[
            CallbackQueryHandler(
                edit_cred_uri,
                pattern="^edit_uri_[0-9a-fA-F]{24}$",
            ),
        ],
        states={
            CRED_URI_EDIT: [
                MessageHandler(
                    filters.TEXT & ~filters.COMMAND & filters.Entity("url"),
                    do_edit_cred_uri,
                ),
            ],
        },
        fallbacks=[
            MessageHandler(filters.COMMAND, silent_cancel),
            CallbackQueryHandler(
                silent_cancel,
                pattern="^dash$",
            ),
        ],
        allow_reentry=True,
        per_chat=False,
    )

    edit_tags_conv_handler = ConversationHandler(
        entry_points=[
            CallbackQueryHandler(
                edit_cred_tags,
                pattern="^edit_tags_[0-9a-fA-F]{24}$",
            ),
        ],
        states={
            CRED_TAGS_EDIT: [
                MessageHandler(
                    MessageHandler(filters.TEXT & ~filters.COMMAND, do_edit_cred_tags),
                    do_edit_cred_tags,
                ),
            ],
        },
        fallbacks=[
            MessageHandler(filters.COMMAND, silent_cancel),
            CallbackQueryHandler(
                silent_cancel,
                pattern="^dash$",
            ),
        ],
        allow_reentry=True,
        per_chat=False,
    )

    edit_note_conv_handler = ConversationHandler(
        entry_points=[
            CallbackQueryHandler(
                edit_cred_note,
                pattern="^edit_note_[0-9a-fA-F]{24}$",
            ),
        ],
        states={
            CRED_NOTE_EDIT: [
                MessageHandler(filters.TEXT & ~filters.COMMAND, do_edit_cred_note),
            ],
        },
        fallbacks=[
            MessageHandler(filters.COMMAND, silent_cancel),
            CallbackQueryHandler(
                silent_cancel,
                pattern="^dash$",
            ),
        ],
        allow_reentry=True,
        per_chat=False,
    )
    # Setting Handlers
    # chosen_result handlers
    # chosen_cred_handler = ChosenInlineResultHandler(
    #     chosen_cred,
    #     pattern="^[0-9a-fA-F]{24}$",
    # )
    # command handlers
    start_handler = CommandHandler("start", start)
    cancel_handler = CommandHandler("cancel", cancel)
    get_cred_handler = CommandHandler("get_cred", get_cred)
    gen_pass_handler = CommandHandler("gen_pass", gen_pass)
    dash_handler = CommandHandler("dash", dash)
    unknown_handler = MessageHandler(filters.COMMAND, unknown)
    # callback handlers
    dash_callback_handler = CallbackQueryHandler(
        dash,
        pattern="^dash$",
    )
    edit_cred_callback_handler = CallbackQueryHandler(
        edit_cred,
        pattern="^edit_[0-9a-fA-F]{24}$",
    )
    return_edit_cred_callback_handler = CallbackQueryHandler(
        return_edit_cred,
        pattern="^return_edit_[0-9a-fA-F]{24}$",
    )
    del_cred_callback_handler = CallbackQueryHandler(
        del_cred,
        pattern="^del_[0-9a-fA-F]{24}$",
    )
    do_del_cred_callback_handler = CallbackQueryHandler(
        do_del_cred,
        pattern="^do_del_[0-9a-fA-F]{24}$",
    )
    return_del_cred_callback_handler = CallbackQueryHandler(
        return_del_cred,
        pattern="^return_del_[0-9a-fA-F]{24}$",
    )
    fav_callback_handler = CallbackQueryHandler(
        fav,
        pattern="^fav$",
    )
    get_pass_callback_handler = CallbackQueryHandler(
        get_pass_callback,
        pattern="^get_pass$",
    )
    done_query_handler = CallbackQueryHandler(
        done_query,
        pattern="^done$",
    )
    # inline handlers
    inline_generator_handler = InlineQueryHandler(inline_generator)
    inline_credential_handler = InlineQueryHandler(
        inline_credential,
        pattern="^🔎: .*$",
    )

    # Adding handlers
    # application.add_handler(chosen_cred_handler)
    application.add_handler(conv_add_cred)
    application.add_handler(edit_name_conv_handler)
    application.add_handler(edit_username_conv_handler)
    application.add_handler(edit_password_conv_handler)
    application.add_handler(edit_uri_conv_handler)
    application.add_handler(edit_tags_conv_handler)
    application.add_handler(edit_note_conv_handler)
    application.add_handler(start_handler, 1)
    application.add_handler(get_pass_callback_handler, 1)
    application.add_handler(fav_callback_handler, 1)
    application.add_handler(dash_callback_handler, 1)
    application.add_handler(edit_cred_callback_handler, 1)
    application.add_handler(return_edit_cred_callback_handler, 1)
    application.add_handler(del_cred_callback_handler, 1)
    application.add_handler(do_del_cred_callback_handler, 1)
    application.add_handler(return_del_cred_callback_handler, 1)
    application.add_handler(done_query_handler, 1)
    application.add_handler(cancel_handler, 1)
    application.add_handler(get_cred_handler, 1)
    application.add_handler(gen_pass_handler, 1)
    application.add_handler(dash_handler, 1)
    application.add_handler(unknown_handler, 2)
    application.add_handler(inline_credential_handler)
    application.add_handler(inline_generator_handler)

    # Poll Application
    application.run_polling()


if __name__ == "__main__":
    main()