server.js

import bodyParser from '@fastify/formbody';
import cors from '@fastify/cors';
import fastify from 'fastify';
import { fastifyRequestContext } from '@fastify/request-context';
import favicon from 'fastify-favicon';
import helmet from '@fastify/helmet';
import { join } from 'node:path';
import serveStatic from '@fastify/static';
import { stat } from 'node:fs/promises';

import { CONTEXT_STORE_KEYS, ENVS } from './constants/index.js';
import globalErrorHandler from './utilities/global-error-handler.js';
import gracefulShutdown from './hooks/graceful-shutdown.js';
import incomingTimestamp from './hooks/incoming-timestamp.js';
import logger from './utilities/logger.js';
import notFoundHandler from './utilities/not-found-handler.js';

import changePasswordAPI from './apis/change-password/index.js';
import deleteAccountAPI from './apis/delete-account/index.js';
import indexAPI from './apis/index/index.js';
import meAPI from './apis/me/index.js';
import refreshTokensAPI from './apis/refresh-tokens/index.js';
import signInAPI from './apis/sign-in/index.js';
import signOutAPI from './apis/sign-out/index.js';
import signOutFullAPI from './apis/sign-out-full/index.js';
import signUpAPI from './apis/sign-up/index.js';
import updateAccountAPI from './apis/update-account/index.js';
import userAPI from './apis/user/index.js';
import usersAPI from './apis/users/index.js';

/**
 * Create Fastify server
 * @param {string} APP_ENV application environment
 * @returns {Promise<FastifyInstance>}
 */
export default async function createServer(APP_ENV) {
  const server = fastify({
    logger: APP_ENV === ENVS.development,
  });

  await server.register(bodyParser);
  await server.register(cors);

  const helmetOptions = {};
  if (APP_ENV !== ENVS.production) {
    helmetOptions.contentSecurityPolicy = false;
  }
  await server.register(helmet, helmetOptions);

  await server.register(
    fastifyRequestContext,
    {
      defaultStoreValues: {
        [CONTEXT_STORE_KEYS.incomingTimestamp]: null,
        [CONTEXT_STORE_KEYS.paginationQueryData]: null,
        [CONTEXT_STORE_KEYS.searchQueryData]: null,
        [CONTEXT_STORE_KEYS.userId]: null,
      },
    },
  );

  await server.register(
    favicon,
    {
      name: 'favicon.ico',
      path: './assets',
    },
  );

  if (APP_ENV === ENVS.development) {
    const documentationPath = join(process.cwd(), 'documentation');
    try {
      await stat(documentationPath);
      await server.register(
        serveStatic,
        {
          cacheControl: false,
          prefix: '/docs/',
          prefixAvoidTrailingSlash: true,
          root: join(process.cwd(), 'documentation'),
        },
      );
      logger('API documentation will be served');
    } catch {
      logger('API documentation will not be served: no static files found');
    }
  }

  server.setErrorHandler(globalErrorHandler);
  server.setNotFoundHandler(notFoundHandler);

  server.addHook('onClose', gracefulShutdown);
  server.addHook('onRequest', incomingTimestamp);

  await Promise.all([
    server.register(changePasswordAPI),
    server.register(deleteAccountAPI),
    server.register(indexAPI),
    server.register(meAPI),
    server.register(refreshTokensAPI),
    server.register(signInAPI),
    server.register(signOutAPI),
    server.register(signOutFullAPI),
    server.register(signUpAPI),
    server.register(updateAccountAPI),
    server.register(userAPI),
    server.register(usersAPI),
  ]);

  process.on(
    'SIGINT',
    async (signal) => {
      logger(`Stopping the server (signal: ${signal})`);
      await server.close();
    },
  );
  process.on(
    'SIGTERM',
    async (signal) => {
      logger(`Stopping the server (signal: ${signal})`);
      await server.close();
    },
  );

  return server;
}