Merge pull request #1 from jkyberneees/initial-commit

initial commit

Author: jkyberneees

.travis.yml

@@ -0,0 +1,6 @@
+
+language: node_js
+node_js:
+  - "10"
+
+install: npm install

README.md

@@ -1,2 +1,40 @@
 # on-http-end
 Allows to capture HTTP response content and headers on request end.
+> Inspired by: https://github.com/kwhitley/apicache/blob/master/src/apicache.js
+
+## Install
+```bash
+npm i on-http-end
+```
+
+## Usage
+```js
+const onEnd = require('on-http-end')
+const http = require('http')
+
+const server = http.createServer((req, res) => {
+  onEnd(res, (payload) => {
+    console.log(payload)
+  })
+
+  res.setHeader('my-header', 'value')
+  res.end('Hello Word!', 'utf-8')
+})
+
+server.listen(3000)
+```
+
+Output:
+```bash
+{
+  status: 200,
+  headers: [Object: null prototype] { 'my-header': 'value' },
+  data: 'Hello Word!',
+  encoding: 'utf-8'
+}
+```
+
+## Want to contribute?
+This is your repo ;)
+
+> Note: We aim to be 100% code coverage, please consider it on your pull requests.

demos/basic.js

@@ -0,0 +1,13 @@
+const onEnd = require('./../index')
+const http = require('http')
+
+const server = http.createServer((req, res) => {
+  onEnd(res, (payload) => {
+    console.log(payload)
+  })
+
+  res.setHeader('my-header', 'value')
+  res.end('Hello Word!', 'utf-8')
+})
+
+server.listen(3000)

index.js

@@ -0,0 +1,58 @@
+module.exports = (res, cb) => {
+  res._parent = {
+    write: res.write,
+    end: res.end,
+    content: undefined,
+    ended: false
+  }
+
+  res.write = function (content) {
+    accumulate(res, content)
+    return res._parent.write.apply(res, arguments)
+  }
+
+  res.end = function (content, encoding) {
+    if (!res._parent.ended) {
+      res._parent.ended = true
+
+      accumulate(res, content)
+      const headers = res.getHeaders()
+      const payload = map(res.statusCode, headers, res._parent.content, encoding)
+
+      setImmediate(() => {
+        cb(payload)
+      })
+    }
+
+    return res._parent.end.apply(res, arguments)
+  }
+}
+
+function map (status, headers, data, encoding) {
+  return {
+    status: status,
+    headers: headers,
+    data: data,
+    encoding: encoding
+  }
+}
+
+function accumulate (res, content) {
+  if (content) {
+    if (typeof content === 'string') {
+      res._parent.content = (res._parent.content || '') + content
+    } else if (Buffer.isBuffer(content)) {
+      let oldContent = res._parent.content
+
+      if (typeof oldContent === 'string') {
+        oldContent = Buffer.from(oldContent)
+      } else if (!oldContent) {
+        oldContent = Buffer.alloc(0)
+      }
+
+      res._parent.content = Buffer.concat([oldContent, content], oldContent.length + content.length)
+    } else {
+      res._parent.content = content
+    }
+  }
+}