Ensure we treeshake and hydrate ASAP with aws-lite

architect · Apr 8, 2024 · c1f143a · c1f143a
1 parent db34cc8
commit c1f143a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/actions/autoinstall/index.js b/src/actions/autoinstall/index.js
@@ -7,6 +7,11 @@ module.exports = function autoinstaller (params) {
   let { dirs, inventory, update, verbose } = params
   if (!dirs.length) return []
 
+  let asap = inventory.inv.http?.find(l => l.arcStaticAssetProxy)
+  if (asap) {
+    dirs.push(asap.src)
+  }
+
   update.start('Finding dependencies')
 
   // Generated manifests to be hydrated later (if there are no parsing failures)
@@ -68,6 +73,10 @@ module.exports = function autoinstaller (params) {
 function getRuntimeDirs (dirs, inventory, runtimeName) {
   let runtimeDirs = dirs.filter(dir => {
     let lambda = inventory.inv.lambdasBySrcDir[dir]
+    if (!lambda) {
+      lambda = inventory.inv.http?.find(l => l.arcStaticAssetProxy)
+      if (!lambda) throw ReferenceError(`Cannot find Lambda at: ${dir}`)
+    }
     if (Array.isArray(lambda)) lambda = lambda[0] // Multi-tenant Lambda check
     let { runtime, hydrate } = lambda.config
     return runtime.startsWith(runtimeName) && hydrate !== false

diff --git a/src/actions/autoinstall/node/index.js b/src/actions/autoinstall/node/index.js
@@ -67,6 +67,10 @@ module.exports = function treeshakeNode (nodeDirs, params) {
   nodeDirs.forEach(dir => {
     projectDirs++
     let lambda = inventory.inv.lambdasBySrcDir[dir]
+    if (!lambda) {
+      lambda = inventory.inv.http?.find(l => l.arcStaticAssetProxy)
+      if (!lambda) throw ReferenceError(`Cannot find Lambda at: ${dir}`)
+    }
     if (Array.isArray(lambda)) lambda = lambda[0] // Multi-tenant Lambda check
     let { config, name, pragma } = lambda
     let { runtime } = config