Initial commit

modules/menu/__init__.py

@@ -0,0 +1 @@
+from .menu import menu

modules/menu/callbacks/config/__init__.py

@@ -0,0 +1 @@
+pass

modules/menu/callbacks/config/callbacks.py

@@ -0,0 +1,16 @@
+from modules.decorators.callback import callback
+from modules.config import config
+
+
+@callback
+def edit_config_callback(field: str, comments: str, default: str = None):
+    print(f'Current value is {config.get(field)}')
+    print(f'Default value is {default}. Press Enter to restore it')
+    config[field] = input(f'{field} ({comments}) > ')
+    if config[field] in ['None', 'none']:
+        del config[field]
+
+    if config[field] == '' and default is not None:
+        config[field] = default
+    elif default is None:
+        del config[field]

modules/menu/callbacks/create_session/callbacks.py

@@ -0,0 +1,33 @@
+from sqlite3 import OperationalError
+from rich import print
+from modules.decorators.callback import async_callback
+from modules.client import GeneratedClient
+from modules.config import sessions
+from modules.config.models import SessionConfig
+
+
+@async_callback
+async def create_session_callback():
+    session_name = input('New session name: ')
+    while session_name in sessions.keys():
+        session_name = input('Session with this name is already saved. Try another name: ')
+
+    try:
+        await (client := GeneratedClient(name=session_name)).start()
+        sessions[session_name] = SessionConfig(
+            id=client.me.id,
+            phone=client.phone_number,
+            profile_name=client.me.first_name + (f' {client.me.last_name}' if client.me.last_name else ''),
+            username=client.me.username
+        ).json()
+        print(f'[green]Created[/] session {session_name}...')
+        input()
+
+    except OperationalError:
+        print('[red]Cannot create session file.[/] Try using different name...')
+        input()
+
+    except Exception as e:
+        print(f'[red]Error:[/] {e}...')
+        input()
+        return

modules/menu/callbacks/get_code/__init__.py

@@ -0,0 +1 @@
+from .get_code import get_code_callback

modules/menu/callbacks/get_code/get_code.py

@@ -0,0 +1,6 @@
+from modules.decorators.callback import async_callback
+
+
+@async_callback
+async def get_code_callback(session_name: str):
+    ...

modules/menu/dynamic/get_code/__init__.py

@@ -0,0 +1 @@
+from .submenu import get_code_submenu_item

modules/menu/dynamic/get_code/generator.py

@@ -0,0 +1,22 @@
+import os
+from cursesmenu import CursesMenu
+from cursesmenu.items import FunctionItem
+
+from modules.config import sessions
+from modules.config.models import SessionConfig
+from ...callbacks.get_code import get_code_callback
+
+
+def generate_get_code_menu() -> CursesMenu:
+    submenu = CursesMenu(
+        title='Sessions'
+    )
+    for session_name in sessions.keys():
+        if os.path.isfile(f'{session_name}.session'):
+            submenu.items.append(FunctionItem(
+                f'{session_name} - {SessionConfig(**sessions[session_name])}',
+                function=get_code_callback,
+                args=[session_name]
+            ))
+
+    return submenu

modules/menu/dynamic/get_code/submenu.py

@@ -0,0 +1,15 @@
+from ...menu import menu
+from cursesmenu import CursesMenu
+from modules.custom_items.DynamicSubmenu import DynamicSubmenuItem
+from .generator import generate_get_code_menu
+
+
+get_code_submenu = CursesMenu(
+    title='Sessions'
+)
+
+get_code_submenu_item = DynamicSubmenuItem(
+    text='Get confirmation code',
+    generator=generate_get_code_menu,
+    menu=menu
+)

modules/menu/fill.py

@@ -0,0 +1,7 @@
+from .menu import menu
+
+
+def fill_menu():
+    from .items.main import items_list
+    for item in items_list:
+        menu.items.append(item)

modules/menu/items/config/__init__.py

@@ -0,0 +1 @@
+from .submenu import config_submenu_item

modules/menu/items/config/fields.py

@@ -0,0 +1,20 @@
+from cursesmenu.items import FunctionItem
+from ...callbacks.config.callbacks import edit_config_callback
+
+
+items_list = [
+    FunctionItem('app_id', function=edit_config_callback,
+                 args=['app_id', 'Telegram app_id, can be obtained on my.telegram.org']),
+    FunctionItem('api_hash', function=edit_config_callback,
+                 args=['api_hash', 'Telegram api_hash, can be obtained on my.telegram.org']),
+    FunctionItem('device_model', function=edit_config_callback,
+                 args=['device_model', 'Device model is first line of information about login in devices settings. ',
+                       'Session Keeper']),
+    FunctionItem('app_version', function=edit_config_callback,
+                 args=['app_version', 'App version is second line of information about login in devices settings.',
+                       'DO NOT DEAUTH THIS SESSION']),
+    FunctionItem('system_version', function=edit_config_callback,
+                 args=['system_version',
+                       'App version is shown in detailed information about login in devices settings. ',
+                       'Bars\'s TelegramSessionKeeper']),
+]

modules/menu/items/config/submenu.py

@@ -0,0 +1,17 @@
+from ...menu import menu
+from cursesmenu import CursesMenu
+from cursesmenu.items import SubmenuItem
+from .fields import items_list
+
+
+config_submenu = CursesMenu(
+    title='Config'
+)
+for item in items_list:
+    config_submenu.items.append(item)
+
+config_submenu_item = SubmenuItem(
+    text='Edit config',
+    submenu=config_submenu,
+    menu=menu
+)

modules/menu/items/create_session/__init__.py

@@ -0,0 +1 @@
+from .fields import create_new_session_item

modules/menu/items/create_session/fields.py

@@ -0,0 +1,7 @@
+from cursesmenu.items import FunctionItem
+from ...callbacks.create_session.callbacks import create_session_callback
+
+
+create_new_session_item = FunctionItem(
+    'Add new session', function=create_session_callback
+)

modules/menu/items/main.py

@@ -0,0 +1,10 @@
+from .config import config_submenu_item
+from .create_session import create_new_session_item
+from ..dynamic.get_code import get_code_submenu_item
+
+
+items_list = [
+    create_new_session_item,
+    get_code_submenu_item,
+    config_submenu_item
+]

modules/menu/menu.py

@@ -0,0 +1,6 @@
+import cursesmenu
+
+
+menu = cursesmenu.CursesMenu(
+    title='SessionKeeper'
+)