tiktok-dl/web/api.ts

import {FastifyInstance, FastifyPluginOptions} from 'fastify';
import {rateLimitConfig} from '../config';
import FastifyRateLimit from 'fastify-rate-limit';
import {client} from '../lib';
import {PingRoute, StoredLinksRoute, DownloadRoute} from './routers';

export default async (
    fastify: FastifyInstance,
    _: FastifyPluginOptions,
    done: (err?: Error) => void) => {
  if (rateLimitConfig.enable) {
    fastify.register(FastifyRateLimit, {
      redis: client,
      global: false,
      max: rateLimitConfig.maxRatelimit,
      ban: rateLimitConfig.maxRatelimit-1,
      cache: rateLimitConfig.ratelimitTime*1_000,
    });
  }

  // routers
  new PingRoute(fastify);
  new StoredLinksRoute(fastify);
  new DownloadRoute(fastify);

  done();
};