diff --git a/assets/2fa.txt b/assets/2fa.txt index ccb86d9..0fb1531 100644 --- a/assets/2fa.txt +++ b/assets/2fa.txt @@ -1,21 +1,19 @@ \033[0m \033[0m - \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;198;65;50m \033[48;2;215;168;34m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;226;88;73m \033[48;2;219;79;62m \033[48;2;176;57;44m \033[48;2;194;152;31m \033[48;2;213;166;35m \033[48;2;205;160;33m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;206;69;54m \033[48;2;227;87;71m \033[48;2;180;57;44m \033[48;2;178;59;44m \033[48;2;174;57;44m \033[48;2;192;150;31m \033[48;2;193;151;31m \033[48;2;213;166;33m \033[48;2;211;166;34m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;174;57;44m \033[48;2;192;150;31m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;174;57;44m \033[48;2;192;150;31m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;174;57;44m \033[48;2;192;150;31m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;174;57;44m \033[48;2;192;150;31m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;162;54;41m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;190;150;31m \033[48;2;188;146;31m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;143;48;38m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;169;132;28m \033[48;2;190;149;31m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;168;54;43m \033[48;2;138;48;37m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;158;125;28m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;155;53;41m \033[48;2;133;47;36m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;144;49;38m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[48;2;0;0;0m \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[0m \033[0m - \033[48;2;255;255;255m \033[0m \033[0m + \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;198;65;50m \033[48;2;215;168;34m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;226;88;73m \033[48;2;219;79;62m \033[48;2;176;57;44m \033[48;2;194;152;31m \033[48;2;213;166;35m \033[48;2;205;160;33m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;206;69;54m \033[48;2;227;87;71m \033[48;2;180;57;44m \033[48;2;178;59;44m \033[48;2;174;57;44m \033[48;2;192;150;31m \033[48;2;193;151;31m \033[48;2;213;166;33m \033[48;2;211;166;34m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;174;57;44m \033[48;2;192;150;31m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;174;57;44m \033[48;2;192;150;31m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;174;57;44m \033[48;2;192;150;31m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;174;57;44m \033[48;2;192;150;31m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;162;54;41m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;190;150;31m \033[48;2;188;146;31m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;143;48;38m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;169;132;28m \033[48;2;190;149;31m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;168;54;43m \033[48;2;138;48;37m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;158;125;28m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;155;53;41m \033[48;2;133;47;36m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;192;150;31m \033[48;2;174;57;44m \033[48;2;144;49;38m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[0m \033[0m + \033[0m + \033[0m \033[0m \ No newline at end of file diff --git a/assets/success.txt b/assets/success.txt new file mode 100644 index 0000000..164488b --- /dev/null +++ b/assets/success.txt @@ -0,0 +1,19 @@ +\033[0m \033[0m + \033[48;2;255;182;11m \033[0m \033[0m + \033[48;2;255;182;11m \033[0m \033[0m + \033[48;2;255;182;11m \033[0m \033[0m + \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;170;196;83m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[0m \033[48;2;28;28;28m \033[48;2;170;196;83m \033[48;2;77;170;87m \033[48;2;170;196;83m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;170;196;83m \033[48;2;28;28;28m \033[0m \033[48;2;28;28;28m \033[48;2;170;196;83m \033[48;2;77;170;87m \033[48;2;52;115;101m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;77;170;87m \033[48;2;170;196;83m \033[48;2;28;28;28m \033[48;2;170;196;83m \033[48;2;77;170;87m \033[48;2;52;115;101m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;52;115;101m \033[48;2;77;170;87m \033[48;2;170;196;83m \033[48;2;77;170;87m \033[48;2;52;115;101m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;52;115;101m \033[48;2;77;170;87m \033[48;2;52;115;101m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;52;115;101m \033[48;2;77;170;87m \033[48;2;52;115;101m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;52;115;101m \033[48;2;77;170;87m \033[48;2;52;115;101m \033[48;2;28;28;28m \033[0m \033[0m + \033[48;2;28;28;28m \033[48;2;52;115;101m \033[48;2;28;28;28m \033[0m \033[48;2;255;182;11m \033[0m \033[0m + \033[48;2;255;182;11m \033[0m \033[48;2;28;28;28m \033[0m \033[48;2;255;182;11m \033[0m \033[0m + \033[48;2;255;182;11m \033[0m \033[48;2;255;182;11m \033[0m \033[0m + \033[48;2;255;182;11m \033[0m \033[0m + \033[0m + \033[0m \ No newline at end of file diff --git a/hikka/_internal.py b/hikka/_internal.py index 75c2703..4ee3a8e 100644 --- a/hikka/_internal.py +++ b/hikka/_internal.py @@ -64,3 +64,19 @@ def restart(): signal.signal(signal.SIGTERM, get_startup_callback()) die() + + +def print_banner(banner: str): + print("\033[2J\033[3;1f") + with open( + os.path.abspath( + os.path.join( + os.path.dirname(__file__), + "..", + "assets", + banner, + ) + ), + "r", + ) as banner: + print(banner.read().replace("\\033", "\033")) diff --git a/hikka/configurator.py b/hikka/configurator.py index 25340ef..2509094 100755 --- a/hikka/configurator.py +++ b/hikka/configurator.py @@ -45,20 +45,14 @@ def tty_input(text: str, tty: bool) -> str: def api_config(tty: typing.Optional[bool] = None): """Request API config from user and set""" from . import main + from ._internal import print_banner if tty is None: print("\033[0;91mThe quick brown fox jumps over the lazy dog\033[0m") tty = input("Is the text above colored? [y/N]").lower() == "y" if tty: - print("\033[2J\033[3;1f") - with open( - os.path.abspath( - os.path.join(os.path.dirname(__file__), "..", "assets", "banner.txt") - ), - "r", - ) as banner: - print(banner.read().replace("\\033", "\033")) + print_banner("banner.txt") tty_print("\033[0;95mWelcome to Hikka Userbot!\033[0m", tty) tty_print("\033[0;96m1. Go to https://my.telegram.org and login\033[0m", tty) diff --git a/hikka/main.py b/hikka/main.py index 5c1d4b5..256c324 100755 --- a/hikka/main.py +++ b/hikka/main.py @@ -61,6 +61,7 @@ from hikkatl.tl.functions.account import GetPasswordRequest from hikkatl.tl.functions.auth import CheckPasswordRequest from . import database, loader, utils, version +from ._internal import print_banner from .dispatcher import CommandDispatcher from .qr import QRCode from .tl_cache import CustomTelegramClient @@ -594,17 +595,7 @@ class Hikka: return False if await qr_login_poll(): - print("\033[2J\033[3;1f") - with open( - os.path.abspath( - os.path.join( - os.path.dirname(__file__), "..", "assets", "2fa.txt" - ) - ), - "r", - ) as banner: - print(banner.read().replace("\\033", "\033")) - + print_banner("2fa.txt") password = await client(GetPasswordRequest()) while True: _2fa = getpass( @@ -643,6 +634,7 @@ class Hikka: else: break + print_banner("success.txt") await self.save_client_session(client) self.clients += [client] return True