add cookie session handler

packages/doorman-api/.env.local

@@ -34,3 +34,5 @@ DOORMAN_URL=http://localhost:3005
 
 # this isn't really a secret, its just to prevent spam on /notify in prod
 NOTIFY_SECRET_KEY=discordnotifyme
+
+COOKIE_PASSWORD=testtesttesttesttesttesttesttest

packages/doorman-api/.env.twiliotemplate

@@ -28,3 +28,5 @@ LOKI_PW=doormanlogs
 NOTIFY_SECRET_KEY=discordnotifyme
 
 DOORMAN_URL=
+
+COOKIE_PASSWORD=testtesttesttesttesttesttesttest

packages/doorman-api/package.json

@@ -21,6 +21,7 @@
     "discord-oauth2": "^2.12.1",
     "discord.js": "^14.19.3",
     "dynabridge": "^0.3.8",
+    "iron-session": "^8.0.4",
     "is-deep-subset": "^0.1.1",
     "prom-client": "^15.1.3",
     "promise.timeout": "^1.2.0",

packages/doorman-api/src/common/DoormanHandler.ts

@@ -1,4 +1,4 @@
-import { ServerlessCallback, ServerlessFunctionSignature } from "@twilio-labs/serverless-runtime-types/types";
+import { ServerlessCallback, ServerlessFunctionSignature, TwilioResponse } from "@twilio-labs/serverless-runtime-types/types";
 import { PrometheusContentType, Registry, Pushgateway, Summary, Counter, register } from "prom-client";
 import { DoormanLambdaContext } from "./DoormanHandlerContext";
 import { shouldBlockRequest } from "../utils/blockUserAgent";
@@ -13,8 +13,7 @@ import pTimeout, { TimeoutError } from "promise.timeout";
 import { ZodError } from "zod";
 import { fromError } from "zod-validation-error";
 import { setResponseJson } from "../utils/responseUtils";
-
+import { BaseEvent, getSessionFromRequest, SessionType, setSessionOnResponse } from "./SessionHandler";
-export type BaseEvent = { request: { cookies: {}; headers: {}; }; }
 
 export type CallbackResult = Parameters<ServerlessCallback>;
 export type FailFastCallback = () => CallbackResult;
@@ -73,7 +72,8 @@ const INNER_HANDLER_MAXIMUM_DURATION_MS: number = 8_750;
  */
 export function withMetrics<T extends DoormanLambdaContext, U extends BaseEvent>(
   functionName: string,
-  handler: DoormanLambda<T, U>
+  handler: DoormanLambda<T, U>,
+  parseSessionCookie?: boolean,
 ): ServerlessFunctionSignature<T, U> {
   return async (context, event, callback) => {
     console.log("[CommonHandler] created loki logger");
@@ -208,6 +208,11 @@ export function withMetrics<T extends DoormanLambdaContext, U extends BaseEvent>
     const startTime = Date.now();
     console.log(`[CommonHandler] started handler at ${startTime}`);
 
+    if (parseSessionCookie) {
+      console.log(`[CommonHandler] unwrapping session cookie`);
+      event.request.session = await getSessionFromRequest(event, context);
+    }
+
     const handlerResponsePromise: Promise<Parameters<ServerlessCallback>> = new Promise(async (resolve, reject) => {
       // intercept the callbackResult
       let callbackResult: Parameters<ServerlessCallback> | undefined;

packages/doorman-api/src/common/DoormanHandlerContext.ts

@@ -15,4 +15,5 @@ export interface DoormanLambdaContext extends EnvironmentVariables {
   LOKI_PW: string;
   NOTIFY_SECRET_KEY: string;
   DOORMAN_URL: string;
+  COOKIE_PASSWORD: string;
 };

packages/doorman-api/src/common/SessionHandler.ts

@@ -0,0 +1,50 @@
+import { TwilioResponse } from "@twilio-labs/serverless-runtime-types/types";
+import { TwilioContext } from "../types/TwilioContext";
+import { sealData, unsealData } from "iron-session";
+import { DoormanLambdaContext } from "./DoormanHandlerContext";
+
+export type BaseEvent = { request: { cookies: { session?: string; }; headers: {}; session?: SessionType }; }
+
+export type SessionType = any;
+export const IRON_SESSION_COOKIE_NAME = "session";
+
+export async function getSessionFromRequest(event: BaseEvent, context: DoormanLambdaContext): Promise<SessionType | undefined> {
+  if (!event.request.cookies[IRON_SESSION_COOKIE_NAME]) {
+    return undefined
+  }
+
+  let session: SessionType | undefined;
+  try {
+    session = await unsealData<any>(event.request.cookies[IRON_SESSION_COOKIE_NAME], {
+      password: context.COOKIE_PASSWORD,
+    });
+  } catch (e: any) {
+    console.error("error unsealing cookie");
+  }
+
+  return session;
+}
+
+export async function setSessionOnResponse(res: TwilioResponse, context: TwilioContext, session: any): Promise<TwilioResponse> {
+  if (!session) {
+    return res;
+  }
+
+  try {
+    const cookie = await sealData(session, {
+      password: context.COOKIE_PASSWORD,
+    });
+    console.log(cookie);
+    res.setCookie(IRON_SESSION_COOKIE_NAME, cookie);
+  } catch (e: any) {
+    console.log("failed to seal cookie");
+    console.log(e.message);
+  }
+
+  return res;
+}
+
+export async function destroySession(res: TwilioResponse): Promise<TwilioResponse> {
+  res.removeCookie(IRON_SESSION_COOKIE_NAME);
+  return res;
+}

packages/doorman-api/src/functions/api/door/info.ts

@@ -14,6 +14,7 @@ import { z } from "zod";
 import { UserAgentHeader } from "../../../utils/blockUserAgent";
 import { setResponseJson } from "../../../utils/responseUtils";
 import { sanitizePhoneNumber } from "../../../utils/phoneUtils";
+import { BaseEvent, destroySession, setSessionOnResponse } from "../../../common/SessionHandler";
 
 export const InfoRequestSchema = z.object({
   door: z.string().optional(),
@@ -23,7 +24,7 @@ export const InfoRequestSchema = z.object({
   .refine(data => data.buzzer || data.door, 'Buzzer or door must be provided');
 
 export type InfoRequest = z.infer<typeof InfoRequestSchema>;
-export interface InfoRequestTwilio extends ServerlessEventObject<InfoRequest, UserAgentHeader> { };
+export type InfoRequestTwilio = ServerlessEventObject<InfoRequest, UserAgentHeader> & BaseEvent;
 
 export const InfoResponseClientSchema = DoorConfigSchema
   .omit({ PK: true, SK: true, pin: true })
@@ -95,5 +96,13 @@ export const handler: ServerlessFunctionSignature<TwilioContext, InfoRequestTwil
   // @ts-ignore
   db.ddbClient.destroy();
 
+  // if (event.request.session) {
+  //   console.log(event.request.session);
+  //   await destroySession(response);
+  // } else {
+  //   await setSessionOnResponse(response, context, { someSecret: new Date() });
+  // }
+
+
   return callback(null, response);
 });

packages/doorman-client/.env.local

@@ -15,4 +15,6 @@ LOKI_URL=https://logs.chromart.cc
 LOKI_USER=doorman
 LOKI_PW=doormanlogs
 
-NOTIFY_SECRET_KEY=discordnotifyme
+NOTIFY_SECRET_KEY=discordnotifyme
+
+COOKIE_PASSWORD=testtesttesttesttesttesttesttest

packages/doorman-client/.env.twiliotemplate

@@ -15,4 +15,6 @@ LOKI_URL=https://logs.chromart.cc
 LOKI_USER=doorman
 LOKI_PW=doormanlogs
 
-NOTIFY_SECRET_KEY=discordnotifyme
+NOTIFY_SECRET_KEY=discordnotifyme
+
+COOKIE_PASSWORD=testtesttesttesttesttesttesttest

packages/doorman-client/package.json

@@ -14,6 +14,7 @@
   "dependencies": {
     "@twilio-labs/serverless-runtime-types": "^4.0.1",
     "@twilio/runtime-handler": "1.3.0",
+    "iron-session": "^8.0.4",
     "node-fetch": "^2.7.0",
     "prom-client": "^15.1.3",
     "prometheus-remote-write": "^0.5.1",