@digitraffic/common 2022.11.23-test-2 → 2022.11.25-1

package/src/aws/infra/api/static-integration.ts

@@ -1,13 +1,17 @@
-import {MockIntegration, PassthroughBehavior, Resource} from "aws-cdk-lib/aws-apigateway";
-import {MediaType} from "../../types/mediatypes";
-import {corsMethod, RESPONSE_CORS_INTEGRATION} from "./responses";
+import {
+    MockIntegration,
+    PassthroughBehavior,
+    Resource,
+} from "aws-cdk-lib/aws-apigateway";
+import { MediaType } from "../../types/mediatypes";
+import { corsMethod, RESPONSE_CORS_INTEGRATION } from "./responses";
 
 const INTEGRATION_RESPONSE_200 = `{
     "statusCode": 200
 }`;
 
 const METHOD_RESPONSE_200 = {
-    statusCode: '200',
+    statusCode: "200",
 };
 
 /**
@@ -19,9 +23,18 @@ const METHOD_RESPONSE_200 = {
  */
 export class DigitrafficStaticIntegration extends MockIntegration {
     constructor(
-        resource: Resource, mediaType: MediaType, response: string, enableCors = true, apiKeyRequired = true,
+        resource: Resource,
+        mediaType: MediaType,
+        response: string,
+        enableCors = true,
+        apiKeyRequired = true
     ) {
-        const integrationResponse = DigitrafficStaticIntegration.createIntegrationResponse(response, mediaType, enableCors);
+        const integrationResponse =
+            DigitrafficStaticIntegration.createIntegrationResponse(
+                response,
+                mediaType,
+                enableCors
+            );
 
         super({
             passthroughBehavior: PassthroughBehavior.WHEN_NO_TEMPLATES,
@@ -31,32 +44,53 @@ export class DigitrafficStaticIntegration extends MockIntegration {
             integrationResponses: [integrationResponse],
         });
 
-        ['GET', 'HEAD'].forEach((httpMethod) => {
+        ["GET", "HEAD"].forEach((httpMethod) => {
             resource.addMethod(httpMethod, this, {
                 apiKeyRequired,
-                methodResponses: [DigitrafficStaticIntegration.createMethodResponse(enableCors)],
+                methodResponses: [
+                    DigitrafficStaticIntegration.createMethodResponse(
+                        enableCors
+                    ),
+                ],
             });
         });
     }
 
-    static json<K>(resource: Resource, response: K, enableCors = true, apiKeyRequired = true) {
+    static json<K>(
+        resource: Resource,
+        response: K,
+        enableCors = true,
+        apiKeyRequired = true
+    ) {
         return new DigitrafficStaticIntegration(
-            resource, MediaType.APPLICATION_JSON, JSON.stringify(response), enableCors, apiKeyRequired,
+            resource,
+            MediaType.APPLICATION_JSON,
+            JSON.stringify(response),
+            enableCors,
+            apiKeyRequired
         );
     }
 
-    private static createIntegrationResponse(response: string, mediaType: MediaType, enableCors: boolean) {
+    private static createIntegrationResponse(
+        response: string,
+        mediaType: MediaType,
+        enableCors: boolean
+    ) {
         const integrationResponse = {
-            statusCode: '200',
+            statusCode: "200",
             responseTemplates: {
                 [mediaType]: response,
             },
         };
 
-        return enableCors ? {...integrationResponse, ...RESPONSE_CORS_INTEGRATION} : integrationResponse;
+        return enableCors
+            ? { ...integrationResponse, ...RESPONSE_CORS_INTEGRATION }
+            : integrationResponse;
     }
 
     private static createMethodResponse(enableCors: boolean) {
-        return enableCors ? corsMethod(METHOD_RESPONSE_200) : METHOD_RESPONSE_200;
+        return enableCors
+            ? corsMethod(METHOD_RESPONSE_200)
+            : METHOD_RESPONSE_200;
     }
 }

package/src/aws/types/lambda-response.ts

@@ -9,6 +9,18 @@ export class LambdaResponse<T> {
         this.fileName = fileName;
     }
 
+    /**
+     * When you want to return binary, use LambdaResponse with base64-encoded string and then binary:
+     * LambdaResponse.ok(base64EncodedBinary).binary()
+     */
+    binary() {
+        return {
+            status: this.status,
+            binary: this.body,
+            fileName: this.fileName,
+        }
+    }
+
     static ok<T>(body: T, fileName?: string) {
         return this.create(200, body, fileName);
     }

package/src/database/database.ts

@@ -1,6 +1,6 @@
 import { IDatabase, ITask } from "pg-promise";
 import { DatabaseEnvironmentKeys } from "../aws/runtime/secrets/dbsecret";
-import { getEnvVariable, getEnvVariableSafe } from "../utils/utils";
+import { getEnvVariable } from "../utils/utils";
 import { envValue } from "../aws/runtime/environment";
 
 // eslint-disable-next-line @typescript-eslint/no-var-requires