@azure/event-hubs 6.0.2-alpha.20251014.1 → 6.0.2-alpha.20251021.1

package/dist/react-native/connectionContext.js

@@ -0,0 +1,310 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { ConnectionEvents } from "rhea-promise";
+import { ConnectionConfig, ConnectionContextBase, Constants, createSasTokenProvider, } from "@azure/core-amqp";
+import { parseEventHubConnectionString } from "./util/connectionStringUtils.js";
+import { ManagementClient } from "./managementClient.js";
+import { isNamedKeyCredential, isSASCredential } from "@azure/core-auth";
+import { logErrorStackTrace, logger } from "./logger.js";
+import { EventHubConnectionConfig } from "./eventhubConnectionConfig.js";
+import { getRuntimeInfo } from "./util/runtimeInfo.js";
+import { isCredential } from "./util/typeGuards.js";
+import { packageJsonInfo } from "./util/constants.js";
+import { createAbortablePromise } from "@azure/core-util";
+/**
+ * @internal
+ */
+export var ConnectionContext;
+(function (ConnectionContext) {
+    /**
+     * The user agent string for the EventHubs client.
+     * See guideline at https://github.com/Azure/azure-sdk/blob/main/docs/design/Telemetry.mdk
+     */
+    const userAgent = `azsdk-js-azureeventhubs/${packageJsonInfo.version} (${getRuntimeInfo()})`;
+    function getUserAgent(options) {
+        const finalUserAgent = options.userAgent ? `${userAgent},${options.userAgent}` : userAgent;
+        if (finalUserAgent.length > Constants.maxUserAgentLength) {
+            throw new Error(`The user-agent string cannot be more than ${Constants.maxUserAgentLength} characters in length.` +
+                `The given user-agent string is: ${finalUserAgent} with length: ${finalUserAgent.length}`);
+        }
+        return finalUserAgent;
+    }
+    ConnectionContext.getUserAgent = getUserAgent;
+    function create(config, tokenCredential, options) {
+        if (!options)
+            options = {};
+        config.webSocket = options.webSocketOptions && options.webSocketOptions.webSocket;
+        config.webSocketEndpointPath = "$servicebus/websocket";
+        config.webSocketConstructorOptions =
+            options.webSocketOptions && options.webSocketOptions.webSocketConstructorOptions;
+        const parameters = {
+            config: config,
+            // re-enabling this will be a post-GA discussion.
+            // dataTransformer: options.dataTransformer,
+            isEntityPathRequired: true,
+            connectionProperties: {
+                product: "MSJSClient",
+                userAgent: getUserAgent(options),
+                version: packageJsonInfo.version,
+            },
+        };
+        // Let us create the base context and then add EventHub specific ConnectionContext properties.
+        const connectionContext = ConnectionContextBase.create(parameters);
+        connectionContext.tokenCredential = tokenCredential;
+        connectionContext.wasConnectionCloseCalled = false;
+        connectionContext.senders = {};
+        connectionContext.receivers = {};
+        const mOptions = {
+            address: options.managementSessionAddress,
+            audience: options.managementSessionAudience,
+        };
+        connectionContext.managementSession = new ManagementClient(connectionContext, mOptions);
+        let waitForConnectionRefreshResolve;
+        let waitForConnectionRefreshPromise;
+        Object.assign(connectionContext, {
+            isConnectionClosing() {
+                // When the connection is not open, but the remote end is open,
+                // then the rhea connection is in the process of terminating.
+                return Boolean(!this.connection.isOpen() && this.connection.isRemoteOpen());
+            },
+            async readyToOpenLink(optionsArg) {
+                // Check that the connection isn't in the process of closing.
+                // This can happen when the idle timeout has been reached but
+                // the underlying socket is waiting to be destroyed.
+                if (this.isConnectionClosing()) {
+                    // Wait for the disconnected event that indicates the underlying socket has closed.
+                    await this.waitForDisconnectedEvent(optionsArg);
+                }
+                // Wait for the connection to be reset.
+                await this.waitForConnectionReset();
+            },
+            waitForDisconnectedEvent(optionsArg) {
+                return createAbortablePromise((resolve) => {
+                    logger.verbose(`[${this.connectionId}] Attempting to reinitialize connection` +
+                        ` but the connection is in the process of closing.` +
+                        ` Waiting for the disconnect event before continuing.`);
+                    this.connection.once(ConnectionEvents.disconnected, resolve);
+                }, optionsArg);
+            },
+            waitForConnectionReset() {
+                // Check if the connection is currently in the process of disconnecting.
+                if (waitForConnectionRefreshPromise) {
+                    return waitForConnectionRefreshPromise;
+                }
+                return Promise.resolve();
+            },
+            async close() {
+                try {
+                    if (this.connection.isOpen()) {
+                        // Close all the senders.
+                        await Promise.all(Object.keys(connectionContext.senders).map((name) => connectionContext.senders[name]?.close().catch(() => {
+                            /* error already logged, swallow it here */
+                        })));
+                        // Close all the receivers.
+                        await Promise.all(Object.keys(connectionContext.receivers).map((name) => connectionContext.receivers[name]?.close().catch(() => {
+                            /* error already logged, swallow it here */
+                        })));
+                        // Close the cbs session;
+                        await this.cbsSession.close();
+                        // Close the management session
+                        await this.managementSession?.close();
+                        await this.connection.close();
+                        this.wasConnectionCloseCalled = true;
+                        logger.info("Closed the amqp connection '%s' on the client.", this.connectionId);
+                    }
+                }
+                catch (err) {
+                    const errorDescription = err instanceof Error ? `${err.name}: ${err.message}` : JSON.stringify(err);
+                    logger.warning(`An error occurred while closing the connection "${this.connectionId}":\n${errorDescription}`);
+                    logErrorStackTrace(err);
+                    throw err;
+                }
+            },
+        });
+        // Define listeners to be added to the connection object for
+        // "connection_open" and "connection_error" events.
+        const onConnectionOpen = () => {
+            connectionContext.wasConnectionCloseCalled = false;
+            logger.verbose("[%s] setting 'wasConnectionCloseCalled' property of connection context to %s.", connectionContext.connection.id, connectionContext.wasConnectionCloseCalled);
+        };
+        async function handleDisconnection(context) {
+            if (waitForConnectionRefreshPromise) {
+                return;
+            }
+            waitForConnectionRefreshPromise = new Promise((resolve) => {
+                waitForConnectionRefreshResolve = resolve;
+            });
+            try {
+                logger.verbose("[%s] 'disconnected' event occurred on the amqp connection.", connectionContext.connection.id);
+                if (context.connection && context.connection.error) {
+                    logger.verbose("[%s] Accompanying error on the context.connection: %O", connectionContext.connection.id, context.connection.error);
+                }
+                if (context.error) {
+                    logger.verbose("[%s] Accompanying error on the context: %O", connectionContext.connection.id, context.error);
+                }
+                const state = {
+                    wasConnectionCloseCalled: connectionContext.wasConnectionCloseCalled,
+                    numSenders: Object.keys(connectionContext.senders).length,
+                    numReceivers: Object.keys(connectionContext.receivers).length,
+                };
+                logger.verbose("[%s] Closing all open senders and receivers in the state: %O", connectionContext.connection.id, state);
+                // Clear internal map maintained by rhea to avoid reconnecting of old links once the
+                // connection is back up.
+                connectionContext.connection.removeAllSessions();
+                // Close the cbs session to ensure all the event handlers are released.
+                await connectionContext.cbsSession?.close().catch(() => {
+                    /* error already logged, swallow it here */
+                });
+                // Close the management session to ensure all the event handlers are released.
+                await connectionContext.managementSession?.close().catch(() => {
+                    /* error already logged, swallow it here */
+                });
+                // Close all senders and receivers to ensure clean up of timers & other resources.
+                if (state.numSenders || state.numReceivers) {
+                    await Promise.all(Object.keys(connectionContext.senders).map((name) => connectionContext.senders[name]?.close().catch(() => {
+                        /* error already logged, swallow it here */
+                    })));
+                    await Promise.all(Object.keys(connectionContext.receivers).map((name) => connectionContext.receivers[name]?.close().catch(() => {
+                        /* error already logged, swallow it here */
+                    })));
+                }
+            }
+            catch (err) {
+                logger.verbose(`[${connectionContext.connectionId}] An error occurred while closing the connection in 'disconnected'. %O`, err);
+            }
+            try {
+                await refreshConnection(connectionContext);
+            }
+            catch (err) {
+                logger.verbose(`[${connectionContext.connectionId}] An error occurred while refreshing the connection in 'disconnected'. %O`, err);
+            }
+            finally {
+                waitForConnectionRefreshResolve();
+                waitForConnectionRefreshPromise = undefined;
+            }
+        }
+        const onDisconnected = (context) => {
+            // Fire-and-forget async operation with error handling
+            handleDisconnection(context).catch((err) => {
+                logger.verbose(`[${connectionContext.connectionId}] Unhandled error in 'disconnected' event handler. %O`, err);
+            });
+        };
+        const protocolError = (context) => {
+            logger.verbose("[%s] 'protocol_error' event occurred on the amqp connection.", connectionContext.connection.id);
+            if (context.connection && context.connection.error) {
+                logger.verbose("[%s] Accompanying error on the context.connection: %O", connectionContext.connection.id, context.connection && context.connection.error);
+            }
+            if (context.error) {
+                logger.verbose("[%s] Accompanying error on the context: %O", connectionContext.connection.id, context.error);
+            }
+        };
+        const error = (context) => {
+            logger.verbose("[%s] 'error' event occurred on the amqp connection.", connectionContext.connection.id);
+            if (context.connection && context.connection.error) {
+                logger.verbose("[%s] Accompanying error on the context.connection: %O", connectionContext.connection.id, context.connection && context.connection.error);
+            }
+            if (context.error) {
+                logger.verbose("[%s] Accompanying error on the context: %O", connectionContext.connection.id, context.error);
+            }
+        };
+        function addConnectionListeners(connection) {
+            // Add listeners on the connection object.
+            connection.on(ConnectionEvents.connectionOpen, onConnectionOpen);
+            connection.on(ConnectionEvents.disconnected, onDisconnected);
+            connection.on(ConnectionEvents.protocolError, protocolError);
+            connection.on(ConnectionEvents.error, error);
+        }
+        function cleanConnectionContext(context) {
+            // Remove listeners from the connection object.
+            context.connection.removeListener(ConnectionEvents.connectionOpen, onConnectionOpen);
+            context.connection.removeListener(ConnectionEvents.disconnected, onDisconnected);
+            context.connection.removeListener(ConnectionEvents.protocolError, protocolError);
+            context.connection.removeListener(ConnectionEvents.error, error);
+            // Close the connection
+            return context.connection.close();
+        }
+        async function refreshConnection(context) {
+            const originalConnectionId = context.connectionId;
+            try {
+                await cleanConnectionContext(context);
+            }
+            catch (err) {
+                logger.verbose(`[${context.connectionId}] There was an error closing the connection before reconnecting: %O`, err);
+            }
+            // Create a new connection, id, locks, and cbs client.
+            context.refreshConnection();
+            addConnectionListeners(context.connection);
+            logger.verbose(`The connection "${originalConnectionId}" has been updated to "${context.connectionId}".`);
+        }
+        addConnectionListeners(connectionContext.connection);
+        logger.verbose("[%s] Created connection context successfully.", connectionContext.connectionId);
+        return connectionContext;
+    }
+    ConnectionContext.create = create;
+})(ConnectionContext || (ConnectionContext = {}));
+/**
+ * Helper method to create a ConnectionContext from the input passed to either
+ * EventHubProducerClient or EventHubConsumerClient constructors
+ *
+ * @internal
+ */
+export function createConnectionContext(hostOrConnectionString, eventHubNameOrOptions, credentialOrOptions, options) {
+    let connectionString;
+    let config;
+    let credential;
+    hostOrConnectionString = String(hostOrConnectionString);
+    if (!isCredential(credentialOrOptions)) {
+        const parsedCS = parseEventHubConnectionString(hostOrConnectionString);
+        if (!(parsedCS.eventHubName ||
+            (typeof eventHubNameOrOptions === "string" && eventHubNameOrOptions))) {
+            throw new TypeError(`Either provide "eventHubName" or the "connectionString": "${hostOrConnectionString}", ` +
+                `must contain "EntityPath=<your-event-hub-name>".`);
+        }
+        if (parsedCS.eventHubName &&
+            typeof eventHubNameOrOptions === "string" &&
+            eventHubNameOrOptions &&
+            parsedCS.eventHubName !== eventHubNameOrOptions) {
+            throw new TypeError(`The entity path "${parsedCS.eventHubName}" in connectionString: "${hostOrConnectionString}" ` +
+                `doesn't match with eventHubName: "${eventHubNameOrOptions}".`);
+        }
+        connectionString = hostOrConnectionString;
+        if (typeof eventHubNameOrOptions !== "string") {
+            // connectionstring and/or options were passed to constructor
+            config = EventHubConnectionConfig.create(connectionString);
+            options = eventHubNameOrOptions;
+        }
+        else {
+            // connectionstring, eventHubName and/or options were passed to constructor
+            const eventHubName = eventHubNameOrOptions;
+            config = EventHubConnectionConfig.create(connectionString, eventHubName);
+            options = credentialOrOptions;
+        }
+        const parsed = parseEventHubConnectionString(connectionString);
+        // Since connectionString was passed, create a TokenProvider.
+        credential = createSasTokenProvider(parsed);
+    }
+    else {
+        // host, eventHubName, a TokenCredential and/or options were passed to constructor
+        const eventHubName = eventHubNameOrOptions;
+        let host = hostOrConnectionString;
+        if (isNamedKeyCredential(credentialOrOptions) || isSASCredential(credentialOrOptions)) {
+            credential = createSasTokenProvider(credentialOrOptions);
+        }
+        else {
+            credential = credentialOrOptions;
+        }
+        if (!eventHubName) {
+            throw new TypeError(`"eventHubName" is missing`);
+        }
+        if (!host.endsWith("/"))
+            host += "/";
+        connectionString = `Endpoint=sb://${host};SharedAccessKeyName=defaultKeyName;SharedAccessKey=defaultKeyValue;EntityPath=${eventHubName}`;
+        config = EventHubConnectionConfig.create(connectionString);
+    }
+    if (options?.customEndpointAddress) {
+        EventHubConnectionConfig.setCustomEndpointAddress(config, options.customEndpointAddress);
+    }
+    ConnectionConfig.validate(config);
+    return ConnectionContext.create(config, credential, options);
+}
+//# sourceMappingURL=connectionContext.js.map

package/dist/react-native/connectionContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"connectionContext.js","sourceRoot":"","sources":["../../src/connectionContext.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAMlC,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAEhD,OAAO,EACL,gBAAgB,EAChB,qBAAqB,EACrB,SAAS,EACT,sBAAsB,GACvB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,6BAA6B,EAAE,MAAM,iCAAiC,CAAC;AAEhF,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAEzD,OAAO,EAAE,oBAAoB,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AACzE,OAAO,EAAE,kBAAkB,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAEzD,OAAO,EAAE,wBAAwB,EAAE,MAAM,+BAA+B,CAAC;AAGzE,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAEtD,OAAO,EAAE,sBAAsB,EAAE,MAAM,kBAAkB,CAAC;AAwG1D;;GAEG;AACH,MAAM,KAAW,iBAAiB,CA4UjC;AA5UD,WAAiB,iBAAiB;IAChC;;;OAGG;IACH,MAAM,SAAS,GAAW,2BACxB,eAAe,CAAC,OAClB,KAAK,cAAc,EAAE,GAAG,CAAC;IAEzB,SAAgB,YAAY,CAAC,OAAiC;QAC5D,MAAM,cAAc,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,SAAS,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3F,IAAI,cAAc,CAAC,MAAM,GAAG,SAAS,CAAC,kBAAkB,EAAE,CAAC;YACzD,MAAM,IAAI,KAAK,CACb,6CAA6C,SAAS,CAAC,kBAAkB,wBAAwB;gBAC/F,mCAAmC,cAAc,iBAAiB,cAAc,CAAC,MAAM,EAAE,CAC5F,CAAC;QACJ,CAAC;QACD,OAAO,cAAc,CAAC;IACxB,CAAC;IATe,8BAAY,eAS3B,CAAA;IAED,SAAgB,MAAM,CACpB,MAAgC,EAChC,eAAmD,EACnD,OAAkC;QAElC,IAAI,CAAC,OAAO;YAAE,OAAO,GAAG,EAAE,CAAC;QAE3B,MAAM,CAAC,SAAS,GAAG,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,gBAAgB,CAAC,SAAS,CAAC;QAClF,MAAM,CAAC,qBAAqB,GAAG,uBAAuB,CAAC;QACvD,MAAM,CAAC,2BAA2B;YAChC,OAAO,CAAC,gBAAgB,IAAI,OAAO,CAAC,gBAAgB,CAAC,2BAA2B,CAAC;QAEnF,MAAM,UAAU,GAA0C;YACxD,MAAM,EAAE,MAAM;YACd,iDAAiD;YACjD,4CAA4C;YAC5C,oBAAoB,EAAE,IAAI;YAC1B,oBAAoB,EAAE;gBACpB,OAAO,EAAE,YAAY;gBACrB,SAAS,EAAE,YAAY,CAAC,OAAO,CAAC;gBAChC,OAAO,EAAE,eAAe,CAAC,OAAO;aACjC;SACF,CAAC;QACF,8FAA8F;QAC9F,MAAM,iBAAiB,GAAG,qBAAqB,CAAC,MAAM,CAAC,UAAU,CAAsB,CAAC;QACxF,iBAAiB,CAAC,eAAe,GAAG,eAAe,CAAC;QACpD,iBAAiB,CAAC,wBAAwB,GAAG,KAAK,CAAC;QACnD,iBAAiB,CAAC,OAAO,GAAG,EAAE,CAAC;QAC/B,iBAAiB,CAAC,SAAS,GAAG,EAAE,CAAC;QACjC,MAAM,QAAQ,GAA4B;YACxC,OAAO,EAAE,OAAO,CAAC,wBAAwB;YACzC,QAAQ,EAAE,OAAO,CAAC,yBAAyB;SAC5C,CAAC;QACF,iBAAiB,CAAC,iBAAiB,GAAG,IAAI,gBAAgB,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC;QAExF,IAAI,+BAA2C,CAAC;QAChD,IAAI,+BAA0D,CAAC;QAE/D,MAAM,CAAC,MAAM,CAA8C,iBAAiB,EAAE;YAC5E,mBAAmB;gBACjB,+DAA+D;gBAC/D,6DAA6D;gBAC7D,OAAO,OAAO,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,IAAI,IAAI,CAAC,UAAU,CAAC,YAAY,EAAE,CAAC,CAAC;YAC9E,CAAC;YACD,KAAK,CAAC,eAAe,CAAC,UAA8C;gBAClE,6DAA6D;gBAC7D,6DAA6D;gBAC7D,oDAAoD;gBACpD,IAAI,IAAI,CAAC,mBAAmB,EAAE,EAAE,CAAC;oBAC/B,mFAAmF;oBACnF,MAAM,IAAI,CAAC,wBAAwB,CAAC,UAAU,CAAC,CAAC;gBAClD,CAAC;gBAED,uCAAuC;gBACvC,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACtC,CAAC;YACD,wBAAwB,CAAC,UAA8C;gBACrE,OAAO,sBAAsB,CAAC,CAAC,OAAO,EAAE,EAAE;oBACxC,MAAM,CAAC,OAAO,CACZ,IAAI,IAAI,CAAC,YAAY,yCAAyC;wBAC5D,mDAAmD;wBACnD,sDAAsD,CACzD,CAAC;oBACF,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;gBAC/D,CAAC,EAAE,UAAU,CAAC,CAAC;YACjB,CAAC;YACD,sBAAsB;gBACpB,wEAAwE;gBACxE,IAAI,+BAA+B,EAAE,CAAC;oBACpC,OAAO,+BAA+B,CAAC;gBACzC,CAAC;gBACD,OAAO,OAAO,CAAC,OAAO,EAAE,CAAC;YAC3B,CAAC;YACD,KAAK,CAAC,KAAK;gBACT,IAAI,CAAC;oBACH,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,EAAE,CAAC;wBAC7B,yBAAyB;wBACzB,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAClD,iBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;4BAClD,2CAA2C;wBAC7C,CAAC,CAAC,CACH,CACF,CAAC;wBACF,2BAA2B;wBAC3B,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACpD,iBAAiB,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;4BACpD,2CAA2C;wBAC7C,CAAC,CAAC,CACH,CACF,CAAC;wBACF,yBAAyB;wBACzB,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;wBAC9B,+BAA+B;wBAC/B,MAAM,IAAI,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAC;wBACtC,MAAM,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;wBAC9B,IAAI,CAAC,wBAAwB,GAAG,IAAI,CAAC;wBACrC,MAAM,CAAC,IAAI,CAAC,gDAAgD,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC;oBACnF,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAQ,EAAE,CAAC;oBAClB,MAAM,gBAAgB,GACpB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,IAAI,KAAK,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;oBAC7E,MAAM,CAAC,OAAO,CACZ,mDAAmD,IAAI,CAAC,YAAY,OAAO,gBAAgB,EAAE,CAC9F,CAAC;oBACF,kBAAkB,CAAC,GAAG,CAAC,CAAC;oBACxB,MAAM,GAAG,CAAC;gBACZ,CAAC;YACH,CAAC;SACF,CAAC,CAAC;QAEH,4DAA4D;QAC5D,mDAAmD;QACnD,MAAM,gBAAgB,GAAgB,GAAG,EAAE;YACzC,iBAAiB,CAAC,wBAAwB,GAAG,KAAK,CAAC;YACnD,MAAM,CAAC,OAAO,CACZ,+EAA+E,EAC/E,iBAAiB,CAAC,UAAU,CAAC,EAAE,EAC/B,iBAAiB,CAAC,wBAAwB,CAC3C,CAAC;QACJ,CAAC,CAAC;QAEF,KAAK,UAAU,mBAAmB,CAAC,OAAqB;YACtD,IAAI,+BAA+B,EAAE,CAAC;gBACpC,OAAO;YACT,CAAC;YACD,+BAA+B,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBACxD,+BAA+B,GAAG,OAAO,CAAC;YAC5C,CAAC,CAAC,CAAC;YACH,IAAI,CAAC;gBACH,MAAM,CAAC,OAAO,CACZ,4DAA4D,EAC5D,iBAAiB,CAAC,UAAU,CAAC,EAAE,CAChC,CAAC;gBAEF,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;oBACnD,MAAM,CAAC,OAAO,CACZ,uDAAuD,EACvD,iBAAiB,CAAC,UAAU,CAAC,EAAE,EAC/B,OAAO,CAAC,UAAU,CAAC,KAAK,CACzB,CAAC;gBACJ,CAAC;gBACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;oBAClB,MAAM,CAAC,OAAO,CACZ,4CAA4C,EAC5C,iBAAiB,CAAC,UAAU,CAAC,EAAE,EAC/B,OAAO,CAAC,KAAK,CACd,CAAC;gBACJ,CAAC;gBACD,MAAM,KAAK,GAIN;oBACH,wBAAwB,EAAE,iBAAiB,CAAC,wBAAwB;oBACpE,UAAU,EAAE,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,MAAM;oBACzD,YAAY,EAAE,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC,MAAM;iBAC9D,CAAC;gBACF,MAAM,CAAC,OAAO,CACZ,8DAA8D,EAC9D,iBAAiB,CAAC,UAAU,CAAC,EAAE,EAC/B,KAAK,CACN,CAAC;gBAEF,oFAAoF;gBACpF,yBAAyB;gBACzB,iBAAiB,CAAC,UAAU,CAAC,iBAAiB,EAAE,CAAC;gBAEjD,uEAAuE;gBACvE,MAAM,iBAAiB,CAAC,UAAU,EAAE,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;oBACrD,2CAA2C;gBAC7C,CAAC,CAAC,CAAC;gBACH,8EAA8E;gBAC9E,MAAM,iBAAiB,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;oBAC5D,2CAA2C;gBAC7C,CAAC,CAAC,CAAC;gBAEH,kFAAkF;gBAClF,IAAI,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;oBAC3C,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAClD,iBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;wBAClD,2CAA2C;oBAC7C,CAAC,CAAC,CACH,CACF,CAAC;oBAEF,MAAM,OAAO,CAAC,GAAG,CACf,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CACpD,iBAAiB,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE;wBACpD,2CAA2C;oBAC7C,CAAC,CAAC,CACH,CACF,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,MAAM,CAAC,OAAO,CACZ,IAAI,iBAAiB,CAAC,YAAY,wEAAwE,EAC1G,GAAG,CACJ,CAAC;YACJ,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;YAC7C,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,MAAM,CAAC,OAAO,CACZ,IAAI,iBAAiB,CAAC,YAAY,2EAA2E,EAC7G,GAAG,CACJ,CAAC;YACJ,CAAC;oBAAS,CAAC;gBACT,+BAA+B,EAAE,CAAC;gBAClC,+BAA+B,GAAG,SAAS,CAAC;YAC9C,CAAC;QACH,CAAC;QAED,MAAM,cAAc,GAAgB,CAAC,OAAqB,EAAE,EAAE;YAC5D,sDAAsD;YACtD,mBAAmB,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACzC,MAAM,CAAC,OAAO,CACZ,IAAI,iBAAiB,CAAC,YAAY,uDAAuD,EACzF,GAAG,CACJ,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,MAAM,aAAa,GAAgB,CAAC,OAAqB,EAAE,EAAE;YAC3D,MAAM,CAAC,OAAO,CACZ,8DAA8D,EAC9D,iBAAiB,CAAC,UAAU,CAAC,EAAE,CAChC,CAAC;YAEF,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;gBACnD,MAAM,CAAC,OAAO,CACZ,uDAAuD,EACvD,iBAAiB,CAAC,UAAU,CAAC,EAAE,EAC/B,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAC/C,CAAC;YACJ,CAAC;YACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAClB,MAAM,CAAC,OAAO,CACZ,4CAA4C,EAC5C,iBAAiB,CAAC,UAAU,CAAC,EAAE,EAC/B,OAAO,CAAC,KAAK,CACd,CAAC;YACJ,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,KAAK,GAAgB,CAAC,OAAqB,EAAE,EAAE;YACnD,MAAM,CAAC,OAAO,CACZ,qDAAqD,EACrD,iBAAiB,CAAC,UAAU,CAAC,EAAE,CAChC,CAAC;YAEF,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;gBACnD,MAAM,CAAC,OAAO,CACZ,uDAAuD,EACvD,iBAAiB,CAAC,UAAU,CAAC,EAAE,EAC/B,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,KAAK,CAC/C,CAAC;YACJ,CAAC;YACD,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;gBAClB,MAAM,CAAC,OAAO,CACZ,4CAA4C,EAC5C,iBAAiB,CAAC,UAAU,CAAC,EAAE,EAC/B,OAAO,CAAC,KAAK,CACd,CAAC;YACJ,CAAC;QACH,CAAC,CAAC;QAEF,SAAS,sBAAsB,CAAC,UAAsB;YACpD,0CAA0C;YAC1C,UAAU,CAAC,EAAE,CAAC,gBAAgB,CAAC,cAAc,EAAE,gBAAgB,CAAC,CAAC;YACjE,UAAU,CAAC,EAAE,CAAC,gBAAgB,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;YAC7D,UAAU,CAAC,EAAE,CAAC,gBAAgB,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;YAC7D,UAAU,CAAC,EAAE,CAAC,gBAAgB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;QAC/C,CAAC;QAED,SAAS,sBAAsB,CAAC,OAA0B;YACxD,+CAA+C;YAC/C,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,gBAAgB,CAAC,cAAc,EAAE,gBAAgB,CAAC,CAAC;YACrF,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,gBAAgB,CAAC,YAAY,EAAE,cAAc,CAAC,CAAC;YACjF,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,gBAAgB,CAAC,aAAa,EAAE,aAAa,CAAC,CAAC;YACjF,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,gBAAgB,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACjE,uBAAuB;YACvB,OAAO,OAAO,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QACpC,CAAC;QAED,KAAK,UAAU,iBAAiB,CAAC,OAA0B;YACzD,MAAM,oBAAoB,GAAG,OAAO,CAAC,YAAY,CAAC;YAClD,IAAI,CAAC;gBACH,MAAM,sBAAsB,CAAC,OAAO,CAAC,CAAC;YACxC,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,MAAM,CAAC,OAAO,CACZ,IAAI,OAAO,CAAC,YAAY,qEAAqE,EAC7F,GAAG,CACJ,CAAC;YACJ,CAAC;YAED,sDAAsD;YACtD,OAAO,CAAC,iBAAiB,EAAE,CAAC;YAC5B,sBAAsB,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC3C,MAAM,CAAC,OAAO,CACZ,mBAAmB,oBAAoB,0BAA0B,OAAO,CAAC,YAAY,IAAI,CAC1F,CAAC;QACJ,CAAC;QAED,sBAAsB,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QAErD,MAAM,CAAC,OAAO,CAAC,+CAA+C,EAAE,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAChG,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAvTe,wBAAM,SAuTrB,CAAA;AACH,CAAC,EA5UgB,iBAAiB,KAAjB,iBAAiB,QA4UjC;AAED;;;;;GAKG;AACH,MAAM,UAAU,uBAAuB,CACrC,sBAA8B,EAC9B,qBAAsD,EACtD,mBAIyB,EACzB,OAA+B;IAE/B,IAAI,gBAAgB,CAAC;IACrB,IAAI,MAAM,CAAC;IACX,IAAI,UAA8C,CAAC;IACnD,sBAAsB,GAAG,MAAM,CAAC,sBAAsB,CAAC,CAAC;IAExD,IAAI,CAAC,YAAY,CAAC,mBAAmB,CAAC,EAAE,CAAC;QACvC,MAAM,QAAQ,GAAG,6BAA6B,CAAC,sBAAsB,CAAC,CAAC;QACvE,IACE,CAAC,CACC,QAAQ,CAAC,YAAY;YACrB,CAAC,OAAO,qBAAqB,KAAK,QAAQ,IAAI,qBAAqB,CAAC,CACrE,EACD,CAAC;YACD,MAAM,IAAI,SAAS,CACjB,6DAA6D,sBAAsB,KAAK;gBACtF,kDAAkD,CACrD,CAAC;QACJ,CAAC;QACD,IACE,QAAQ,CAAC,YAAY;YACrB,OAAO,qBAAqB,KAAK,QAAQ;YACzC,qBAAqB;YACrB,QAAQ,CAAC,YAAY,KAAK,qBAAqB,EAC/C,CAAC;YACD,MAAM,IAAI,SAAS,CACjB,oBAAoB,QAAQ,CAAC,YAAY,2BAA2B,sBAAsB,IAAI;gBAC5F,qCAAqC,qBAAqB,IAAI,CACjE,CAAC;QACJ,CAAC;QACD,gBAAgB,GAAG,sBAAsB,CAAC;QAC1C,IAAI,OAAO,qBAAqB,KAAK,QAAQ,EAAE,CAAC;YAC9C,6DAA6D;YAC7D,MAAM,GAAG,wBAAwB,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAC3D,OAAO,GAAG,qBAAqB,CAAC;QAClC,CAAC;aAAM,CAAC;YACN,2EAA2E;YAC3E,MAAM,YAAY,GAAG,qBAAqB,CAAC;YAC3C,MAAM,GAAG,wBAAwB,CAAC,MAAM,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;YACzE,OAAO,GAAG,mBAAmB,CAAC;QAChC,CAAC;QAED,MAAM,MAAM,GAAG,6BAA6B,CAAC,gBAAgB,CAG5D,CAAC;QACF,6DAA6D;QAC7D,UAAU,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;IAC9C,CAAC;SAAM,CAAC;QACN,kFAAkF;QAClF,MAAM,YAAY,GAAG,qBAAqB,CAAC;QAC3C,IAAI,IAAI,GAAG,sBAAsB,CAAC;QAClC,IAAI,oBAAoB,CAAC,mBAAmB,CAAC,IAAI,eAAe,CAAC,mBAAmB,CAAC,EAAE,CAAC;YACtF,UAAU,GAAG,sBAAsB,CAAC,mBAAmB,CAAC,CAAC;QAC3D,CAAC;aAAM,CAAC;YACN,UAAU,GAAG,mBAAmB,CAAC;QACnC,CAAC;QACD,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,MAAM,IAAI,SAAS,CAAC,2BAA2B,CAAC,CAAC;QACnD,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;YAAE,IAAI,IAAI,GAAG,CAAC;QACrC,gBAAgB,GAAG,iBAAiB,IAAI,kFAAkF,YAAY,EAAE,CAAC;QACzI,MAAM,GAAG,wBAAwB,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;IAC7D,CAAC;IAED,IAAI,OAAO,EAAE,qBAAqB,EAAE,CAAC;QACnC,wBAAwB,CAAC,wBAAwB,CAAC,MAAM,EAAE,OAAO,CAAC,qBAAqB,CAAC,CAAC;IAC3F,CAAC;IAED,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAElC,OAAO,iBAAiB,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;AAC/D,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\n/* eslint-disable @typescript-eslint/no-namespace */\n/* eslint-disable no-inner-declarations */\n\nimport type { Connection, Dictionary, EventContext, OnAmqpEvent } from \"rhea-promise\";\nimport { ConnectionEvents } from \"rhea-promise\";\nimport type { CreateConnectionContextBaseParameters, SasTokenProvider } from \"@azure/core-amqp\";\nimport {\n  ConnectionConfig,\n  ConnectionContextBase,\n  Constants,\n  createSasTokenProvider,\n} from \"@azure/core-amqp\";\nimport type { EventHubConnectionStringProperties } from \"./util/connectionStringUtils.js\";\nimport { parseEventHubConnectionString } from \"./util/connectionStringUtils.js\";\nimport type { ManagementClientOptions } from \"./managementClient.js\";\nimport { ManagementClient } from \"./managementClient.js\";\nimport type { NamedKeyCredential, SASCredential, TokenCredential } from \"@azure/core-auth\";\nimport { isNamedKeyCredential, isSASCredential } from \"@azure/core-auth\";\nimport { logErrorStackTrace, logger } from \"./logger.js\";\nimport type { EventHubClientOptions } from \"./models/public.js\";\nimport { EventHubConnectionConfig } from \"./eventhubConnectionConfig.js\";\nimport type { PartitionReceiver } from \"./partitionReceiver.js\";\nimport type { EventHubSender } from \"./eventHubSender.js\";\nimport { getRuntimeInfo } from \"./util/runtimeInfo.js\";\nimport { isCredential } from \"./util/typeGuards.js\";\nimport { packageJsonInfo } from \"./util/constants.js\";\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport { createAbortablePromise } from \"@azure/core-util\";\n\n/**\n * @internal\n * Provides contextual information like the underlying amqp connection, cbs session, management session,\n * tokenProvider, senders, receivers, etc. about the EventHub client.\n */\nexport interface ConnectionContext extends ConnectionContextBase {\n  /**\n   * The EventHub connection config that is created after\n   * parsing the connection string.\n   */\n  readonly config: EventHubConnectionConfig;\n  /**\n   * The credential to be used for Authentication.\n   * Default value: SasTokenProvider.\n   */\n  tokenCredential: SasTokenProvider | TokenCredential;\n  /**\n   * Indicates whether the close() method was\n   * called on theconnection object.\n   */\n  wasConnectionCloseCalled: boolean;\n  /**\n   * A dictionary of the EventHub Receivers associated with this client.\n   */\n  receivers: Dictionary<PartitionReceiver>;\n  /**\n   * A dictionary of the EventHub Senders associated with this client.\n   */\n  senders: Dictionary<EventHubSender>;\n  /**\n   * A reference to the management session ($management endpoint) on\n   * the underlying amqp connection for the EventHub Client.\n   */\n  managementSession?: ManagementClient;\n  /**\n   * Function returning a promise that resolves once the connectionContext is ready to open an AMQP link.\n   * ConnectionContext will be ready to open an AMQP link when:\n   * - The AMQP connection is already open on both sides.\n   * - The AMQP connection has been closed or disconnected. In this case, a new AMQP connection is expected\n   * to be created first.\n   * An AMQP link cannot be opened if the AMQP connection\n   * is in the process of closing or disconnecting.\n   */\n  readyToOpenLink(options?: { abortSignal?: AbortSignalLike }): Promise<void>;\n  /**\n   * Closes all AMQP links, sessions and connection.\n   */\n  close(): Promise<void>;\n}\n\n/**\n * Describes the members on the ConnectionContext that are only\n * used by it internally.\n * @internal\n */\nexport interface ConnectionContextInternalMembers extends ConnectionContext {\n  /**\n   * Indicates whether the connection is in the process of closing.\n   * When this returns `true`, a `disconnected` event will be received\n   * after the connection is closed.\n   *\n   */\n  isConnectionClosing(): boolean;\n  /**\n   * Resolves once the context's connection emits a `disconnected` event.\n   */\n  waitForDisconnectedEvent(options?: { abortSignal?: AbortSignalLike }): Promise<void>;\n  /**\n   * Resolves once the connection has finished being reset.\n   * Connections are reset as part of reacting to a `disconnected` event.\n   */\n  waitForConnectionReset(): Promise<void>;\n}\n\n/**\n * @internal\n */\nexport interface ConnectionContextOptions extends EventHubClientOptions {\n  managementSessionAddress?: string;\n  managementSessionAudience?: string;\n}\n\n/**\n * Helper type to get the names of all the functions on an object.\n */\ntype FunctionPropertyNames<T> = { [K in keyof T]: T[K] extends Function ? K : never }[keyof T]; // eslint-disable-line @typescript-eslint/no-unsafe-function-type\n/**\n * Helper type to get the types of all the functions on an object.\n */\ntype FunctionProperties<T> = Pick<T, FunctionPropertyNames<T>>;\n/**\n * Helper type to get the types of all the functions on ConnectionContext\n * and the internal methods from ConnectionContextInternalMembers.\n * Note that this excludes the functions that ConnectionContext inherits.\n * Each function also has its `this` type set as `ConnectionContext`.\n */\ntype ConnectionContextMethods = Omit<\n  FunctionProperties<ConnectionContextInternalMembers>,\n  FunctionPropertyNames<ConnectionContextBase>\n> &\n  ThisType<ConnectionContextInternalMembers>;\n\n/**\n * @internal\n */\nexport namespace ConnectionContext {\n  /**\n   * The user agent string for the EventHubs client.\n   * See guideline at https://github.com/Azure/azure-sdk/blob/main/docs/design/Telemetry.mdk\n   */\n  const userAgent: string = `azsdk-js-azureeventhubs/${\n    packageJsonInfo.version\n  } (${getRuntimeInfo()})`;\n\n  export function getUserAgent(options: ConnectionContextOptions): string {\n    const finalUserAgent = options.userAgent ? `${userAgent},${options.userAgent}` : userAgent;\n    if (finalUserAgent.length > Constants.maxUserAgentLength) {\n      throw new Error(\n        `The user-agent string cannot be more than ${Constants.maxUserAgentLength} characters in length.` +\n          `The given user-agent string is: ${finalUserAgent} with length: ${finalUserAgent.length}`,\n      );\n    }\n    return finalUserAgent;\n  }\n\n  export function create(\n    config: EventHubConnectionConfig,\n    tokenCredential: SasTokenProvider | TokenCredential,\n    options?: ConnectionContextOptions,\n  ): ConnectionContext {\n    if (!options) options = {};\n\n    config.webSocket = options.webSocketOptions && options.webSocketOptions.webSocket;\n    config.webSocketEndpointPath = \"$servicebus/websocket\";\n    config.webSocketConstructorOptions =\n      options.webSocketOptions && options.webSocketOptions.webSocketConstructorOptions;\n\n    const parameters: CreateConnectionContextBaseParameters = {\n      config: config,\n      // re-enabling this will be a post-GA discussion.\n      // dataTransformer: options.dataTransformer,\n      isEntityPathRequired: true,\n      connectionProperties: {\n        product: \"MSJSClient\",\n        userAgent: getUserAgent(options),\n        version: packageJsonInfo.version,\n      },\n    };\n    // Let us create the base context and then add EventHub specific ConnectionContext properties.\n    const connectionContext = ConnectionContextBase.create(parameters) as ConnectionContext;\n    connectionContext.tokenCredential = tokenCredential;\n    connectionContext.wasConnectionCloseCalled = false;\n    connectionContext.senders = {};\n    connectionContext.receivers = {};\n    const mOptions: ManagementClientOptions = {\n      address: options.managementSessionAddress,\n      audience: options.managementSessionAudience,\n    };\n    connectionContext.managementSession = new ManagementClient(connectionContext, mOptions);\n\n    let waitForConnectionRefreshResolve: () => void;\n    let waitForConnectionRefreshPromise: Promise<void> | undefined;\n\n    Object.assign<ConnectionContext, ConnectionContextMethods>(connectionContext, {\n      isConnectionClosing() {\n        // When the connection is not open, but the remote end is open,\n        // then the rhea connection is in the process of terminating.\n        return Boolean(!this.connection.isOpen() && this.connection.isRemoteOpen());\n      },\n      async readyToOpenLink(optionsArg?: { abortSignal?: AbortSignalLike }) {\n        // Check that the connection isn't in the process of closing.\n        // This can happen when the idle timeout has been reached but\n        // the underlying socket is waiting to be destroyed.\n        if (this.isConnectionClosing()) {\n          // Wait for the disconnected event that indicates the underlying socket has closed.\n          await this.waitForDisconnectedEvent(optionsArg);\n        }\n\n        // Wait for the connection to be reset.\n        await this.waitForConnectionReset();\n      },\n      waitForDisconnectedEvent(optionsArg?: { abortSignal?: AbortSignalLike }) {\n        return createAbortablePromise((resolve) => {\n          logger.verbose(\n            `[${this.connectionId}] Attempting to reinitialize connection` +\n              ` but the connection is in the process of closing.` +\n              ` Waiting for the disconnect event before continuing.`,\n          );\n          this.connection.once(ConnectionEvents.disconnected, resolve);\n        }, optionsArg);\n      },\n      waitForConnectionReset() {\n        // Check if the connection is currently in the process of disconnecting.\n        if (waitForConnectionRefreshPromise) {\n          return waitForConnectionRefreshPromise;\n        }\n        return Promise.resolve();\n      },\n      async close() {\n        try {\n          if (this.connection.isOpen()) {\n            // Close all the senders.\n            await Promise.all(\n              Object.keys(connectionContext.senders).map((name) =>\n                connectionContext.senders[name]?.close().catch(() => {\n                  /* error already logged, swallow it here */\n                }),\n              ),\n            );\n            // Close all the receivers.\n            await Promise.all(\n              Object.keys(connectionContext.receivers).map((name) =>\n                connectionContext.receivers[name]?.close().catch(() => {\n                  /* error already logged, swallow it here */\n                }),\n              ),\n            );\n            // Close the cbs session;\n            await this.cbsSession.close();\n            // Close the management session\n            await this.managementSession?.close();\n            await this.connection.close();\n            this.wasConnectionCloseCalled = true;\n            logger.info(\"Closed the amqp connection '%s' on the client.\", this.connectionId);\n          }\n        } catch (err: any) {\n          const errorDescription =\n            err instanceof Error ? `${err.name}: ${err.message}` : JSON.stringify(err);\n          logger.warning(\n            `An error occurred while closing the connection \"${this.connectionId}\":\\n${errorDescription}`,\n          );\n          logErrorStackTrace(err);\n          throw err;\n        }\n      },\n    });\n\n    // Define listeners to be added to the connection object for\n    // \"connection_open\" and \"connection_error\" events.\n    const onConnectionOpen: OnAmqpEvent = () => {\n      connectionContext.wasConnectionCloseCalled = false;\n      logger.verbose(\n        \"[%s] setting 'wasConnectionCloseCalled' property of connection context to %s.\",\n        connectionContext.connection.id,\n        connectionContext.wasConnectionCloseCalled,\n      );\n    };\n\n    async function handleDisconnection(context: EventContext): Promise<void> {\n      if (waitForConnectionRefreshPromise) {\n        return;\n      }\n      waitForConnectionRefreshPromise = new Promise((resolve) => {\n        waitForConnectionRefreshResolve = resolve;\n      });\n      try {\n        logger.verbose(\n          \"[%s] 'disconnected' event occurred on the amqp connection.\",\n          connectionContext.connection.id,\n        );\n\n        if (context.connection && context.connection.error) {\n          logger.verbose(\n            \"[%s] Accompanying error on the context.connection: %O\",\n            connectionContext.connection.id,\n            context.connection.error,\n          );\n        }\n        if (context.error) {\n          logger.verbose(\n            \"[%s] Accompanying error on the context: %O\",\n            connectionContext.connection.id,\n            context.error,\n          );\n        }\n        const state: Readonly<{\n          wasConnectionCloseCalled: boolean;\n          numSenders: number;\n          numReceivers: number;\n        }> = {\n          wasConnectionCloseCalled: connectionContext.wasConnectionCloseCalled,\n          numSenders: Object.keys(connectionContext.senders).length,\n          numReceivers: Object.keys(connectionContext.receivers).length,\n        };\n        logger.verbose(\n          \"[%s] Closing all open senders and receivers in the state: %O\",\n          connectionContext.connection.id,\n          state,\n        );\n\n        // Clear internal map maintained by rhea to avoid reconnecting of old links once the\n        // connection is back up.\n        connectionContext.connection.removeAllSessions();\n\n        // Close the cbs session to ensure all the event handlers are released.\n        await connectionContext.cbsSession?.close().catch(() => {\n          /* error already logged, swallow it here */\n        });\n        // Close the management session to ensure all the event handlers are released.\n        await connectionContext.managementSession?.close().catch(() => {\n          /* error already logged, swallow it here */\n        });\n\n        // Close all senders and receivers to ensure clean up of timers & other resources.\n        if (state.numSenders || state.numReceivers) {\n          await Promise.all(\n            Object.keys(connectionContext.senders).map((name) =>\n              connectionContext.senders[name]?.close().catch(() => {\n                /* error already logged, swallow it here */\n              }),\n            ),\n          );\n\n          await Promise.all(\n            Object.keys(connectionContext.receivers).map((name) =>\n              connectionContext.receivers[name]?.close().catch(() => {\n                /* error already logged, swallow it here */\n              }),\n            ),\n          );\n        }\n      } catch (err: any) {\n        logger.verbose(\n          `[${connectionContext.connectionId}] An error occurred while closing the connection in 'disconnected'. %O`,\n          err,\n        );\n      }\n\n      try {\n        await refreshConnection(connectionContext);\n      } catch (err: any) {\n        logger.verbose(\n          `[${connectionContext.connectionId}] An error occurred while refreshing the connection in 'disconnected'. %O`,\n          err,\n        );\n      } finally {\n        waitForConnectionRefreshResolve();\n        waitForConnectionRefreshPromise = undefined;\n      }\n    }\n\n    const onDisconnected: OnAmqpEvent = (context: EventContext) => {\n      // Fire-and-forget async operation with error handling\n      handleDisconnection(context).catch((err) => {\n        logger.verbose(\n          `[${connectionContext.connectionId}] Unhandled error in 'disconnected' event handler. %O`,\n          err,\n        );\n      });\n    };\n\n    const protocolError: OnAmqpEvent = (context: EventContext) => {\n      logger.verbose(\n        \"[%s] 'protocol_error' event occurred on the amqp connection.\",\n        connectionContext.connection.id,\n      );\n\n      if (context.connection && context.connection.error) {\n        logger.verbose(\n          \"[%s] Accompanying error on the context.connection: %O\",\n          connectionContext.connection.id,\n          context.connection && context.connection.error,\n        );\n      }\n      if (context.error) {\n        logger.verbose(\n          \"[%s] Accompanying error on the context: %O\",\n          connectionContext.connection.id,\n          context.error,\n        );\n      }\n    };\n\n    const error: OnAmqpEvent = (context: EventContext) => {\n      logger.verbose(\n        \"[%s] 'error' event occurred on the amqp connection.\",\n        connectionContext.connection.id,\n      );\n\n      if (context.connection && context.connection.error) {\n        logger.verbose(\n          \"[%s] Accompanying error on the context.connection: %O\",\n          connectionContext.connection.id,\n          context.connection && context.connection.error,\n        );\n      }\n      if (context.error) {\n        logger.verbose(\n          \"[%s] Accompanying error on the context: %O\",\n          connectionContext.connection.id,\n          context.error,\n        );\n      }\n    };\n\n    function addConnectionListeners(connection: Connection): void {\n      // Add listeners on the connection object.\n      connection.on(ConnectionEvents.connectionOpen, onConnectionOpen);\n      connection.on(ConnectionEvents.disconnected, onDisconnected);\n      connection.on(ConnectionEvents.protocolError, protocolError);\n      connection.on(ConnectionEvents.error, error);\n    }\n\n    function cleanConnectionContext(context: ConnectionContext): Promise<void> {\n      // Remove listeners from the connection object.\n      context.connection.removeListener(ConnectionEvents.connectionOpen, onConnectionOpen);\n      context.connection.removeListener(ConnectionEvents.disconnected, onDisconnected);\n      context.connection.removeListener(ConnectionEvents.protocolError, protocolError);\n      context.connection.removeListener(ConnectionEvents.error, error);\n      // Close the connection\n      return context.connection.close();\n    }\n\n    async function refreshConnection(context: ConnectionContext): Promise<void> {\n      const originalConnectionId = context.connectionId;\n      try {\n        await cleanConnectionContext(context);\n      } catch (err: any) {\n        logger.verbose(\n          `[${context.connectionId}] There was an error closing the connection before reconnecting: %O`,\n          err,\n        );\n      }\n\n      // Create a new connection, id, locks, and cbs client.\n      context.refreshConnection();\n      addConnectionListeners(context.connection);\n      logger.verbose(\n        `The connection \"${originalConnectionId}\" has been updated to \"${context.connectionId}\".`,\n      );\n    }\n\n    addConnectionListeners(connectionContext.connection);\n\n    logger.verbose(\"[%s] Created connection context successfully.\", connectionContext.connectionId);\n    return connectionContext;\n  }\n}\n\n/**\n * Helper method to create a ConnectionContext from the input passed to either\n * EventHubProducerClient or EventHubConsumerClient constructors\n *\n * @internal\n */\nexport function createConnectionContext(\n  hostOrConnectionString: string,\n  eventHubNameOrOptions?: string | EventHubClientOptions,\n  credentialOrOptions?:\n    | TokenCredential\n    | NamedKeyCredential\n    | SASCredential\n    | EventHubClientOptions,\n  options?: EventHubClientOptions,\n): ConnectionContext {\n  let connectionString;\n  let config;\n  let credential: TokenCredential | SasTokenProvider;\n  hostOrConnectionString = String(hostOrConnectionString);\n\n  if (!isCredential(credentialOrOptions)) {\n    const parsedCS = parseEventHubConnectionString(hostOrConnectionString);\n    if (\n      !(\n        parsedCS.eventHubName ||\n        (typeof eventHubNameOrOptions === \"string\" && eventHubNameOrOptions)\n      )\n    ) {\n      throw new TypeError(\n        `Either provide \"eventHubName\" or the \"connectionString\": \"${hostOrConnectionString}\", ` +\n          `must contain \"EntityPath=<your-event-hub-name>\".`,\n      );\n    }\n    if (\n      parsedCS.eventHubName &&\n      typeof eventHubNameOrOptions === \"string\" &&\n      eventHubNameOrOptions &&\n      parsedCS.eventHubName !== eventHubNameOrOptions\n    ) {\n      throw new TypeError(\n        `The entity path \"${parsedCS.eventHubName}\" in connectionString: \"${hostOrConnectionString}\" ` +\n          `doesn't match with eventHubName: \"${eventHubNameOrOptions}\".`,\n      );\n    }\n    connectionString = hostOrConnectionString;\n    if (typeof eventHubNameOrOptions !== \"string\") {\n      // connectionstring and/or options were passed to constructor\n      config = EventHubConnectionConfig.create(connectionString);\n      options = eventHubNameOrOptions;\n    } else {\n      // connectionstring, eventHubName and/or options were passed to constructor\n      const eventHubName = eventHubNameOrOptions;\n      config = EventHubConnectionConfig.create(connectionString, eventHubName);\n      options = credentialOrOptions;\n    }\n\n    const parsed = parseEventHubConnectionString(connectionString) as Required<\n      | Pick<EventHubConnectionStringProperties, \"sharedAccessKey\" | \"sharedAccessKeyName\">\n      | Pick<EventHubConnectionStringProperties, \"sharedAccessSignature\">\n    >;\n    // Since connectionString was passed, create a TokenProvider.\n    credential = createSasTokenProvider(parsed);\n  } else {\n    // host, eventHubName, a TokenCredential and/or options were passed to constructor\n    const eventHubName = eventHubNameOrOptions;\n    let host = hostOrConnectionString;\n    if (isNamedKeyCredential(credentialOrOptions) || isSASCredential(credentialOrOptions)) {\n      credential = createSasTokenProvider(credentialOrOptions);\n    } else {\n      credential = credentialOrOptions;\n    }\n    if (!eventHubName) {\n      throw new TypeError(`\"eventHubName\" is missing`);\n    }\n\n    if (!host.endsWith(\"/\")) host += \"/\";\n    connectionString = `Endpoint=sb://${host};SharedAccessKeyName=defaultKeyName;SharedAccessKey=defaultKeyValue;EntityPath=${eventHubName}`;\n    config = EventHubConnectionConfig.create(connectionString);\n  }\n\n  if (options?.customEndpointAddress) {\n    EventHubConnectionConfig.setCustomEndpointAddress(config, options.customEndpointAddress);\n  }\n\n  ConnectionConfig.validate(config);\n\n  return ConnectionContext.create(config, credential, options);\n}\n"]}

package/dist/react-native/dataTransformer.d.ts

@@ -0,0 +1,53 @@
+/**
+ * The allowed AMQP message body types.
+ * @internal
+ */
+export type BodyTypes = "data" | "value" | "sequence";
+/** @internal */
+export declare const dataSectionTypeCode: 117;
+/** @internal */
+export declare const sequenceSectionTypeCode: 118;
+/** @internal */
+export declare const valueSectionTypeCode: 119;
+/**
+ * The default data transformer that will be used by the Azure SDK.
+ * @internal
+ */
+export declare const defaultDataTransformer: {
+    /**
+     * A function that takes the body property from an EventData object
+     * and returns an encoded body (some form of AMQP type).
+     *
+     * @param body - The AMQP message body
+     * @param bodyType - The AMQP section to story the body in.
+     * @returns The encoded AMQP message body as an AMQP Data/Sequence/Value section.
+     */
+    encode(body: unknown, bodyType: BodyTypes): any;
+    /**
+     * A function that takes the body property from an AMQP message, which can come from either
+     * the 'data', 'value' or 'sequence' sections of an AMQP message.
+     *
+     * If the body is not a JSON string the the raw contents will be returned, along with the bodyType
+     * indicating which part of the AMQP message the body was decoded from.
+     *
+     * @param body - The AMQP message body as received from rhea.
+     * @param skipParsingBodyAsJson - Boolean to skip running JSON.parse() on message body when body type is `content`.
+     * @returns The decoded/raw body and the body type.
+     */
+    decode(body: unknown | RheaAmqpSection, skipParsingBodyAsJson: boolean): {
+        body: unknown;
+        bodyType: BodyTypes;
+    };
+};
+/**
+ * Mirror of the internal Section interface in rhea.
+ *
+ * @internal
+ */
+export interface RheaAmqpSection {
+    typecode: typeof dataSectionTypeCode | typeof sequenceSectionTypeCode | typeof valueSectionTypeCode;
+    content: any;
+}
+/** @internal */
+export declare function isRheaAmqpSection(possibleSection: any | RheaAmqpSection): possibleSection is RheaAmqpSection;
+//# sourceMappingURL=dataTransformer.d.ts.map

package/dist/react-native/dataTransformer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"dataTransformer.d.ts","sourceRoot":"","sources":["../../src/dataTransformer.ts"],"names":[],"mappings":"AAQA;;;GAGG;AACH,MAAM,MAAM,SAAS,GAAG,MAAM,GAAG,OAAO,GAAG,UAAU,CAAC;AAEtD,gBAAgB;AAChB,eAAO,MAAM,mBAAmB,EAAG,GAAa,CAAC;AACjD,gBAAgB;AAChB,eAAO,MAAM,uBAAuB,EAAG,GAAa,CAAC;AACrD,gBAAgB;AAChB,eAAO,MAAM,oBAAoB,EAAG,GAAa,CAAC;AAElD;;;GAGG;AACH,eAAO,MAAM,sBAAsB;IACjC;;;;;;;OAOG;iBACU,OAAO,YAAY,SAAS,GAAG,GAAG;IAmC/C;;;;;;;;;;OAUG;iBAEK,OAAO,GAAG,eAAe,yBACR,OAAO,GAC7B;QAAE,IAAI,EAAE,OAAO,CAAC;QAAC,QAAQ,EAAE,SAAS,CAAA;KAAE;CA6B1C,CAAC;AA2BF;;;;GAIG;AACH,MAAM,WAAW,eAAe;IAC9B,QAAQ,EACJ,OAAO,mBAAmB,GAC1B,OAAO,uBAAuB,GAC9B,OAAO,oBAAoB,CAAC;IAChC,OAAO,EAAE,GAAG,CAAC;CACd;AAED,gBAAgB;AAChB,wBAAgB,iBAAiB,CAC/B,eAAe,EAAE,GAAG,GAAG,eAAe,GACrC,eAAe,IAAI,eAAe,CAQpC"}

package/dist/react-native/dataTransformer.js

@@ -0,0 +1,133 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { logErrorStackTrace, logger } from "./logger.js";
+import { Buffer } from "buffer";
+import isBuffer from "is-buffer";
+import { message } from "rhea-promise";
+/** @internal */
+export const dataSectionTypeCode = 0x75;
+/** @internal */
+export const sequenceSectionTypeCode = 0x76;
+/** @internal */
+export const valueSectionTypeCode = 0x77;
+/**
+ * The default data transformer that will be used by the Azure SDK.
+ * @internal
+ */
+export const defaultDataTransformer = {
+    /**
+     * A function that takes the body property from an EventData object
+     * and returns an encoded body (some form of AMQP type).
+     *
+     * @param body - The AMQP message body
+     * @param bodyType - The AMQP section to story the body in.
+     * @returns The encoded AMQP message body as an AMQP Data/Sequence/Value section.
+     */
+    encode(body, bodyType) {
+        let result;
+        // string, undefined, null, boolean, array, object, number should end up here
+        // coercing undefined to null as that will ensure that null value will be given to the
+        // customer on receive.
+        if (body === undefined)
+            body = null;
+        if (bodyType === "value") {
+            // TODO: Expose value_section from `rhea` similar to the data_section and sequence_section.
+            // Right now there isn't a way to create a value section officially.
+            result = message.data_section(body);
+            result.typecode = valueSectionTypeCode;
+        }
+        else if (bodyType === "sequence") {
+            result = message.sequence_section(body);
+        }
+        else if (isBuffer(body) || body instanceof Uint8Array) {
+            result = message.data_section(body);
+        }
+        else if (body === null && bodyType === "data") {
+            result = message.data_section(null);
+        }
+        else {
+            try {
+                const bodyStr = JSON.stringify(body);
+                result = message.data_section(Buffer.from(bodyStr, "utf8"));
+            }
+            catch (err) {
+                const msg = `An error occurred while executing JSON.stringify() on the given body ` +
+                    body +
+                    `${err ? err.stack : JSON.stringify(err)}`;
+                logger.warning("[encode] " + msg);
+                logErrorStackTrace(err);
+                throw new Error(msg);
+            }
+        }
+        return result;
+    },
+    /**
+     * A function that takes the body property from an AMQP message, which can come from either
+     * the 'data', 'value' or 'sequence' sections of an AMQP message.
+     *
+     * If the body is not a JSON string the the raw contents will be returned, along with the bodyType
+     * indicating which part of the AMQP message the body was decoded from.
+     *
+     * @param body - The AMQP message body as received from rhea.
+     * @param skipParsingBodyAsJson - Boolean to skip running JSON.parse() on message body when body type is `content`.
+     * @returns The decoded/raw body and the body type.
+     */
+    decode(body, skipParsingBodyAsJson) {
+        try {
+            if (isRheaAmqpSection(body)) {
+                switch (body.typecode) {
+                    case dataSectionTypeCode:
+                        return {
+                            body: skipParsingBodyAsJson ? body.content : tryToJsonDecode(body.content),
+                            bodyType: "data",
+                        };
+                    case sequenceSectionTypeCode:
+                        return { body: body.content, bodyType: "sequence" };
+                    case valueSectionTypeCode:
+                        return { body: body.content, bodyType: "value" };
+                }
+            }
+            else {
+                if (isBuffer(body)) {
+                    return { body: skipParsingBodyAsJson ? body : tryToJsonDecode(body), bodyType: "data" };
+                }
+                return { body, bodyType: "value" };
+            }
+        }
+        catch (err) {
+            logger.verbose("[decode] An error occurred while decoding the received message body. The error is: %O", err);
+            throw err;
+        }
+    },
+};
+/**
+ * Attempts to decode 'body' as a JSON string. If it fails it returns body
+ * verbatim.
+ *
+ * @param body - An AMQP message body.
+ * @returns A JSON decoded object, or body if body was not a JSON string.
+ *
+ * @internal
+ */
+function tryToJsonDecode(body) {
+    let processedBody = body;
+    try {
+        // Trying to stringify and JSON.parse() anything else will fail flat and we shall return
+        // the original type back
+        const bodyStr = processedBody.toString("utf8");
+        processedBody = JSON.parse(bodyStr);
+    }
+    catch (err) {
+        logger.verbose("[decode] An error occurred while trying JSON.parse() on the received body. The error is %O", err);
+    }
+    return processedBody;
+}
+/** @internal */
+export function isRheaAmqpSection(possibleSection) {
+    return (possibleSection != null &&
+        typeof possibleSection.typecode === "number" &&
+        (possibleSection.typecode === dataSectionTypeCode ||
+            possibleSection.typecode === valueSectionTypeCode ||
+            possibleSection.typecode === sequenceSectionTypeCode));
+}
+//# sourceMappingURL=dataTransformer.js.map

package/dist/react-native/dataTransformer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dataTransformer.js","sourceRoot":"","sources":["../../src/dataTransformer.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,kBAAkB,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAChC,OAAO,QAAQ,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAQvC,gBAAgB;AAChB,MAAM,CAAC,MAAM,mBAAmB,GAAG,IAAa,CAAC;AACjD,gBAAgB;AAChB,MAAM,CAAC,MAAM,uBAAuB,GAAG,IAAa,CAAC;AACrD,gBAAgB;AAChB,MAAM,CAAC,MAAM,oBAAoB,GAAG,IAAa,CAAC;AAElD;;;GAGG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG;IACpC;;;;;;;OAOG;IACH,MAAM,CAAC,IAAa,EAAE,QAAmB;QACvC,IAAI,MAAW,CAAC;QAChB,6EAA6E;QAC7E,sFAAsF;QACtF,uBAAuB;QACvB,IAAI,IAAI,KAAK,SAAS;YAAE,IAAI,GAAG,IAAI,CAAC;QAEpC,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;YACzB,2FAA2F;YAC3F,oEAAoE;YACpE,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACpC,MAAM,CAAC,QAAQ,GAAG,oBAAoB,CAAC;QACzC,CAAC;aAAM,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;YACnC,MAAM,GAAG,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;QAC1C,CAAC;aAAM,IAAI,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,YAAY,UAAU,EAAE,CAAC;YACxD,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC;aAAM,IAAI,IAAI,KAAK,IAAI,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;YAChD,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QACtC,CAAC;aAAM,CAAC;YACN,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;gBACrC,MAAM,GAAG,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;YAC9D,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,MAAM,GAAG,GACP,uEAAuE;oBACvE,IAAI;oBACJ,GAAG,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC7C,MAAM,CAAC,OAAO,CAAC,WAAW,GAAG,GAAG,CAAC,CAAC;gBAClC,kBAAkB,CAAC,GAAG,CAAC,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC;YACvB,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;;;;OAUG;IACH,MAAM,CACJ,IAA+B,EAC/B,qBAA8B;QAE9B,IAAI,CAAC;YACH,IAAI,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC5B,QAAQ,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACtB,KAAK,mBAAmB;wBACtB,OAAO;4BACL,IAAI,EAAE,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC;4BAC1E,QAAQ,EAAE,MAAM;yBACjB,CAAC;oBACJ,KAAK,uBAAuB;wBAC1B,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC;oBACtD,KAAK,oBAAoB;wBACvB,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;gBACrD,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;oBACnB,OAAO,EAAE,IAAI,EAAE,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;gBAC1F,CAAC;gBAED,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;YACrC,CAAC;QACH,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,CAAC,OAAO,CACZ,uFAAuF,EACvF,GAAG,CACJ,CAAC;YACF,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;CACF,CAAC;AAEF;;;;;;;;GAQG;AACH,SAAS,eAAe,CAAC,IAAa;IACpC,IAAI,aAAa,GAAQ,IAAI,CAAC;IAC9B,IAAI,CAAC;QACH,wFAAwF;QACxF,yBAAyB;QACzB,MAAM,OAAO,GAAW,aAAa,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QACvD,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC;IAAC,OAAO,GAAQ,EAAE,CAAC;QAClB,MAAM,CAAC,OAAO,CACZ,4FAA4F,EAC5F,GAAG,CACJ,CAAC;IACJ,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC;AAeD,gBAAgB;AAChB,MAAM,UAAU,iBAAiB,CAC/B,eAAsC;IAEtC,OAAO,CACL,eAAe,IAAI,IAAI;QACvB,OAAO,eAAe,CAAC,QAAQ,KAAK,QAAQ;QAC5C,CAAC,eAAe,CAAC,QAAQ,KAAK,mBAAmB;YAC/C,eAAe,CAAC,QAAQ,KAAK,oBAAoB;YACjD,eAAe,CAAC,QAAQ,KAAK,uBAAuB,CAAC,CACxD,CAAC;AACJ,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { logErrorStackTrace, logger } from \"./logger.js\";\nimport { Buffer } from \"buffer\";\nimport isBuffer from \"is-buffer\";\nimport { message } from \"rhea-promise\";\n\n/**\n * The allowed AMQP message body types.\n * @internal\n */\nexport type BodyTypes = \"data\" | \"value\" | \"sequence\";\n\n/** @internal */\nexport const dataSectionTypeCode = 0x75 as const;\n/** @internal */\nexport const sequenceSectionTypeCode = 0x76 as const;\n/** @internal */\nexport const valueSectionTypeCode = 0x77 as const;\n\n/**\n * The default data transformer that will be used by the Azure SDK.\n * @internal\n */\nexport const defaultDataTransformer = {\n  /**\n   * A function that takes the body property from an EventData object\n   * and returns an encoded body (some form of AMQP type).\n   *\n   * @param body - The AMQP message body\n   * @param bodyType - The AMQP section to story the body in.\n   * @returns The encoded AMQP message body as an AMQP Data/Sequence/Value section.\n   */\n  encode(body: unknown, bodyType: BodyTypes): any {\n    let result: any;\n    // string, undefined, null, boolean, array, object, number should end up here\n    // coercing undefined to null as that will ensure that null value will be given to the\n    // customer on receive.\n    if (body === undefined) body = null;\n\n    if (bodyType === \"value\") {\n      // TODO: Expose value_section from `rhea` similar to the data_section and sequence_section.\n      // Right now there isn't a way to create a value section officially.\n      result = message.data_section(body);\n      result.typecode = valueSectionTypeCode;\n    } else if (bodyType === \"sequence\") {\n      result = message.sequence_section(body);\n    } else if (isBuffer(body) || body instanceof Uint8Array) {\n      result = message.data_section(body);\n    } else if (body === null && bodyType === \"data\") {\n      result = message.data_section(null);\n    } else {\n      try {\n        const bodyStr = JSON.stringify(body);\n        result = message.data_section(Buffer.from(bodyStr, \"utf8\"));\n      } catch (err: any) {\n        const msg =\n          `An error occurred while executing JSON.stringify() on the given body ` +\n          body +\n          `${err ? err.stack : JSON.stringify(err)}`;\n        logger.warning(\"[encode] \" + msg);\n        logErrorStackTrace(err);\n        throw new Error(msg);\n      }\n    }\n    return result;\n  },\n\n  /**\n   * A function that takes the body property from an AMQP message, which can come from either\n   * the 'data', 'value' or 'sequence' sections of an AMQP message.\n   *\n   * If the body is not a JSON string the the raw contents will be returned, along with the bodyType\n   * indicating which part of the AMQP message the body was decoded from.\n   *\n   * @param body - The AMQP message body as received from rhea.\n   * @param skipParsingBodyAsJson - Boolean to skip running JSON.parse() on message body when body type is `content`.\n   * @returns The decoded/raw body and the body type.\n   */\n  decode(\n    body: unknown | RheaAmqpSection,\n    skipParsingBodyAsJson: boolean,\n  ): { body: unknown; bodyType: BodyTypes } {\n    try {\n      if (isRheaAmqpSection(body)) {\n        switch (body.typecode) {\n          case dataSectionTypeCode:\n            return {\n              body: skipParsingBodyAsJson ? body.content : tryToJsonDecode(body.content),\n              bodyType: \"data\",\n            };\n          case sequenceSectionTypeCode:\n            return { body: body.content, bodyType: \"sequence\" };\n          case valueSectionTypeCode:\n            return { body: body.content, bodyType: \"value\" };\n        }\n      } else {\n        if (isBuffer(body)) {\n          return { body: skipParsingBodyAsJson ? body : tryToJsonDecode(body), bodyType: \"data\" };\n        }\n\n        return { body, bodyType: \"value\" };\n      }\n    } catch (err: any) {\n      logger.verbose(\n        \"[decode] An error occurred while decoding the received message body. The error is: %O\",\n        err,\n      );\n      throw err;\n    }\n  },\n};\n\n/**\n * Attempts to decode 'body' as a JSON string. If it fails it returns body\n * verbatim.\n *\n * @param body - An AMQP message body.\n * @returns A JSON decoded object, or body if body was not a JSON string.\n *\n * @internal\n */\nfunction tryToJsonDecode(body: unknown): unknown {\n  let processedBody: any = body;\n  try {\n    // Trying to stringify and JSON.parse() anything else will fail flat and we shall return\n    // the original type back\n    const bodyStr: string = processedBody.toString(\"utf8\");\n    processedBody = JSON.parse(bodyStr);\n  } catch (err: any) {\n    logger.verbose(\n      \"[decode] An error occurred while trying JSON.parse() on the received body. The error is %O\",\n      err,\n    );\n  }\n  return processedBody;\n}\n\n/**\n * Mirror of the internal Section interface in rhea.\n *\n * @internal\n */\nexport interface RheaAmqpSection {\n  typecode:\n    | typeof dataSectionTypeCode\n    | typeof sequenceSectionTypeCode\n    | typeof valueSectionTypeCode;\n  content: any;\n}\n\n/** @internal */\nexport function isRheaAmqpSection(\n  possibleSection: any | RheaAmqpSection,\n): possibleSection is RheaAmqpSection {\n  return (\n    possibleSection != null &&\n    typeof possibleSection.typecode === \"number\" &&\n    (possibleSection.typecode === dataSectionTypeCode ||\n      possibleSection.typecode === valueSectionTypeCode ||\n      possibleSection.typecode === sequenceSectionTypeCode)\n  );\n}\n"]}

package/dist/react-native/diagnostics/instrumentEventData.d.ts

@@ -0,0 +1,28 @@
+import type { EventData } from "../eventData.js";
+import type { TracingContext } from "@azure/core-tracing";
+import type { AmqpAnnotatedMessage } from "@azure/core-amqp";
+import type { OperationOptions } from "../util/operationOptions.js";
+import type { MessagingOperationNames } from "./tracing.js";
+/**
+ * @internal
+ */
+export declare const TRACEPARENT_PROPERTY = "Diagnostic-Id";
+/**
+ * Populates the `EventData` with `SpanContext` info to support trace propagation.
+ * Creates and returns a copy of the passed in `EventData` unless the `EventData`
+ * has already been instrumented.
+ * @param eventData - The `EventData` or `AmqpAnnotatedMessage` to instrument.
+ * @param span - The `Span` containing the context to propagate tracing information.
+ * @param operation - The type of the operation being performed.
+ */
+export declare function instrumentEventData(eventData: EventData | AmqpAnnotatedMessage, options: OperationOptions, entityPath: string, host: string, operation?: MessagingOperationNames): {
+    event: EventData;
+    spanContext: TracingContext | undefined;
+};
+/**
+ * Extracts the `SpanContext` from an `EventData` if the context exists.
+ * @param eventData - An individual `EventData` object.
+ * @internal
+ */
+export declare function extractSpanContextFromEventData(eventData: EventData): TracingContext | undefined;
+//# sourceMappingURL=instrumentEventData.d.ts.map

package/dist/react-native/diagnostics/instrumentEventData.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"instrumentEventData.d.ts","sourceRoot":"","sources":["../../../src/diagnostics/instrumentEventData.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAEjD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AAC7D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAC;AACpE,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,cAAc,CAAC;AAG5D;;GAEG;AACH,eAAO,MAAM,oBAAoB,kBAAkB,CAAC;AAEpD;;;;;;;GAOG;AACH,wBAAgB,mBAAmB,CACjC,SAAS,EAAE,SAAS,GAAG,oBAAoB,EAC3C,OAAO,EAAE,gBAAgB,EACzB,UAAU,EAAE,MAAM,EAClB,IAAI,EAAE,MAAM,EACZ,SAAS,CAAC,EAAE,uBAAuB,GAClC;IAAE,KAAK,EAAE,SAAS,CAAC;IAAC,WAAW,EAAE,cAAc,GAAG,SAAS,CAAA;CAAE,CA+C/D;AAED;;;;GAIG;AACH,wBAAgB,+BAA+B,CAAC,SAAS,EAAE,SAAS,GAAG,cAAc,GAAG,SAAS,CAOhG"}