From 2b664589dc79a55f362ccb7c714097d6e2cc3e70 Mon Sep 17 00:00:00 2001
From: Sam Hartman <hartmans@debian.org>
Date: Fri, 25 Aug 2023 13:57:03 -0600
Subject: [PATCH] Entangle FailedInstantiations

Send out objects for all failed instantiations.
---
 .../dependency_injection/introspection.py     |  5 +++
 carthage/entanglement/instrumentation.py      | 35 +++++++++++++++++--
 2 files changed, 38 insertions(+), 2 deletions(-)

diff --git a/carthage/dependency_injection/introspection.py b/carthage/dependency_injection/introspection.py
index 0ec78d5d..2e3e87b4 100644
--- a/carthage/dependency_injection/introspection.py
+++ b/carthage/dependency_injection/introspection.py
@@ -323,6 +323,11 @@ def failed_instantiation(context, exception):
     all_instantiation_failures[id(injector), key] = failure
     if len(dependency) == 0:
         failed_instantiation_leaves[id(injector), key] = failure
+    injector.emit_event(
+        key, "dependency_instantiation_failed",
+        failure,
+        adl_keys={base.InjectionKey(base.Injector)})
+    
 
 __all__ += ['failed_instantiation']
 
diff --git a/carthage/entanglement/instrumentation.py b/carthage/entanglement/instrumentation.py
index 7a6cf0c0..cd670401 100644
--- a/carthage/entanglement/instrumentation.py
+++ b/carthage/entanglement/instrumentation.py
@@ -11,6 +11,7 @@
 import dataclasses
 import enum
 from pathlib import Path
+import traceback
 import uuid
 import yaml
 
@@ -124,7 +125,14 @@ async def on_task_event(self, event, target, task, **kwargs):
                     should_run=should_run,
                     exception=exception,
                     running=running)
-            
+
+    def on_instantiation_failed(self, target, scope, target_key, **kwargs):
+        failed_instantiation = self.get_or_create(FailedInstantiation,
+                                                  id(scope), target_key)
+        failed_instantiation.exception = "".join(traceback.format_exception(target.exception))
+        failed_instantiation.dependency = target.dependency
+        self.store_synchronize(failed_instantiation)
+        
     def dump(self, path):
         path = Path(path)
         result: dict[str, list] = {}
@@ -135,7 +143,7 @@ def dump(self, path):
                 sync_repr['_sync_type'] = o.sync_type
                 objects.append(sync_repr)
             result[t.sync_type] = objects
-        path.write_text(yaml.dump(result, default_flow_style=False))
+        path.write_text(yaml.dump(result, default_flow_style=False ))
         
 
     def instrument_injector(self, injector):
@@ -146,6 +154,8 @@ def instrument_injector(self, injector):
             InjectionKey(SetupTaskMixin),
             {'task_ran', 'task_start', 'task_fail', 'task_should_run'},
             self.on_task_event)
+        injector.add_event_listener(
+            InjectionKey(Injector), {'dependency_instantiation_failed'}, self.on_instantiation_failed)
         for k, inspector in injector.inspect():
             self.on_add_provider(scope=injector, target_key=inspector.key, other_keys=set(),
                                  inspector=inspector
@@ -223,3 +233,24 @@ class TaskInfo(StoreInSyncStoreMixin):
     sync_registry = carthage_registry
     
 __all__ += ['TaskInfo']
+
+def encode_fi_dependency(val):
+    return [encode_injection_key(x) for x in val]
+
+def decode_fi_dependency(val):
+    return [decode_injection_key(x) for x in val]
+
+@dataclasses.dataclass
+class FailedInstantiation(StoreInSyncStoreMixin):
+
+    injector_id: int = sync_property(constructor=True)
+    key: InjectionKey = sync_property(constructor=True)
+    exception: str = sync_property("")
+    dependency: list[InjectionKey] = sync_property(
+        dataclasses.field(default_factory=lambda: []),
+        encoder=encode_fi_dependency,
+        decoder=decode_fi_dependency)
+    sync_primary_keys = ('injector_id', 'key')
+    sync_registry = carthage_registry
+
+__all__ += ['FailedInstantiation']