Function initHttpHook

packages/room-server/src/shared/adapters/adapters.init.ts:141–192

(app: INestApplication)

};

export const initHttpHook = (app: INestApplication) => {
  const fastify = app.getHttpAdapter().getInstance() as FastifyInstance;
  fastify.decorateRequest(USER_HTTP_DECORATE, null);
  fastify.decorateRequest(DATASHEET_META_HTTP_DECORATE, null);
  fastify.decorateRequest(DATASHEET_LINKED, null);
  fastify.decorateRequest(DATASHEET_ENRICH_SELECT_FIELD, null);
  fastify.decorateRequest(DATASHEET_MEMBER_FIELD, null);
  // TODO: REQUEST_ID should be returned by api-gateway, so that we could trace all the services
  // TODO: support multiple languages
  fastify.decorateRequest(REQUEST_ID, null);
  fastify.decorateRequest(REQUEST_AT, null);

  fastify.addHook('preHandler', async (request) => {
    request[REQUEST_AT] = Date.now();
    request[REQUEST_ID] = generateRandomString();
    if (request.headers.authorization && request.headers.authorization.startsWith(AUTHORIZATION_PREFIX)) {
      const developerService = app.select(DatabaseModule).get(DeveloperService);
      const apiKey = request.headers.authorization.slice(AUTHORIZATION_PREFIX.length);
      request[USER_HTTP_DECORATE] = await developerService.getUserInfoByApiKey(apiKey);
    }
    if ((request.params as any)['spaceId']) {
      request[SPACE_ID_HTTP_DECORATE] = (request.params as any)['spaceId'];
    }
    if ((request.params as any)['nodeId']) {
      const nodeRepository = app.select(DatabaseModule).get(NodeRepository);
      const nodeInfo = await nodeRepository.getNodeInfo((request.params as any)['nodeId']);
      if (nodeInfo) {
        request[NODE_INFO] = nodeInfo;
        request[SPACE_ID_HTTP_DECORATE] = nodeInfo.spaceId;
      }
    }
    if (request.body && request.body['folderId']) {
      const nodeRepository = app.select(DatabaseModule).get(NodeRepository);
      const folderId = request.body['folderId'];
      request[REQUEST_HOOK_FOLDER] = await nodeRepository.getNodeInfo(folderId);
    }
    if (request.body && request.body['preNodeId']) {
      const nodeRepository = app.select(DatabaseModule).get(NodeRepository);
      const preNodeId = request.body['preNodeId'];
      request[REQUEST_HOOK_PRE_NODE] = await nodeRepository.getNodeInfo(preNodeId);
    }
    return;
  });
  fastify.addHook('onSend', (request, reply, _payload, done) => {
    // add request-id to Headers
    // TODO: REQUEST_ID should be returned by api-gateway, so that we could trace all the services
    void reply.header(REQUEST_ID, request[REQUEST_ID]);
    const serverTime = Date.now() - request[REQUEST_AT];
    void reply.header(SERVER_TIME, 'total;dur=' + serverTime);
    done();
  });
};

export const initSentry = (_app: INestApplication) => {
  const sentryDsn = process.env.SENTRY_DSN;