@teamkeel/functions-runtime 0.407.0 → 0.408.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@teamkeel/functions-runtime",
-  "version": "0.407.0",
+  "version": "0.408.0",
   "description": "Internal package used by @teamkeel/sdk",
   "main": "src/index.js",
   "scripts": {
@@ -16,7 +16,7 @@
   },
   "devDependencies": {
     "prettier": "3.1.1",
-    "vitest": "^3.0.7"
+    "vitest": "3.0.8"
   },
   "dependencies": {
     "@aws-sdk/client-s3": "~3.722.0",

package/src/handleRoute.js

@@ -0,0 +1,112 @@
+const {
+  createJSONRPCErrorResponse,
+  createJSONRPCSuccessResponse,
+  JSONRPCErrorCode,
+} = require("json-rpc-2.0");
+const { createDatabaseClient, withDatabase } = require("./database");
+const { withAuditContext } = require("./auditing");
+const { errorToJSONRPCResponse, RuntimeErrors } = require("./errors");
+const opentelemetry = require("@opentelemetry/api");
+const { withSpan } = require("./tracing");
+
+async function handleRoute(request, config) {
+  // Try to extract trace context from caller
+  const activeContext = opentelemetry.propagation.extract(
+    opentelemetry.context.active(),
+    request.meta?.tracing
+  );
+
+  // Run the whole request with the extracted context
+  return opentelemetry.context.with(activeContext, () => {
+    // Wrapping span for the whole request
+    return withSpan(request.method, async (span) => {
+      let db = null;
+
+      try {
+        const { createContextAPI, functions } = config;
+
+        if (!(request.method in functions)) {
+          const message = `no route function found for '${request.method}'`;
+          span.setStatus({
+            code: opentelemetry.SpanStatusCode.ERROR,
+            message: message,
+          });
+          return createJSONRPCErrorResponse(
+            request.id,
+            JSONRPCErrorCode.MethodNotFound,
+            message
+          );
+        }
+
+        // For route functions context doesn't need request headers or the response object as this is handled by
+        // params and the function response respectively
+        const {
+          headers,
+          response: __,
+          ...ctx
+        } = createContextAPI({
+          responseHeaders: new Headers(),
+          meta: request.meta,
+        });
+
+        // Add request headers to params
+        request.params.headers = headers;
+
+        db = createDatabaseClient({
+          connString: request.meta?.secrets?.KEEL_DB_CONN,
+        });
+        const routeHandler = functions[request.method];
+
+        const result = await withDatabase(db, false, () => {
+          return withAuditContext(request, () => {
+            return routeHandler(request.params, ctx);
+          });
+        });
+
+        if (result instanceof Error) {
+          span.recordException(result);
+          span.setStatus({
+            code: opentelemetry.SpanStatusCode.ERROR,
+            message: result.message,
+          });
+          return errorToJSONRPCResponse(request, result);
+        }
+
+        const response = createJSONRPCSuccessResponse(request.id, result);
+
+        return response;
+      } catch (e) {
+        if (e instanceof Error) {
+          span.recordException(e);
+          span.setStatus({
+            code: opentelemetry.SpanStatusCode.ERROR,
+            message: e.message,
+          });
+          return errorToJSONRPCResponse(request, e);
+        }
+
+        const message = JSON.stringify(e);
+
+        span.setStatus({
+          code: opentelemetry.SpanStatusCode.ERROR,
+          message: message,
+        });
+
+        return createJSONRPCErrorResponse(
+          request.id,
+          RuntimeErrors.UnknownError,
+          message
+        );
+      } finally {
+        if (db) {
+          await db.destroy();
+        }
+      }
+    });
+  });
+}
+
+module.exports = {
+  handleRoute,
+  RuntimeErrors,
+};

package/src/index.js

@@ -3,6 +3,7 @@ const { RequestHeaders } = require("./RequestHeaders");
 const { handleRequest } = require("./handleRequest");
 const { handleJob } = require("./handleJob");
 const { handleSubscriber } = require("./handleSubscriber");
+const { handleRoute } = require("./handleRoute");
 const KSUID = require("ksuid");
 const { useDatabase } = require("./database");
 const {
@@ -21,6 +22,7 @@ module.exports = {
   handleRequest,
   handleJob,
   handleSubscriber,
+  handleRoute,
   useDatabase,
   Duration,
   InlineFile,