Add entry fields to inferred types

alineacms · Nov 12, 2024 · d1f59c9 · d1f59c9
1 parent e2d35ad
commit d1f59c9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 11 deletions.
diff --git a/src/core/Graph.ts b/src/core/Graph.ts
@@ -77,9 +77,7 @@ type InferResult<Selection, Types, Include> = Selection extends Expr<
   : Selection extends undefined
   ? Types extends undefined
     ? EntryFields & (Include extends undefined ? {} : InferSelection<Include>)
-    : Infer<Types> &
-        EntryFields &
-        (Include extends undefined ? {} : InferSelection<Include>)
+    : Infer<Types> & (Include extends undefined ? {} : InferSelection<Include>)
   : InferSelection<Selection>
 
 type QueryResult<Selection, Types, Include> = Expand<

diff --git a/src/core/Infer.ts b/src/core/Infer.ts
@@ -1,24 +1,19 @@
 import {Expand, UnionOfValues} from 'alinea/core/util/Types'
+import {EntryFields} from './EntryFields.js'
 import {Expr} from './Expr.js'
 import {Field} from './Field.js'
 import {Type} from './Type.js'
 
 type QueryList<T> = Expand<
   UnionOfValues<{
-    [K in keyof T]: {_type: K} & QueryRow<T[K]>
+    [K in keyof T]: {_type: K} & Type.Infer<T[K]>
   }>
 >
 
-export type QueryRow<Definition> = {
-  [K in keyof Definition as Definition[K] extends Expr<any>
-    ? K
-    : never]: Definition[K] extends Expr<infer T> ? T : never
-}
-
 export type InferQueryValue<T> = T extends Array<Type<infer X>>
   ? InferQueryValue<X>
   : T extends Type<infer Fields>
-  ? QueryRow<Fields>
+  ? Type.Infer<Fields> & EntryFields
   : T extends Expr<infer QueryValue>
   ? QueryValue
   : T extends Record<string, Type>