From b790a97500e6761990bdf536fdb110138cc6f266 Mon Sep 17 00:00:00 2001
From: SantaSpeen <santaspeen@gmail.com>
Date: Tue, 14 Nov 2023 01:21:35 +0300
Subject: [PATCH] Perms system

---
 config.json            | 21 -------------
 src/config.json        | 12 ++++++++
 src/help_message.txt   |  4 +++
 main.py => src/main.py | 70 +++++++++++++++++++++++++++++++-----------
 src/permissions.json   | 10 ++++++
 5 files changed, 78 insertions(+), 39 deletions(-)
 delete mode 100644 config.json
 create mode 100644 src/config.json
 create mode 100644 src/help_message.txt
 rename main.py => src/main.py (64%)
 create mode 100644 src/permissions.json

diff --git a/config.json b/config.json
deleted file mode 100644
index 48ce364..0000000
--- a/config.json
+++ /dev/null
@@ -1,21 +0,0 @@
-{
-  "rcon": {
-    "host": "",
-    "port": 0,
-    "password": ""
-  },
-  "vk": {
-    "token": "",
-    "admins": []
-  },
-  "permissions": {
-    "admin": {
-      "lvl": 100,
-      "allow": true
-    },
-    "helper": {
-      "lvl": 1,
-      "allow": ["say"]
-    }
-  }
-}
\ No newline at end of file
diff --git a/src/config.json b/src/config.json
new file mode 100644
index 0000000..fe8a888
--- /dev/null
+++ b/src/config.json
@@ -0,0 +1,12 @@
+{
+  "rcon": {
+    "host": "",
+    "port": 0,
+    "password": ""
+  },
+  "vk": {
+    "token": "",
+    "help_file": "help_message.txt"
+  },
+  "permission_file": "permissions.json"
+}
\ No newline at end of file
diff --git a/src/help_message.txt b/src/help_message.txt
new file mode 100644
index 0000000..f9be79d
--- /dev/null
+++ b/src/help_message.txt
@@ -0,0 +1,4 @@
+Тебе не нужна помощь, ты и так беспомощный, кожаный ублюдок. Так уж и быть, подскажу пару команд...
+!help - Вывести эту "справку"
+!online - Показать текущий онлайн сервере
+Бот сделан кожанным петухом - админом, все вопросы к нему, я не причём.
\ No newline at end of file
diff --git a/main.py b/src/main.py
similarity index 64%
rename from main.py
rename to src/main.py
index 89a1085..80064a8 100644
--- a/main.py
+++ b/src/main.py
@@ -16,17 +16,54 @@ 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']
-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 = {}
+
+    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
+
+
+perms = Permissions(config['permission_file'])
+
+
 def fix_rcon_text(_srt):
     try:
         _srt = list(_srt)
@@ -63,35 +100,32 @@ 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}")
+    a, r = perms.is_allow(from_id, cmd.split()[0])
+    if a:
+        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)
+    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]
diff --git a/src/permissions.json b/src/permissions.json
new file mode 100644
index 0000000..dfc6dde
--- /dev/null
+++ b/src/permissions.json
@@ -0,0 +1,10 @@
+{
+  "admin": {
+    "allow": true,
+    "members": [123]
+  },
+  "helper": {
+    "allow": ["say", "warn"],
+    "members": [111, 432]
+  }
+}
\ No newline at end of file