1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 | 1× 1× 1× 1× 1× 1× 1× 9× 1× 9× 9× 11× 128× 128× 27× 27× 24× 16× 16× 16× 16× 12× 16× 16× 16× 109× 109× 109× 109× 4× 4× 4× 105× 105× 49× 48× 48× 48× 36× 36× 48× 48× 48× 48× 48× 48× 48× 48× 48× 1× 105× 105× 16× 105× 9× 7× 1× 48× 4× 4× 1× 16× 16× 16× 16× 16× 16× 10× 3× 3× 3× 3× 7× 10× 16× 13× 13× 13× 1× 12× 12× 12× 12× 12× 12× 12× 12× 12× 12× 1× 13× | 'use strict'; module.exports = init; var debug = require('debug')('swagger'); var debugContent = require('debug')('swagger:content'); var _ = require('lodash'); var util = require('util'); var EventEmitter = require('events').EventEmitter; function init(runner) { return new Middleware(runner); } function Middleware(runner) { this.runner = runner; this.middleware = function middleware() { return function middleware(req, res, next) { // flow back to connect pipe var operation = runner.getOperation(req); if (!operation) { var path = runner.getPath(req); if (!path) { return next(); } if (!path['x-swagger-pipe']) { var msg = util.format('Path [%s] defined in Swagger, but %s operation is not.', path.path, req.method); var err = new Error(msg); err.statusCode = 405; var allowedMethods = _.map(path.operationObjects, function(operation) { return operation.method.toUpperCase(); }); err.allowedMethods = allowedMethods; res.setHeader('Allow', allowedMethods.sort().join(', ')); return next(err); } } runner.applyMetadata(req, operation, function(err) { Iif (err) { /* istanbul ignore next */ return next(err); } var pipe = runner.getPipe(req); if (!pipe) { var err = new Error('No implementation found for this path.'); err.statusCode = 405; return next(err); } var context = { // system values _errorHandler: runner.defaultErrorHandler(), request: req, response: res, // user-modifiable values input: undefined, statusCode: undefined, headers: {}, output: undefined }; context._finish = function finishConnect(ignore1, ignore2) { // must have arity of 2 if (context.error) { return next(context.error); } try { var response = context.response; if (context.statusCode) { debug('setting response statusCode: %d', context.statusCode); response.statusCode = context.statusCode; } Eif (context.headers) { debugContent('setting response headers: %j', context.headers); _.each(context.headers, function(value, name) { response.setHeader(name, value); }); } Eif (context.output) { var body = translate(context.output, response.getHeader('content-type')); debugContent('sending response body: %s', body); response.end(body); } next(); } catch (err) { /* istanbul ignore next */ next(err); } }; /* istanbul ignore next */ var listenerCount = (runner.listenerCount) ? runner.listenerCount('responseValidationError') : // Node >= 4.0 EventEmitter.listenerCount(runner, 'responseValidationError'); // Node < 4.0 if (listenerCount) { hookResponseForValidation(context, runner); } runner.bagpipes.play(pipe, context); }); }; }; this.register = function register(app) { app.use(this.middleware()); }; } function translate(output, mimeType) { if (typeof output !== 'object') { return output; } switch(true) { case /json/.test(mimeType): return JSON.stringify(output); default: return util.inspect(output) } } function hookResponseForValidation(context, eventEmitter) { debug('add response validation hook'); var res = context.response; var end = res.end; var write = res.write; var written; res.write = function hookWrite(chunk, encoding, callback) { if (written) { written = ''; res.write = write; res.end = end; debug('multiple writes, will not validate response'); } else { written = chunk; } write.apply(res, arguments); }; res.end = function hookEnd(data, encoding, callback) { res.write = write; res.end = end; if (written && data) { debug('multiple writes, will not validate response'); } else { debug('validating response'); try { var headers = res._headers || res.headers || {}; var body = data || written; debugContent('response body type: %s value: %s', typeof body, body); var validateResult = context.request.swagger.operation.validateResponse(res.statusCode, headers, body); debug('validation result:', validateResult); Eif (validateResult.errors.length || validateResult.warnings.length) { debug('emitting responseValidationError'); eventEmitter.emit('responseValidationError', validateResult, context.request, res); } } catch (err) { /* istanbul ignore next */ console.error(err.stack); } } end.apply(res, arguments); }; } |