Add support for inferring types from refs

hayes · Sep 22, 2023 · f8fb4e6 · f8fb4e6 · vercel · Sep 22, 2023
1 parent 3207dcc
commit f8fb4e6
Show file tree

Hide file tree

Showing 54 changed files with 245 additions and 98 deletions.
diff --git a/.changeset/smart-wasps-look.md b/.changeset/smart-wasps-look.md
@@ -0,0 +1,6 @@
+---
+'@pothos/core': minor
+'@pothos/deno': minor
+---
+
+Add support for $inferType and $inferInput helpers on Refs
diff --git a/packages/core/src/refs/enum.ts b/packages/core/src/refs/enum.ts
@@ -7,6 +7,10 @@ export default class EnumRef<T, U = T>
 {
   override kind = 'Enum' as const;
 
+  $inferType!: T;
+
+  $inferInput!: U;
+
   [outputShapeKey]!: T;
 
   [inputShapeKey]!: U;

diff --git a/packages/core/src/refs/field.ts b/packages/core/src/refs/field.ts
@@ -7,6 +7,8 @@ export default class FieldRef<T = unknown, Kind extends FieldKind = FieldKind> {
 
   fieldName?: string;
 
+  $inferType!: T;
+
   [outputFieldShapeKey]!: T;
 
   constructor(kind: Kind, parentTypename: string) {

diff --git a/packages/core/src/refs/input-field.ts b/packages/core/src/refs/input-field.ts
@@ -13,6 +13,8 @@ export default class InputFieldRef<
 
   argFor?: FieldRef | InputFieldRef;
 
+  $inferInput!: T;
+
   [inputFieldShapeKey]!: T;
 
   constructor(kind: Kind, parentTypename: string) {

diff --git a/packages/core/src/refs/input-list.ts b/packages/core/src/refs/input-list.ts
@@ -1,14 +1,18 @@
 import { inputShapeKey, InputTypeParam, SchemaTypes } from '../types';
 import BaseTypeRef from './base';
 
-export default class InputObjectRef<Types extends SchemaTypes, T>
+export default class InputListRef<Types extends SchemaTypes, T>
   extends BaseTypeRef
   implements PothosSchemaTypes.InputListRef<Types, T>
 {
   override kind = 'InputList' as const;
 
   [inputShapeKey]!: T;
+
+  $inferInput!: T;
+
   listType: InputTypeParam<Types>;
+
   required: boolean;
 
   constructor(listType: InputTypeParam<Types>, required: boolean) {

diff --git a/packages/core/src/refs/input-object.ts b/packages/core/src/refs/input-object.ts
@@ -14,6 +14,8 @@ export default class InputObjectRef<T>
 {
   override kind = 'InputObject' as const;
 
+  $inferInput!: T;
+
   [inputShapeKey]!: T;
 
   constructor(name: string) {

diff --git a/packages/core/src/refs/input.ts b/packages/core/src/refs/input.ts
@@ -4,6 +4,8 @@ import BaseTypeRef from './base';
 export default class InputTypeRef<T> extends BaseTypeRef {
   override kind;
 
+  $inferInput!: T;
+
   [inputShapeKey]!: T;
 
   constructor(kind: 'Enum' | 'InputObject' | 'Scalar', name: string) {

diff --git a/packages/core/src/refs/interface.ts b/packages/core/src/refs/interface.ts
@@ -16,7 +16,10 @@ export default class InterfaceRef<T, P = T>
 {
   override kind = 'Interface' as const;
 
+  $inferType!: T;
+
   [outputShapeKey]!: T;
+
   [parentShapeKey]!: P;
 
   constructor(name: string) {

diff --git a/packages/core/src/refs/list.ts b/packages/core/src/refs/list.ts
@@ -7,10 +7,14 @@ export default class ListRef<Types extends SchemaTypes, T, P = T>
 {
   override kind = 'List' as const;
 
+  $inferType!: T;
+
   [outputShapeKey]!: T;
+
   [parentShapeKey]!: P;
 
   listType: TypeParam<Types>;
+
   nullable: boolean;
 
   constructor(listType: TypeParam<Types>, nullable: boolean) {

diff --git a/packages/core/src/refs/object.ts b/packages/core/src/refs/object.ts
@@ -15,7 +15,10 @@ export default class ObjectRef<T, P = T>
 {
   override kind = 'Object' as const;
 
+  $inferType!: T;
+
   [outputShapeKey]!: T;
+
   [parentShapeKey]!: P;
 
   constructor(name: string) {

diff --git a/packages/core/src/refs/output.ts b/packages/core/src/refs/output.ts
@@ -4,7 +4,10 @@ import BaseTypeRef from './base';
 export default class OutputTypeRef<T, P = T> extends BaseTypeRef {
   override kind;
 
+  $inferType!: T;
+
   [outputShapeKey]!: T;
+
   [parentShapeKey]!: P;
 
   constructor(kind: 'Enum' | 'Interface' | 'Object' | 'Scalar' | 'Union', name: string) {

diff --git a/packages/core/src/refs/scalar.ts b/packages/core/src/refs/scalar.ts
@@ -7,7 +7,12 @@ export default class ScalarRef<T, U, P = T>
 {
   override kind = 'Scalar' as const;
 
+  $inferType!: T;
+
+  $inferInput!: U;
+
   [outputShapeKey]!: T;
+
   [parentShapeKey]!: P;
 
   [inputShapeKey]!: U;

diff --git a/packages/core/src/refs/union.ts b/packages/core/src/refs/union.ts
@@ -7,7 +7,10 @@ export default class UnionRef<T, P = T>
 {
   override kind = 'Union' as const;
 
+  $inferType!: T;
+
   [outputShapeKey]!: T;
+
   [parentShapeKey]!: P;
 
   constructor(name: string) {

diff --git a/packages/deno/packages/core/builder.ts b/packages/deno/packages/core/builder.ts
diff --git a/packages/deno/packages/core/refs/enum.ts b/packages/deno/packages/core/refs/enum.ts
diff --git a/packages/deno/packages/core/refs/field.ts b/packages/deno/packages/core/refs/field.ts
diff --git a/packages/deno/packages/core/refs/input-field.ts b/packages/deno/packages/core/refs/input-field.ts
diff --git a/packages/deno/packages/core/refs/input-list.ts b/packages/deno/packages/core/refs/input-list.ts
diff --git a/packages/deno/packages/core/refs/input-object.ts b/packages/deno/packages/core/refs/input-object.ts
diff --git a/packages/deno/packages/core/refs/input.ts b/packages/deno/packages/core/refs/input.ts
diff --git a/packages/deno/packages/core/refs/interface.ts b/packages/deno/packages/core/refs/interface.ts
diff --git a/packages/deno/packages/core/refs/list.ts b/packages/deno/packages/core/refs/list.ts
diff --git a/packages/deno/packages/core/refs/object.ts b/packages/deno/packages/core/refs/object.ts
diff --git a/packages/deno/packages/core/refs/output.ts b/packages/deno/packages/core/refs/output.ts
diff --git a/packages/deno/packages/core/refs/scalar.ts b/packages/deno/packages/core/refs/scalar.ts
diff --git a/packages/deno/packages/core/refs/union.ts b/packages/deno/packages/core/refs/union.ts
diff --git a/packages/deno/packages/core/types/builder-options.ts b/packages/deno/packages/core/types/builder-options.ts
diff --git a/packages/deno/packages/core/types/global/type-options.ts b/packages/deno/packages/core/types/global/type-options.ts
diff --git a/packages/deno/packages/core/types/utils.ts b/packages/deno/packages/core/types/utils.ts
diff --git a/packages/deno/packages/core/utils/base64.ts b/packages/deno/packages/core/utils/base64.ts
diff --git a/packages/deno/packages/plugin-add-graphql/global-types.ts b/packages/deno/packages/plugin-add-graphql/global-types.ts
diff --git a/packages/deno/packages/plugin-add-graphql/schema-builder.ts b/packages/deno/packages/plugin-add-graphql/schema-builder.ts