@pcg/postgres-pubsub 1.0.0-alpha.0

package/dist/index.d.ts

@@ -0,0 +1,70 @@
+import { OnModuleInit } from "@nestjs/common";
+import { EventEmitter } from "events";
+import { PubSubEngine } from "graphql-subscriptions";
+import { Client, ClientConfig } from "pg";
+import { BaseEntity, DataSource } from "typeorm";
+import { Logger } from "@pcg/core";
+import { MaybeNull } from "@pcg/predicates";
+import { PostgresConnectionOptions } from "typeorm/driver/postgres/PostgresConnectionOptions.js";
+
+//#region src/postgres-pubsub.d.ts
+type PubSubSubscriptionCallback = (...args: unknown[]) => void;
+interface PubSubSubscription {
+  id: number;
+  channel: string;
+  callback: PubSubSubscriptionCallback;
+}
+/**
+ * Entity reference
+ * @example
+ * {
+ *  __ref: 'Media#jsv:12jg839hkvgs'
+ * }
+ */
+interface EntityRef {
+  __ref: string;
+}
+interface EntityWithId extends BaseEntity {
+  id: string;
+}
+interface ObjectWithId {
+  id: string;
+}
+declare class PostgresPubSub extends PubSubEngine implements OnModuleInit {
+  protected readonly dbConfig: PostgresConnectionOptions;
+  private readonly loggerFactory;
+  protected logger: Logger;
+  protected readonly dataSource: DataSource;
+  protected dbClient: Client;
+  protected ee: EventEmitter<[never]>;
+  protected subscriptions: PubSubSubscription[];
+  protected config: ClientConfig;
+  protected readonly retryLimit = 5;
+  protected isReinitializing: boolean;
+  onModuleInit(): Promise<void>;
+  keepDbConnectionAlive(): Promise<void>;
+  protected reinit(): Promise<void>;
+  protected addDbClientEventListeners(): void;
+  protected reconnect(): Promise<Client>;
+  publish(triggerName: string, payload: any, retryCount?: number): Promise<void>;
+  subscribe(triggerName: string, callback: PubSubSubscriptionCallback): Promise<number>;
+  unsubscribe(subId: number): Promise<void>;
+  private processNotification;
+  protected toRef(object: object): MaybeNull<EntityRef>;
+  protected minify(payload: any): any;
+  protected unref(entityRef: EntityRef): Promise<BaseEntity | ObjectWithId>;
+  protected unminify(payload: any): Promise<any>;
+  protected isEntityRef(value: unknown): value is EntityRef;
+  protected isEntityWithId(value: unknown): value is EntityWithId;
+  asyncIterator<T>(triggers: string | readonly string[]): AsyncIterator<T>;
+}
+//#endregion
+//#region src/postgres-pubsub.module.d.ts
+/**
+ * PostgresPubSubModule provides PostgreSQL NOTIFY/LISTEN based PubSub engine
+ * for GraphQL subscriptions in NestJS applications.
+ */
+declare class PostgresPubSubModule {}
+//#endregion
+export { EntityRef, EntityWithId, ObjectWithId, PostgresPubSub, PostgresPubSubModule, PubSubSubscription, PubSubSubscriptionCallback };
+//# sourceMappingURL=index.d.ts.map

package/dist/index.js

@@ -0,0 +1,265 @@
+import { Global, Injectable, Module } from "@nestjs/common";
+import { Interval } from "@nestjs/schedule";
+import { InjectDataSource } from "@nestjs/typeorm";
+import { EventEmitter } from "events";
+import { PubSubEngine } from "graphql-subscriptions";
+import { Client, escapeIdentifier, escapeLiteral } from "pg";
+import { BaseEntity, DataSource } from "typeorm";
+import { InjectDbConfig, InjectLoggerFactory, LoggerFactory, wait } from "@pcg/core";
+import { isObject } from "@pcg/predicates";
+
+//#region \0@oxc-project+runtime@0.95.0/helpers/decorateMetadata.js
+function __decorateMetadata(k, v) {
+	if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+}
+
+//#endregion
+//#region \0@oxc-project+runtime@0.95.0/helpers/decorate.js
+function __decorate(decorators, target, key, desc) {
+	var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+	if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+	else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+	return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+
+//#endregion
+//#region src/postgres-pubsub.ts
+var _ref, _ref2;
+let PostgresPubSub = class PostgresPubSub$1 extends PubSubEngine {
+	dbClient;
+	ee = new EventEmitter();
+	subscriptions = [];
+	config;
+	retryLimit = 5;
+	isReinitializing = false;
+	async onModuleInit() {
+		this.logger = this.loggerFactory.create({ scope: this.constructor.name });
+		this.config = {
+			host: this.dbConfig.host,
+			port: this.dbConfig.port,
+			database: this.dbConfig.database,
+			user: this.dbConfig.username,
+			password: this.dbConfig.password,
+			ssl: this.dbConfig.ssl,
+			connectionTimeoutMillis: 3e4,
+			query_timeout: 3e4,
+			statement_timeout: 3e4,
+			keepAlive: true,
+			keepAliveInitialDelayMillis: 1e4
+		};
+		await this.reinit();
+	}
+	async keepDbConnectionAlive() {
+		if (this.isReinitializing) return;
+		try {
+			await this.dbClient.query("SELECT pg_backend_pid()");
+		} catch (error) {
+			if (error instanceof Error) this.logger.error("PubSub PostgreSQL connection check failed:", error);
+			await this.reinit();
+		}
+	}
+	async reinit() {
+		this.isReinitializing = true;
+		this.logger.info("🔌 Initializing PubSub PostgreSQL client...");
+		if (this.dbClient) {
+			this.dbClient.removeAllListeners();
+			this.dbClient.once("error", (error) => {
+				this.logger.error(`Previous DB client errored after reconnecting already:`, error);
+			});
+			this.dbClient.end();
+		}
+		this.dbClient = await this.reconnect();
+		this.logger.info("✅ PubSub PostgreSQL client connected");
+		this.addDbClientEventListeners();
+		this.isReinitializing = false;
+	}
+	addDbClientEventListeners() {
+		this.dbClient.on("notification", (message) => {
+			this.processNotification(message);
+		});
+		this.dbClient.on("end", () => {
+			this.logger.warn("⛓️‍💥 PubSub PostgreSQL client ended");
+			if (!this.isReinitializing) this.reinit();
+		});
+		this.dbClient.on("error", (error) => {
+			this.logger.error("🔴 PubSub PostgreSQL client error", error);
+			if (!this.isReinitializing) this.reinit();
+		});
+	}
+	async reconnect() {
+		this.logger.info("🔌 Connecting to PubSub PostgreSQL for notification streaming");
+		const startTime = Date.now();
+		const retryTimeout = 3e3;
+		for (let attempt = 1; attempt < this.retryLimit; attempt++) {
+			this.logger.info(`🔌 PostgreSQL connection attempt #${String(attempt)}...`);
+			try {
+				const dbClient = new Client(this.config);
+				const connecting = new Promise((resolve, reject) => {
+					dbClient.once("connect", resolve);
+					dbClient.once("end", () => {
+						reject(Error("Connection ended."));
+					});
+					dbClient.once("error", reject);
+				});
+				await Promise.all([dbClient.connect(), connecting]);
+				this.logger.info("✅ PostgreSQL connection succeeded");
+				return dbClient;
+			} catch (error) {
+				if (error instanceof Error) this.logger.error("🔌 PostgreSQL connection attempt failed:", error);
+				await wait(500);
+				if (Date.now() - startTime > retryTimeout) throw new Error(`🚫 Stopping PostgreSQL connection attempts after ${String(retryTimeout)}ms timeout has been reached.`);
+			}
+		}
+		throw new Error("🔴 Reconnecting notification client to PostgreSQL database failed.");
+	}
+	async publish(triggerName, payload, retryCount = 1) {
+		if (!payload) throw new Error("Payload is required argument");
+		const channel = escapeIdentifier(triggerName);
+		const message = escapeLiteral(JSON.stringify(this.minify(payload)));
+		try {
+			await this.dbClient.query(`NOTIFY ${channel}, ${message}`);
+		} catch (error) {
+			if (error instanceof Error) {
+				this.logger.error(`Can't publish PostgreSQL notification to channel "${channel}"`, error, { payload });
+				if (/connection/i.exec(error.message)) {
+					if (retryCount < 3) {
+						this.logger.error(`Try to reconnect to PubSub PostgreSQL Client...(Retry count: ${String(retryCount)})`);
+						await this.dbClient.end();
+						await this.dbClient.connect();
+						await this.publish(triggerName, payload, retryCount + 1);
+					}
+				}
+			}
+		}
+		await Promise.resolve();
+	}
+	async subscribe(triggerName, callback) {
+		if (!this.subscriptions.some((s) => s.channel === triggerName)) {
+			const channel = escapeIdentifier(triggerName);
+			await this.dbClient.query(`LISTEN ${channel}`);
+		}
+		this.ee.on(triggerName, callback);
+		const subId = Math.floor(Math.random() * 1e5);
+		this.subscriptions.push({
+			id: subId,
+			channel: triggerName,
+			callback
+		});
+		return await Promise.resolve(subId);
+	}
+	async unsubscribe(subId) {
+		const index = this.subscriptions.findIndex((s) => s.id === subId);
+		if (index !== -1) {
+			const subscription = this.subscriptions[index];
+			this.subscriptions.splice(index, 1);
+			this.ee.removeListener(subscription.channel, subscription.callback);
+			if (!this.subscriptions.some((s) => s.channel === subscription.channel)) {
+				const channel = escapeIdentifier(subscription.channel);
+				await this.dbClient.query(`UNLISTEN ${channel}`);
+			}
+		}
+	}
+	async processNotification(message) {
+		try {
+			const minifiedPayload = JSON.parse(message.payload ?? "");
+			if (minifiedPayload.id) if (Object.keys(minifiedPayload).length === 1) this.ee.emit(message.channel, minifiedPayload);
+			else this.ee.emit(message.channel, await this.unminify(minifiedPayload));
+			else if (Array.isArray(minifiedPayload)) {
+				const payload = await this.unminify(minifiedPayload);
+				this.ee.emit(message.channel, payload);
+			} else {
+				const payload = {};
+				for (const [key, value] of Object.entries(minifiedPayload)) payload[key] = await this.unminify(value);
+				this.ee.emit(message.channel, payload);
+			}
+		} catch (error) {
+			if (error instanceof Error) this.logger.error(`Can't extract PostgreSQL notificaton from channel ${message.channel}`, error, { message });
+		}
+	}
+	toRef(object) {
+		if (this.isEntityWithId(object)) return { __ref: `${object.constructor.name}#${object.id}` };
+		return null;
+	}
+	minify(payload) {
+		if (Array.isArray(payload)) return payload.map((item) => this.minify(item));
+		if (!isObject(payload)) return payload;
+		if (this.isEntityWithId(payload)) return this.toRef(payload) ?? payload;
+		const minifiedObject = {};
+		for (const key of Object.keys(payload)) {
+			const value = payload[key];
+			if (value instanceof Date) minifiedObject[key] = value.toISOString();
+			else if (Array.isArray(value)) minifiedObject[key] = value.map((item) => this.minify(item));
+			else if (isObject(value)) minifiedObject[key] = this.minify(value);
+			else minifiedObject[key] = value;
+		}
+		return minifiedObject;
+	}
+	async unref(entityRef) {
+		const logger = this.logger.child({
+			action: this.unref.name,
+			entityRef
+		});
+		const [entityName, id] = entityRef.__ref.split("#");
+		if (!entityName || !id) {
+			logger.error(`Invalid ref ${entityRef.__ref}`);
+			return { id: entityRef.__ref };
+		}
+		const entityMetadata = this.dataSource.entityMetadatas.find((entityMetadata$1) => entityMetadata$1.name === entityName);
+		if (!entityMetadata?.target) {
+			logger.error(`Entity ${entityName} not found in registry`);
+			return { id };
+		}
+		const instance = await this.dataSource.getRepository(entityMetadata.target).findOneBy({ id });
+		if (!instance) {
+			logger.error(`Entity ${entityName} with id ${id} not found in database`);
+			return { id };
+		}
+		return instance;
+	}
+	async unminify(payload) {
+		if (Array.isArray(payload)) return await Promise.all(payload.map((item) => this.unminify(item)));
+		if (this.isEntityRef(payload)) return await this.unref(payload);
+		if (!isObject(payload)) return payload;
+		const unminifiedObject = {};
+		for (const key of Object.keys(payload)) {
+			const value = payload[key];
+			if (typeof value === "string" && !isNaN(Date.parse(value))) unminifiedObject[key] = new Date(value);
+			else if (this.isEntityRef(value)) unminifiedObject[key] = await this.unref(value);
+			else if (Array.isArray(value)) unminifiedObject[key] = await Promise.all(value.map((item) => this.unminify(item)));
+			else if (isObject(value)) unminifiedObject[key] = await this.unminify(value);
+			else unminifiedObject[key] = value;
+		}
+		return unminifiedObject;
+	}
+	isEntityRef(value) {
+		return typeof value === "object" && value !== null && "__ref" in value;
+	}
+	isEntityWithId(value) {
+		return value instanceof BaseEntity && "id" in value;
+	}
+	asyncIterator(triggers) {
+		return super.asyncIterableIterator(triggers);
+	}
+};
+__decorate([InjectDbConfig(), __decorateMetadata("design:type", Object)], PostgresPubSub.prototype, "dbConfig", void 0);
+__decorate([InjectLoggerFactory(), __decorateMetadata("design:type", typeof (_ref = typeof LoggerFactory !== "undefined" && LoggerFactory) === "function" ? _ref : Object)], PostgresPubSub.prototype, "loggerFactory", void 0);
+__decorate([InjectDataSource(), __decorateMetadata("design:type", typeof (_ref2 = typeof DataSource !== "undefined" && DataSource) === "function" ? _ref2 : Object)], PostgresPubSub.prototype, "dataSource", void 0);
+__decorate([
+	Interval("KEEP_DB_CONNECTION_ALIVE", 5 * 1e3),
+	__decorateMetadata("design:type", Function),
+	__decorateMetadata("design:paramtypes", []),
+	__decorateMetadata("design:returntype", Promise)
+], PostgresPubSub.prototype, "keepDbConnectionAlive", null);
+PostgresPubSub = __decorate([Injectable()], PostgresPubSub);
+
+//#endregion
+//#region src/postgres-pubsub.module.ts
+let PostgresPubSubModule = class PostgresPubSubModule$1 {};
+PostgresPubSubModule = __decorate([Global(), Module({
+	providers: [PostgresPubSub],
+	exports: [PostgresPubSub]
+})], PostgresPubSubModule);
+
+//#endregion
+export { PostgresPubSub, PostgresPubSubModule };
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":["PostgresPubSub","payload: any","minifiedObject: Record<string, unknown>","entityMetadata","unminifiedObject: Record<string, unknown>","PostgresPubSubModule"],"sources":["../src/postgres-pubsub.ts","../src/postgres-pubsub.module.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-unsafe-return */\n/* eslint-disable @typescript-eslint/no-unsafe-member-access */\n/* eslint-disable @typescript-eslint/no-unsafe-argument */\n/* eslint-disable @typescript-eslint/no-unsafe-assignment */\n/* eslint-disable @typescript-eslint/no-explicit-any */\nimport { Injectable, OnModuleInit } from '@nestjs/common';\nimport { Interval } from '@nestjs/schedule';\nimport { InjectDataSource } from '@nestjs/typeorm';\nimport { EventEmitter } from 'events';\nimport { PubSubEngine } from 'graphql-subscriptions';\nimport {\n  Client, ClientConfig, Notification, escapeIdentifier, escapeLiteral,\n} from 'pg';\nimport {\n  BaseEntity, DataSource, FindOptionsWhere,\n} from 'typeorm';\n\nimport {\n  InjectDbConfig,\n  InjectLoggerFactory,\n  Logger,\n  LoggerFactory,\n  wait,\n} from '@pcg/core';\nimport { isObject, MaybeNull } from '@pcg/predicates';\n\nimport type { PostgresConnectionOptions } from 'typeorm/driver/postgres/PostgresConnectionOptions.js';\n\nexport type PubSubSubscriptionCallback = (...args: unknown[]) => void;\n\nexport interface PubSubSubscription {\n  id: number;\n  channel: string;\n  callback: PubSubSubscriptionCallback;\n}\n\n/**\n * Entity reference\n * @example\n * {\n *  __ref: 'Media#jsv:12jg839hkvgs'\n * }\n */\nexport interface EntityRef {\n  __ref: string;\n}\n\nexport interface EntityWithId extends BaseEntity {\n  id: string;\n}\n\nexport interface ObjectWithId {\n  id: string;\n}\n\n@Injectable()\nexport class PostgresPubSub extends PubSubEngine implements OnModuleInit {\n  @InjectDbConfig() declare protected readonly dbConfig: PostgresConnectionOptions;\n\n  @InjectLoggerFactory() declare private readonly loggerFactory: LoggerFactory;\n  declare protected logger: Logger;\n\n  @InjectDataSource() declare protected readonly dataSource: DataSource;\n\n  protected dbClient!: Client;\n  protected ee = new EventEmitter();\n  protected subscriptions: PubSubSubscription[] = [];\n\n  protected config!: ClientConfig;\n  protected readonly retryLimit = 5;\n  protected isReinitializing = false;\n\n  async onModuleInit() {\n    this.logger = this.loggerFactory.create({\n      scope: this.constructor.name,\n    });\n\n    this.config = {\n      host: this.dbConfig.host,\n      port: this.dbConfig.port,\n      database: this.dbConfig.database,\n      user: this.dbConfig.username,\n      password: this.dbConfig.password,\n      ssl: this.dbConfig.ssl as boolean | undefined,\n\n      // Pool configuration\n      connectionTimeoutMillis: 30000, // 30 seconds to establish connection\n\n      // Query timeout settings\n      query_timeout: 30000, // 30 seconds for queries to complete\n      statement_timeout: 30000, // 30 seconds for statements\n\n      // Automatic reconnection\n      keepAlive: true,\n      keepAliveInitialDelayMillis: 10000, // Start keep-alive after 10 seconds\n    };\n\n    await this.reinit();\n  }\n\n  // @see fix suggestion [here](https://community.fly.io/t/postgresql-connection-issues-have-returned/6424/6)\n  @Interval('KEEP_DB_CONNECTION_ALIVE', 5 * 1000) // each 5s\n  async keepDbConnectionAlive() {\n    if (this.isReinitializing) {\n      return;\n    }\n\n    try {\n      await this.dbClient.query('SELECT pg_backend_pid()');\n    } catch (error) {\n      if (error instanceof Error) {\n        this.logger.error('PubSub PostgreSQL connection check failed:', error);\n      }\n\n      await this.reinit();\n    }\n  }\n\n  protected async reinit() {\n    this.isReinitializing = true;\n\n    this.logger.info('🔌 Initializing PubSub PostgreSQL client...');\n\n    // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition -- dbClient may be undefined on first call\n    if (this.dbClient) {\n      this.dbClient.removeAllListeners();\n      this.dbClient.once('error', (error) => {\n        this.logger.error(`Previous DB client errored after reconnecting already:`, error);\n      });\n      void this.dbClient.end();\n    }\n\n    this.dbClient = await this.reconnect();\n\n    this.logger.info('✅ PubSub PostgreSQL client connected');\n\n    this.addDbClientEventListeners();\n\n    this.isReinitializing = false;\n  }\n\n  protected addDbClientEventListeners() {\n    this.dbClient.on('notification', (message: Notification) => {\n      void this.processNotification(message);\n    });\n\n    this.dbClient.on('end', () => {\n      this.logger.warn('⛓️‍💥 PubSub PostgreSQL client ended');\n      if (!this.isReinitializing) {\n        void this.reinit();\n      }\n    });\n\n    this.dbClient.on('error', (error: Error) => {\n      this.logger.error('🔴 PubSub PostgreSQL client error', error);\n\n      if (!this.isReinitializing) {\n        void this.reinit();\n      }\n    });\n  }\n\n  protected async reconnect() {\n    this.logger.info('🔌 Connecting to PubSub PostgreSQL for notification streaming');\n    const startTime = Date.now();\n    const retryTimeout = 3000;\n\n    for (let attempt = 1; attempt < this.retryLimit; attempt++) {\n      this.logger.info(`🔌 PostgreSQL connection attempt #${String(attempt)}...`);\n\n      try {\n        const dbClient = new Client(this.config);\n        const connecting = new Promise((resolve, reject) => {\n          dbClient.once('connect', resolve);\n          dbClient.once('end', () => {\n            reject(Error('Connection ended.'));\n          });\n          dbClient.once('error', reject);\n        });\n        await Promise.all([\n          dbClient.connect(),\n          connecting,\n        ]);\n        this.logger.info('✅ PostgreSQL connection succeeded');\n\n        return dbClient;\n      } catch (error) {\n        if (error instanceof Error) {\n          this.logger.error('🔌 PostgreSQL connection attempt failed:', error);\n        }\n        await wait(500);\n\n        if ((Date.now() - startTime) > retryTimeout) {\n          throw new Error(`🚫 Stopping PostgreSQL connection attempts after ${String(retryTimeout)}ms timeout has been reached.`);\n        }\n      }\n    }\n\n    throw new Error('🔴 Reconnecting notification client to PostgreSQL database failed.');\n  }\n\n  public async publish(triggerName: string, payload: any, retryCount = 1): Promise<void> {\n    if (!payload) {\n      throw new Error('Payload is required argument');\n    }\n\n    const channel = escapeIdentifier(triggerName);\n\n    const message = escapeLiteral(JSON.stringify(this.minify(payload)));\n\n    // this.logger.info('Publish PostgreSQL notification', {\n    //   channel,\n    //   message,\n    //   payload,\n    // });\n\n    try {\n      await this.dbClient.query(`NOTIFY ${channel}, ${message}`);\n    } catch (error) {\n      if (error instanceof Error) {\n        this.logger.error(`Can't publish PostgreSQL notification to channel \"${channel}\"`, error, {\n          payload,\n        });\n\n        if (/connection/i.exec(error.message)) {\n          if (retryCount < 3) {\n            this.logger.error(`Try to reconnect to PubSub PostgreSQL Client...(Retry count: ${String(retryCount)})`);\n            await this.dbClient.end();\n            await this.dbClient.connect();\n            await this.publish(triggerName, payload, retryCount + 1);\n          }\n        }\n      }\n    }\n\n    await Promise.resolve();\n  }\n\n  public async subscribe(triggerName: string, callback: PubSubSubscriptionCallback): Promise<number> {\n    if (!this.subscriptions.some((s) => s.channel === triggerName)) {\n      const channel = escapeIdentifier(triggerName);\n      await this.dbClient.query(`LISTEN ${channel}`);\n    }\n\n    this.ee.on(triggerName, callback);\n\n    const subId = Math.floor(Math.random() * 100000);\n\n    this.subscriptions.push({\n      id: subId,\n      channel: triggerName,\n      callback,\n    });\n\n    return await Promise.resolve(subId);\n  }\n\n  public async unsubscribe(subId: number) {\n    const index = this.subscriptions.findIndex((s) => s.id === subId);\n    if (index !== -1) {\n      const subscription = this.subscriptions[index];\n      this.subscriptions.splice(index, 1);\n      this.ee.removeListener(subscription.channel, subscription.callback);\n\n      if (!this.subscriptions.some((s) => s.channel === subscription.channel)) {\n        const channel = escapeIdentifier(subscription.channel);\n        await this.dbClient.query(`UNLISTEN ${channel}`);\n      }\n    }\n  }\n\n  private async processNotification(message: Notification) {\n    try {\n      const minifiedPayload = JSON.parse(message.payload ?? '');\n\n      // this.logger.info('Process PostgreSQL notification', {\n      //   channel: message.channel,\n      //   minifiedObject,\n      // });\n\n      if (minifiedPayload.id) {\n        if (Object.keys(minifiedPayload).length === 1) {\n          this.ee.emit(message.channel, minifiedPayload);\n        } else {\n          this.ee.emit(message.channel, await this.unminify(minifiedPayload));\n        }\n      } else if (Array.isArray(minifiedPayload)) {\n        const payload = await this.unminify(minifiedPayload);\n\n        this.ee.emit(message.channel, payload);\n      } else {\n        const payload: any = {\n        };\n        for (const [key, value] of Object.entries(minifiedPayload)) {\n          payload[key] = await this.unminify(value);\n        }\n\n        this.ee.emit(message.channel, payload);\n      }\n    } catch (error) {\n      if (error instanceof Error) {\n        this.logger.error(`Can't extract PostgreSQL notificaton from channel ${message.channel}`, error, {\n          message,\n        });\n      }\n    }\n  }\n\n  protected toRef(object: object): MaybeNull<EntityRef> {\n    // this.logger.info('Create reference', {\n    //   object,\n    //   isEntityWithId: this.isEntityWithId(object),\n    // });\n\n    if (this.isEntityWithId(object)) {\n      return {\n        __ref: `${object.constructor.name}#${object.id}`,\n      };\n    }\n\n    return null;\n  }\n\n  protected minify(payload: any): any {\n    if (Array.isArray(payload)) {\n      return payload.map((item) => this.minify(item));\n    }\n\n    if (!isObject(payload)) {\n      return payload;\n    }\n\n    // this.logger.info('Minifying object', {\n    //   object,\n    //   isEntityWithId: this.isEntityWithId(object),\n    // });\n\n    if (this.isEntityWithId(payload)) {\n      const ref = this.toRef(payload);\n\n      return (ref ?? payload);\n    }\n\n    const minifiedObject: Record<string, unknown> = {\n    };\n\n    for (const key of Object.keys(payload)) {\n      const value = payload[key];\n\n      if (value instanceof Date) {\n        minifiedObject[key] = value.toISOString();\n      } else if (Array.isArray(value)) {\n        minifiedObject[key] = value.map((item) => this.minify(item));\n      } else if (isObject(value)) {\n        minifiedObject[key] = this.minify(value);\n      } else {\n        minifiedObject[key] = value;\n      }\n    }\n\n    return minifiedObject;\n  }\n\n  protected async unref(entityRef: EntityRef): Promise<BaseEntity | ObjectWithId> {\n    const logger = this.logger.child({\n      action: this.unref.name,\n      entityRef,\n    });\n\n    // logger.info('Unref entity');\n\n    const [entityName, id] = entityRef.__ref.split('#');\n\n    if (!entityName || !id) {\n      logger.error(`Invalid ref ${entityRef.__ref}`);\n\n      return {\n        id: entityRef.__ref,\n      };\n    }\n\n    const entityMetadata = this.dataSource.entityMetadatas.find(\n      (entityMetadata) => entityMetadata.name === entityName,\n    );\n\n    if (!entityMetadata?.target) {\n      logger.error(`Entity ${entityName} not found in registry`);\n\n      return {\n        id,\n      };\n    }\n\n    const instance = await this.dataSource.getRepository(entityMetadata.target).findOneBy({\n      id,\n    } as FindOptionsWhere<EntityWithId>);\n\n    if (!instance) {\n      logger.error(`Entity ${entityName} with id ${id} not found in database`);\n\n      return {\n        id,\n      };\n    }\n\n    // logger.info('Unref entity success', {\n    //   instance,\n    // });\n\n    return instance as BaseEntity;\n  }\n\n  protected async unminify(payload: any): Promise<any> {\n    // this.logger.info('Unminifying object', {\n    //   object,\n    //   isEntityRef: this.isEntityRef(object),\n    // });\n\n    if (Array.isArray(payload)) {\n      return await Promise.all(payload.map((item) => this.unminify(item)));\n    }\n\n    if (this.isEntityRef(payload)) {\n      return await this.unref(payload);\n    }\n\n    if (!isObject(payload)) {\n      return payload;\n    }\n\n    const unminifiedObject: Record<string, unknown> = {\n    };\n\n    for (const key of Object.keys(payload)) {\n      const value = payload[key];\n\n      if (typeof value === 'string' && !isNaN(Date.parse(value))) {\n        unminifiedObject[key] = new Date(value);\n      } else if (this.isEntityRef(value)) {\n        unminifiedObject[key] = await this.unref(value);\n      } else if (Array.isArray(value)) {\n        unminifiedObject[key] = await Promise.all(value.map((item) => this.unminify(item)));\n      } else if (isObject(value)) {\n        unminifiedObject[key] = await this.unminify(value);\n      } else {\n        unminifiedObject[key] = value;\n      }\n    }\n\n    return unminifiedObject;\n  }\n\n  protected isEntityRef(value: unknown): value is EntityRef {\n    return typeof value === 'object' && value !== null && '__ref' in value;\n  }\n\n  protected isEntityWithId(value: unknown): value is EntityWithId {\n    return value instanceof BaseEntity && 'id' in value;\n  }\n\n  public asyncIterator<T>(triggers: string | readonly string[]) {\n    return super.asyncIterableIterator(triggers) as AsyncIterator<T>;\n  }\n}\n","import { Global, Module } from '@nestjs/common';\n\nimport { PostgresPubSub } from './postgres-pubsub.js';\n\n/**\n * PostgresPubSubModule provides PostgreSQL NOTIFY/LISTEN based PubSub engine\n * for GraphQL subscriptions in NestJS applications.\n */\n@Global()\n@Module({\n  providers: [\n    PostgresPubSub,\n  ],\n  exports: [\n    PostgresPubSub,\n  ],\n})\nexport class PostgresPubSubModule {}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;AAwDO,2BAAMA,yBAAuB,aAAqC;CAQvE,AAAU;CACV,AAAU,KAAK,IAAI,cAAc;CACjC,AAAU,gBAAsC,EAAE;CAElD,AAAU;CACV,AAAmB,aAAa;CAChC,AAAU,mBAAmB;CAE7B,MAAM,eAAe;AACnB,OAAK,SAAS,KAAK,cAAc,OAAO,EACtC,OAAO,KAAK,YAAY,MACzB,CAAC;AAEF,OAAK,SAAS;GACZ,MAAM,KAAK,SAAS;GACpB,MAAM,KAAK,SAAS;GACpB,UAAU,KAAK,SAAS;GACxB,MAAM,KAAK,SAAS;GACpB,UAAU,KAAK,SAAS;GACxB,KAAK,KAAK,SAAS;GAGnB,yBAAyB;GAGzB,eAAe;GACf,mBAAmB;GAGnB,WAAW;GACX,6BAA6B;GAC9B;AAED,QAAM,KAAK,QAAQ;;CAIrB,MACM,wBAAwB;AAC5B,MAAI,KAAK,iBACP;AAGF,MAAI;AACF,SAAM,KAAK,SAAS,MAAM,0BAA0B;WAC7C,OAAO;AACd,OAAI,iBAAiB,MACnB,MAAK,OAAO,MAAM,8CAA8C,MAAM;AAGxE,SAAM,KAAK,QAAQ;;;CAIvB,MAAgB,SAAS;AACvB,OAAK,mBAAmB;AAExB,OAAK,OAAO,KAAK,8CAA8C;AAG/D,MAAI,KAAK,UAAU;AACjB,QAAK,SAAS,oBAAoB;AAClC,QAAK,SAAS,KAAK,UAAU,UAAU;AACrC,SAAK,OAAO,MAAM,0DAA0D,MAAM;KAClF;AACF,GAAK,KAAK,SAAS,KAAK;;AAG1B,OAAK,WAAW,MAAM,KAAK,WAAW;AAEtC,OAAK,OAAO,KAAK,uCAAuC;AAExD,OAAK,2BAA2B;AAEhC,OAAK,mBAAmB;;CAG1B,AAAU,4BAA4B;AACpC,OAAK,SAAS,GAAG,iBAAiB,YAA0B;AAC1D,GAAK,KAAK,oBAAoB,QAAQ;IACtC;AAEF,OAAK,SAAS,GAAG,aAAa;AAC5B,QAAK,OAAO,KAAK,uCAAuC;AACxD,OAAI,CAAC,KAAK,iBACR,CAAK,KAAK,QAAQ;IAEpB;AAEF,OAAK,SAAS,GAAG,UAAU,UAAiB;AAC1C,QAAK,OAAO,MAAM,qCAAqC,MAAM;AAE7D,OAAI,CAAC,KAAK,iBACR,CAAK,KAAK,QAAQ;IAEpB;;CAGJ,MAAgB,YAAY;AAC1B,OAAK,OAAO,KAAK,gEAAgE;EACjF,MAAM,YAAY,KAAK,KAAK;EAC5B,MAAM,eAAe;AAErB,OAAK,IAAI,UAAU,GAAG,UAAU,KAAK,YAAY,WAAW;AAC1D,QAAK,OAAO,KAAK,qCAAqC,OAAO,QAAQ,CAAC,KAAK;AAE3E,OAAI;IACF,MAAM,WAAW,IAAI,OAAO,KAAK,OAAO;IACxC,MAAM,aAAa,IAAI,SAAS,SAAS,WAAW;AAClD,cAAS,KAAK,WAAW,QAAQ;AACjC,cAAS,KAAK,aAAa;AACzB,aAAO,MAAM,oBAAoB,CAAC;OAClC;AACF,cAAS,KAAK,SAAS,OAAO;MAC9B;AACF,UAAM,QAAQ,IAAI,CAChB,SAAS,SAAS,EAClB,WACD,CAAC;AACF,SAAK,OAAO,KAAK,oCAAoC;AAErD,WAAO;YACA,OAAO;AACd,QAAI,iBAAiB,MACnB,MAAK,OAAO,MAAM,4CAA4C,MAAM;AAEtE,UAAM,KAAK,IAAI;AAEf,QAAK,KAAK,KAAK,GAAG,YAAa,aAC7B,OAAM,IAAI,MAAM,oDAAoD,OAAO,aAAa,CAAC,8BAA8B;;;AAK7H,QAAM,IAAI,MAAM,qEAAqE;;CAGvF,MAAa,QAAQ,aAAqB,SAAc,aAAa,GAAkB;AACrF,MAAI,CAAC,QACH,OAAM,IAAI,MAAM,+BAA+B;EAGjD,MAAM,UAAU,iBAAiB,YAAY;EAE7C,MAAM,UAAU,cAAc,KAAK,UAAU,KAAK,OAAO,QAAQ,CAAC,CAAC;AAQnE,MAAI;AACF,SAAM,KAAK,SAAS,MAAM,UAAU,QAAQ,IAAI,UAAU;WACnD,OAAO;AACd,OAAI,iBAAiB,OAAO;AAC1B,SAAK,OAAO,MAAM,qDAAqD,QAAQ,IAAI,OAAO,EACxF,SACD,CAAC;AAEF,QAAI,cAAc,KAAK,MAAM,QAAQ,EACnC;SAAI,aAAa,GAAG;AAClB,WAAK,OAAO,MAAM,gEAAgE,OAAO,WAAW,CAAC,GAAG;AACxG,YAAM,KAAK,SAAS,KAAK;AACzB,YAAM,KAAK,SAAS,SAAS;AAC7B,YAAM,KAAK,QAAQ,aAAa,SAAS,aAAa,EAAE;;;;;AAMhE,QAAM,QAAQ,SAAS;;CAGzB,MAAa,UAAU,aAAqB,UAAuD;AACjG,MAAI,CAAC,KAAK,cAAc,MAAM,MAAM,EAAE,YAAY,YAAY,EAAE;GAC9D,MAAM,UAAU,iBAAiB,YAAY;AAC7C,SAAM,KAAK,SAAS,MAAM,UAAU,UAAU;;AAGhD,OAAK,GAAG,GAAG,aAAa,SAAS;EAEjC,MAAM,QAAQ,KAAK,MAAM,KAAK,QAAQ,GAAG,IAAO;AAEhD,OAAK,cAAc,KAAK;GACtB,IAAI;GACJ,SAAS;GACT;GACD,CAAC;AAEF,SAAO,MAAM,QAAQ,QAAQ,MAAM;;CAGrC,MAAa,YAAY,OAAe;EACtC,MAAM,QAAQ,KAAK,cAAc,WAAW,MAAM,EAAE,OAAO,MAAM;AACjE,MAAI,UAAU,IAAI;GAChB,MAAM,eAAe,KAAK,cAAc;AACxC,QAAK,cAAc,OAAO,OAAO,EAAE;AACnC,QAAK,GAAG,eAAe,aAAa,SAAS,aAAa,SAAS;AAEnE,OAAI,CAAC,KAAK,cAAc,MAAM,MAAM,EAAE,YAAY,aAAa,QAAQ,EAAE;IACvE,MAAM,UAAU,iBAAiB,aAAa,QAAQ;AACtD,UAAM,KAAK,SAAS,MAAM,YAAY,UAAU;;;;CAKtD,MAAc,oBAAoB,SAAuB;AACvD,MAAI;GACF,MAAM,kBAAkB,KAAK,MAAM,QAAQ,WAAW,GAAG;AAOzD,OAAI,gBAAgB,GAClB,KAAI,OAAO,KAAK,gBAAgB,CAAC,WAAW,EAC1C,MAAK,GAAG,KAAK,QAAQ,SAAS,gBAAgB;OAE9C,MAAK,GAAG,KAAK,QAAQ,SAAS,MAAM,KAAK,SAAS,gBAAgB,CAAC;YAE5D,MAAM,QAAQ,gBAAgB,EAAE;IACzC,MAAM,UAAU,MAAM,KAAK,SAAS,gBAAgB;AAEpD,SAAK,GAAG,KAAK,QAAQ,SAAS,QAAQ;UACjC;IACL,MAAMC,UAAe,EACpB;AACD,SAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,gBAAgB,CACxD,SAAQ,OAAO,MAAM,KAAK,SAAS,MAAM;AAG3C,SAAK,GAAG,KAAK,QAAQ,SAAS,QAAQ;;WAEjC,OAAO;AACd,OAAI,iBAAiB,MACnB,MAAK,OAAO,MAAM,qDAAqD,QAAQ,WAAW,OAAO,EAC/F,SACD,CAAC;;;CAKR,AAAU,MAAM,QAAsC;AAMpD,MAAI,KAAK,eAAe,OAAO,CAC7B,QAAO,EACL,OAAO,GAAG,OAAO,YAAY,KAAK,GAAG,OAAO,MAC7C;AAGH,SAAO;;CAGT,AAAU,OAAO,SAAmB;AAClC,MAAI,MAAM,QAAQ,QAAQ,CACxB,QAAO,QAAQ,KAAK,SAAS,KAAK,OAAO,KAAK,CAAC;AAGjD,MAAI,CAAC,SAAS,QAAQ,CACpB,QAAO;AAQT,MAAI,KAAK,eAAe,QAAQ,CAG9B,QAFY,KAAK,MAAM,QAAQ,IAEhB;EAGjB,MAAMC,iBAA0C,EAC/C;AAED,OAAK,MAAM,OAAO,OAAO,KAAK,QAAQ,EAAE;GACtC,MAAM,QAAQ,QAAQ;AAEtB,OAAI,iBAAiB,KACnB,gBAAe,OAAO,MAAM,aAAa;YAChC,MAAM,QAAQ,MAAM,CAC7B,gBAAe,OAAO,MAAM,KAAK,SAAS,KAAK,OAAO,KAAK,CAAC;YACnD,SAAS,MAAM,CACxB,gBAAe,OAAO,KAAK,OAAO,MAAM;OAExC,gBAAe,OAAO;;AAI1B,SAAO;;CAGT,MAAgB,MAAM,WAA0D;EAC9E,MAAM,SAAS,KAAK,OAAO,MAAM;GAC/B,QAAQ,KAAK,MAAM;GACnB;GACD,CAAC;EAIF,MAAM,CAAC,YAAY,MAAM,UAAU,MAAM,MAAM,IAAI;AAEnD,MAAI,CAAC,cAAc,CAAC,IAAI;AACtB,UAAO,MAAM,eAAe,UAAU,QAAQ;AAE9C,UAAO,EACL,IAAI,UAAU,OACf;;EAGH,MAAM,iBAAiB,KAAK,WAAW,gBAAgB,MACpD,qBAAmBC,iBAAe,SAAS,WAC7C;AAED,MAAI,CAAC,gBAAgB,QAAQ;AAC3B,UAAO,MAAM,UAAU,WAAW,wBAAwB;AAE1D,UAAO,EACL,IACD;;EAGH,MAAM,WAAW,MAAM,KAAK,WAAW,cAAc,eAAe,OAAO,CAAC,UAAU,EACpF,IACD,CAAmC;AAEpC,MAAI,CAAC,UAAU;AACb,UAAO,MAAM,UAAU,WAAW,WAAW,GAAG,wBAAwB;AAExE,UAAO,EACL,IACD;;AAOH,SAAO;;CAGT,MAAgB,SAAS,SAA4B;AAMnD,MAAI,MAAM,QAAQ,QAAQ,CACxB,QAAO,MAAM,QAAQ,IAAI,QAAQ,KAAK,SAAS,KAAK,SAAS,KAAK,CAAC,CAAC;AAGtE,MAAI,KAAK,YAAY,QAAQ,CAC3B,QAAO,MAAM,KAAK,MAAM,QAAQ;AAGlC,MAAI,CAAC,SAAS,QAAQ,CACpB,QAAO;EAGT,MAAMC,mBAA4C,EACjD;AAED,OAAK,MAAM,OAAO,OAAO,KAAK,QAAQ,EAAE;GACtC,MAAM,QAAQ,QAAQ;AAEtB,OAAI,OAAO,UAAU,YAAY,CAAC,MAAM,KAAK,MAAM,MAAM,CAAC,CACxD,kBAAiB,OAAO,IAAI,KAAK,MAAM;YAC9B,KAAK,YAAY,MAAM,CAChC,kBAAiB,OAAO,MAAM,KAAK,MAAM,MAAM;YACtC,MAAM,QAAQ,MAAM,CAC7B,kBAAiB,OAAO,MAAM,QAAQ,IAAI,MAAM,KAAK,SAAS,KAAK,SAAS,KAAK,CAAC,CAAC;YAC1E,SAAS,MAAM,CACxB,kBAAiB,OAAO,MAAM,KAAK,SAAS,MAAM;OAElD,kBAAiB,OAAO;;AAI5B,SAAO;;CAGT,AAAU,YAAY,OAAoC;AACxD,SAAO,OAAO,UAAU,YAAY,UAAU,QAAQ,WAAW;;CAGnE,AAAU,eAAe,OAAuC;AAC9D,SAAO,iBAAiB,cAAc,QAAQ;;CAGhD,AAAO,cAAiB,UAAsC;AAC5D,SAAO,MAAM,sBAAsB,SAAS;;;YApZ7C,gBAAgB;YAEhB,qBAAqB;YAGrB,kBAAkB;;CAuClB,SAAS,4BAA4B,IAAI,IAAK;;;;;6BA9ChD,YAAY;;;;ACtCN,iCAAMC,uBAAqB;mCATjC,QAAQ,EACR,OAAO;CACN,WAAW,CACT,eACD;CACD,SAAS,CACP,eACD;CACF,CAAC"}

package/package.json

@@ -0,0 +1,41 @@
+{
+  "name": "@pcg/postgres-pubsub",
+  "version": "1.0.0-alpha.0",
+  "description": "PostgreSQL NOTIFY/LISTEN PubSub engine for NestJS and GraphQL subscriptions",
+  "license": "MIT",
+  "author": {
+    "email": "code@deepvision.team",
+    "name": "DeepVision Code"
+  },
+  "contributors": [
+    "Vitaliy Angolenko <v.angolenko@deepvision.software>",
+    "Sergii Sadovyi <s.sadovyi@deepvision.software>"
+  ],
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist"
+  ],
+  "dependencies": {
+    "graphql-subscriptions": "^3.0.0",
+    "pg": "^8.16.3",
+    "@pcg/core": "1.0.0-alpha.1",
+    "@pcg/predicates": "1.0.0-alpha.1"
+  },
+  "devDependencies": {
+    "@types/pg": "^8.16.0"
+  },
+  "peerDependencies": {
+    "@nestjs/common": "^11.0.0",
+    "@nestjs/config": "^4.0.0",
+    "@nestjs/schedule": "^5.0.0",
+    "@nestjs/typeorm": "^11.0.0",
+    "typeorm": "^0.3.0"
+  },
+  "scripts": {
+    "dev": "tsdown --watch",
+    "build": "tsdown",
+    "lint": "eslint \"src/**/*.ts\" --fix"
+  }
+}