Add types for typescript

Author: 9at8

index.d.ts

@@ -0,0 +1,44 @@
+declare module 'apollo-datasource-mongodb' {
+  import { DataSource } from 'apollo-datasource'
+  import { Collection as MongoCollection, ObjectId } from 'mongodb'
+  import {
+    Collection as MongooseCollection,
+    Document,
+    Model as MongooseModel
+  } from 'mongoose'
+
+  export type Collection<T> = T extends Document
+    ? MongooseCollection
+    : MongoCollection<T>
+
+  export type Model<T> = T extends Document ? MongooseModel<T> : never
+
+  export type ModelOrCollection<T> = T extends Document
+    ? Model<T>
+    : Collection<T>
+
+  export interface Options {
+    ttl: number
+  }
+
+  export class MongoDataSource<TData, TContext = any> extends DataSource<
+    TContext
+  > {
+    protected collection: Collection<TData>
+    protected model: Model<TData>
+
+    constructor(modelOrCollection: ModelOrCollection<TData>)
+
+    findOneById(
+      id: ObjectId,
+      options?: Options
+    ): Promise<TData | null | undefined>
+
+    findManyByIds(
+      ids: ObjectId[],
+      options?: Options
+    ): Promise<(TData | null | undefined)[]>
+
+    deleteFromCacheById(id: ObjectId): void
+  }
+}

package.json

@@ -3,6 +3,7 @@
   "version": "0.2.1",
   "description": "Apollo data source for MongoDB",
   "main": "dist/index.js",
+  "types": "index.d.ts",
   "scripts": {
     "build": "babel src -d dist --ignore **/*.test.js",
     "dev": "babel src -d dist --watch --ignore **/*.test.js",
@@ -24,6 +25,8 @@
     "@babel/cli": "^7.4.3",
     "@babel/core": "^7.4.3",
     "@babel/preset-env": "^7.4.3",
+    "@types/mongodb": "^3.5.3",
+    "@types/mongoose": "^5.7.7",
     "babel-jest": "^24.7.1",
     "graphql": "^14.2.1",
     "jest": "^24.7.1",