1 files changed, 36 insertions, 30 deletions
diff --git a/src/openwrt_log_analyzer.py b/src/openwrt_log_analyzer.py
index c1c4fbe..96023cd 100755
--- a/src/openwrt_log_analyzer.py
+++ b/src/openwrt_log_analyzer.py
@@ -1,33 +1,39 @@
 #!/usr/bin/env python3
 import home.telegram as telegram
 
-from home.config import config
+from home.telegram.config import TelegramChatsConfig
+from home.util import validate_mac_address
+from typing import Optional
+from home.config import config, AppConfigUnit
 from home.database import BotsDatabase, SimpleState
 
-"""
-config.toml example:
 
-[simple_state]
-file = "/home/user/.config/openwrt_log_analyzer/state.txt"
-
-[mysql]
-host = "localhost"
-database = ".."
-user = ".."
-password = ".."
-
-[devices]
-Device1 = "00:00:00:00:00:00"
-Device2 = "01:01:01:01:01:01"
-
-[telegram]
-chat_id = ".."
-token = ".."
-parse_mode = "HTML"
-
-[openwrt_log_analyzer]
-limit = 10
-"""
+class OpenwrtLogAnalyzerConfig(AppConfigUnit):
+    @classmethod
+    def schema(cls) -> Optional[dict]:
+        return {
+            'database_name': {'type': 'string', 'required': True},
+            'devices': {
+                'type': 'dict',
+                'keysrules': {'type': 'string'},
+                'valuesrules': {
+                    'type': 'string',
+                    'check_with': validate_mac_address
+                }
+            },
+            'limit': {'type': 'integer'},
+            'telegram_chat': {'type': 'string'},
+            'aps': {
+                'type': 'list',
+                'schema': {'type': 'integer'}
+            }
+        }
+
+    @staticmethod
+    def custom_validator(data):
+        chats = TelegramChatsConfig()
+        if data['telegram_chat'] not in chats:
+            return ValueError(f'unknown telegram chat {data["telegram_chat"]}')
 
 
 def main(mac: str,
@@ -48,18 +54,18 @@ def main(mac: str,
             max_id = log.id
 
     text = '\n'.join(map(lambda s: str(s), data))
-    telegram.send_message(f'<b>{title} (AP #{ap})</b>\n\n' + text)
+    telegram.send_message(f'<b>{title} (AP #{ap})</b>\n\n' + text, config.app_config['telegram_chat'])
 
     return max_id
 
 
 if __name__ == '__main__':
-    config.load_app('openwrt_log_analyzer')
-    for ap in config['openwrt_log_analyzer']['aps']:
-        state_file = config['simple_state']['file']
-        state_file = state_file.replace('.txt', f'-{ap}.txt')
+    config.load_app(OpenwrtLogAnalyzerConfig)
+    for ap in config.app_config['aps']:
+        dbname = config.app_config['database_name']
+        dbname = dbname.replace('.txt', f'-{ap}.txt')
 
-        state = SimpleState(name=state_file,
+        state = SimpleState(name=dbname,
                             default={'last_id': 0})
 
         max_last_id = 0