fix: handle interface with factory

Author: msp5382

.changeset/rude-beds-argue.md

@@ -0,0 +1,5 @@
+---
+"@softnetics/what-the-dep": minor
+---
+
+Reduce bundle size and bug fixes

src/walk/1-create-container.ts

@@ -3,7 +3,7 @@ import { DependencyGraph } from "./graph";
 import { globalTypeChecker, globalContext } from ".";
 import { listDependenciesOfClass } from "./2-list-dependency";
 import { hashSymbol, isWhatTheDepMethod } from "./utils";
-import { handleGet } from "./3-modify-get";
+import { getFactoryDependencies, handleGet } from "./3-modify-get";
 import { createContextFromGraph } from "./4-create-context";
 
 type ContainerRegisterMethod = "register" | "registerSingleton";
@@ -89,7 +89,7 @@ export const handleContainer = (
             if (registerMethod === "get") {
               // case 2 of get
               handleGet(node, transformList);
-              return node;
+              return ts.visitEachChild(node, visitor, globalContext);
             }
 
             if (!isContainerRegisterMethod(registerMethod)) {
@@ -116,10 +116,18 @@ export const handleContainer = (
                   `Could not find symbol of ${identifier.getText()}`
                 );
               }
-              const dependencies = listDependenciesOfClass(
-                OriginalClassSymbol,
-                identifier
-              );
+
+              let dependencies: string[] = [];
+
+              if (factoryNode) {
+                // dont care a class dependency and check a get of a factory
+                dependencies = getFactoryDependencies(factoryNode);
+              } else {
+                dependencies = listDependenciesOfClass(
+                  OriginalClassSymbol,
+                  identifier
+                );
+              }
 
               const graphRegisterMethod =
                 registerMethod === "register" ? "transient" : "singleton";

src/walk/3-modify-get.ts

@@ -1,6 +1,6 @@
 import ts from "typescript";
 import { DependencyGraph } from "./graph";
-import { globalTypeChecker } from ".";
+import { globalTypeChecker, globalContext } from ".";
 import { hashSymbol } from "./utils";
 
 export const handleGet = (
@@ -21,3 +21,33 @@ export const handleGet = (
   );
   transformList.set(node, newNode);
 };
+
+export const getFactoryDependencies = (factory: ts.Expression) => {
+  const dependencies: string[] = [];
+
+  let getSymbol: ts.Symbol | undefined;
+  const visitor = (node: ts.Node): ts.Node => {
+    if (!getSymbol && ts.isParameter(node)) {
+      getSymbol = globalTypeChecker.getTypeAtLocation(node).getSymbol();
+    }
+    if (ts.isCallExpression(node)) {
+      if (ts.isIdentifier(node.expression)) {
+        const symbol = globalTypeChecker
+          .getTypeAtLocation(node.expression)
+          .getSymbol();
+        if (symbol && symbol === getSymbol) {
+          const classOrInterface = node.typeArguments![0];
+          const hash = hashSymbol(
+            globalTypeChecker.getTypeAtLocation(classOrInterface).getSymbol()!
+          );
+          dependencies.push(hash);
+        }
+      }
+    }
+    return ts.visitEachChild(node, visitor, globalContext);
+  };
+
+  ts.visitNode(factory, visitor);
+
+  return dependencies;
+};

src/walk/4-create-context.ts

@@ -12,6 +12,7 @@ import {
 } from "./source-templates";
 
 export const createContextFromGraph = (graph: DependencyGraph) => {
+  graph.print();
   const properties = graph.topologicalSort().map((hash) => {
     const resolved = graph.resolve(hash);
     if (!resolved) {

src/walk/index.ts

@@ -15,7 +15,11 @@ export const transformerFactory = (
   return (context) => (rootNode) => {
     globalContext = context;
     const visitor = (node: ts.Node): ts.Node => {
-      //1. find `const ModuleName = new Container();`
+      // Find container initialization
+      // 1. const container = new Container();
+      // 2. const container = (new Container()).register();
+
+      // case 1
       if (
         ts.isVariableStatement(node) &&
         node.declarationList.declarations.length === 1
@@ -58,6 +62,45 @@ export const transformerFactory = (
           }
         }
       }
+
+      // case 2
+      if (
+        ts.isCallExpression(node) &&
+        ts.isPropertyAccessExpression(node.expression)
+      ) {
+        // dig into expression to find the identifier
+        let expressionChildren = node.expression.getChildren();
+        while (!ts.isIdentifier(expressionChildren[0])) {
+          if (ts.isParenthesizedExpression(expressionChildren[0])) {
+            // (new Container()).register()
+            if (ts.isNewExpression(expressionChildren[0].expression)) {
+              const containerNode =
+                expressionChildren[0].expression.getChildren()[1];
+              handleContainer(
+                rootNode,
+                containerNode,
+                globalTypeChecker.getSymbolAtLocation(containerNode)!,
+                program,
+                transformList
+              );
+            }
+          }
+          if (ts.isNewExpression(expressionChildren[0])) {
+            const containerNode = expressionChildren[0].getChildren()[1];
+            handleContainer(
+              rootNode,
+              containerNode,
+              globalTypeChecker.getSymbolAtLocation(containerNode)!,
+              program,
+              transformList
+            );
+          }
+          expressionChildren = expressionChildren[0].getChildren();
+          if (expressionChildren.length <= 2) {
+            break;
+          }
+        }
+      }
       return ts.visitEachChild(node, visitor, context);
     };
 

tsup.config.ts

@@ -1,7 +1,7 @@
 import { defineConfig } from "tsup";
 
 export default defineConfig({
-  entry: ["src/index.ts", "src/create.ts"],
+  entry: ["src/plugin.ts", "src/container.ts"],
   splitting: true,
   minify: true,
   clean: true,