feat: Make injection tokens display more pretty in dependency graphing

ogre-works · Feb 21, 2022 · dc55d12 · dc55d12
1 parent 1fbdf74
commit dc55d12
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/...injectable/src/dependency-injection-container/createContainer.dependency-graphing.test.js b/...injectable/src/dependency-injection-container/createContainer.dependency-graphing.test.js
@@ -55,6 +55,7 @@ describe('createContainer.dependency-graph', () => {
         '@startuml',
         '"Setup(some-setuppable)" ..up* "some-child-injectable" : Setup',
         '"some-child-injectable" ..up* "some-injection-token" : Setup',
+        'class "some-injection-token" << (T,#FF7700) >>',
         '"some-injection-token" ..up* "some-token-injectable" : Setup',
         '"Setup(some-setuppable)" ..up* "some-setuppable" : Setup',
         '"some-parent-injectable" --up* "some-child-injectable"',

diff --git a/packages/injectable/src/dependency-injection-container/createContainer.js b/packages/injectable/src/dependency-injection-container/createContainer.js
@@ -76,7 +76,10 @@ export default (...listOfGetRequireContexts) => {
         find({ id: injectionToken.id }),
       );
 
-      const newContext = [...oldContext, { id: injectionToken.id }];
+      const newContext = [
+        ...oldContext,
+        { id: injectionToken.id, isInjectionToken: true },
+      ];
 
       if (injectableCausingCycle) {
         throw new Error(

diff --git a/.../src/dependency-injection-container/extensions/dependency-graphing/dependency-graphing.js b/.../src/dependency-injection-container/extensions/dependency-graphing/dependency-graphing.js
@@ -37,6 +37,10 @@ const plantUmlExtractorInjectable = getInjectable({
 
     return ({ context }) => {
       context.reduce((parent, dependency) => {
+        if (parent.isInjectionToken) {
+          plantUmlState.add(`class "${parent.id}" << (T,#FF7700) >>`);
+        }
+
         if (parent.isChildOfSetup === true) {
           plantUmlState.add(`"${parent.id}" ..up* "${dependency.id}" : Setup`);