mirror of https://github.com/coddrago/Heroku
97 lines
3.2 KiB
Python
Executable File
97 lines
3.2 KiB
Python
Executable File
# Friendly Telegram (telegram userbot)
|
|
# Copyright (C) 2018-2021 The Authors
|
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
# ©️ Dan Gazizullin, 2021-2023
|
|
# This file is a part of Hikka Userbot
|
|
# 🌐 https://github.com/hikariatama/Hikka
|
|
# You can redistribute it and/or modify it under the terms of the GNU AGPLv3
|
|
# 🔑 https://www.gnu.org/licenses/agpl-3.0.html
|
|
|
|
import os
|
|
import re
|
|
import string
|
|
import typing
|
|
|
|
|
|
def tty_print(text: str, tty: bool):
|
|
"""
|
|
Print text to terminal if tty is True,
|
|
otherwise removes all ANSI escape sequences
|
|
"""
|
|
print(text if tty else re.sub(r"\033\[[0-9;]*m", "", text))
|
|
|
|
|
|
def tty_input(text: str, tty: bool) -> str:
|
|
"""
|
|
Print text to terminal if tty is True,
|
|
otherwise removes all ANSI escape sequences
|
|
"""
|
|
return input(text if tty else re.sub(r"\033\[[0-9;]*m", "", text))
|
|
|
|
|
|
def api_config(tty: typing.Optional[bool] = None):
|
|
"""Request API config from user and set"""
|
|
from . import main
|
|
|
|
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"))
|
|
|
|
tty_print("\033[0;96mWelcome to Hikka Userbot!\033[0m", tty)
|
|
tty_print("\033[0;96m1. Go to https://my.telegram.org\033[0m", tty)
|
|
tty_print("\033[0;96m2. Login using your Telegram account\033[0m", tty)
|
|
tty_print("\033[0;96m3. Click on API development tools\033[0m", tty)
|
|
tty_print(
|
|
"\033[0;96m4. Create a new application, by entering the required details\033[0m",
|
|
tty,
|
|
)
|
|
tty_print("\033[0;96m5. Copy your API ID and API hash\033[0m", tty)
|
|
|
|
while api_id := tty_input("\033[0;96mEnter API ID: \033[0m", tty):
|
|
if api_id.isdigit():
|
|
break
|
|
|
|
tty_print("\033[0;91mInvalid ID\033[0m", tty)
|
|
|
|
if not api_id:
|
|
tty_print("\033[0;91mCancelled\033[0m", tty)
|
|
exit(0)
|
|
|
|
while api_hash := tty_input("\033[0;96mEnter API hash: \033[0m", tty):
|
|
if len(api_hash) == 32 and all(
|
|
symbol in string.hexdigits for symbol in api_hash
|
|
):
|
|
break
|
|
|
|
tty_print("\033[0;91mInvalid hash\033[0m", tty)
|
|
|
|
if not api_hash:
|
|
tty_print("\033[0;91mCancelled\033[0m", tty)
|
|
exit(0)
|
|
|
|
(main.BASE_PATH / "api_token.txt").write_text(api_id + "\n" + api_hash)
|
|
tty_print("\033[0;92mAPI config saved\033[0m", tty)
|