ws-handler.ts

export const handler: Handler<APIGatewayEvent> = async (event, context, callback) => {
  if (!event.requestContext) {
    return {
      statusCode: 500,
      body: ''
    };
  }
  
  const { eventType, connectionId } = event.requestContext;
  
  if (eventType === 'CONNECT') {
    // handle connect
    return {
      statusCode: 200,
      body: ''
    };
  }
  
  if (eventType === 'DISCONNECT') {
    // handle disconnect
    return {
      statusCode: 200,
      body: ''
    };
  }
  
  if (eventType === 'MESSAGE') {
    // handle message
    return {
      statusCode: 200,
      body: ''
    };
  }
  
  return {
    statusCode: 500,
    body: ''
  };
};