Initial commit

bot.py

@@ -0,0 +1,130 @@
+# -*- coding: utf-8 -*-
+import os
+import subprocess
+
+from sys import exit
+from modules.functions import *
+from modules.functions_bot import *
+from pyrogram import Client, filters, idle
+from pyrogram.types import ForceReply, BotCommand, BotCommandScopeChat
+from pyrogram.enums.chat_action import ChatAction
+
+if configGet("bot_token") != "12345678:asdfghjklzxcvbnm":
+    pid = os.getpid()
+    app = Client("auto_zoom_public_bot", api_id=configGet("api_id"), api_hash=configGet("api_hash"), bot_token=configGet("bot_token"))
+else:
+    logWrite("Could not start the bot. Please, configure token in config.json")
+    exit()
+
+def botSend(userid, message="Test message"):
+    app = Client("auto_zoom_public_bot")
+    app.send_message(userid, message)
+
+
+@app.on_message(~ filters.scheduled & filters.command(["link", "start"], prefixes="/"))
+def start(app, msg):
+
+    logWrite(f'Got command start/link from {msg.from_user.id}')
+
+    app.send_chat_action(chat_id=msg.chat.id, action=ChatAction.TYPING)
+
+    if f"{msg.from_user.id}.json" not in os.listdir("data/users/"):
+        logWrite(f'Creating blank data file for {msg.from_user.id}')
+        jsonSave( f"data/users/{msg.from_user.id}.json", {"api_key": None, "linked": False, "context": {"action": None, "data": None}} )
+
+    if not userGet(msg.from_user.id, "linked"):
+        msg.reply_text(locale("link_input", "msg"), reply_markup=ForceReply(placeholder=locale("link", "fry")))
+        userSet(msg.chat.id, "context", "link_key")
+    else:
+        msg.reply_text(locale("already_linked", "msg"))
+
+@app.on_message(~ filters.scheduled & filters.command(["unlink"], prefixes="/"))
+def unlink(app, msg):
+
+    logWrite(f'Got command ulink from {msg.from_user.id}')
+
+    app.send_chat_action(chat_id=msg.chat.id, action=ChatAction.TYPING)
+
+    if not userGet(msg.from_user.id, "linked"):
+        msg.reply_text(locale("not_linked", "msg"))
+    else:
+        try:
+            keys_storage = jsonLoad("data/keys_storage.json")
+            del keys_storage[userGet(msg.from_user.id, "api_key")]
+            jsonSave("data/keys_storage.json", keys_storage)
+        except:
+            pass
+        userClear(msg.from_user.id, "api_key")
+        userSet(msg.chat.id, "linked", False)
+        msg.reply_text(locale("unlinked", "msg"))
+
+@app.on_message(~ filters.scheduled & filters.command(["cancel"], prefixes="/"))
+def cancel(app, msg):
+
+    app.send_chat_action(chat_id=msg.chat.id, action=ChatAction.TYPING)
+
+    if userGet(msg.from_user.id, "context") is not None:
+        userClear(msg.from_user.id, "context")
+        userClear(msg.from_user.id, "context_content")
+        msg.reply_text(locale("cancel", "msg"))
+    else:
+        msg.reply_text(locale("cancel_empty", "msg"))
+
+
+@app.on_message(filters.command(["kill", "die", "shutdown", "reboot"], prefixes="/"))
+def kill(app, msg):
+
+    if msg.from_user.id == configGet("admin"):
+        msg.reply_text(f"Shutting down bot with pid `{pid}`")
+        os.system('kill -9 '+str(pid))
+
+
+@app.on_message(~ filters.scheduled)
+def any_message_handler(app, msg):
+
+    if userGet(msg.from_user.id, "context") == "link_key":
+
+        if msg.text in jsonLoad(configGet("api_keys"))["autozoom"]:
+            msg.reply_text(locale("key_correct", "msg"))
+            userSet(msg.from_user.id, "api_key", msg.text)
+            userSet(msg.from_user.id, "linked", True)
+            keys_storage = jsonLoad("data/keys_storage.json")
+            keys_storage[msg.text] = msg.from_user.id
+            jsonSave("data/keys_storage.json", keys_storage)
+            logWrite(f"Added apikey {msg.text} for user {msg.from_user.id}")
+        else:
+            logWrite(f"User {msg.from_user.id} tried to pair with invalid apikey {msg.text}")
+            msg.reply_text(locale("key_wrong", "msg"))
+
+        userClear(msg.from_user.id, "context")
+        userClear(msg.from_user.id, "context_content")
+
+
+if __name__ == "__main__":
+
+    logWrite(f'Starting with PID {str(pid)}')
+
+    app.start()
+    app.send_message(configGet("admin"), f"Starting bot with pid `{pid}`")
+    
+    all_commands = locale("cmd")
+
+    commands_list = []
+    for command in all_commands["general"]:
+        commands_list.append(BotCommand(command, all_commands["general"][command]))
+    app.set_bot_commands(commands_list)
+
+    # Registering admin commands
+    commands_admin_list = []
+    for command in all_commands["general"]:
+        commands_admin_list.append(BotCommand(command, all_commands["general"][command]))
+    for command in all_commands["admin"]:
+        commands_admin_list.append(BotCommand(command, all_commands["admin"][command]))
+    app.set_bot_commands(commands_admin_list, scope=BotCommandScopeChat(chat_id=configGet("admin")))
+
+    idle()
+
+    app.send_message(configGet("admin"), f"Shutting down bot with pid `{pid}`")
+    logWrite(f'Shutting down with PID {pid}')
+
+    subprocess.call(f'kill -9 {pid}', shell=True)