massa_acheta/main.py at main · dex2code/massa_acheta · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
from loguru import logger
logger.add(
    "main.log",
    format="\"{time}\", \"{level}\", \"{file}:{line}\", \"{module}:{function}\", \"{message}\"",
    level="INFO",
    rotation="1 week",
    compression="zip",
    enqueue=True,
    backtrace=True,
    diagnose=True
)

import asyncio
from aiogram.utils.formatting import as_list
from aiogram.types import BotCommand
from pathlib import Path
from sys import exit as sys_exit

from app_config import app_config
import app_globals

from remotes.monitor import monitor as remote_monitor
from remotes.massa import massa as remote_massa
from remotes.heartbeat import heartbeat as remote_heartbeat

from telegram.queue import queue_telegram_message, operate_telegram_queue

from telegram.handlers import start
from telegram.handlers import cancel
from telegram.handlers import view_config, view_node, view_wallet, view_address, clean_address, view_credits, view_earnings, view_id, chart_wallet
from telegram.handlers import add_node, add_wallet
from telegram.handlers import delete_node, delete_wallet
from telegram.handlers import massa_info, massa_chart, acheta_release
from telegram.handlers import reset
from telegram.handlers import unknown

from tools import save_app_stat, save_public_dir


@logger.catch
async def main() -> None:
    logger.debug(f"-> Enter Def")

    public_obj = Path("public")
    if not public_obj.exists():
        logger.info(f"No file '{public_obj}' exists. Private menu applied.")
        bot_commands = [
            BotCommand(command="/help", description="Show help info"),
            BotCommand(command="/view_config", description="View service config"),
            BotCommand(command="/view_node", description="View node status"),
            BotCommand(command="/view_wallet", description="View wallet info"),
            BotCommand(command="/chart_wallet", description="View wallet chart"),
            BotCommand(command="/view_address", description="View any wallet info"),
            BotCommand(command="/clean_address", description="Clean remembered address"),
            BotCommand(command="/view_credits", description="View any wallet credits"),
            BotCommand(command="/view_earnings", description="View rewards for staking"),
            BotCommand(command="/add_node", description="Add node to bot"),
            BotCommand(command="/add_wallet", description="Add wallet to bot"),
            BotCommand(command="/delete_node", description="Delete node from bot"),
            BotCommand(command="/delete_wallet", description="Delete wallet from bot"),
            BotCommand(command="/massa_info", description="Show MASSA network info"),
            BotCommand(command="/massa_chart", description="Show MASSA network chart"),
            BotCommand(command="/acheta_release", description="Actual Acheta release"),
            BotCommand(command="/view_id", description="Show your TG ID"),
            BotCommand(command="/cancel", description="Cancel ongoing scenario"),
            BotCommand(command="/reset", description="Reset bot configuration")
        ]

    else:
        logger.info(f"File '{public_obj}' exists. Public menu applied.")
        bot_commands = [
            BotCommand(command="/help", description="Show help info"),
            BotCommand(command="/view_address", description="View any wallet info"),
            BotCommand(command="/clean_address", description="Clean remembered address"),
            BotCommand(command="/view_credits", description="View any wallet credits"),
            BotCommand(command="/view_earnings", description="View rewards for staking"),
            BotCommand(command="/massa_info", description="Show MASSA network info"),
            BotCommand(command="/massa_chart", description="Show MASSA network chart"),
            BotCommand(command="/view_id", description="Show your TG ID"),
            BotCommand(command="/cancel", description="Cancel ongoing scenario")
        ]

    await app_globals.tg_bot.set_my_commands(bot_commands)

    nodes_list = []

    if len(app_globals.app_results) == 0:
        nodes_list.append("\n⭕ Node list is empty")
    else:
        for node_name in app_globals.app_results:
            nodes_list.append(f"\n🏠 Node: \"{node_name}\"")
            nodes_list.append(f"📍 {app_globals.app_results[node_name]['url']}")
            nodes_list.append(f"👛 {len(app_globals.app_results[node_name]['wallets'])} wallet(s) attached")


    t = as_list(
        "🔆 Service successfully started to watch the following nodes:",
        *nodes_list, "",
        "👉 Try /help to learn bot commands", "",
        f"⏳ Main loop period: {app_config['service']['main_loop_period_min']} minutes",
        f"⚡ Probe timeout: {app_config['service']['http_probe_timeout_sec']} seconds"
    )
    await queue_telegram_message(message_text=t.as_html())

    try:
        asyncio.create_task(operate_telegram_queue())
        asyncio.create_task(remote_monitor())
        asyncio.create_task(remote_massa())
        asyncio.create_task(remote_heartbeat())


        app_globals.tg_dp.include_router(start.router)

        app_globals.tg_dp.include_router(cancel.router)

        app_globals.tg_dp.include_router(view_config.router)
        app_globals.tg_dp.include_router(view_node.router)
        app_globals.tg_dp.include_router(view_wallet.router)
        app_globals.tg_dp.include_router(chart_wallet.router)
        app_globals.tg_dp.include_router(view_address.router)
        app_globals.tg_dp.include_router(clean_address.router)
        app_globals.tg_dp.include_router(view_credits.router)
        app_globals.tg_dp.include_router(view_earnings.router)
        app_globals.tg_dp.include_router(view_id.router)

        app_globals.tg_dp.include_router(add_node.router)
        app_globals.tg_dp.include_router(add_wallet.router)

        app_globals.tg_dp.include_router(delete_node.router)
        app_globals.tg_dp.include_router(delete_wallet.router)

        app_globals.tg_dp.include_router(massa_info.router)
        app_globals.tg_dp.include_router(massa_chart.router)
        app_globals.tg_dp.include_router(acheta_release.router)

        app_globals.tg_dp.include_router(reset.router)

        app_globals.tg_dp.include_router(unknown.router)

        await app_globals.tg_bot.delete_webhook(drop_pending_updates=True)
        await app_globals.tg_dp.start_polling(app_globals.tg_bot)

    except BaseException as E:
        logger.error(f"Exception {str(E)} ({E})")

    finally:
        logger.error(f"<- Quit Def")

    return


if __name__ == "__main__":
    logger.info(f"MASSA Acheta started at {app_globals.acheta_start_time}")

    try:
        asyncio.run(main())

    except BaseException as E:
        logger.error(f"Exception {str(E)} ({E})")

    finally:
        save_app_stat()
        save_public_dir()

        logger.critical("Service terminated")
        sys_exit()