@atri-bot/lib-db 1.1.1 → 1.2.0

package/dist/index.d.mts

@@ -0,0 +1,29 @@
+import * as _atri_bot_core0 from "@atri-bot/core";
+import { AnyRelations, DrizzleConfig, EmptyRelations } from "drizzle-orm";
+import * as drizzle_orm_bun_sql_postgres_driver0 from "drizzle-orm/bun-sql/postgres/driver";
+import { MigrationConfig } from "drizzle-orm/migrator";
+
+//#region src/index.d.ts
+interface InitDbPluginOptions {
+  connectString: string;
+  config: DrizzleConfig;
+}
+declare function InitDbPlugin(options: InitDbPluginOptions): _atri_bot_core0.definePluginReturnType<object, {
+  pluginName: string;
+  install(): Promise<void>;
+  uninstall(): void;
+}>;
+interface DbPluginOptions<TSchema extends Record<string, unknown>, TRelations extends AnyRelations> {
+  config: DrizzleConfig<TSchema, TRelations>;
+  migration?: MigrationConfig;
+}
+declare function DbPlugin<TSchema extends Record<string, unknown>, TRelations extends AnyRelations = EmptyRelations>(options: DbPluginOptions<TSchema, TRelations>): _atri_bot_core0.definePluginReturnType<object, {
+  pluginName: string;
+  drizzle: drizzle_orm_bun_sql_postgres_driver0.BunSQLDatabase<TSchema, TRelations> & {
+    $client: Bun.SQL;
+  };
+  install(): Promise<void>;
+  uninstall(): void;
+}>;
+//#endregion
+export { DbPlugin, DbPluginOptions, InitDbPlugin, InitDbPluginOptions };

package/dist/index.mjs

@@ -0,0 +1 @@
+import e from"node:process";import{definePlugin as t}from"@atri-bot/core";import{sql as n}from"drizzle-orm";import{drizzle as r}from"drizzle-orm/bun-sql";import{migrate as i}from"drizzle-orm/bun-sql/migrator";var a=`@atri-bot/lib-db`;let o=null;function s(i){return t({pluginName:`${a}-init-db`,async install(){try{await r(i.connectString,i.config).execute(n`SELECT 1;`),o=i.connectString,this.logger.INFO(`测试数据库连接成功`)}catch(t){this.logger.ERROR(`测试数据库连接失败`,t),e.exit(1)}},uninstall(){}})}function c(e){if(!o)throw Error(`请先通过 InitDbPlugin 插件初始化数据库连接`);return t({pluginName:`${a}-db`,drizzle:r(o??``,e.config),async install(){e.migration&&await i(this.drizzle,e.migration)},uninstall(){}})}export{c as DbPlugin,s as InitDbPlugin};

package/package.json

@@ -1,25 +1,23 @@
 {
   "name": "@atri-bot/lib-db",
-  "version": "1.1.1",
+  "type": "module",
+  "version": "1.2.0",
   "description": "db lib for atri framework",
   "author": "huan_kong",
   "license": "MIT",
-  "type": "module",
-  "main": "./dist/index.js",
-  "types": "./dist/index.d.ts",
+  "main": "./dist/index.mjs",
+  "types": "./dist/index.d.mts",
   "files": [
     "./dist/**/*"
   ],
   "scripts": {
-    "build": "pnpm run type-check && pnpm run lint && tsup",
-    "clean": "rimraf dist",
+    "build": "tsdown",
     "lint": "eslint",
-    "lint:fix": "eslint --fix",
-    "type-check": "tsc --noEmit -p tsconfig.json --composite false"
+    "typecheck": "tsc --noEmit -p tsconfig.json --composite false"
   },
   "dependencies": {
-    "@huan_kong/logger": "^1.0.6",
-    "sequelize": "^6.37.7",
-    "sqlite3": "^5.1.7"
+    "@atri-bot/core": "^2.0.0",
+    "@huan_kong/logger": "^2.0.2",
+    "drizzle-orm": "^1.0.0-beta.18-7eb39f0"
   }
 }

package/dist/index.d.ts

@@ -1,17 +0,0 @@
-import { LogLevel, Logger } from '@huan_kong/logger';
-import { Options, Sequelize } from 'sequelize';
-
-type DbConfig = Options & {
-    debug?: boolean;
-    logLevel?: LogLevel;
-    baseDir: string;
-};
-declare class Db {
-    config: DbConfig;
-    logger: Logger;
-    sequelize: Sequelize;
-    dbs: Record<string, Sequelize>;
-    constructor(config: DbConfig);
-}
-
-export { Db, type DbConfig };

package/dist/index.js

@@ -1 +0,0 @@
-import{Logger as l,LogLevel as r}from"@huan_kong/logger";import g from"crypto";import n from"path";import{Sequelize as d}from"sequelize";var b={},s=class{config;logger;sequelize;dbs=b;constructor(e){this.config=e,this.logger=new l({title:"Db",level:e.logLevel??(e.debug?r.DEBUG:void 0)}),"logger"in e||(e.logging=e.debug?o=>this.logger.DEBUG(o):!1);let i={dialect:"sqlite",storage:n.join(e.baseDir,"./db.sqlite"),...e};i.dialect==="sqlite"&&"timezone"in i&&delete i.timezone;let t=g.createHash("sha256").update(JSON.stringify(i)).digest("hex");this.sequelize=this.dbs[t]??new d(i),this.dbs[t]=this.sequelize}};export{s as Db};