From aca7db07f9ff95adf1572d746861c7740a9754df Mon Sep 17 00:00:00 2001
From: "Eshan Roy (Eshanized)" <m.eshanized@gmail.com>
Date: Tue, 23 Apr 2024 12:41:15 +0530
Subject: [PATCH] =?UTF-8?q?=E2=8F=B3=20@eshanized=20updated=20the=20reposi?=
 =?UTF-8?q?tory!!!?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 blackbox/Functions.py | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/blackbox/Functions.py b/blackbox/Functions.py
index 49d8b11..92a5ed2 100644
--- a/blackbox/Functions.py
+++ b/blackbox/Functions.py
@@ -1,7 +1,12 @@
 #!/bin/python
 
 import os
+from os import makedirs
+import sys
 import gi
+import subprocess
+# import logging
+from logging import Logger
 
 gi.require_version("Gtk" "3.0") # GTK 2.0 is dead!
 from gi.repository import GLib, Gtk
@@ -47,3 +52,40 @@ event_log_file = "%s/event.log" % log_dir
 export_dir = "%s/blackbox-exports" % home
 
 # NOTE: Permissions specified here
+
+def permissions(dst):
+    try:
+        # NOTE : Use try-catch block so that we can trace any error!
+        groups = subprocess.run(
+            ["sh", "-c", "id " + sudo_username],
+            shell=False,
+            stdout=subprocess.PIPE, # NOTE: Standard Output
+            stderr=subprocess.STDOUT, # NOTE: Standard Error Output
+        )
+        for i in groups.stdout.decode().split(" "):
+            if "gid" in i:
+                g = i.split("(")[1]
+                group = g.replace(")", "").strip() # NOTE: replace with nothing!
+        subprocess.call(["chown", "-R", sudo_username + ":" + group, dst], shell=False)
+    except Exception as e:
+        Logger.error(e)
+
+# NOTE: Creating Log, Export and Config Directory:
+try:
+    if not os.path.exists(log_dir):
+        makedirs(log_dir)
+    if not os.path.exists(export_dir):
+        makedirs(export_dir)
+    if not os.path.exists(config_dir):
+        makedirs(config_dir)
+    
+    permissions(export_dir)
+    permissions(config_dir)
+
+    print("[INFO] Log Directory: %s" % log_dir)
+    print("[INFO] Export Directory: %s" % export_dir)
+    print("[INFO] Config Directory: %s" % config_dir)
+
+except os.error as oserror:
+    print("[ERROR] Exception: %s" % oserror)
+    sys.exit(1)
\ No newline at end of file