feat: add a useTreatment hook

newrelic · Sep 3, 2020 · e9c8aac · e9c8aac
1 parent 503e3c6
commit e9c8aac
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/hooks/useTreatment.js b/src/hooks/useTreatment.js
@@ -0,0 +1,20 @@
+import { useMemo } from 'react';
+import { useTreatments } from '@splitsoftware/splitio-react';
+
+const useTreatment = (name) => {
+  const treatments = useTreatments([name]);
+  const { treatment, config } = treatments[name];
+
+  return {
+    treatment,
+    config: useMemo(() => {
+      try {
+        return JSON.parse(config);
+      } catch (e) {
+        return {};
+      }
+    }, [config]),
+  };
+};
+
+export default useTreatment;