feat: query parser: add types for count (#498)

Closes #447, #479.
supabase · Mar 3, 2024 · 5ea5163 · 5ea5163
1 parent 14910bf
commit 5ea5163
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/select-query-parser.ts b/src/select-query-parser.ts
@@ -267,6 +267,8 @@ type ConstructFieldDefinition<
   : Field extends { name: string; original: string }
   ? Field['original'] extends keyof Row
     ? { [K in Field['name']]: Row[Field['original']] }
+    : Field['original'] extends 'count'
+    ? { count: number }
     : SelectQueryError<`Referencing missing column \`${Field['original']}\``>
   : Record<string, unknown>
 

diff --git a/test/index.test-d.ts b/test/index.test-d.ts
@@ -88,6 +88,15 @@ const postgrest = new PostgrestClient<Database>(REST_URL)
   expectType<{ message: string | null }>(data)
 }
 
+// `count` in embedded resource
+{
+  const { data, error } = await postgrest.from('messages').select('message, users(count)').single()
+  if (error) {
+    throw new Error(error.message)
+  }
+  expectType<{ message: string | null; users: { count: number } | null }>(data)
+}
+
 // json accessor in select query
 {
   const { data, error } = await postgrest