connection_init.ts

import { ConnectionInitMessage, MessageType } from 'graphql-ws';
import { MessageHandler } from '../types';
import { postToConnection } from '../utils/postToConnection';
import { deleteConnection } from '../utils/deleteConnection';
import { defaultTTL } from '../utils/defaultTTL';

/** Handler function for 'connection_init' message. */
export const connection_init: MessageHandler<ConnectionInitMessage> =
  async ({ server, event, message }) => {
    try {
      const payload = (await server.onConnectionInit?.({ event, message })) ?? message.payload ?? {}

      if (server.pingpong) {
        console.error('Missing implementation for pingpong');
        // await new StepFunctions()
        //   .startExecution({
        //     stateMachineArn: server.pingpong.machine,
        //     name: event.requestContext.connectionId,
        //     input: JSON.stringify({
        //       connectionId: event.requestContext.connectionId,
        //       domainName: event.requestContext.domainName,
        //       stage: event.requestContext.stage,
        //       state: 'PING',
        //       choice: 'WAIT',
        //       seconds: server.pingpong.delay,
        //     } as StateFunctionInput),
        //   })
        //   .promise()
      }

      // Write to persistence
      await server.models.connection.put({
        id: event.requestContext.connectionId,
        createdAt: Date.now(),
        requestContext: event.requestContext,
        payload,
        hasPonged: false,
        ttl: defaultTTL(),
      })
      return postToConnection(server)({
        ...event.requestContext,
        message: { type: MessageType.ConnectionAck },
      })
    } catch (err) {
      await server.onError?.(err, { event, message })
      await deleteConnection(server)(event.requestContext)
    }
  }