From 2d46fcf95548ec3ef69ace0cc70aa98b387e28a5 Mon Sep 17 00:00:00 2001
From: Loic Guegan <manzerbredes@mailbox.org>
Date: Thu, 26 Oct 2023 17:50:34 +0200
Subject: [PATCH] Minor changes

---
 clusterman/config.py | 20 +++++++++++++++++++-
 pyproject.toml       |  2 +-
 2 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/clusterman/config.py b/clusterman/config.py
index 645f4d1..94294be 100644
--- a/clusterman/config.py
+++ b/clusterman/config.py
@@ -1,5 +1,6 @@
 from pathlib import Path
-import os, json
+import os, json, sys
+from jsonschema import validate
 
 
 class Config:
@@ -16,6 +17,18 @@ class Config:
         "plugins": { "ls": "ls -al" },
         "timeout": 0.5
     }
+    SCHEMA_CONFIG = {
+        "type": "object",
+        "properties": {
+            "timeout": {"type": "number"},
+            "plugins": {"type": "object"},
+            "cluster": {"type": "object", "properties":{
+                "ip4_from": {"type": "string"},
+                "ip4_to": {"type": "string"},
+                "ip4_ignore": {"type": "array", "items":{"type": "string"}}
+            }}
+        }
+    }
     
     def __init__(self):
         Path(self.CONF_DIR).mkdir(parents=True, exist_ok=True)
@@ -27,6 +40,11 @@ class Config:
         if os.path.exists(self.CONF_FILE):
             with open(self.CONF_FILE) as f:
                 self.config=json.load(f)
+                try:
+                    validate(instance=self.config, schema=self.SCHEMA_CONFIG)
+                except:
+                    print("Invalid configuration file")
+                    sys.exit(1)
         else:
             self.save()
             
diff --git a/pyproject.toml b/pyproject.toml
index d18be07..a565d86 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "clusterman"
 dynamic = ["version"]
-dependencies = ["psutil"]
+dependencies = ["psutil", "jsonschema"]
 description = "Simple cluster management utility"
 readme = "README.md"