@adonisjs/limiter/build/src/stores/database.js

Rate limiting package for AdonisJS framework

import {
  RateLimiterBridge,
  debug_default
} from "../../chunk-3YOYZ7DU.js";
import "../../chunk-MLKGABMK.js";

// src/stores/database.ts
import string from "@adonisjs/core/helpers/string";
import { RuntimeException } from "@adonisjs/core/exceptions";
import { RateLimiterMySQL, RateLimiterPostgres, RateLimiterSQLite } from "rate-limiter-flexible";
var SUPPORTED_CLIENTS = [
  "mysql",
  "postgres",
  "better-sqlite3",
  "sqlite3"
];
var LimiterDatabaseStore = class extends RateLimiterBridge {
  #config;
  #client;
  get name() {
    return "database";
  }
  constructor(client, config) {
    const dialectName = client.dialect.name;
    if (!SUPPORTED_CLIENTS.includes(dialectName)) {
      throw new RuntimeException(
        `Unsupported database "${dialectName}". The limiter can only work with PostgreSQL, MySQL, and SQLite databases`
      );
    }
    debug_default("creating %s limiter store %O", dialectName, config);
    switch (dialectName) {
      case "mysql":
        super(
          new RateLimiterMySQL({
            storeType: "knex",
            storeClient: client.getWriteClient(),
            tableCreated: true,
            dbName: config.dbName,
            tableName: config.tableName,
            keyPrefix: config.keyPrefix,
            execEvenly: config.execEvenly,
            points: config.requests,
            clearExpiredByTimeout: config.clearExpiredByTimeout,
            duration: string.seconds.parse(config.duration),
            inMemoryBlockOnConsumed: config.inMemoryBlockOnConsumed,
            blockDuration: config.blockDuration ? string.seconds.parse(config.blockDuration) : void 0,
            inMemoryBlockDuration: config.inMemoryBlockDuration ? string.seconds.parse(config.inMemoryBlockDuration) : void 0
          })
        );
        this.#client = client;
        this.#config = config;
        break;
      case "postgres":
        super(
          new RateLimiterPostgres({
            storeType: "knex",
            schemaName: config.schemaName,
            storeClient: client.getWriteClient(),
            tableCreated: true,
            dbName: config.dbName,
            tableName: config.tableName,
            keyPrefix: config.keyPrefix,
            execEvenly: config.execEvenly,
            points: config.requests,
            clearExpiredByTimeout: config.clearExpiredByTimeout,
            duration: string.seconds.parse(config.duration),
            inMemoryBlockOnConsumed: config.inMemoryBlockOnConsumed,
            blockDuration: config.blockDuration ? string.seconds.parse(config.blockDuration) : void 0,
            inMemoryBlockDuration: config.inMemoryBlockDuration ? string.seconds.parse(config.inMemoryBlockDuration) : void 0
          })
        );
        this.#client = client;
        this.#config = config;
        break;
      case "better-sqlite3":
      case "sqlite3":
        super(
          new RateLimiterSQLite({
            storeType: "knex",
            storeClient: client.getWriteClient(),
            tableCreated: true,
            dbName: config.dbName,
            tableName: config.tableName,
            keyPrefix: config.keyPrefix,
            execEvenly: config.execEvenly,
            points: config.requests,
            clearExpiredByTimeout: config.clearExpiredByTimeout,
            duration: string.seconds.parse(config.duration),
            inMemoryBlockOnConsumed: config.inMemoryBlockOnConsumed,
            blockDuration: config.blockDuration ? string.seconds.parse(config.blockDuration) : void 0,
            inMemoryBlockDuration: config.inMemoryBlockDuration ? string.seconds.parse(config.inMemoryBlockDuration) : void 0
          })
        );
        this.#client = client;
        this.#config = config;
        break;
    }
  }
  /**
   * Truncates the database table, removing all rate limit data.
   *
   * **Warning**: Use a dedicated table for each limiter configuration
   * to avoid accidentally clearing other limiter data.
   */
  async clear() {
    debug_default("truncating database table %s", this.#config.tableName);
    this.deleteInMemoryBlockedKeys();
    await this.#client.dialect.truncate(this.#config.tableName, true);
  }
};
export {
  LimiterDatabaseStore as default
};