Add types to finder methods

Author: garymardell

packages/api-client-core/src/GadgetRecordList.ts

@@ -13,17 +13,17 @@ type PaginationConfig = {
 };
 
 /** Represents a list of objects returned from the API. Facilitates iterating and paginating. */
-export class GadgetRecordList<Shape extends RecordShape> extends Array<GadgetRecord<Shape>> {
+export class GadgetRecordList<Shape extends RecordShape, RecordType extends GadgetRecord<Shape> = GadgetRecord<any>> extends Array<RecordType> {
   modelManager!: AnyModelManager | InternalModelManager<Shape>;
   pagination!: PaginationConfig;
 
   /** Internal method used to create a list. Should not be used by applications. */
-  static boot<Shape extends RecordShape>(
+  static boot<Shape extends RecordShape, RecordType extends GadgetRecord<Shape> = GadgetRecord<any>>(
     modelManager: AnyModelManager | InternalModelManager<Shape>,
-    records: GadgetRecord<Shape>[],
+    records: RecordType[],
     pagination: PaginationConfig
   ) {
-    const list = new GadgetRecordList<Shape>();
+    const list = new GadgetRecordList<Shape, RecordType>();
     list.push(...records);
     list.modelManager = modelManager;
     list.pagination = pagination;
@@ -71,7 +71,7 @@ export class GadgetRecordList<Shape extends RecordShape> extends Array<GadgetRec
       ...options,
       after: this.pagination.pageInfo.endCursor,
       first: first || last,
-    }) as Promise<GadgetRecordList<Shape>>;
+    }) as Promise<GadgetRecordList<Shape, RecordType>>;
     return await nextPage;
   }
 
@@ -86,7 +86,7 @@ export class GadgetRecordList<Shape extends RecordShape> extends Array<GadgetRec
       ...options,
       before: this.pagination.pageInfo.startCursor,
       last: last || first,
-    }) as Promise<GadgetRecordList<Shape>>;
+    }) as Promise<GadgetRecordList<Shape, RecordType>>;
     return await prevPage;
   }
 }

packages/api-client-core/src/ModelManager.ts

@@ -24,9 +24,9 @@ export type AnyModelFinderMetadata = {
  * This is a generic interface. Concrete ones are generated by Gadget, */
 export interface AnyModelManager {
   connection: GadgetConnection;
-  findOne: ((id: string, options: any) => Promise<GadgetRecord<any>>) & AnyModelFinderMetadata;
-  findMany: ((options: any) => Promise<GadgetRecordList<any>>) & AnyModelFinderMetadata;
-  findFirst: ((options: any) => Promise<GadgetRecord<any>>) & AnyModelFinderMetadata;
-  maybeFindFirst(options: any): Promise<GadgetRecord<any> | null>;
-  maybeFindOne(id: string, options: any): Promise<GadgetRecord<any> | null>;
+  findOne: (<RecordType extends GadgetRecord<any> = GadgetRecord<any>>(id: string, options: any) => Promise<RecordType>) & AnyModelFinderMetadata;
+  findMany: (<RecordType extends GadgetRecord<any> = GadgetRecord<any>>(options: any) => Promise<GadgetRecordList<any, RecordType>>) & AnyModelFinderMetadata;
+  findFirst: (<RecordType extends GadgetRecord<any> = GadgetRecord<any>>(options: any) => Promise<RecordType>) & AnyModelFinderMetadata;
+  maybeFindFirst<RecordType extends GadgetRecord<any> = GadgetRecord<any>>(options: any): Promise<RecordType | null>;
+  maybeFindOne<RecordType extends GadgetRecord<any> = GadgetRecord<any>>(id: string, options: any): Promise<RecordType| null>;
 }