Clean up callback methods, support getting headers from response

Author: zackify

package.json

@@ -1,6 +1,6 @@
 {
   "name": "legible",
-  "version": "0.2.6",
+  "version": "0.2.7",
   "description": "cleanly code your api requests",
   "main": "dist/index.js",
   "scripts": {

src/utilities/fetch.js

@@ -3,7 +3,10 @@ import fetch from 'isomorphic-fetch'
 export default (url, options = {}) => {
   return new Promise((resolve, reject) => {
     fetch(url, options)
-    .then(response => response.json())
+    .then(response => {
+      if (options.onResponse) options.onResponse(response)
+      return response.json()
+    })
     .then(json => resolve(json))
     .catch(error => reject(error))
   })

src/utilities/normalize.js

@@ -1,10 +1,18 @@
 import processBody from './body'
 
+/*
+  Methods that can be callbacks, ex:
+  headers: ${partialHeaders => }
+*/
+const callbackMethods = {
+  headers: ({ value, partial }) => value(partial),
+  url: ({ value, partial }) => value(partial.url)
+}
+
 /*
   Take in raw query string and
   return a fetch api compatible object
 */
-
 const buildObjectFromTag = (strings, vars, partial) => {
   const namespace = 'legible-request-var-'
   return strings
@@ -31,14 +39,15 @@ const buildObjectFromTag = (strings, vars, partial) => {
       if (!value.startsWith(namespace)) return [key, value]
       // Get the index at the end of the namespaced string
       const index = parseInt(value.replace(namespace, ''), 10)
-      // Return an array of the object key and replaced value from `vars`
-      if (partial.url && key === 'url' && typeof vars[index] === 'function') {
-        return [key, vars[index](partial.url)]
-      }
 
-      if (key === 'headers' && typeof vars[index] === 'function') {
-        return [key, vars[index](partial)]
+      // run through any callback methods
+      if (callbackMethods[key] && typeof vars[index] === 'function') {
+        return [key, callbackMethods[key]({
+          value: vars[index],
+          partial
+        })]
       }
+
       return [key, vars[index]]
     })
     // Convert to object

tests/unit/request.js

@@ -7,4 +7,13 @@ describe('request', () => {
     `
     expect(response.country_code).to.equal('US')
   })
+
+  it('returns headers on response', async function () {
+    await request`
+      url: https://freegeoip.net/json/github.com
+      onResponse: ${response => {
+        expect(response.headers.get('content-type')).to.equal('application/json')
+      }}
+    `
+  })
 })