Class DataConnection

src/data/data-connection.ts:26–61

 * @typeParam TSchema - The merged Drizzle schema object.
 */
export class DataConnection<
  TSchema extends Record<string, unknown> = Record<string, unknown>,
> {
  private _sqlite: SQLiteDatabase;
  private _db: DrizzleDB<TSchema>;
  private _logger: Logger;

  constructor(schemas: TSchema) {
    this._logger = createLogger("database");

    const dbPath = config.paths.resolveDataFilePath("agentara.db");
    this._sqlite = new SQLiteDatabase(dbPath, { create: true });
    this._sqlite.run("PRAGMA journal_mode = WAL");
    this._sqlite.run("PRAGMA foreign_keys = ON");

    this._db = drizzle(this._sqlite, { schema: schemas });

    const migrationsFolder = join(import.meta.dir, "..", "..", "drizzle");
    migrate(this._db, { migrationsFolder });

    this._logger.info(`Database "${dbPath}" opened`);
  }

  /** The Drizzle database instance for executing typed queries. */
  get db(): DrizzleDB<TSchema> {
    return this._db;
  }

  /**
   * Close the database connection.
   */
  close(): void {
    this._sqlite.close();
    this._logger.info("database closed");
  }
}