@mhubcloud/tms_core 3.4.1 → 3.4.2

package/dist/index.js

@@ -31,12 +31,23 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 var index_exports = {};
 __export(index_exports, {
   DefaultPublisher: () => defaultPublisher,
+  broker: () => broker_exports,
   publisher: () => publisher,
   testUtils: () => testUtils_exports,
   utils: () => utils_exports
 });
 module.exports = __toCommonJS(index_exports);
 
+// src/broker.ts
+var broker_exports = {};
+__export(broker_exports, {
+  dispatchEvent: () => dispatchEvent,
+  logToCosmosDb: () => logToCosmosDb,
+  resolveEvent: () => resolveEvent
+});
+var import_cosmos = require("@azure/cosmos");
+var import_service_bus = require("@azure/service-bus");
+
 // src/utils/utils.ts
 var utils_exports = {};
 __export(utils_exports, {
@@ -44,19 +55,19 @@ __export(utils_exports, {
   encodeEmail: () => encodeEmail,
   extractAdCode: () => extractAdCode,
   flatten: () => flatten,
+  formatError: () => formatError,
   handlePlatformRequest: () => handlePlatformRequest,
   handleResponse: () => handleResponse,
   logEvent: () => logEvent,
   logException: () => logException,
   logMessage: () => logMessage,
-  parseErrorMessage: () => parseErrorMessage,
   prepareTestRequestHeaders: () => prepareTestRequestHeaders,
   removeUndefined: () => removeUndefined
 });
 var crypto = __toESM(require("crypto"));
 var appInsights = __toESM(require("applicationinsights"));
 var aiClient = null;
-if (process.env.APPLICATIONINSIGHTS_CONNECTION_STRING || process.env.APPINSIGHTS_INSTRUMENTATIONKEY) {
+if (process.env.APPLICATIONINSIGHTS_CONNECTION_STRING) {
   appInsights.setup().setAutoCollectDependencies(false).setAutoCollectConsole(true, false).setSendLiveMetrics(false).setAutoCollectRequests(false).setAutoCollectPerformance(false).setUseDiskRetryCaching(false).start();
   aiClient = appInsights.defaultClient;
 }
@@ -89,15 +100,17 @@ function logMessage(level, message, properties) {
     properties
   });
 }
-function logEvent(publisher2, event, region) {
+function logEvent(publisher2, event, region, status, payload) {
   const properties = {
     publisher: publisher2,
     event,
-    region
+    region,
+    status,
+    payload
   };
   logMessage(1, "Event dispatched to platform", properties);
 }
-function logException(log2, exception) {
+function logException(exception) {
   aiClient?.trackException({ exception });
 }
 function removeUndefined(originalValue) {
@@ -211,7 +224,10 @@ async function handlePlatformRequest({
     // eslint-disable-next-line @typescript-eslint/naming-convention
     headers: { ...requestOptions.headers }
   };
-  const rawPlatformResponse = await fetch(fullUrl, fetchRequestOptions);
+  const rawPlatformResponse = await fetch(
+    fullUrl,
+    fetchRequestOptions
+  );
   if (process.env.PUBLISHER_TESTURL) {
     await handleTestRequest({
       fetchRequestOptions,
@@ -225,10 +241,22 @@ async function handlePlatformRequest({
       rawPlatformResponse,
       event.event
     );
-    logEvent(source, event.event, event.region);
+    logEvent(
+      source,
+      event.event,
+      event.region,
+      rawPlatformResponse.status,
+      requestOptions.body ? JSON.stringify(requestOptions.body) : void 0
+    );
     return processedPlatformResponse;
   } else {
-    logEvent(source, event.event, event.region);
+    logEvent(
+      source,
+      event.event,
+      event.region,
+      rawPlatformResponse.status,
+      requestOptions.body ? JSON.stringify(requestOptions.body) : void 0
+    );
   }
 }
 async function handleTestRequest(params) {
@@ -251,18 +279,137 @@ async function handleTestRequest(params) {
   } catch {
   }
 }
-function parseErrorMessage(error) {
-  let errorMessage;
+function formatError(error) {
+  let formattedError;
   if (error instanceof Error) {
+    formattedError = error;
     if (error.cause) {
-      errorMessage = `${error.message} Cause: ${error.cause.toString()}`;
+      formattedError.message = `${error.message} Cause: ${error.cause.toString()}`;
     } else {
-      errorMessage = error.message;
+      formattedError.message = error.message;
     }
   } else {
-    errorMessage = error;
+    formattedError = new Error(error);
+  }
+  return formattedError;
+}
+
+// src/broker.ts
+var serviceBusConnectionString = process.env.ServiceBusConnectionString;
+var cosmosConnectionString = process.env.CosmosDbConnectionString;
+var cosmosDatabaseName = process.env.CosmosDbDatabaseName;
+var cosmosContainerName = process.env.CosmosDbEventsContainerName;
+var serviceBusClient;
+var cosmosClient;
+var senders = {};
+function resolveEvent(event, publisherConfig) {
+  if (!publisherConfig) {
+    const envConfig = process.env.Broker_PublisherConfig;
+    publisherConfig = envConfig ? JSON.parse(envConfig) : {};
+  }
+  const publishers = [];
+  for (const publisher2 of Object.keys(publisherConfig)) {
+    const publisherEvents = publisherConfig[publisher2];
+    if (publisherEvents) {
+      if (publisherEvents[0] === "*") {
+        let matchesExcluded = false;
+        if (Array.isArray(publisherEvents[1]) && publisherEvents[1].length > 0) {
+          for (const publisherEvent of publisherEvents[1]) {
+            if (matchesExcluded) {
+              break;
+            }
+            matchesExcluded = matchEvent(event, publisherEvent);
+          }
+        }
+        if (!matchesExcluded) {
+          publishers.push(publisher2);
+        }
+      } else {
+        let match = false;
+        for (const publisherEvent of publisherEvents) {
+          if (Array.isArray(publisherEvent)) {
+            continue;
+          }
+          match = matchEvent(event, publisherEvent);
+          if (match) {
+            publishers.push(publisher2);
+          }
+        }
+      }
+    }
+  }
+  if (publishers.length === 0) {
+    return null;
+  }
+  return publishers;
+}
+function matchEvent(event, publisherEvent) {
+  if (typeof publisherEvent === "string") {
+    const regex = new RegExp(publisherEvent);
+    const match = regex.test(event.code);
+    return match;
+  } else if (typeof publisherEvent === "object") {
+    let paramMatch = true;
+    for (const key of Object.keys(publisherEvent)) {
+      const publisherEventParam = publisherEvent[key];
+      const eventParam = event[key];
+      if (typeof publisherEventParam === "string" && typeof eventParam === "string") {
+        const regex = new RegExp(publisherEventParam);
+        paramMatch = regex.test(eventParam);
+        if (!paramMatch) {
+          return false;
+        }
+      } else {
+        throw new Error("Unexpected formatting in publisher configuration");
+      }
+    }
+    return true;
+  } else {
+    throw new Error("Unexpected formatting in publisher configuration");
+  }
+}
+async function dispatchEvent(message, queues) {
+  if (!serviceBusClient) {
+    if (!serviceBusConnectionString) {
+      throw new Error("Service Bus connection string is missing.");
+    }
+    serviceBusClient = new import_service_bus.ServiceBusClient(serviceBusConnectionString);
+  }
+  await Promise.all(
+    queues.map(async (queue) => {
+      try {
+        let sender;
+        if (senders[queue]) {
+          sender = senders[queue];
+        } else {
+          senders[queue] = serviceBusClient.createSender(queue);
+          sender = senders[queue];
+        }
+        await sender.sendMessages({ body: message });
+      } catch (e) {
+        console.log(`Sending to queue ${queue} failed`);
+      }
+    })
+  );
+}
+async function logToCosmosDb(message) {
+  if (!cosmosClient) {
+    if (!cosmosConnectionString) {
+      throw new Error("Cosmos DB connection string is missing.");
+    }
+    cosmosClient = new import_cosmos.CosmosClient(cosmosConnectionString);
+  }
+  if (!cosmosDatabaseName) {
+    throw new Error("Cosmos DB database name is missing.");
+  }
+  if (!cosmosContainerName) {
+    throw new Error("Cosmos DB events container name is missing.");
+  }
+  try {
+    await cosmosClient.database(cosmosDatabaseName).container(cosmosContainerName).items.create(message);
+  } catch (e) {
+    logException(e);
   }
-  return errorMessage;
 }
 
 // src/defaultPublisher.ts
@@ -283,15 +430,21 @@ async function defaultPublisher(azureFunctionContext, serviceBusMessage, source,
   for (const result of results) {
     if (result.status === "rejected") {
       if (result.reason instanceof Error) {
-        const errorMessage = parseErrorMessage(result.reason);
-        logException(azureFunctionContext.log, new Error(errorMessage));
-        throw new Error(errorMessage);
+        const formattedError = formatError(result.reason);
+        logException(
+          azureFunctionContext.log,
+          new Error(formattedError.message)
+        );
+        throw formattedError;
       } else {
-        const errorMessage = parseErrorMessage(
+        const formattedError = formatError(
           typeof result.reason === "object" ? JSON.stringify(result.reason) : result.reason
         );
-        logException(azureFunctionContext.log, new Error(errorMessage));
-        throw new Error(errorMessage);
+        logException(
+          azureFunctionContext.log,
+          new Error(formattedError.message)
+        );
+        throw formattedError;
       }
     }
   }
@@ -532,7 +685,13 @@ var googleAds = {
     } else if (responseJson.error) {
       throw new Error(responseJson.error.message);
     } else {
-      logEvent(this.source, event.event, event.region);
+      logEvent(
+        this.source,
+        event.event,
+        event.region,
+        response.status,
+        requestOptions.body
+      );
     }
   },
   async getToken(config) {
@@ -754,6 +913,7 @@ var publisher = {
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   DefaultPublisher,
+  broker,
   publisher,
   testUtils,
   utils

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mhubcloud/tms_core",
-  "version": "3.4.1",
+  "version": "3.4.2",
   "description": "Core function for mHub Cloud TMS publishers",
   "homepage": "https://mhubcloud.com/",
   "license": "Apache 2.0",