@earbug/db-models 0.0.24 → 0.0.26

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+export * from "./init-models";
+export { initDb } from "./initDb";
+export type { InitDbOptions } from "./initDb";

package/dist/index.js

@@ -0,0 +1,20 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.initDb = void 0;
+__exportStar(require("./init-models"), exports);
+var initDb_1 = require("./initDb");
+Object.defineProperty(exports, "initDb", { enumerable: true, get: function () { return initDb_1.initDb; } });

package/dist/initDb.d.ts

@@ -0,0 +1,7 @@
+import { Sequelize, Options } from "sequelize";
+export interface InitDbOptions {
+    pool?: Options["pool"];
+    retry?: Options["retry"];
+    logging?: Options["logging"];
+}
+export declare function initDb(options?: InitDbOptions): Sequelize;

package/dist/initDb.js

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.initDb = initDb;
+const sequelize_1 = require("sequelize");
+const init_models_1 = require("./init-models");
+const REQUIRED_ENV_VARS = ["PG_DATABASE", "PG_USER", "PG_HOST"];
+function initDb(options = {}) {
+    var _a, _b, _c;
+    const missing = REQUIRED_ENV_VARS.filter((v) => !process.env[v]);
+    if (missing.length > 0) {
+        throw new Error(`Missing required environment variables: ${missing.join(", ")}`);
+    }
+    console.log(`Initializing DB connection to ${process.env.PG_HOST}:${process.env.PG_PORT || "5432"}/${process.env.PG_DATABASE}...`);
+    const sequelize = new sequelize_1.Sequelize(process.env.PG_DATABASE, process.env.PG_USER, process.env.PG_PASSWORD, {
+        host: process.env.PG_HOST,
+        port: parseInt(process.env.PG_PORT || "5432"),
+        dialect: "postgres",
+        schema: "eb",
+        minifyAliases: true,
+        logging: (_a = options.logging) !== null && _a !== void 0 ? _a : false,
+        pool: (_b = options.pool) !== null && _b !== void 0 ? _b : {
+            max: 10,
+            min: 1,
+            acquire: 30000,
+            idle: 10000,
+            evict: 1000,
+        },
+        retry: (_c = options.retry) !== null && _c !== void 0 ? _c : {
+            max: 2,
+            match: [
+                /ConnectionError/,
+                /ConnectionRefused/,
+                /ConnectionAcquireTimeout/,
+                /ECONNRESET/,
+                /ETIMEDOUT/,
+            ],
+        },
+    });
+    (0, init_models_1.initModels)(sequelize);
+    console.log("DB connection initialized.");
+    return sequelize;
+}

package/models/index.ts

@@ -0,0 +1,3 @@
+export * from "./init-models";
+export { initDb } from "./initDb";
+export type { InitDbOptions } from "./initDb";

package/models/initDb.ts

@@ -0,0 +1,60 @@
+import { Sequelize, Options } from "sequelize";
+import { initModels } from "./init-models";
+
+const REQUIRED_ENV_VARS = ["PG_DATABASE", "PG_USER", "PG_HOST"] as const;
+
+export interface InitDbOptions {
+  pool?: Options["pool"];
+  retry?: Options["retry"];
+  logging?: Options["logging"];
+}
+
+export function initDb(options: InitDbOptions = {}): Sequelize {
+  const missing = REQUIRED_ENV_VARS.filter((v) => !process.env[v]);
+  if (missing.length > 0) {
+    throw new Error(
+      `Missing required environment variables: ${missing.join(", ")}`,
+    );
+  }
+
+  console.log(
+    `Initializing DB connection to ${process.env.PG_HOST}:${process.env.PG_PORT || "5432"}/${process.env.PG_DATABASE}...`,
+  );
+
+  const sequelize = new Sequelize(
+    process.env.PG_DATABASE!,
+    process.env.PG_USER!,
+    process.env.PG_PASSWORD,
+    {
+      host: process.env.PG_HOST,
+      port: parseInt(process.env.PG_PORT || "5432"),
+      dialect: "postgres",
+      schema: "eb",
+      minifyAliases: true,
+      logging: options.logging ?? false,
+      pool: options.pool ?? {
+        max: 10,
+        min: 1,
+        acquire: 30000,
+        idle: 10000,
+        evict: 1000,
+      },
+      retry: options.retry ?? {
+        max: 2,
+        match: [
+          /ConnectionError/,
+          /ConnectionRefused/,
+          /ConnectionAcquireTimeout/,
+          /ECONNRESET/,
+          /ETIMEDOUT/,
+        ],
+      },
+    },
+  );
+
+  initModels(sequelize);
+
+  console.log("DB connection initialized.");
+
+  return sequelize;
+}

package/package.json

@@ -1,8 +1,8 @@
 {
   "name": "@earbug/db-models",
-  "version": "0.0.24",
-  "main": "dist/init-models.js",
-  "types": "dist/init-models.d.ts",
+  "version": "0.0.26",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
   "scripts": {
     "generate-models": "sequelize-auto -l ts --cm=p --cf=p --cp=c -o \"./models\" -a ./model_options.json -d earbug_live -h localhost -u postgres -p 5432 -e postgres -T harvested_article article_to_artist_relation",
     "build": "tsc"