feat(): pg-query-stream typescript

.eslintrc

@@ -2,7 +2,7 @@
   "plugins": ["prettier"],
   "parser": "@typescript-eslint/parser",
   "extends": ["plugin:prettier/recommended", "prettier/@typescript-eslint"],
-  "ignorePatterns": ["node_modules", "coverage", "packages/pg-protocol/dist/**/*"],
+  "ignorePatterns": ["node_modules", "coverage", "packages/pg-protocol/dist/**/*", "packages/pg-query-stream/dist/**/*"],
   "parserOptions": {
     "ecmaVersion": 2017,
     "sourceType": "module"

package.json

@@ -11,7 +11,8 @@
   ],
   "scripts": {
     "test": "yarn lerna exec yarn test",
-    "build": "yarn lerna exec --scope pg-protocol yarn build",
+    "build": "tsc --build",
+    "build:watch": "tsc --build --watch",
     "pretest": "yarn build",
     "lint": "eslint '*/**/*.{js,ts,tsx}'"
   },
@@ -23,7 +24,8 @@
     "eslint-plugin-node": "^11.1.0",
     "eslint-plugin-prettier": "^3.1.4",
     "lerna": "^3.19.0",
-    "prettier": "2.1.2"
+    "prettier": "2.1.2",
+    "typescript": "^4.0.3"
   },
   "prettier": {
     "semi": false,

packages/pg-protocol/package.json

@@ -13,13 +13,17 @@
     "chunky": "^0.0.0",
     "mocha": "^7.1.2",
     "ts-node": "^8.5.4",
-    "typescript": "^3.7.3"
+    "typescript": "^4.0.3"
   },
   "scripts": {
     "test": "mocha dist/**/*.test.js",
     "build": "tsc",
     "build:watch": "tsc --watch",
     "prepublish": "yarn build",
     "pretest": "yarn build"
-  }
+  },
+  "files": [
+    "/dist/*{js,ts,map}",
+    "/src"
+  ]
 }

packages/pg-protocol/tsconfig.json

@@ -9,6 +9,7 @@
     "moduleResolution": "node",
     "sourceMap": true,
     "outDir": "dist",
+    "incremental": true,
     "baseUrl": ".",
     "declaration": true,
     "paths": {

packages/pg-query-stream/package.json

@@ -2,9 +2,10 @@
   "name": "pg-query-stream",
   "version": "3.3.2",
   "description": "Postgres query result returned as readable stream",
-  "main": "index.js",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
   "scripts": {
-    "test": "mocha"
+    "test": "mocha -r ts-node/register test/**/*.ts"
   },
   "repository": {
     "type": "git",
@@ -16,20 +17,30 @@
     "query",
     "stream"
   ],
+  "files": [
+    "/dist/*{js,ts,map}",
+    "/src"
+  ],
   "author": "Brian M. Carlson",
   "license": "MIT",
   "bugs": {
     "url": "https://github.com/brianc/node-postgres/issues"
   },
   "devDependencies": {
+    "@types/node": "^14.0.0",
+    "@types/pg": "^7.14.5",
+    "@types/chai": "^4.2.13",
+    "@types/mocha": "^8.0.3",
     "JSONStream": "~0.7.1",
     "concat-stream": "~1.0.1",
     "eslint-plugin-promise": "^3.5.0",
     "mocha": "^7.1.2",
     "pg": "^8.4.2",
     "stream-spec": "~0.3.5",
     "stream-tester": "0.0.5",
-    "through": "~2.3.4"
+    "through": "~2.3.4",
+    "ts-node": "^8.5.4",
+    "typescript": "^4.0.3"
   },
   "dependencies": {
     "pg-cursor": "^2.4.2"

packages/pg-query-stream/index.js → packages/pg-query-stream/src/index.ts

@@ -1,11 +1,30 @@
-const { Readable } = require('stream')
-const Cursor = require('pg-cursor')
+import { Readable } from 'stream'
+import { Submittable, Connection } from 'pg'
+import Cursor from 'pg-cursor'
 
-class PgQueryStream extends Readable {
-  constructor(text, values, config = {}) {
+interface QueryStreamConfig {
+  batchSize?: number
+  highWaterMark?: number
+  rowMode?: 'array'
+  types?: any
+}
+
+class QueryStream extends Readable implements Submittable {
+  cursor: any
+  _result: any
+
+  handleRowDescription: Function
+  handleDataRow: Function
+  handlePortalSuspended: Function
+  handleCommandComplete: Function
+  handleReadyForQuery: Function
+  handleError: Function
+  handleEmptyQuery: Function
+
+  public constructor(text: string, values?: any[], config: QueryStreamConfig = {}) {
     const { batchSize, highWaterMark = 100 } = config
-    // https://nodejs.org/api/stream.html#stream_new_stream_readable_options
-    super({ objectMode: true, emitClose: true, autoDestroy: true, highWaterMark: batchSize || highWaterMark })
+
+    super({ objectMode: true, autoDestroy: true, highWaterMark: batchSize || highWaterMark })
     this.cursor = new Cursor(text, values, config)
 
     // delegate Submittable callbacks to cursor
@@ -21,19 +40,19 @@ class PgQueryStream extends Readable {
     this._result = this.cursor._result
   }
 
-  submit(connection) {
+  public submit(connection: Connection): void {
     this.cursor.submit(connection)
   }
 
-  _destroy(_err, cb) {
-    this.cursor.close((err) => {
+  public _destroy(_err: Error, cb: Function) {
+    this.cursor.close((err?: Error) => {
       cb(err || _err)
     })
   }
 
   // https://nodejs.org/api/stream.html#stream_readable_read_size_1
-  _read(size) {
-    this.cursor.read(size, (err, rows, result) => {
+  public _read(size: number) {
+    this.cursor.read(size, (err: Error, rows: any[]) => {
       if (err) {
         // https://nodejs.org/api/stream.html#stream_errors_while_reading
         this.destroy(err)
@@ -45,4 +64,4 @@ class PgQueryStream extends Readable {
   }
 }
 
-module.exports = PgQueryStream
+export = QueryStream