feat(simulator): support sub-package resource definitions (#332)

jianzs · web-flow · commit 75e5a8973725 · 2024-08-22T13:59:52.000+08:00
Allow resource types to be defined within sub-packages, extending the previous limitation of only allowing definitions in the root package. This enhancement enables the simulator to process resources that are defined in sub-packages.
diff --git a/.changeset/new-items-stare.md b/.changeset/new-items-stare.md
@@ -0,0 +1,7 @@
+---
+"@plutolang/simulator-adapter": patch
+---
+
+feat(simulator): support sub-package resource definitions
+
+Allow resource types to be defined within sub-packages, extending the previous limitation of only allowing definitions in the root package. This enhancement enables the simulator to process resources that are defined in sub-packages.
diff --git a/components/adapters/simulator/src/simulator.ts b/components/adapters/simulator/src/simulator.ts
@@ -78,16 +78,24 @@ export class Simulator {
 
   private async createResource(resource: arch.Resource): Promise<simulator.IResourceInstance> {
     const resourceTypeFqn = resource.type;
-    const dotPos = resourceTypeFqn.lastIndexOf(".");
+
+    const dotPos = resourceTypeFqn.indexOf(".");
     const pkgName = resourceTypeFqn.substring(0, dotPos);
     const resourceType = resourceTypeFqn.substring(dotPos + 1);
-    // TODO: check if the package exists, and import from user project
+
     const infraPkg = (await import(resolvePkg(`${pkgName}-infra`))) as any;
-    const resourceInfraClass = infraPkg[resourceType];
-    if (!resourceInfraClass) {
-      throw new Error(
-        "Cannot find the infrastructure implementation class of the resource type " + resourceType
-      );
+    const importLevels = resourceType.split(".");
+
+    let resourceInfraClass: any;
+    for (let i = 0; i < importLevels.length; i++) {
+      resourceInfraClass = i == 0 ? infraPkg[importLevels[i]] : resourceInfraClass[importLevels[i]];
+
+      if (!resourceInfraClass) {
+        throw new Error(
+          "Cannot find the infrastructure implementation class of the resource type " +
+            resourceTypeFqn
+        );
+      }
     }
 
     const args = new Array(resource.arguments.length);
