swarm-debug 0.1.0__py3-none-any.whl

backend/core/models/project_scanner.py

@@ -0,0 +1,153 @@
+import os
+import json
+import colorsys
+from typing import Union
+from backend.core.models.Directory import Directory
+from backend.core.DEFAULTS import DEFAULT_COLOR, DEFAULT_TOGGLED, DEFAULT_SET_MANUALLY, DEFAULT_SET_MANUALLY_EMOJI, TOGGLE_FILE, DEFAULT_EMOJI
+from backend.core.models.DebugFile import DebugFile
+from collections import OrderedDict
+
+def merge_directories(json_dir: Directory, scanned_dir: Directory):
+    """
+    Merges two Directory instances: one loaded from JSON (json_dir) and one built from scanning (scanned_dir).
+    The values from json_dir take precedence where attributes overlap.
+    It matches based on full directory and file structure, not just file names.
+    """
+    # print(f"Merging JSON_DIR: {json_dir.path}\n with SCAN_DIR: {scanned_dir.path}")
+    json_abspaths, json_instances = json_dir.get_ordered_abspaths_and_instances()
+    # print(f"json_abspaths: {json_abspaths}")
+    scanned_abspaths, scanned_instances = scanned_dir.get_ordered_abspaths_and_instances()
+    # print(f"scanned_abspaths: {scanned_abspaths}")
+
+    def find_matching_in_structure(scanned_child: Union[DebugFile, Directory], json_dir: Directory):
+        assert json_dir in json_instances, f"JSON_DIR: {json_dir.path} not in json_instances"
+        assert scanned_child in scanned_instances, f"SCANNED_CHILD: {scanned_child.path} not in scanned_instances"
+        scanned_id = scanned_instances.index(scanned_child)
+        scanned_abspath = scanned_abspaths[scanned_id]
+        json_instance = None
+        try:
+            json_id = json_abspaths.index(scanned_abspath)
+            json_instance = json_instances[json_id]
+            # print(f"Match found: {scanned_child.path} == {json_instance.path}")
+        except ValueError:
+            # print(f"SCANNED_ABSPATH: {scanned_abspath} not in JSON_ABSPATHS")
+            pass
+        return json_instance
+
+    def construct_merged_dir(json_dir: Directory, scanned_dir: Directory):
+        for scanned_child in scanned_dir.children:
+            # Use the new recursive function to find the corresponding child in the JSON directory structure
+            matching_json_child = find_matching_in_structure(scanned_child, json_dir)
+            
+            if isinstance(scanned_child, DebugFile) and matching_json_child:
+                scanned_child.color = matching_json_child.color
+                scanned_child.is_toggled = matching_json_child.is_toggled
+                scanned_child.set_manually = matching_json_child.set_manually
+                scanned_child.set_manually_emoji = matching_json_child.set_manually_emoji
+                scanned_child.emoji = matching_json_child.emoji
+
+            elif isinstance(scanned_child, Directory) and matching_json_child:
+                scanned_child.color = matching_json_child.color
+                scanned_child.is_toggled = matching_json_child.is_toggled
+                scanned_child.set_manually = matching_json_child.set_manually
+                scanned_child.set_manually_emoji = matching_json_child.set_manually_emoji
+                scanned_child.emoji = matching_json_child.emoji
+
+                # Recursively merge the subdirectories
+                construct_merged_dir(matching_json_child, scanned_child)
+            else:
+                scanned_child.color = DEFAULT_COLOR
+                scanned_child.is_toggled = DEFAULT_TOGGLED
+                scanned_child.set_manually = DEFAULT_SET_MANUALLY
+                scanned_child.set_manually_emoji = DEFAULT_SET_MANUALLY_EMOJI
+                scanned_child.emoji = DEFAULT_EMOJI
+    
+    construct_merged_dir(json_dir, scanned_dir)
+
+
+def update_debug_toggles(save_to_file=True) -> Directory:
+    # print(f"[update_debug_toggles]: START")
+    json_loaded_dir = None
+    if os.path.exists(TOGGLE_FILE):
+        with open(TOGGLE_FILE, 'r', encoding='utf-8') as file:
+            try:
+                json_data = json.load(file)
+                json_loaded_dir = Directory("")
+                json_loaded_dir = Directory(path="",
+                                            color=json_data[0].get('color', DEFAULT_COLOR),
+                                            is_toggled=json_data[0].get('is_toggled', DEFAULT_TOGGLED),
+                                            set_manually=json_data[0].get('set_manually', DEFAULT_SET_MANUALLY),
+                                            set_manually_emoji=json_data[0].get('set_manually_emoji', DEFAULT_SET_MANUALLY_EMOJI),
+                                            emoji=json_data[0].get('emoji', DEFAULT_EMOJI)
+                                            )
+                # print(f"Root: {json_loaded_dir}")
+                # print("Json Children 1:")
+                # [print(child.path) for child in json_loaded_dir.children]
+
+                json_loaded_dir.load_from_json(json_data[0]['children'])  # Assuming the root is in json_data[0]
+                # print("Json Children 2:")
+                # [print(child.path) for child in json_loaded_dir.children]
+
+            except json.JSONDecodeError:
+                ValueError("Error: JSON file could not be decoded.")
+    else:
+        print("No JSON file found")
+    # 1. Create a directory structure from the filesystem scan
+    # print("Scanning directory...")
+    scanned_dir = Directory(path="", 
+                            color=json_loaded_dir.color if json_loaded_dir else DEFAULT_COLOR, 
+                            is_toggled=json_loaded_dir.is_toggled if json_loaded_dir else DEFAULT_TOGGLED, 
+                            set_manually=json_loaded_dir.set_manually if json_loaded_dir else DEFAULT_SET_MANUALLY,
+                            set_manually_emoji=json_loaded_dir.set_manually_emoji if json_loaded_dir else DEFAULT_SET_MANUALLY_EMOJI,
+                            emoji=json_loaded_dir.emoji if json_loaded_dir else DEFAULT_EMOJI
+                            )
+    # print(f"\n\nNum Children 1: {len(scanned_dir.children)}")
+    # [print(child.path) for child in scanned_dir.children]
+    scanned_dir.build_structure()
+    # print(f"\n\nNum Children 2: {len(scanned_dir.children)}")
+    # [print(child.path) for child in scanned_dir.children]
+    scanned_dir.prune_empty()
+    # print(f"\n\nNum Children 3: {len(scanned_dir.children)}")
+    # [print(child.path) for child in scanned_dir.children]
+
+    # print("1.1 Merged Dir First Child: ", scanned_dir.children[0])
+    # 4. Propagate the toggled state and color through the merged structure
+    scanned_dir.propagate_toggled_state()
+    # print(f"\n\nNum Children 4: {len(scanned_dir.children)}")
+    # [print(child.path) for child in scanned_dir.children]
+
+    # 3. Merge the two directory structures
+    if json_loaded_dir:
+        merge_directories(json_loaded_dir, scanned_dir)
+    
+    # print(f"\n\nNum Children 5: {len(scanned_dir.children)}")
+    scanned_dir.propagate_color()
+    output = dir_to_output_format(scanned_dir)
+    # print(f"\n\nNum Children 6: {len(scanned_dir.children)}")
+    
+
+    # 5. Write the updated structure back to the JSON file
+    if save_to_file:
+        with open(TOGGLE_FILE, 'w', encoding='utf-8') as file:
+             json.dump(output, file, ensure_ascii=False, indent=4)
+    # print(f"[update_debug_toggles]: END")
+    return scanned_dir
+
+def dir_to_output_format(input_dir):
+    root_node = {
+        "name": "root",
+        "color": input_dir.color,
+        "is_toggled": input_dir.is_toggled,
+        "set_manually": input_dir.set_manually,
+        "set_manually_emoji": input_dir.set_manually_emoji,
+        "emoji": input_dir.emoji,
+        "children": input_dir.to_dict()["children"]
+    }
+    return [ordered(root_node)]
+
+def ordered(obj):
+    if isinstance(obj, dict):
+        return OrderedDict((k, ordered(v)) for k, v in obj.items())
+    if isinstance(obj, list):
+        return [ordered(x) for x in obj]
+    return obj

backend/core/utils/__init__.py

@@ -0,0 +1,3 @@
+from backend.core.utils.color_adjuster import rgb_to_ansi, bold_and_italicize_text, hex_to_rgb
+from backend.core.utils.debug_arg_parser import is_text, is_error, is_fstring
+from backend.core.utils.path_mngr import get_abspath, get_root_rel_path

backend/core/utils/color_adjuster.py

@@ -0,0 +1,13 @@
+
+def rgb_to_ansi(rgb):
+    return '\033[38;2;{};{};{}m'.format(*rgb)
+
+def bold_and_italicize_text(text):
+    return f"\033[1m\033[3m{text}\033[0m"
+
+def hex_to_rgb(hex_code):
+    # Remove the '#' symbol if it exists
+    hex_code = hex_code.lstrip('#')
+    
+    # Convert the hex code to RGB
+    return tuple(int(hex_code[i:i+2], 16) for i in (0, 2, 4))

backend/core/utils/debug_arg_parser.py

@@ -0,0 +1,22 @@
+x = "y"
+
+def is_fstring(arg_name):
+    if not isinstance(arg_name, str):
+        return False
+    # print(f"arg_name: {arg_name}")
+    fstring_start_values = ["f'", "f\""]
+    num_start_matches = sum(arg_name.startswith(start_value) for start_value in fstring_start_values)
+    conditions = [num_start_matches == 1]
+    return all(conditions)
+
+def is_text(arg_value, arg_name):
+    arg_is_text = isinstance(arg_value, str) and len(arg_name) > 2 and arg_name[1:len(arg_name)-1] == arg_value and not arg_name.endswith(")")
+    if not arg_is_text:
+        arg_is_text = is_fstring(arg_name)
+    # print(f"is_text: {arg_is_text}")
+    return arg_is_text
+
+def is_error(arg_value, arg_name):
+    arg_is_error = isinstance(arg_value, Exception) or "error" in str(arg_value).lower() or "error" in str(arg_name).lower()
+    return arg_is_error
+

backend/core/utils/path_mngr.py

@@ -0,0 +1,15 @@
+import os
+from backend.core.DEFAULTS import get_root_dir
+
+
+def get_abspath(path: str):
+    return os.path.join(get_root_dir(), path)
+
+
+def get_root_rel_path(path: str):
+    root = get_root_dir()
+    assert path.startswith(root)
+    path = path[len(root):]
+    while path.startswith(os.sep):
+        path = path[1:]
+    return path