fix: Support querying hash data with 'not' condition (#95)

Co-authored-by: Percy Chen <[email protected]>

Author: chenpercy

src/encryption.ts

@@ -212,8 +212,8 @@ function rewriteHashedFieldPath(
   hashField: string
 ) {
   const items = path.split('.').reverse()
-  // Special case for `where field equals` clause
-  if (items.includes('where') && items[1] === field && items[0] === 'equals') {
+  // Special case for `where field equals or not` clause
+  if (items.includes('where') && items[1] === field && ['equals', 'not'].includes(items[0])) {
     items[1] = hashField
     return items.reverse().join('.')
   }

src/tests/integration.test.ts

@@ -392,14 +392,32 @@ describe.each(clients)('integration ($type)', ({ client }) => {
       expect(post).toBeNull()
       return
     }
-    // Should be unreacheable
+    // Should be unreachable
     const reached = true
     expect(reached).toBe(false)
   })
+
   test("Doesn't work with the Fluent API", async () => {
     const posts = await client.user.findUnique({ where: { email } }).posts()
     for (const post of posts!) {
       expect(post.content).toMatch(cloakedStringRegex)
     }
   })
+
+  test("query entries with non-empty name", async () => {
+    const fakeName = 'f@keU$er'
+    await client.user.create({
+       data: {
+         name: '',
+         email: '[email protected]'
+      }
+    });
+    const users = await client.user.findMany();
+    // assume active user with nonempty name
+    const activeUserCount = await client.user.count({ where: { name: { not: '' } } })
+    // use fakeName to pretend unique name
+    const existingUsers = await client.user.findMany({ where: { name: { not: fakeName } } })
+    expect(activeUserCount).toBe(users.length - 1);
+    expect(existingUsers).toEqual(users);
+  })
 })

src/visitor.ts

@@ -84,7 +84,7 @@ export function visitInputTargetFields<
 ) {
   traverseTree(
     params.args,
-    makeVisitor(models, visitor, ['equals', 'set'], debug.encryption),
+    makeVisitor(models, visitor, ['equals', 'set', 'not'], debug.encryption),
     {
       currentModel: params.model!
     }