fix(ts-client): no select root types if not in schema

Author: jasonkuhrt

src/entrypoints/alpha/client.ts

@@ -1,3 +1,2 @@
 export * from '../../client/client.js'
-// todo need to export a generated version
-export { create as createSelect } from '../../select.js'
+export { create as createSelect, select } from '../../select.js'

src/select.test-d.ts

@@ -0,0 +1,32 @@
+import { describe, expect, expectTypeOf, it, test } from 'vitest'
+import type { Index } from '../tests/_/schema/schema.js'
+import type * as SchemaQueryOnly from '../tests/_/schemaQueryOnly/generated/Index.js'
+import type { SelectionSet } from './client/SelectionSet/__.js'
+import { create } from './select.js'
+
+describe(`select`, () => {
+  const select = create<Index>()
+  it(`returns the input for any method name`, () => {
+    const s = select as any // eslint-disable-line
+    expect(s.anything(1)).toEqual(1) // eslint-disable-line
+  })
+
+  it(`has type safe methods`, () => {
+    // @ts-expect-error ID issue
+    type $Parameters = Parameters<typeof select.Bar>
+    expectTypeOf<$Parameters>().toEqualTypeOf<[SelectionSet.Object<Index['objects']['Bar'], Index>]>()
+    expectTypeOf(select.Bar({ int: true })).toEqualTypeOf<{ int: true }>()
+  })
+})
+
+describe(`create`, () => {
+  const select = create<SchemaQueryOnly.Index>()
+  test(`does not have root types if not in schema`, () => {
+    // fine
+    select.Query
+    // @ts-expect-error no mutation in schema
+    select.Mutation
+    // @ts-expect-error no mutation in schema
+    select.Subscription
+  })
+})

src/select.test.ts

@@ -1,13 +1,13 @@
-import { expect, it } from 'vitest'
-import type { Index } from '../tests/ts/_/schema/generated/Index.js'
+import { expect, test } from 'vitest'
+import type { Index } from '../tests/_/schema/schema.js'
 import { create } from './select.js'
 
-it(`returns the input for any method name`, () => {
-  const select = create() as any // eslint-disable-line
-  expect(select.anything(1)).toEqual(1) // eslint-disable-line
+const select = create<Index>()
+test(`returns the input for any method name`, () => {
+  const s = select as any // eslint-disable-line
+  expect(s.anything(1)).toEqual(1) // eslint-disable-line
 })
 
-it(`has type safe methods`, () => {
-  const select = create<Index>()
+test(`has type safe methods`, () => {
   expect(select.Bar({ ___: { $defer: true, int: true } })).toEqual({ ___: { $defer: true, int: true } })
 })

src/select.ts

@@ -1,12 +1,12 @@
 import type { SelectionSet } from './client/SelectionSet/__.js'
+import type { RootTypeName } from './lib/graphql.js'
 import type { Exact } from './lib/prelude.js'
 import type { Schema } from './Schema/__.js'
 
-// todo test
 // dprint-ignore
 type TypeSelectionSets<$Index extends Schema.Index> =
 & {
-		[$RootTypeName in Schema.RootTypeName]:
+		[$RootTypeName in RootTypeName as $RootTypeName extends keyof $Index['Root'] ? $Index['Root'][$RootTypeName] extends null ? never : $RootTypeName:never ]:
 			<$SelectionSet extends object>(selectionSet: Exact<$SelectionSet, SelectionSet.Root<$Index, $RootTypeName>>) =>
 				$SelectionSet
 	}
@@ -33,3 +33,7 @@ export const create = <$Index extends Schema.Index>(): TypeSelectionSets<$Index>
 const idProxy = new Proxy({}, {
   get: () => (value: unknown) => value,
 })
+
+// eslint-disable-next-line
+// @ts-ignore generated types
+export const select: TypeSelectionSets<NamedSchemas['default']['index']> = idProxy