@azure/notification-hubs 1.0.0-beta.1 → 1.0.0-beta.2

package/dist/{index.js → index.cjs}

@@ -3,13 +3,144 @@
 Object.defineProperty(exports, '__esModule', { value: true });
 
 var coreRestPipeline = require('@azure/core-rest-pipeline');
-var coreAmqp = require('@azure/core-amqp');
+var coreAuth = require('@azure/core-auth');
+var coreUtil = require('@azure/core-util');
+var crypto = require('crypto');
 var coreXml = require('@azure/core-xml');
 var coreClient = require('@azure/core-client');
 var coreTracing = require('@azure/core-tracing');
 var tslib = require('tslib');
 
 // Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+const SDK_VERSION = "1.0.0-beta.2";
+const JSON_CONTENT_TYPE = "application/json;charset=utf-8";
+const XML_CONTENT_TYPE = "application/xml";
+const STREAM_CONTENT_TYPE = "application/octet-stream";
+const WNS_TYPE_NAME = "X-WNS-Type";
+const WNS_RAW = "wns/raw";
+const WNS_BADGE = "wns/badge";
+const WNS_TITLE = "wns/tile";
+const WNS_TOAST = "wns/toast";
+
+// Copyright (c) Microsoft Corporation.
+async function signString(key, toSign) {
+    const hmac = crypto.createHmac("sha256", key).update(toSign).digest("base64");
+    return encodeURIComponent(hmac);
+}
+
+// Copyright (c) Microsoft Corporation.
+/**
+ * Creates a token provider from the provided shared access data.
+ * @param data - The sharedAccessKeyName/sharedAccessKey pair or the sharedAccessSignature.
+ * @hidden
+ */
+function createSasTokenProvider(data) {
+    if (coreAuth.isNamedKeyCredential(data) || coreAuth.isSASCredential(data)) {
+        return new SasTokenProviderImpl(data);
+    }
+    else if (coreUtil.isObjectWithProperties(data, ["sharedAccessKeyName", "sharedAccessKey"])) {
+        return new SasTokenProviderImpl({
+            name: data.sharedAccessKeyName,
+            key: data.sharedAccessKey,
+        });
+    }
+    else {
+        return new SasTokenProviderImpl({ signature: data.sharedAccessSignature });
+    }
+}
+/**
+ * A TokenProvider that generates a Sas token:
+ * `SharedAccessSignature sr=<resource>&sig=<signature>&se=<expiry>&skn=<keyname>`
+ *
+ * @internal
+ */
+class SasTokenProviderImpl {
+    /**
+     * Initializes a new instance of SasTokenProvider
+     * @param credential - The source `NamedKeyCredential` or `SASCredential`.
+     */
+    constructor(credential) {
+        this._credential = credential;
+    }
+    /**
+     * Property used to distinguish TokenProvider from TokenCredential.
+     */
+    get isSasTokenProvider() {
+        return true;
+    }
+    /**
+     * Gets the sas token for the specified audience
+     * @param audience - The audience for which the token is desired.
+     */
+    async getToken(audience) {
+        if (coreAuth.isNamedKeyCredential(this._credential)) {
+            return createToken(this._credential.name, this._credential.key, Math.floor(Date.now() / 1000) + 3600, audience);
+        }
+        else {
+            return {
+                token: this._credential.signature,
+                expiresOnTimestamp: 0,
+            };
+        }
+    }
+}
+/**
+ * Creates the sas token based on the provided information.
+ * @param keyName - The shared access key name.
+ * @param key - The shared access key.
+ * @param expiry - The time period in unix time after which the token will expire.
+ * @param audience - The audience for which the token is desired.
+ * @internal
+ */
+async function createToken(keyName, key, expiry, audience) {
+    audience = encodeURIComponent(audience.toLowerCase());
+    keyName = encodeURIComponent(keyName);
+    const stringToSign = audience + "\n" + expiry;
+    const sig = await signString(key, stringToSign);
+    return {
+        token: `SharedAccessSignature sr=${audience}&sig=${sig}&se=${expiry}&skn=${keyName}`,
+        expiresOnTimestamp: expiry,
+    };
+}
+
+// Copyright (c) Microsoft Corporation.
+/**
+ * Parses the connection string and returns an object of type T.
+ *
+ * Connection strings have the following syntax:
+ *
+ * ConnectionString ::= `Part { ";" Part } [ ";" ] [ WhiteSpace ]`
+ * Part             ::= [ PartLiteral [ "=" PartLiteral ] ]
+ * PartLiteral      ::= [ WhiteSpace ] Literal [ WhiteSpace ]
+ * Literal          ::= ? any sequence of characters except ; or = or WhiteSpace ?
+ * WhiteSpace       ::= ? all whitespace characters including `\r` and `\n` ?
+ *
+ * @param connectionString - The connection string to be parsed.
+ * @returns ParsedOutput<T>.
+ */
+function parseConnectionString(connectionString) {
+    const output = {};
+    const parts = connectionString.trim().split(";");
+    for (let part of parts) {
+        part = part.trim();
+        if (part === "") {
+            // parts can be empty
+            continue;
+        }
+        const splitIndex = part.indexOf("=");
+        if (splitIndex === -1) {
+            throw new Error("Connection string malformed: each part of the connection string must have an `=` assignment.");
+        }
+        const key = part.substring(0, splitIndex).trim();
+        if (key === "") {
+            throw new Error("Connection string malformed: missing key for assignment");
+        }
+        const value = part.substring(splitIndex + 1).trim();
+        output[key] = value;
+    }
+    return output;
+}
 /**
  * Creates a SasTokenProvider from a shared access key and shared access key name.
  * @param sharedAccessKey - The shared access key value.
@@ -17,7 +148,7 @@ var tslib = require('tslib');
  * @returns A SasTokenProvider with the given shared access token information.
  */
 function createTokenProviderFromConnection(sharedAccessKey, sharedAccessKeyName) {
-    return coreAmqp.createSasTokenProvider({ sharedAccessKey, sharedAccessKeyName });
+    return createSasTokenProvider({ sharedAccessKey, sharedAccessKeyName });
 }
 /**
  * Parses given connection string into the different properties applicable to Azure Service Bus.
@@ -26,7 +157,7 @@ function createTokenProviderFromConnection(sharedAccessKey, sharedAccessKeyName)
  * for the Service Bus namespace, queue or topic.
  */
 function parseNotificationHubsConnectionString(connectionString) {
-    const parsedResult = coreAmqp.parseConnectionString(connectionString);
+    const parsedResult = parseConnectionString(connectionString);
     if (!parsedResult.Endpoint) {
         throw new Error("Connection string should have an Endpoint key.");
     }
@@ -61,17 +192,20 @@ class NotificationHubsServiceClient$1 extends coreClient.ServiceClient {
                 parseXML: coreXml.parseXML,
             }, serializationOptions: {
                 stringifyXML: coreXml.stringifyXML,
+            }, userAgentOptions: {
+                userAgentPrefix: `azsdk-js-messaging-notificationhubs/${SDK_VERSION}`,
             } }, options));
         this.hubName = hubName;
         const parsedConnection = parseNotificationHubsConnectionString(connectionString);
         this.baseUrl = parsedConnection.endpoint;
         this.sasTokenProvider = createTokenProviderFromConnection(parsedConnection.sharedAccessKey, parsedConnection.sharedAccessKeyName);
     }
-    createHeaders() {
-        const authorization = this.sasTokenProvider.getToken(this.baseUrl);
+    async createHeaders(operationName) {
+        const authorization = await this.sasTokenProvider.getToken(this.baseUrl);
         const headers = coreRestPipeline.createHttpHeaders();
         headers.set("Authorization", authorization.token);
         headers.set("x-ms-version", API_VERSION);
+        headers.set("x-ms-azsdk-telemetry", `class=NotificationHubsServiceClient;method=${operationName}`);
         return headers;
     }
     requestUrl() {
@@ -361,10 +495,11 @@ async function sendRequest(context, request, successStatusCode) {
 const tracingClient = coreTracing.createTracingClient({
     namespace: "Microsoft.NotificationHubs",
     packageName: "@azure/notification-hubs",
-    packageVersion: "1.0.0",
+    packageVersion: SDK_VERSION,
 });
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$h = "cancelScheduledNotification";
 /**
  * Cancels the scheduled notification with the given notification ID.
  * NOTE: This is only available in Standard SKU Azure Notification Hubs.
@@ -374,10 +509,10 @@ const tracingClient = coreTracing.createTracingClient({
  * @returns A notification hub response with correlation ID and tracking ID.
  */
 function cancelScheduledNotification(context, notificationId, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-cancelScheduledNotification", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$h}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += `/schedulednotifications/${notificationId}`;
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$h);
         const request = createRequest(endpoint, "DELETE", headers, updatedOptions);
         const response = await sendRequest(context, request, 200);
         return parseNotificationSendResponse(response);
@@ -385,6 +520,7 @@ function cancelScheduledNotification(context, notificationId, options = {}) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$g = "createOrUpdateInstallation";
 /**
  * Creates or overwrites an installation to a Notification Hub.
  * @param context - The Notification Hubs client.
@@ -393,10 +529,10 @@ function cancelScheduledNotification(context, notificationId, options = {}) {
  * @returns A NotificationHubResponse with the tracking ID, correlation ID and location.
  */
 function createOrUpdateInstallation(context, installation, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-createOrUpdateInstallation", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$g}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += `/installations/${installation.installationId}`;
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$g);
         headers.set("Content-Type", "application/json");
         const request = createRequest(endpoint, "PUT", headers, updatedOptions);
         request.body = JSON.stringify(installation);
@@ -791,7 +927,7 @@ async function createOrUpdateRegistrationDescription(context, registration, oper
     // Clear out readonly properties
     registration.registrationId = undefined;
     registration.etag = undefined;
-    const headers = context.createHeaders();
+    const headers = await context.createHeaders(`${operationName}Registration`);
     headers.set("Content-Type", "application/atom+xml;type=entry;charset=utf-8");
     if (operationName === "update") {
         headers.set("If-Match", isDefined(etag) ? `"${etag}"` : "*");
@@ -803,6 +939,7 @@ async function createOrUpdateRegistrationDescription(context, registration, oper
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$f = "createOrUpdateRegistration";
 /**
  * Creates or updates a registration.
  * @param context - The Notification Hubs client.
@@ -811,12 +948,13 @@ async function createOrUpdateRegistrationDescription(context, registration, oper
  * @returns The created or updated registration description.
  */
 function createOrUpdateRegistration(context, registration, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-createOrUpdateRegistration", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$f}`, options, async (updatedOptions) => {
         return createOrUpdateRegistrationDescription(context, registration, "createOrUpdate", updatedOptions);
     });
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$e = "createRegistrationId";
 /**
  * Creates a new registration ID.
  * @param context - The Notification Hubs client.
@@ -824,10 +962,10 @@ function createOrUpdateRegistration(context, registration, options = {}) {
  * @returns The newly created registration ID.
  */
 function createRegistrationId(context, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-createRegistrationId", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$e}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += "/registrationIDs";
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$e);
         headers.set("Content-Type", "application/xml;type=entry;charset=utf-8");
         const request = createRequest(endpoint, "POST", headers, updatedOptions);
         const response = await sendRequest(context, request, 201);
@@ -840,6 +978,7 @@ function createRegistrationId(context, options = {}) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$d = "createRegistration";
 /**
  * Creates a new registration. This method generates a registration ID,
  * which you can subsequently use to retrieve, update, and delete this registration.
@@ -849,7 +988,7 @@ function createRegistrationId(context, options = {}) {
  * @returns The newly created registration description.
  */
 function createRegistration(context, registration, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-createRegistration", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$d}`, options, async (updatedOptions) => {
         if (registration.registrationId) {
             throw new coreRestPipeline.RestError("registrationId must not be set during a create operation", {
                 statusCode: 400,
@@ -860,6 +999,7 @@ function createRegistration(context, registration, options = {}) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$c = "deleteInstallation";
 /**
  * Deletes an installation from a Notification Hub.
  * @param context - The Notification Hubs client.
@@ -868,10 +1008,10 @@ function createRegistration(context, registration, options = {}) {
  * @returns A NotificationHubResponse with the tracking ID, correlation ID and location.
  */
 function deleteInstallation(context, installationId, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-deleteInstallation", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$c}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += `/installations/${installationId}`;
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$c);
         const request = createRequest(endpoint, "DELETE", headers, updatedOptions);
         const response = await sendRequest(context, request, 204);
         return parseNotificationResponse(response);
@@ -879,6 +1019,7 @@ function deleteInstallation(context, installationId, options = {}) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$b = "getFeedbackContainerUrl";
 /**
  * Retrieves an Azure Storage container URL. The container has feedback data for the notification hub.
  * The caller can then use the Azure Storage Services SDK to retrieve the contents of the container.
@@ -887,10 +1028,10 @@ function deleteInstallation(context, installationId, options = {}) {
  * @returns The URL of the Azure Storage Container containing the feedback data.
  */
 function getFeedbackContainerUrl(context, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-getFeedbackContainerUrl", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$b}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += "/feedbackcontainer";
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$b);
         headers.set("Content-Type", "application/xml;type=entry;charset=utf-8");
         const request = createRequest(endpoint, "GET", headers, updatedOptions);
         const response = await sendRequest(context, request, 200);
@@ -899,6 +1040,7 @@ function getFeedbackContainerUrl(context, options = {}) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$a = "getInstallation";
 /**
  * Gets an Azure Notification Hub installation by the installation ID.
  * @param context - The Notification Hubs client.
@@ -907,10 +1049,10 @@ function getFeedbackContainerUrl(context, options = {}) {
  * @returns The installation that matches the installation ID.
  */
 function getInstallation(context, installationId, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-getInstallation", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$a}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += `/installations/${installationId}`;
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$a);
         headers.set("Content-Type", "application/json");
         const request = createRequest(endpoint, "GET", headers, updatedOptions);
         const response = await sendRequest(context, request, 200);
@@ -989,6 +1131,7 @@ function createNotificationHubJob(content) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$9 = "getNotificationHubJob";
 /**
  * Gets a Notification Hub Job by the ID.
  * @param context - The Notification Hubs client.
@@ -997,10 +1140,10 @@ function createNotificationHubJob(content) {
  * @returns The Notification Hub Job with the matching ID.
  */
 function getNotificationHubJob(context, jobId, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-getNotificationHubJob", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$9}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += `/jobs/${jobId}`;
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$9);
         headers.set("Content-Type", "application/atom+xml;type=entry;charset=utf-8");
         const request = createRequest(endpoint, "GET", headers, updatedOptions);
         const response = await sendRequest(context, request, 200);
@@ -1064,6 +1207,7 @@ function parseOutcomeCounts(counts) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$8 = "getNotificationOutcomeDetails";
 /**
  * Retrieves the results of a send operation. This can retrieve intermediate results if the send is being processed
  * or final results if the Send* has completed. This API can only be called for Standard SKU and above.
@@ -1073,10 +1217,10 @@ function parseOutcomeCounts(counts) {
  * @returns The results of the send operation.
  */
 function getNotificationOutcomeDetails(context, notificationId, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-getNotificationOutcomeDetails", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$8}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += `/messages/${notificationId}`;
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$8);
         const request = createRequest(endpoint, "GET", headers, updatedOptions);
         const response = await sendRequest(context, request, 200);
         return parseNotificationDetails(response.bodyAsText);
@@ -1084,6 +1228,7 @@ function getNotificationOutcomeDetails(context, notificationId, options = {}) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$7 = "getRegistration";
 /**
  * Gets a registration by the given registration ID.
  * @param context - The Notification Hubs client.
@@ -1092,10 +1237,10 @@ function getNotificationOutcomeDetails(context, notificationId, options = {}) {
  * @returns A RegistrationDescription that has the given registration ID.
  */
 function getRegistration(context, registrationId, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-getRegistration", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$7}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += `/registrations/${registrationId}`;
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$7);
         headers.set("Content-Type", "application/xml;type=entry;charset=utf-8");
         const request = createRequest(endpoint, "GET", headers, updatedOptions);
         const response = await sendRequest(context, request, 200);
@@ -1104,6 +1249,7 @@ function getRegistration(context, registrationId, options = {}) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$6 = "listNotificationHubJobs";
 /**
  * Gets all Notification Hub Jobs for this Notification Hub.
  * @param context - The Notification Hubs client.xs
@@ -1111,10 +1257,10 @@ function getRegistration(context, registrationId, options = {}) {
  * @returns An array of all Notification Hub Jobs for this Notification Hub.
  */
 function listNotificationHubJobs(context, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-getNotificationHubJobs", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$6}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += "/jobs";
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$6);
         headers.set("Content-Type", "application/atom+xml;type=entry;charset=utf-8");
         const request = createRequest(endpoint, "GET", headers, updatedOptions);
         const response = await sendRequest(context, request, 200);
@@ -1123,6 +1269,7 @@ function listNotificationHubJobs(context, options = {}) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$5 = "listRegistrationsByTag";
 /**
  * Lists all registrations with the matching tag.
  * @param context - The Notification Hubs client.
@@ -1131,7 +1278,7 @@ function listNotificationHubJobs(context, options = {}) {
  * @returns A paged async iterable containing the matching registrations for the notification hub.
  */
 function listRegistrationsByTag(context, tag, options = {}) {
-    const { span, updatedOptions } = tracingClient.startSpan("NotificationHubsClientContext-listRegistrationsByTag", options);
+    const { span, updatedOptions } = tracingClient.startSpan(`NotificationHubsClientContext-${OPERATION_NAME$5}`, options);
     try {
         const iter = listRegistrationsByTagAll(context, tag, updatedOptions);
         return {
@@ -1193,7 +1340,7 @@ async function _listRegistrationsByTag(context, tag, options, continuationToken)
     if (continuationToken !== undefined) {
         endpoint.searchParams.set("continuationtoken", continuationToken);
     }
-    const headers = context.createHeaders();
+    const headers = await context.createHeaders(OPERATION_NAME$5);
     const request = createRequest(endpoint, "GET", headers, options);
     const response = await sendRequest(context, request, 200);
     const registrations = await registrationDescriptionParser.parseRegistrationFeed(response.bodyAsText);
@@ -1205,6 +1352,7 @@ async function _listRegistrationsByTag(context, tag, options, continuationToken)
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$4 = "listRegistrations";
 /**
  * Gets all registrations for the notification hub with the given query options.
  * @param context - The Notification Hubs client.
@@ -1212,7 +1360,7 @@ async function _listRegistrationsByTag(context, tag, options, continuationToken)
  * @returns A paged async iterable containing all of the registrations for the notification hub.
  */
 function listRegistrations(context, options = {}) {
-    const { span, updatedOptions } = tracingClient.startSpan("NotificationHubsClientContext-listRegistrations", options);
+    const { span, updatedOptions } = tracingClient.startSpan(`NotificationHubsClientContext-${OPERATION_NAME$4}`, options);
     try {
         const iter = listRegistrationsAll(context, updatedOptions);
         return {
@@ -1277,7 +1425,7 @@ async function _listRegistrations(context, options, continuationToken) {
     if (continuationToken !== undefined) {
         endpoint.searchParams.set("continuationtoken", continuationToken);
     }
-    const headers = context.createHeaders();
+    const headers = await context.createHeaders(OPERATION_NAME$4);
     const request = createRequest(endpoint, "GET", headers, options);
     const response = await sendRequest(context, request, 200);
     const registrations = await registrationDescriptionParser.parseRegistrationFeed(response.bodyAsText);
@@ -1289,14 +1437,15 @@ async function _listRegistrations(context, options, continuationToken) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$3 = "scheduleNotificationPayload";
 /**
  * @internal
  */
 function scheduleNotificationPayload(context, scheduledTime, tags, notification, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-$scheduleNotification", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$3}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += "/schedulednotifications/";
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$3);
         if (notification.headers) {
             for (const headerName of Object.keys(notification.headers)) {
                 headers.set(headerName, notification.headers[headerName]);
@@ -1364,7 +1513,7 @@ function sendNotificationPayload(context, notification, method, pushHandle, tags
         if (options.enableTestSend) {
             endpoint.searchParams.append("debug", "true");
         }
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(method);
         if (notification.headers) {
             for (const headerName of Object.keys(notification.headers)) {
                 headers.set(headerName, notification.headers[headerName]);
@@ -1443,6 +1592,7 @@ function sendNotification(context, tags, notification, options = {}) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$2 = "submitNotificationHubJob";
 /**
  * Submits a Notification Hub Job.
  * Note: this is available to Standard SKU namespace and above.
@@ -1452,10 +1602,10 @@ function sendNotification(context, tags, notification, options = {}) {
  * @returns The notification hub job details including job ID and status.
  */
 function submitNotificationHubJob(context, job, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-submitNotificationHubJob", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$2}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += "/jobs";
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$2);
         headers.set("Content-Type", "application/atom+xml;type=entry;charset=utf-8");
         const request = createRequest(endpoint, "POST", headers, updatedOptions);
         request.body = serializeNotificationHubJobEntry(job);
@@ -1465,6 +1615,7 @@ function submitNotificationHubJob(context, job, options = {}) {
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME$1 = "updateInstallation";
 /**
  * Updates an installation using the JSON-Patch standard in RFC6902.
  * @param context - The Notification Hubs client.
@@ -1474,10 +1625,10 @@ function submitNotificationHubJob(context, job, options = {}) {
  * @returns A NotificationHubResponse with the tracking ID, correlation ID and location.
  */
 function updateInstallation(context, installationId, installationPatches, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-updateInstallation", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME$1}`, options, async (updatedOptions) => {
         const endpoint = context.requestUrl();
         endpoint.pathname += `/installations/${installationId}`;
-        const headers = context.createHeaders();
+        const headers = await context.createHeaders(OPERATION_NAME$1);
         headers.set("Content-Type", "application/json");
         const request = createRequest(endpoint, "PATCH", headers, updatedOptions);
         request.body = JSON.stringify(installationPatches);
@@ -1487,6 +1638,7 @@ function updateInstallation(context, installationId, installationPatches, option
 }
 
 // Copyright (c) Microsoft Corporation.
+const OPERATION_NAME = "updateRegistration";
 /**
  * Updates an existing registration.
  * @param context - The Notification Hubs client.
@@ -1495,7 +1647,7 @@ function updateInstallation(context, installationId, installationPatches, option
  * @returns The updated registration description.
  */
 function updateRegistration(context, registration, options = {}) {
-    return tracingClient.withSpan("NotificationHubsClientContext-updateRegistration", options, async (updatedOptions) => {
+    return tracingClient.withSpan(`NotificationHubsClientContext-${OPERATION_NAME}`, options, async (updatedOptions) => {
         if (!registration.etag) {
             throw new coreRestPipeline.RestError("ETag is required for registration update", { statusCode: 400 });
         }
@@ -1781,17 +1933,6 @@ function createBrowserInstallation(installation) {
     return Object.assign(Object.assign({}, installation), { platform: "browser" });
 }
 
-// Copyright (c) Microsoft Corporation.
-// Licensed under the MIT license.
-const JSON_CONTENT_TYPE = "application/json;charset=utf-8";
-const XML_CONTENT_TYPE = "application/xml";
-const STREAM_CONTENT_TYPE = "application/octet-stream";
-const WNS_TYPE_NAME = "X-WNS-Type";
-const WNS_RAW = "wns/raw";
-const WNS_BADGE = "wns/badge";
-const WNS_TITLE = "wns/tile";
-const WNS_TOAST = "wns/toast";
-
 // Copyright (c) Microsoft Corporation.
 /**
  * Creates a notification to send to an Apple device.