decode obfuscated bandit

Eppo-exp · Jan 16, 2025 · 5bd5fd2 · 5bd5fd2
1 parent 8ce8737
commit 5bd5fd2
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 4 deletions.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@eppo/js-client-sdk-common",
-  "version": "4.8.2",
+  "version": "4.8.3-alpha.1",
   "description": "Common library for Eppo JavaScript SDKs (web, react native, and node)",
   "main": "dist/index.js",
   "files": [

diff --git a/src/client/eppo-precomputed-client.ts b/src/client/eppo-precomputed-client.ts
@@ -18,7 +18,7 @@ import {
   MAX_EVENT_QUEUE_SIZE,
   PRECOMPUTED_BASE_URL,
 } from '../constants';
-import { decodePrecomputedFlag } from '../decoding';
+import { decodePrecomputedBandit, decodePrecomputedFlag } from '../decoding';
 import { FlagEvaluationWithoutDetails } from '../evaluator';
 import FetchHttpClient from '../http-client';
 import {
@@ -361,10 +361,13 @@ export default class EppoPrecomputedClient {
     return this.getObfuscatedPrecomputedBandit(banditKey);
   }
 
-  private getObfuscatedPrecomputedBandit(banditKey: string): IObfuscatedPrecomputedBandit | null {
+  private getObfuscatedPrecomputedBandit(banditKey: string): IPrecomputedBandit | null {
     const salt = this.precomputedBanditStore?.salt;
     const saltedAndHashedBanditKey = getMD5Hash(banditKey, salt);
-    return this.precomputedBanditStore?.get(saltedAndHashedBanditKey) ?? null;
+    const precomputedBandit: IObfuscatedPrecomputedBandit | null = this.precomputedBanditStore?.get(
+      saltedAndHashedBanditKey,
+    ) as IObfuscatedPrecomputedBandit;
+    return precomputedBandit ? decodePrecomputedBandit(precomputedBandit) : null;
   }
 
   public isInitialized() {

diff --git a/src/decoding.ts b/src/decoding.ts
@@ -11,6 +11,8 @@ import {
   ObfuscatedSplit,
   PrecomputedFlag,
   DecodedPrecomputedFlag,
+  IPrecomputedBandit,
+  IObfuscatedPrecomputedBandit,
 } from './interfaces';
 import { decodeBase64 } from './obfuscation';
 
@@ -88,3 +90,16 @@ export function decodePrecomputedFlag(precomputedFlag: PrecomputedFlag): Decoded
     extraLogging: decodeObject(precomputedFlag.extraLogging ?? {}),
   };
 }
+
+export function decodePrecomputedBandit(
+  precomputedBandit: IObfuscatedPrecomputedBandit,
+): IPrecomputedBandit {
+  return {
+    ...precomputedBandit,
+    banditKey: decodeBase64(precomputedBandit.banditKey),
+    action: decodeBase64(precomputedBandit.action),
+    modelVersion: decodeBase64(precomputedBandit.modelVersion),
+    actionNumericAttributes: decodeObject(precomputedBandit.actionNumericAttributes ?? {}),
+    actionCategoricalAttributes: decodeObject(precomputedBandit.actionCategoricalAttributes ?? {}),
+  };
+}