From adadab1d104962d9bccc90a25c3961ce088a37d5 Mon Sep 17 00:00:00 2001
From: Jerel Miller <jerelmiller@gmail.com>
Date: Wed, 8 Jan 2025 15:09:44 -0700
Subject: [PATCH] Use mode instead of enabled

---
 src/masking/types.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/masking/types.ts b/src/masking/types.ts
index 26aa434745..138b4e0e6d 100644
--- a/src/masking/types.ts
+++ b/src/masking/types.ts
@@ -44,7 +44,8 @@ export type MaybeMasked<TData> =
     // prevent "Type instantiation is excessively deep and possibly infinite."
     true extends IsAny<TData> ? TData
     : TData extends { __masked?: true } ? Prettify<RemoveMaskedMarker<TData>>
-    : DataMasking extends { enabled: false } ?
+    : DataMasking extends { mode: "preserveTypes" } ? TData
+    : DataMasking extends { mode: "unmask" } ?
       true extends ContainsFragmentsRefs<TData> ?
         Unmasked<TData>
       : TData