Merge pull request #111 from trayio/CN-39/f/function-afterheaders

CN-39: handling threadneedle update for afterHeaders support for function ops

README.md

@@ -13,6 +13,7 @@ as on the operational level.
 * [Project structuring](#project-structuring)
   * [Schema](#schema)
   * [Model](#model)
+  	* [after_headers](#after_headers)
   * [Sample response](#sample-response)
 * [Global models](#global-models)
 * [Global message schemas](#global-message-schemas)
@@ -152,6 +153,15 @@ module.exports = {
 
 Variables passed in the input schema will be passed into a Mustache template system.
 
+#### after_headers
+For function models (only), the afterHeaders function can be specified in a `after_headers.js` file, exporting a function accepting the arguments `error`, `body`, and `params`.
+```js
+module.exports = function (error, params, body) {
+	return {
+	//Must return an object
+	};
+};
+```
 
 ### Sample response
 

exampleTwo/connectors.json

@@ -99,6 +99,19 @@
         },
         "dynamic_output": false
       },
+      {
+        "name": "test_op_function",
+        "title": "Test function operation",
+        "input_schema": {
+          "type": "object",
+          "properties": {},
+          "required": [],
+          "advanced": [],
+          "$schema": "http://json-schema.org/draft-04/schema#",
+          "additionalProperties": false
+        },
+        "dynamic_output": false
+      },
       {
         "name": "test_op",
         "title": "Test operation",

exampleTwo/connectors/testconnector/test_op_function/after_headers.js

@@ -0,0 +1,7 @@
+module.exports = function (error, params, body) {
+	return (
+		params.ahFlag ?
+		when.resolve({ gotHeader: true, modelError: !!error }) :
+		when.reject(new Error('Failed afterHeader'))
+	);
+};

exampleTwo/connectors/testconnector/test_op_function/model.js

@@ -0,0 +1,7 @@
+module.exports = function (params) {
+	return (
+		params.modelFlag ?
+		when.resolve(params) :
+		when.reject(new Error('Failed function'))
+	);
+};

exampleTwo/connectors/testconnector/test_op_function/schema.js

@@ -0,0 +1,12 @@
+module.exports = {
+
+	title: 'Test function operation',
+
+	globals: false,
+
+	input: {
+
+	},
+
+
+};

exampleTwo/docs.html

@@ -22,6 +22,15 @@ <h2>Configuration</h2>
 			</td>
 		</tr>
 
+		<tr>
+			<td>
+				<em>Test function operation</em>
+			</td>
+			<td>
+
+			</td>
+		</tr>
+
 		<tr>
 			<td>
 				<em>Test operation</em>

lib/bindConnectors/bindMessage.js

@@ -26,7 +26,7 @@ module.exports = function (message, threadneedle, connectorConfig) {
 	message = interceptAfterSuccess(message);
 
 	// Add the threadneedle method
-	threadneedle.addMethod(methodName, message.model);
+	threadneedle.addMethod(methodName, message.model, message.afterHeaders);
 
 	/*	Add the threadneedle method to the falafel global -
 		The falafel version of the operation should expose only

lib/getConnectors/index.js

@@ -117,10 +117,16 @@ module.exports = function (directory) {
 						output.schema.responseSample = requireFile(messageDir + '/response.sample.json', '/response.sample.json');
 				}
 
+				//Check if output.js is available
 				if (files.indexOf('output.js') !== -1) {
 					output['dynamicOutput'] = requireFile(messageDir + '/output', 'output.js');
 				}
 
+				//Check if afterHeaders.js is available
+				if (files.indexOf('after_headers.js') !== -1) {
+					output['afterHeaders'] = requireFile(messageDir + '/after_headers', 'after_headers.js');
+				}
+
 				// Set the name on the top level
 				output.name = _.get(output, 'schema.name', message);