del none

.idea/Rcon-VK-Bot.iml

@@ -1,8 +1,11 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <module type="PYTHON_MODULE" version="4">
   <component name="NewModuleRootManager">
-    <content url="file://$MODULE_DIR$" />
-    <orderEntry type="jdk" jdkName="Python 3.10" jdkType="Python SDK" />
+    <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
+      <excludeFolder url="file://$MODULE_DIR$/venv" />
+    </content>
+    <orderEntry type="jdk" jdkName="Python 3.10 (Rcon-VK-Bot)" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
 </module>

.idea/misc.xml

@@ -1,4 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10" project-jdk-type="Python SDK" />
+  <component name="Black">
+    <option name="sdkName" value="Python 3.10 (Rcon-VK-Bot)" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10 (Rcon-VK-Bot)" project-jdk-type="Python SDK" />
 </project>

README.md

@@ -4,13 +4,32 @@
 
 ## Что умеет: 
 
-* Доступные тольно админам
+* Доступные разрешённым людям
   * **.rcon <*commnd*>** - Исполняет <*commnd*> и отсылает ответ сервера
-  * **.wl <*nick*>** - По сути ссылка на whitelist add <*nick*>
 * Доступные всем
   * **!help** - Выводит станичку с командами (строки 99-103 main.py)
   * **!online** - Парсит rcon cmd *list* и выводит онлайн сервера
 
+
+## Система permissions
+
+В файле permissions.json указаны все пользователи с "повышенным" уровнем доступа к боту
+Пример
+```json
+{
+  "<role>": {
+    "allow": ["<command_1>", "<command_2>"],
+    "members": [123]
+  },
+  "helper": {
+    "allow": ["say", "warn"],
+    "members": [111, 432]
+  }
+}
+```
+
+Если `"allow": true` - разрешены все команды
+
 ## Как запустить?
 
 1. Должен установлен быть Python3 (Если не установлен, не лезь, попроси друга айтишника помочь)

src/config.json

@@ -6,7 +6,7 @@
   },
   "vk": {
     "token": "",
-    "admins": [],
-    "stoplist": []
-  }
+    "help_file": "help_message.txt"
+  },
+  "permission_file": "permissions.json"
 }

src/help_message.txt

@@ -0,0 +1,4 @@
+Тебе не нужна помощь, ты и так беспомощный, кожаный ублюдок. Так уж и быть, подскажу пару команд...
+!help - Вывести эту "справку"
+!online - Показать текущий онлайн сервере
+Бот сделан кожанным петухом - админом, все вопросы к нему, я не причём.

src/main.py

@@ -16,18 +16,61 @@ log("Starting..")
 with open('config-t.json' if "-t" in sys.argv else 'config.json') as f:
     config = json.load(f)
 
+with open('help_message.txt') as f:
+    help_message = f.read()
+
 vk = vk.API(access_token=config['vk']['token'], v=5.131)
 group_id = vk.groups.getById()[0]['id']
 log(f"Group id: {group_id}")
-admins = config['vk']['admins']
-fake_admins = []
-stoplist = config['vk']['stoplist']
 
 host = config['rcon']['host']
 port = config['rcon']['port']
 password = config['rcon']['password']
 
 
+class Permissions:
+
+    def __init__(self, permission_file):
+        self.permission_file = permission_file
+        self._raw_file = {}
+        self._members = {}
+        self._parse_file()
+
+    def _parse_file(self):
+        with open(self.permission_file) as pf:
+            self._raw_file = json.load(pf)
+
+        for role, role_data in self._raw_file.items():
+            members = role_data.get("members", [])
+            allow = role_data.get("allow", [])
+            for member in members:
+                self._members[member] = {
+                    "role": role,
+                    "allow": allow
+                }
+
+    def is_allow(self, vk_id, cmd):
+        u = self._members.get(vk_id)
+        if u is not None:
+            role = u['role']
+            allow = u['allow']
+            if allow is True:
+                return True, role
+            elif cmd in allow:
+                return True, role
+            return False, role
+        return False, None
+
+    def get_role(self, vk_id):
+        u = self._members.get(vk_id)
+        if u is not None:
+            return u['role']
+        return None
+
+
+perms = Permissions(config['permission_file'])
+
+
 def fix_rcon_text(_srt):
     try:
         _srt = list(_srt)
@@ -64,52 +107,41 @@ def write(peer_id, message):
             if i > 30:
                 log("Found very long message...", 1)
                 break
-            write(peer_id, message[:4095*i])
+            write(peer_id, message[:4095 * i])
     else:
         vk.messages.send(message=message, peer_id=peer_id, random_id=0)
 
 
-def rcon_cmd_handle(cmd, from_id, peer_id, _write=True):
-    answer = rcon(cmd)
-    log(f"User: {from_id} in Chat: {peer_id} use RCON cmd: \"{cmd}\", with answer: \"{answer}\"")
-    if _write:
-        write(peer_id, f"RCON:\n{answer}")
+def rcon_cmd_handle(cmd, from_id, peer_id, _write=True, _allow=False):
+    a, r = perms.is_allow(from_id, cmd.split()[0])
+    if _allow:
+        r = cmd
+    if a or _allow:
+        answer = rcon(cmd)
+        log(f"User: {from_id}({r}) in Chat: {peer_id} use RCON cmd: \"{cmd}\", with answer: \"{answer}\"")
+        if _write:
+            write(peer_id, f"RCON:\n{answer}")
+        else:
+            return answer
     else:
-        return answer
+        log(f"User: {from_id}({r}) in Chat: {peer_id} no have rights RCON cmd: \"{cmd}\".")
 
 
 def message_handle(message):
     from_id = message['from_id']
     peer_id = message['peer_id']
     text = message['text']
-    if from_id in admins:
-        if text.startswith(".rcon "):
-            rcon_cmd_handle(text[6:], from_id, peer_id)
-        elif text.startswith(".wl "):
-            rcon_cmd_handle(f'whitelist add {text[4:]}', from_id, peer_id)
-        elif text.startswith(".add "):
-            try:
-                i = int(text[5:])
-                fake_admins.append(i)
-                write(peer_id, f"Админ успешно добавлен: @id{i}")
-            except Exception as e:
-                write(peer_id, f"Ошибка добавления: {e}")
-    if from_id in fake_admins:
-        if text.startswith(".rcon "):
-            if text[5:].split()[0] in stoplist:
-                write(peer_id, "У вас нет доступа к этой команде")
-            else:
-                rcon_cmd_handle(text[6:], from_id, peer_id, False)
+    if text.startswith(".rcon "):
+        rcon_cmd_handle(text[6:], from_id, peer_id)
     if text == "!help":
-        write(peer_id, "Тебе не нужна помощь, ты и так беспомощный, кожаный ублюдок."
-                       "Так уж и быть, подскажу пару команд..\n\n"
-                       "!help - Вывести эту \"справку\"\n"
-                       "!online - Показать текущий онлайн сервере\n\n"
-                       "Бот сделан кожанным петухом - админом, все вопросы к нему, я не причём.")
+        write(peer_id, help_message)
     elif text == "!online":
-        text = rcon_cmd_handle('list', from_id, peer_id, False).replace("\n", "")
-        now = text[10:11]
-        write(peer_id, f"Сейчас играет {now} человек" + ("" if now == "0" else f": {text[43:]}"))
+        text = rcon_cmd_handle('list', from_id, peer_id, True, True).replace("\n", "")
+        write(peer_id, text)
+        # now = text[10:11]
+        # write(peer_id, f"Сейчас играет {now} человек" + ("" if now == "0" else f": {text[43:]}"))
+    elif text == "!id":
+        write(peer_id, f"Твой ID: {from_id}\nРоль в боте: {perms.get_role(from_id) or 'Отсутствует'}")
 
 
 def listen():

src/permissions.json

@@ -0,0 +1,10 @@
+{
+  "admin": {
+    "allow": true,
+    "members": [123]
+  },
+  "helper": {
+    "allow": ["say", "warn"],
+    "members": [111, 432]
+  }
+}