Add file read support

porsager · porsager · commit b8dbeb06a7c5 · 2019-11-28T10:03:51.000+01:00
diff --git a/lib/index.js b/lib/index.js
@@ -1,4 +1,5 @@
 import os from 'os'
+import fs from 'fs'
 import crypto from 'crypto'
 import Url from 'url'
 import Connection from './connection.js'
@@ -35,6 +36,7 @@ export default function Postgres(url, options) {
       , queries = Queue()
       , listeners = {}
       , typeArrayMap = {}
+      , files = {}
 
   function postgres(xs, ...args) {
     return query({}, getConnection(), xs, args)
@@ -142,7 +144,6 @@ export default function Postgres(url, options) {
     const promise = new Promise((resolve, reject) => {
       query.resolve = resolve
       query.reject = reject
-
       ended !== null
         ? reject(errors.connection('ENDED', options))
         : ready
@@ -266,9 +267,10 @@ export default function Postgres(url, options) {
       notify,
       unsafe,
       array,
+      file,
+      json,
       rows,
-      row,
-      json
+      row
     })
 
     function notify(channel, payload) {
@@ -283,6 +285,25 @@ export default function Postgres(url, options) {
       return query({ raw: true, simple: options.simple }, connection || getConnection(), xs, args)
     }
 
+    function file(path) {
+      const file = files[path]
+
+      if (typeof file === 'string')
+        return query({ raw: true, simple: true }, connection || getConnection(), files[path])
+
+      const promise = (file || new Promise((resolve, reject) => fs.readFile(path, 'utf8', (err, str) => {
+        if (err)
+          return reject(err)
+
+        files[path] = str
+        const q = query({ raw: true, simple: true }, connection || getConnection(), files[path])
+        promise.stream = q.stream
+        resolve(q)
+      })))
+
+      return promise
+    }
+
     options.types && Object.entries(options.types).forEach(([name, type]) => {
       if (name in sql)
         throw errors.generic({ message: name + ' is a reserved method name', code: 'RESERVED_METHOD_NAME' })
