From 55a891f94c041cc0af54eab345259f4052b16952 Mon Sep 17 00:00:00 2001
From: John Lancaster <32917998+jsl12@users.noreply.github.com>
Date: Thu, 8 Aug 2024 00:10:12 -0500
Subject: [PATCH] dependency manager prototype

---
 dependencies/__init__.py |  0
 dependencies/manager.py  | 80 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 80 insertions(+)
 create mode 100644 dependencies/__init__.py
 create mode 100644 dependencies/manager.py

diff --git a/dependencies/__init__.py b/dependencies/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/dependencies/manager.py b/dependencies/manager.py
new file mode 100644
index 0000000..2a4eea4
--- /dev/null
+++ b/dependencies/manager.py
@@ -0,0 +1,80 @@
+from abc import ABC
+from dataclasses import dataclass, field
+from pathlib import Path
+from typing import Dict, Iterable, Self, Set
+
+import appdaemon.dependency as deps
+from appdaemon.models.app_config import AllAppConfig, AppConfig
+from appdaemon.models.internal.file_check import FileCheck
+
+
+@dataclass
+class Dependencies(ABC):
+    files: FileCheck = field(repr=False)
+    ext: str = field(init=False)
+    dep_graph: Dict[str, Set[str]] = field(init=False)
+    rev_graph: Dict[str, Set[str]] = field(init=False)
+
+    def __post_init__(self):
+        self.rev_graph = deps.reverse_graph(self.dep_graph)
+
+    @classmethod
+    def from_path(cls: Self, path: Path):
+        return cls.from_paths(path.rglob(f'*{cls.ext}'))
+
+    @classmethod
+    def from_paths(cls: Self, paths: Iterable[Path]):
+        return cls(files=FileCheck.from_paths(paths))
+
+    def get_dependents(self, items: Iterable[str]) -> Set[str]:
+        items = items if isinstance(items, set) else set(items)
+        items |= deps.find_all_dependents(items, self.rev_graph)
+        return items
+
+
+@dataclass
+class PythonDeps(Dependencies):
+    ext: str = '.py'
+
+    def __post_init__(self):
+        self.dep_graph = deps.get_dependency_graph(self.files)
+        super().__post_init__()
+
+
+@dataclass
+class AppDeps(Dependencies):
+    app_config: AllAppConfig = field(init=False, repr=False)
+    ext: str = '.yaml'
+
+    def __post_init__(self):
+        self.app_config = AllAppConfig.from_paths(self.files)
+        self.dep_graph = self.app_config.depedency_graph()
+        super().__post_init__()
+
+    def direct_app_deps(self, modules: Iterable[str]):
+        """Find the apps that directly depend on any of the given modules"""
+        return set(
+            app_name
+            for app_name, app_cfg in self.app_config.root.items()
+            if isinstance(app_cfg, AppConfig) and app_cfg.module_name in modules
+        )
+
+    def cascade_modules(self, modules: Iterable[str]) -> Set[str]:
+        """Find all the apps that depend on the given modules, even indirectly"""
+        return self.get_dependents(self.direct_app_deps(modules))
+
+
+@dataclass
+class DependencyManager:
+    app_dir: Path
+    python_deps: PythonDeps = field(init=False)
+    app_deps: AppDeps = field(init=False)
+
+    def __post_init__(self):
+        self.python_deps = PythonDeps.from_path(self.app_dir)
+        self.app_deps = AppDeps.from_path(self.app_dir)
+
+    def get_dependent_apps(self, modules: Iterable[str]) -> Set[str]:
+        """Finds all of the apps that depend on the given modules, even indirectly"""
+        modules |= self.python_deps.get_dependents(modules)
+        return self.app_deps.cascade_modules(modules)