chore(core): add setTokenFeature

Author: trivikr

packages/core/src/submodules/client/index.ts

@@ -1,3 +1,4 @@
 export * from "./emitWarningIfUnsupportedVersion";
 export * from "./setCredentialFeature";
 export * from "./setFeature";
+export * from "./setTokenFeature";

packages/core/src/submodules/client/setTokenFeature.spec.ts

@@ -0,0 +1,17 @@
+import { AttributedTokenIdentity } from "@aws-sdk/types";
+import { describe, expect, test as it } from "vitest";
+
+import { setTokenFeature } from "./setTokenFeature";
+
+describe(setTokenFeature.name, () => {
+  it("should create the data path if it does't exist", () => {
+    const token = { token: "" } as AttributedTokenIdentity;
+
+    expect(setTokenFeature(token, "BEARER_SERVICE_ENV_VARS", "3")).toEqual({
+      token: "",
+      $source: {
+        BEARER_SERVICE_ENV_VARS: "3",
+      },
+    });
+  });
+});

packages/core/src/submodules/client/setTokenFeature.ts

@@ -0,0 +1,18 @@
+import type { AttributedTokenIdentity, AwsSdkTokenFeatures } from "@aws-sdk/types";
+
+/**
+ * @internal
+ *
+ * @returns the token with source feature attribution.
+ */
+export function setTokenFeature<F extends keyof AwsSdkTokenFeatures>(
+  token: AttributedTokenIdentity,
+  feature: F,
+  value: AwsSdkTokenFeatures[F]
+): AttributedTokenIdentity {
+  if (!token.$source) {
+    token.$source = {};
+  }
+  token.$source![feature] = value;
+  return token;
+}