opticedge-cloud-utils 1.1.17 → 1.1.18

package/dist/index.d.ts

@@ -7,6 +7,7 @@ export * from './auth';
 export * from './chunk';
 export * from './env';
 export * from './parser';
+export * from './pub';
 export * from './regex';
 export * from './retry';
 export * from './secrets';

package/dist/index.js

@@ -23,6 +23,7 @@ __exportStar(require("./auth"), exports);
 __exportStar(require("./chunk"), exports);
 __exportStar(require("./env"), exports);
 __exportStar(require("./parser"), exports);
+__exportStar(require("./pub"), exports);
 __exportStar(require("./regex"), exports);
 __exportStar(require("./retry"), exports);
 __exportStar(require("./secrets"), exports);

package/dist/pub.d.ts

@@ -0,0 +1 @@
+export declare function publishMessage(projectId: string, topicName: string, envelope: Record<string, unknown>): Promise<string>;

package/dist/pub.js

@@ -0,0 +1,38 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.publishMessage = publishMessage;
+// src/pub.ts
+const pubsub_1 = require("@google-cloud/pubsub");
+const pubsubCache = new Map();
+function getPubSub(projectId) {
+    if (!projectId)
+        throw new Error('projectId is required');
+    let ps = pubsubCache.get(projectId);
+    if (!ps) {
+        ps = new pubsub_1.PubSub({ projectId });
+        pubsubCache.set(projectId, ps);
+    }
+    return ps;
+}
+async function publishMessage(projectId, topicName, envelope) {
+    if (!projectId)
+        throw new Error('projectId is required');
+    if (!topicName)
+        throw new Error('topicName is required');
+    if (envelope === undefined)
+        throw new Error('envelope is required');
+    try {
+        const pubsub = getPubSub(projectId);
+        const topic = pubsub.topic(topicName);
+        const data = Buffer.from(JSON.stringify(envelope));
+        const messageId = await topic.publishMessage({
+            data
+        });
+        console.info(`INFO: Pub/Sub publish project=${projectId} topic=${topicName} msg_id=${messageId}`);
+        return messageId;
+    }
+    catch (err) {
+        console.error('ERROR: publish failed:', err);
+        throw err;
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opticedge-cloud-utils",
-  "version": "1.1.17",
+  "version": "1.1.18",
   "description": "Common utilities for cloud functions",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -16,6 +16,7 @@
   "author": "Evans Musonda",
   "license": "MIT",
   "dependencies": {
+    "@google-cloud/pubsub": "^5.2.0",
     "@google-cloud/secret-manager": "^6.0.1",
     "@google-cloud/tasks": "^6.1.0",
     "axios": "^1.10.0",

package/src/index.ts

@@ -7,6 +7,7 @@ export * from './auth'
 export * from './chunk'
 export * from './env'
 export * from './parser'
+export * from './pub'
 export * from './regex'
 export * from './retry'
 export * from './secrets'

package/src/pub.ts

@@ -0,0 +1,41 @@
+// src/pub.ts
+import { PubSub } from '@google-cloud/pubsub'
+
+const pubsubCache = new Map<string, PubSub>()
+
+function getPubSub(projectId: string): PubSub {
+  if (!projectId) throw new Error('projectId is required')
+  let ps = pubsubCache.get(projectId)
+  if (!ps) {
+    ps = new PubSub({ projectId })
+    pubsubCache.set(projectId, ps)
+  }
+  return ps
+}
+
+export async function publishMessage(
+  projectId: string,
+  topicName: string,
+  envelope: Record<string, unknown>
+): Promise<string> {
+  if (!projectId) throw new Error('projectId is required')
+  if (!topicName) throw new Error('topicName is required')
+  if (envelope === undefined) throw new Error('envelope is required')
+
+  try {
+    const pubsub = getPubSub(projectId)
+    const topic = pubsub.topic(topicName)
+    const data = Buffer.from(JSON.stringify(envelope))
+    const messageId = await topic.publishMessage({
+      data
+    })
+
+    console.info(
+      `INFO: Pub/Sub publish project=${projectId} topic=${topicName} msg_id=${messageId}`
+    )
+    return messageId
+  } catch (err) {
+    console.error('ERROR: publish failed:', err)
+    throw err
+  }
+}

package/tests/pub.test.ts

@@ -0,0 +1,85 @@
+// tests/pub.test.ts
+jest.mock('@google-cloud/pubsub', () => {
+  const instances: any[] = []
+
+  class PubSub {
+    opts: any
+    topics = new Map<string, any>()
+    constructor(opts?: any) {
+      this.opts = opts
+      instances.push(this)
+    }
+    topic(name: string) {
+      if (!this.topics.has(name)) {
+        const publishMessage = jest.fn(({ data }: { data: Buffer }) =>
+          Promise.resolve(`${this.opts?.projectId}-${name}`)
+        )
+        this.topics.set(name, { publishMessage })
+      }
+      return this.topics.get(name)
+    }
+  }
+
+  return { PubSub, __instances: instances }
+})
+
+const path = '../src/pub'
+
+describe('publishMessage', () => {
+  beforeEach(() => {
+    // Ensure a fresh module load so the mocked PubSub and src/pub share the same mock instance
+    jest.resetModules()
+  })
+
+  test('publishes and returns messageId, sends JSON buffer', async () => {
+    const mockPubsub = require('@google-cloud/pubsub')
+    const { publishMessage } = require(path)
+    const envelope = { hello: 'world' }
+
+    const msgId = await publishMessage('project-1', 'topic-a', envelope)
+    expect(msgId).toBe('project-1-topic-a')
+
+    const instance = mockPubsub.__instances[0]
+    expect(instance).toBeDefined()
+
+    const publishMock = instance.topics.get('topic-a').publishMessage
+    expect(publishMock).toHaveBeenCalledTimes(1)
+
+    const callArg = publishMock.mock.calls[0][0]
+    expect(callArg).toHaveProperty('data')
+    expect(callArg.data.toString()).toBe(JSON.stringify(envelope))
+  })
+
+  test('throws on missing arguments', async () => {
+    const { publishMessage } = require(path)
+    await expect(publishMessage('', 't', {})).rejects.toThrow('projectId is required')
+    await expect(publishMessage('p', '', {})).rejects.toThrow('topicName is required')
+    await expect(publishMessage('p', 't', undefined as any)).rejects.toThrow('envelope is required')
+  })
+
+  test('caches one PubSub instance per project', async () => {
+    const mockPubsub = require('@google-cloud/pubsub')
+    const { publishMessage } = require(path)
+
+    await publishMessage('project-1', 'topic-x', { a: 1 })
+    await publishMessage('project-1', 'topic-x', { b: 2 })
+
+    const instances = mockPubsub.__instances
+    expect(instances.length).toBe(1)
+    const publishMock = instances[0].topics.get('topic-x').publishMessage
+    expect(publishMock).toHaveBeenCalledTimes(2)
+  })
+
+  test('creates separate PubSub instances for different projects', async () => {
+    const mockPubsub = require('@google-cloud/pubsub')
+    const { publishMessage } = require(path)
+
+    await publishMessage('project-A', 't1', { x: 1 })
+    await publishMessage('project-B', 't2', { y: 2 })
+
+    const instances = mockPubsub.__instances
+    expect(instances.length).toBe(2)
+    expect(instances[0].opts).toEqual({ projectId: 'project-A' })
+    expect(instances[1].opts).toEqual({ projectId: 'project-B' })
+  })
+})