@celerity-sdk/cli 0.3.1 → 0.4.0

package/dist/extract/cli.js

@@ -35,10 +35,12 @@ function buildScannedModule(rootModule) {
   const graph = buildModuleGraph(rootModule);
   const controllerClasses = [];
   const functionHandlers = [];
+  const guardClasses = [];
+  const functionGuards = [];
   const providers = [];
   const seenTokens = /* @__PURE__ */ new Set();
   for (const [moduleClass, node] of graph) {
-    debug("scan: module %s \u2014 %d providers, %d controllers", moduleClass.name, node.providers.length, node.controllers.length);
+    debug("scan: module %s \u2014 %d providers, %d controllers, %d guards", moduleClass.name, node.providers.length, node.controllers.length, node.guards.length);
     for (const provider of node.providers) {
       const scanned = scanProvider(provider, seenTokens);
       if (scanned) providers.push(scanned);
@@ -54,11 +56,28 @@ function buildScannedModule(rootModule) {
         });
       }
     }
+    for (const guard of node.guards) {
+      if (typeof guard === "function") {
+        guardClasses.push(guard);
+        if (!seenTokens.has(guard)) {
+          seenTokens.add(guard);
+          providers.push({
+            token: guard,
+            providerType: "class",
+            dependencies: getClassDependencyTokens(guard)
+          });
+        }
+      } else {
+        functionGuards.push(guard);
+      }
+    }
     functionHandlers.push(...node.functionHandlers);
   }
   return {
     controllerClasses,
     functionHandlers,
+    guardClasses,
+    functionGuards,
     providers
   };
 }
@@ -102,7 +121,7 @@ __name(serializeToken, "serializeToken");
 
 // src/extract/serializer.ts
 import "reflect-metadata";
-import { CONTROLLER_METADATA, HTTP_METHOD_METADATA, ROUTE_PATH_METADATA, GUARD_PROTECTEDBY_METADATA, GUARD_CUSTOM_METADATA, PUBLIC_METADATA, CUSTOM_METADATA } from "@celerity-sdk/core";
+import { CONTROLLER_METADATA, HTTP_METHOD_METADATA, ROUTE_PATH_METADATA, WEBSOCKET_CONTROLLER_METADATA, WEBSOCKET_EVENT_METADATA, CONSUMER_METADATA, CONSUMER_HANDLER_METADATA, SCHEDULE_HANDLER_METADATA, INVOKE_METADATA, GUARD_PROTECTEDBY_METADATA, GUARD_CUSTOM_METADATA, PUBLIC_METADATA, CUSTOM_METADATA, USE_RESOURCE_METADATA } from "@celerity-sdk/core";
 
 // src/extract/path-utils.ts
 import { joinHandlerPath } from "@celerity-sdk/common";
@@ -143,8 +162,9 @@ __name(deriveCodeLocation, "deriveCodeLocation");
 function serializeManifest(scanned, sourceFile, options) {
   const handlers = [];
   const functionHandlers = [];
+  const guardHandlers = [];
   for (const controllerClass of scanned.controllerClasses) {
-    const entries = serializeClassHandler(controllerClass, sourceFile, options);
+    const entries = serializeClassHandlers(controllerClass, sourceFile, options);
     handlers.push(...entries);
   }
   for (const fnHandler of scanned.functionHandlers) {
@@ -153,40 +173,76 @@ function serializeManifest(scanned, sourceFile, options) {
       functionHandlers.push(entry);
     }
   }
+  for (const guardClass of scanned.guardClasses) {
+    const entry = serializeClassGuard(guardClass, sourceFile, options);
+    if (entry) {
+      guardHandlers.push(entry);
+    }
+  }
+  for (const fnGuard of scanned.functionGuards) {
+    const entry = serializeFunctionGuard(fnGuard, sourceFile, options);
+    if (entry) {
+      guardHandlers.push(entry);
+    }
+  }
   return {
     version: "1.0.0",
     handlers,
     functionHandlers,
+    guardHandlers,
     dependencyGraph: serializeDependencyGraph(scanned)
   };
 }
 __name(serializeManifest, "serializeManifest");
-function extractClassMeta(controllerClass) {
-  const controllerMeta = Reflect.getOwnMetadata(CONTROLLER_METADATA, controllerClass);
-  if (!controllerMeta) return null;
+function extractControllerMeta(controllerClass) {
+  const httpMeta = Reflect.getOwnMetadata(CONTROLLER_METADATA, controllerClass);
+  if (httpMeta) {
+    return {
+      controllerType: "http",
+      prefix: httpMeta.prefix ?? "",
+      ...extractSharedClassMeta(controllerClass)
+    };
+  }
+  const isWebSocket = Reflect.getOwnMetadata(WEBSOCKET_CONTROLLER_METADATA, controllerClass);
+  if (isWebSocket) {
+    return {
+      controllerType: "websocket",
+      prefix: "",
+      ...extractSharedClassMeta(controllerClass)
+    };
+  }
+  const consumerMeta = Reflect.getOwnMetadata(CONSUMER_METADATA, controllerClass);
+  if (consumerMeta) {
+    return {
+      controllerType: "consumer",
+      prefix: "",
+      sourceId: consumerMeta.sourceId,
+      ...extractSharedClassMeta(controllerClass)
+    };
+  }
+  return null;
+}
+__name(extractControllerMeta, "extractControllerMeta");
+function extractSharedClassMeta(controllerClass) {
   return {
-    prefix: controllerMeta.prefix ?? "",
     protectedBy: Reflect.getOwnMetadata(GUARD_PROTECTEDBY_METADATA, controllerClass) ?? [],
     customGuardName: Reflect.getOwnMetadata(GUARD_CUSTOM_METADATA, controllerClass),
-    customMetadata: Reflect.getOwnMetadata(CUSTOM_METADATA, controllerClass) ?? {}
+    customMetadata: Reflect.getOwnMetadata(CUSTOM_METADATA, controllerClass) ?? {},
+    resourceRefs: Reflect.getOwnMetadata(USE_RESOURCE_METADATA, controllerClass) ?? []
   };
 }
-__name(extractClassMeta, "extractClassMeta");
-function buildMethodAnnotations(classMeta, prototype, methodName, httpMethod, fullPath) {
-  const annotations = {};
-  annotations["celerity.handler.http"] = true;
-  annotations["celerity.handler.http.method"] = httpMethod;
-  annotations["celerity.handler.http.path"] = fullPath;
+__name(extractSharedClassMeta, "extractSharedClassMeta");
+function appendSharedAnnotations(annotations, meta, prototype, methodName) {
   const methodProtectedBy = Reflect.getOwnMetadata(GUARD_PROTECTEDBY_METADATA, prototype, methodName) ?? [];
   const allProtectedBy = [
-    ...classMeta.protectedBy,
+    ...meta.protectedBy,
     ...methodProtectedBy
   ];
   if (allProtectedBy.length > 0) {
     annotations["celerity.handler.guard.protectedBy"] = allProtectedBy;
   }
-  if (classMeta.customGuardName) {
-    annotations["celerity.handler.guard.custom"] = classMeta.customGuardName;
+  if (meta.customGuardName) {
+    annotations["celerity.handler.guard.custom"] = meta.customGuardName;
   }
   const isPublic = Reflect.getOwnMetadata(PUBLIC_METADATA, prototype, methodName) === true;
   if (isPublic) {
@@ -194,57 +250,206 @@ function buildMethodAnnotations(classMeta, prototype, methodName, httpMethod, fu
   }
   const methodCustomMetadata = Reflect.getOwnMetadata(CUSTOM_METADATA, prototype, methodName) ?? {};
   const customMetadata = {
-    ...classMeta.customMetadata,
+    ...meta.customMetadata,
     ...methodCustomMetadata
   };
   for (const [key, value] of Object.entries(customMetadata)) {
     if (value === void 0) continue;
     annotations[`celerity.handler.metadata.${key}`] = serializeAnnotationValue(value);
   }
+  const methodResourceRefs = Reflect.getOwnMetadata(USE_RESOURCE_METADATA, prototype, methodName) ?? [];
+  const allResourceRefs = [
+    .../* @__PURE__ */ new Set([
+      ...meta.resourceRefs,
+      ...methodResourceRefs
+    ])
+  ];
+  if (allResourceRefs.length > 0) {
+    annotations["celerity.handler.resource.ref"] = allResourceRefs;
+  }
+}
+__name(appendSharedAnnotations, "appendSharedAnnotations");
+function buildHttpAnnotations(meta, prototype, methodName, httpMethod, fullPath) {
+  const annotations = {};
+  annotations["celerity.handler.http"] = true;
+  annotations["celerity.handler.http.method"] = httpMethod;
+  annotations["celerity.handler.http.path"] = fullPath;
+  appendSharedAnnotations(annotations, meta, prototype, methodName);
+  return annotations;
+}
+__name(buildHttpAnnotations, "buildHttpAnnotations");
+function buildWebSocketAnnotations(meta, prototype, methodName, wsEvent) {
+  const annotations = {};
+  annotations["celerity.handler.websocket"] = true;
+  annotations["celerity.handler.websocket.route"] = wsEvent.route;
+  annotations["celerity.handler.websocket.eventType"] = wsEvent.eventType;
+  appendSharedAnnotations(annotations, meta, prototype, methodName);
+  return annotations;
+}
+__name(buildWebSocketAnnotations, "buildWebSocketAnnotations");
+function buildConsumerAnnotations(meta, prototype, methodName, consumerHandler) {
+  const annotations = {};
+  annotations["celerity.handler.consumer"] = true;
+  if (meta.sourceId) {
+    annotations["celerity.handler.consumer.sourceId"] = meta.sourceId;
+  }
+  if (consumerHandler.route) {
+    annotations["celerity.handler.consumer.route"] = consumerHandler.route;
+  }
+  appendSharedAnnotations(annotations, meta, prototype, methodName);
+  return annotations;
+}
+__name(buildConsumerAnnotations, "buildConsumerAnnotations");
+function buildScheduleAnnotations(meta, prototype, methodName, scheduleMeta) {
+  const annotations = {};
+  annotations["celerity.handler.schedule"] = true;
+  if (scheduleMeta.scheduleId) {
+    annotations["celerity.handler.schedule.scheduleId"] = scheduleMeta.scheduleId;
+  }
+  if (scheduleMeta.schedule) {
+    annotations["celerity.handler.schedule.expression"] = scheduleMeta.schedule;
+  }
+  appendSharedAnnotations(annotations, meta, prototype, methodName);
+  return annotations;
+}
+__name(buildScheduleAnnotations, "buildScheduleAnnotations");
+function buildCustomAnnotations(meta, prototype, methodName, invokeMeta) {
+  const annotations = {};
+  annotations["celerity.handler.custom"] = true;
+  annotations["celerity.handler.custom.name"] = invokeMeta.name;
+  appendSharedAnnotations(annotations, meta, prototype, methodName);
   return annotations;
 }
-__name(buildMethodAnnotations, "buildMethodAnnotations");
-function serializeClassHandler(controllerClass, sourceFile, options) {
-  const classMeta = extractClassMeta(controllerClass);
-  if (!classMeta) return [];
+__name(buildCustomAnnotations, "buildCustomAnnotations");
+function serializeClassHandlers(controllerClass, sourceFile, options) {
+  const meta = extractControllerMeta(controllerClass);
+  if (!meta) return [];
   const className = controllerClass.name;
   const prototype = controllerClass.prototype;
   const methods = Object.getOwnPropertyNames(prototype).filter((n) => n !== "constructor");
   const entries = [];
   for (const methodName of methods) {
-    const httpMethod = Reflect.getOwnMetadata(HTTP_METHOD_METADATA, prototype, methodName);
-    if (!httpMethod) continue;
-    const routePath = Reflect.getOwnMetadata(ROUTE_PATH_METADATA, prototype, methodName) ?? "/";
-    const fullPath = joinHandlerPath(classMeta.prefix, routePath);
-    const annotations = buildMethodAnnotations(classMeta, prototype, methodName, httpMethod, fullPath);
-    entries.push({
-      resourceName: deriveClassResourceName(className, methodName),
-      className,
-      methodName,
-      sourceFile,
-      handlerType: "http",
-      annotations,
-      spec: {
-        handlerName: deriveClassHandlerName(className, methodName),
-        codeLocation: deriveCodeLocation(sourceFile, options.projectRoot),
-        handler: deriveClassHandlerFunction(sourceFile, className, methodName)
-      }
-    });
+    const typeEntry = serializeControllerTypeMethod(meta, className, prototype, methodName, sourceFile, options);
+    if (typeEntry) entries.push(typeEntry);
+    const scheduleMeta = Reflect.getOwnMetadata(SCHEDULE_HANDLER_METADATA, prototype, methodName);
+    if (scheduleMeta) {
+      entries.push({
+        resourceName: deriveClassResourceName(className, methodName),
+        className,
+        methodName,
+        sourceFile,
+        handlerType: "schedule",
+        annotations: buildScheduleAnnotations(meta, prototype, methodName, scheduleMeta),
+        spec: {
+          handlerName: deriveClassHandlerName(className, methodName),
+          codeLocation: deriveCodeLocation(sourceFile, options.projectRoot),
+          handler: deriveClassHandlerFunction(sourceFile, className, methodName)
+        }
+      });
+    }
+    const invokeMeta = Reflect.getOwnMetadata(INVOKE_METADATA, prototype, methodName);
+    if (invokeMeta) {
+      entries.push({
+        resourceName: deriveClassResourceName(className, methodName),
+        className,
+        methodName,
+        sourceFile,
+        handlerType: "custom",
+        annotations: buildCustomAnnotations(meta, prototype, methodName, invokeMeta),
+        spec: {
+          handlerName: deriveClassHandlerName(className, methodName),
+          codeLocation: deriveCodeLocation(sourceFile, options.projectRoot),
+          handler: deriveClassHandlerFunction(sourceFile, className, methodName)
+        }
+      });
+    }
   }
   return entries;
 }
-__name(serializeClassHandler, "serializeClassHandler");
+__name(serializeClassHandlers, "serializeClassHandlers");
+function serializeControllerTypeMethod(meta, className, prototype, methodName, sourceFile, options) {
+  switch (meta.controllerType) {
+    case "http":
+      return serializeHttpMethod(meta, className, prototype, methodName, sourceFile, options);
+    case "websocket":
+      return serializeWebSocketMethod(meta, className, prototype, methodName, sourceFile, options);
+    case "consumer":
+      return serializeConsumerMethod(meta, className, prototype, methodName, sourceFile, options);
+    default:
+      return null;
+  }
+}
+__name(serializeControllerTypeMethod, "serializeControllerTypeMethod");
+function serializeHttpMethod(meta, className, prototype, methodName, sourceFile, options) {
+  const httpMethod = Reflect.getOwnMetadata(HTTP_METHOD_METADATA, prototype, methodName);
+  if (!httpMethod) return null;
+  const routePath = Reflect.getOwnMetadata(ROUTE_PATH_METADATA, prototype, methodName) ?? "/";
+  const fullPath = joinHandlerPath(meta.prefix, routePath);
+  return {
+    resourceName: deriveClassResourceName(className, methodName),
+    className,
+    methodName,
+    sourceFile,
+    handlerType: "http",
+    annotations: buildHttpAnnotations(meta, prototype, methodName, httpMethod, fullPath),
+    spec: {
+      handlerName: deriveClassHandlerName(className, methodName),
+      codeLocation: deriveCodeLocation(sourceFile, options.projectRoot),
+      handler: deriveClassHandlerFunction(sourceFile, className, methodName)
+    }
+  };
+}
+__name(serializeHttpMethod, "serializeHttpMethod");
+function serializeWebSocketMethod(meta, className, prototype, methodName, sourceFile, options) {
+  const wsEvent = Reflect.getOwnMetadata(WEBSOCKET_EVENT_METADATA, prototype, methodName);
+  if (!wsEvent) return null;
+  return {
+    resourceName: deriveClassResourceName(className, methodName),
+    className,
+    methodName,
+    sourceFile,
+    handlerType: "websocket",
+    annotations: buildWebSocketAnnotations(meta, prototype, methodName, wsEvent),
+    spec: {
+      handlerName: deriveClassHandlerName(className, methodName),
+      codeLocation: deriveCodeLocation(sourceFile, options.projectRoot),
+      handler: deriveClassHandlerFunction(sourceFile, className, methodName)
+    }
+  };
+}
+__name(serializeWebSocketMethod, "serializeWebSocketMethod");
+function serializeConsumerMethod(meta, className, prototype, methodName, sourceFile, options) {
+  const consumerHandler = Reflect.getOwnMetadata(CONSUMER_HANDLER_METADATA, prototype, methodName);
+  if (!consumerHandler) return null;
+  return {
+    resourceName: deriveClassResourceName(className, methodName),
+    className,
+    methodName,
+    sourceFile,
+    handlerType: "consumer",
+    annotations: buildConsumerAnnotations(meta, prototype, methodName, consumerHandler),
+    spec: {
+      handlerName: deriveClassHandlerName(className, methodName),
+      codeLocation: deriveCodeLocation(sourceFile, options.projectRoot),
+      handler: deriveClassHandlerFunction(sourceFile, className, methodName)
+    }
+  };
+}
+__name(serializeConsumerMethod, "serializeConsumerMethod");
 function serializeFunctionHandler(definition, sourceFile, options) {
+  const supported = [
+    "http",
+    "websocket",
+    "consumer",
+    "schedule",
+    "custom"
+  ];
+  if (!supported.includes(definition.type)) return null;
   const exportName = definition.metadata.handlerName ?? "handler";
   const customMetadata = definition.metadata.customMetadata ?? {};
+  const handlerType = definition.type;
   const annotations = {};
-  const path = definition.metadata.path;
-  const method = definition.metadata.method;
-  if (path !== void 0 && method !== void 0) {
-    annotations["celerity.handler.http"] = true;
-    annotations["celerity.handler.http.method"] = method;
-    annotations["celerity.handler.http.path"] = path;
-  }
+  buildFunctionTypeAnnotations(annotations, definition);
   for (const [key, value] of Object.entries(customMetadata)) {
     if (value === void 0) continue;
     annotations[`celerity.handler.metadata.${key}`] = serializeAnnotationValue(value);
@@ -253,6 +458,7 @@ function serializeFunctionHandler(definition, sourceFile, options) {
     resourceName: deriveFunctionResourceName(exportName),
     exportName,
     sourceFile,
+    handlerType,
     ...Object.keys(annotations).length > 0 ? {
       annotations
     } : {},
@@ -264,6 +470,122 @@ function serializeFunctionHandler(definition, sourceFile, options) {
   };
 }
 __name(serializeFunctionHandler, "serializeFunctionHandler");
+function buildFunctionTypeAnnotations(annotations, definition) {
+  const meta = definition.metadata;
+  switch (definition.type) {
+    case "http": {
+      const path = meta.path;
+      const method = meta.method;
+      if (path !== void 0 && method !== void 0) {
+        annotations["celerity.handler.http"] = true;
+        annotations["celerity.handler.http.method"] = method;
+        annotations["celerity.handler.http.path"] = path;
+      }
+      break;
+    }
+    case "websocket": {
+      annotations["celerity.handler.websocket"] = true;
+      const route = meta.route;
+      if (route) {
+        annotations["celerity.handler.websocket.route"] = route;
+      }
+      break;
+    }
+    case "consumer": {
+      annotations["celerity.handler.consumer"] = true;
+      const route = meta.route;
+      if (route) {
+        annotations["celerity.handler.consumer.route"] = route;
+      }
+      break;
+    }
+    case "schedule": {
+      annotations["celerity.handler.schedule"] = true;
+      const scheduleId = meta.scheduleId;
+      if (scheduleId) {
+        annotations["celerity.handler.schedule.scheduleId"] = scheduleId;
+      }
+      const schedule = meta.schedule;
+      if (schedule) {
+        annotations["celerity.handler.schedule.expression"] = schedule;
+      }
+      break;
+    }
+    case "custom": {
+      annotations["celerity.handler.custom"] = true;
+      const name = meta.name;
+      if (name) {
+        annotations["celerity.handler.custom.name"] = name;
+      }
+      break;
+    }
+  }
+}
+__name(buildFunctionTypeAnnotations, "buildFunctionTypeAnnotations");
+function extractGuardMeta(guardClass) {
+  const guardName = Reflect.getOwnMetadata(GUARD_CUSTOM_METADATA, guardClass);
+  if (!guardName) return null;
+  return {
+    guardName,
+    customMetadata: Reflect.getOwnMetadata(CUSTOM_METADATA, guardClass) ?? {}
+  };
+}
+__name(extractGuardMeta, "extractGuardMeta");
+function serializeClassGuard(guardClass, sourceFile, options) {
+  const meta = extractGuardMeta(guardClass);
+  if (!meta) return null;
+  const className = guardClass.name;
+  const methodName = "check";
+  const annotations = {
+    "celerity.handler.guard.custom": meta.guardName
+  };
+  for (const [key, value] of Object.entries(meta.customMetadata)) {
+    if (value === void 0) continue;
+    annotations[`celerity.handler.metadata.${key}`] = serializeAnnotationValue(value);
+  }
+  return {
+    resourceName: deriveClassResourceName(className, methodName),
+    guardName: meta.guardName,
+    sourceFile,
+    guardType: "class",
+    className,
+    annotations,
+    spec: {
+      handlerName: deriveClassHandlerName(className, methodName),
+      codeLocation: deriveCodeLocation(sourceFile, options.projectRoot),
+      handler: deriveClassHandlerFunction(sourceFile, className, methodName)
+    }
+  };
+}
+__name(serializeClassGuard, "serializeClassGuard");
+function serializeFunctionGuard(definition, sourceFile, options) {
+  const guardName = definition.name;
+  if (!guardName) return null;
+  const meta = definition.metadata ?? {};
+  const customMetadata = meta.customMetadata ?? {};
+  const annotations = {
+    "celerity.handler.guard.custom": guardName
+  };
+  for (const [key, value] of Object.entries(customMetadata)) {
+    if (value === void 0) continue;
+    annotations[`celerity.handler.metadata.${key}`] = serializeAnnotationValue(value);
+  }
+  const exportName = guardName;
+  return {
+    resourceName: deriveFunctionResourceName(exportName),
+    guardName,
+    sourceFile,
+    guardType: "function",
+    exportName,
+    annotations,
+    spec: {
+      handlerName: exportName,
+      codeLocation: deriveCodeLocation(sourceFile, options.projectRoot),
+      handler: deriveFunctionHandlerFunction(sourceFile, exportName)
+    }
+  };
+}
+__name(serializeFunctionGuard, "serializeFunctionGuard");
 function serializeAnnotationValue(value) {
   if (typeof value === "boolean") return value;
   if (typeof value === "string") return value;