@digitraffic/common 2024.3.22-1 → 2024.3.22-2

package/dist/__test__/api/static-integration.test.d.mts

@@ -0,0 +1 @@
+export {};

package/dist/__test__/api/static-integration.test.mjs

@@ -0,0 +1,26 @@
+import { DigitrafficStaticIntegration } from "../../aws/infra/api/static-integration.mjs";
+import { MediaType } from "../../aws/types/mediatypes.mjs";
+describe("response tests", () => {
+    it("createIntegrationResponse works", () => {
+        const integrationResponse = DigitrafficStaticIntegration.createIntegrationResponse("FakeResource", MediaType.APPLICATION_JSON, { "test-header": "test-value" });
+        expect(integrationResponse).toEqual({
+            responseParameters: {
+                "method.response.header.test-header": "'test-value'"
+            },
+            responseTemplates: {
+                "application/json": "FakeResource"
+            },
+            statusCode: "200"
+        });
+    });
+    it("createMethodResponse works", () => {
+        const methodResponse = DigitrafficStaticIntegration.createMethodResponse({ "test-header": "test-value" });
+        expect(methodResponse).toEqual({
+            responseParameters: {
+                "method.response.header.test-header": true
+            },
+            statusCode: "200"
+        });
+    });
+});
+//# sourceMappingURL=static-integration.test.mjs.map

package/dist/aws/infra/api/static-integration.d.mts

@@ -8,8 +8,17 @@ import { MediaType } from "../../types/mediatypes.mjs";
  * @param response
  */
 export declare class DigitrafficStaticIntegration extends MockIntegration {
-    constructor(resource: Resource, mediaType: MediaType, response: string, enableCors?: boolean, apiKeyRequired?: boolean);
-    static json<K>(resource: Resource, response: K, enableCors?: boolean, apiKeyRequired?: boolean): DigitrafficStaticIntegration;
-    private static createIntegrationResponse;
-    private static createMethodResponse;
+    constructor(resource: Resource, mediaType: MediaType, response: string, enableCors?: boolean, apiKeyRequired?: boolean, headers?: Record<string, string>);
+    static json<K>(resource: Resource, response: K, enableCors?: boolean, apiKeyRequired?: boolean, headers?: Record<string, string>): DigitrafficStaticIntegration;
+    static createIntegrationResponse(response: string, mediaType: MediaType, headers?: Record<string, string>): {
+        statusCode: string;
+        responseTemplates: {
+            [x: string]: string;
+        };
+        responseParameters: Record<string, string>;
+    };
+    static createMethodResponse(headers: Record<string, string>): {
+        statusCode: string;
+        responseParameters: Record<string, boolean>;
+    };
 }

package/dist/aws/infra/api/static-integration.mjs

@@ -1,12 +1,8 @@
-import { MockIntegration, PassthroughBehavior, Resource, } from "aws-cdk-lib/aws-apigateway";
+import { MockIntegration, PassthroughBehavior, Resource } from "aws-cdk-lib/aws-apigateway";
 import { MediaType } from "../../types/mediatypes.mjs";
-import { RESPONSE_CORS_INTEGRATION } from "./responses.mjs";
 const INTEGRATION_RESPONSE_200 = `{
     "statusCode": 200
 }`;
-const METHOD_RESPONSE_200 = {
-    statusCode: "200",
-};
 /**
  * Static integration, that returns the given response with given mediaType from given resource.
  *
@@ -15,8 +11,11 @@ const METHOD_RESPONSE_200 = {
  * @param response
  */
 export class DigitrafficStaticIntegration extends MockIntegration {
-    constructor(resource, mediaType, response, enableCors = true, apiKeyRequired = true) {
-        const integrationResponse = DigitrafficStaticIntegration.createIntegrationResponse(response, mediaType, enableCors);
+    constructor(resource, mediaType, response, enableCors = true, apiKeyRequired = true, headers = {}) {
+        if (enableCors) {
+            headers = { ...headers, "Access-Control-Allow-Origin": "*" };
+        }
+        const integrationResponse = DigitrafficStaticIntegration.createIntegrationResponse(response, mediaType, headers);
         super({
             passthroughBehavior: PassthroughBehavior.WHEN_NO_TEMPLATES,
             requestTemplates: {
@@ -28,39 +27,44 @@ export class DigitrafficStaticIntegration extends MockIntegration {
             resource.addMethod(httpMethod, this, {
                 apiKeyRequired,
                 methodResponses: [
-                    DigitrafficStaticIntegration.createMethodResponse(enableCors),
+                    DigitrafficStaticIntegration.createMethodResponse(headers),
                 ],
             });
         });
     }
-    static json(resource, response, enableCors = true, apiKeyRequired = true) {
-        return new DigitrafficStaticIntegration(resource, MediaType.APPLICATION_JSON, JSON.stringify(response), enableCors, apiKeyRequired);
+    static json(resource, response, enableCors = true, apiKeyRequired = true, headers = {}) {
+        return new DigitrafficStaticIntegration(resource, MediaType.APPLICATION_JSON, JSON.stringify(response), enableCors, apiKeyRequired, headers);
     }
-    static createIntegrationResponse(response, mediaType, enableCors) {
-        const integrationResponse = {
+    static createIntegrationResponse(response, mediaType, headers = {}) {
+        const params = mapRecord(headers, (entry) => ["method.response.header." + entry[0], `'${entry[1]}'`]);
+        return {
             statusCode: "200",
             responseTemplates: {
                 [mediaType]: response,
             },
+            responseParameters: params
         };
-        return enableCors
-            ? { ...integrationResponse, ...RESPONSE_CORS_INTEGRATION }
-            : integrationResponse;
     }
-    static createMethodResponse(enableCors) {
-        return enableCors
-            ? corsMethod(METHOD_RESPONSE_200)
-            : METHOD_RESPONSE_200;
+    static createMethodResponse(headers) {
+        const allowedHeaders = Object.keys(headers);
+        const entries = Object.fromEntries(allowedHeaders.map((key) => [key, true]));
+        return {
+            statusCode: "200",
+            responseParameters: prefixKeys("method.response.header.", entries)
+        };
     }
 }
-function corsMethod(response) {
-    return {
-        ...response,
-        ...{
-            responseParameters: {
-                "method.response.header.Access-Control-Allow-Origin": true,
-            },
-        },
-    };
+function mapRecord(obj, func) {
+    const mappedEntries = Object.entries(obj).map((entry) => func(entry));
+    return Object.fromEntries(mappedEntries);
+}
+/**
+ * Create a new Record with prefix added to each of the keys.
+ *
+ * @param prefix
+ * @param obj
+ */
+function prefixKeys(prefix, obj) {
+    return mapRecord(obj, (entry) => [prefix + entry[0], entry[1]]);
 }
 //# sourceMappingURL=static-integration.mjs.map

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@digitraffic/common",
-  "version": "2024.3.22-1",
+  "version": "2024.3.22-2",
   "description": "",
   "type": "module",
   "repository": {