index.js 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285
  1. /*!
  2. * Morgan | Connect - logger
  3. * Copyright(c) 2010 Sencha Inc.
  4. * Copyright(c) 2011 TJ Holowaychuk
  5. * MIT Licensed
  6. */
  7. /**
  8. * Module dependencies.
  9. */
  10. var bytes = require('bytes');
  11. /*!
  12. * Default log buffer duration.
  13. */
  14. var defaultBufferDuration = 1000;
  15. /**
  16. * Log requests with the given `options` or a `format` string.
  17. *
  18. * See README.md for documentation of options and formatting.
  19. *
  20. * @param {String|Function|Object} format or options
  21. * @return {Function} middleware
  22. * @api public
  23. */
  24. exports = module.exports = function logger(options) {
  25. if (options && typeof options !== 'object') {
  26. options = { format: options };
  27. } else {
  28. options = options || {};
  29. }
  30. // output on request instead of response
  31. var immediate = options.immediate;
  32. // check if log entry should be skipped
  33. var skip = options.skip || function () { return false; };
  34. // format name
  35. var fmt = exports[options.format] || options.format || exports.default;
  36. // compile format
  37. if ('function' != typeof fmt) fmt = compile(fmt);
  38. // options
  39. var stream = options.stream || process.stdout
  40. , buffer = options.buffer;
  41. // buffering support
  42. if (buffer) {
  43. var realStream = stream
  44. var buf = []
  45. var timer = null
  46. var interval = 'number' == typeof buffer
  47. ? buffer
  48. : defaultBufferDuration
  49. // flush function
  50. var flush = function(){
  51. timer = null
  52. if (buf.length) {
  53. realStream.write(buf.join(''));
  54. buf.length = 0;
  55. }
  56. }
  57. // swap the stream
  58. stream = {
  59. write: function(str){
  60. if (timer === null) {
  61. timer = setTimeout(flush, interval)
  62. }
  63. buf.push(str);
  64. }
  65. };
  66. }
  67. return function logger(req, res, next) {
  68. req._startAt = process.hrtime();
  69. req._startTime = new Date;
  70. req._remoteAddress = req.connection && req.connection.remoteAddress;
  71. function logRequest(){
  72. res.removeListener('finish', logRequest);
  73. res.removeListener('close', logRequest);
  74. if (skip(req, res)) return;
  75. var line = fmt(exports, req, res);
  76. if (null == line) return;
  77. stream.write(line + '\n');
  78. };
  79. // immediate
  80. if (immediate) {
  81. logRequest();
  82. // proxy end to output logging
  83. } else {
  84. res.on('finish', logRequest);
  85. res.on('close', logRequest);
  86. }
  87. next();
  88. };
  89. };
  90. /**
  91. * Compile `fmt` into a function.
  92. *
  93. * @param {String} fmt
  94. * @return {Function}
  95. * @api private
  96. */
  97. function compile(fmt) {
  98. fmt = fmt.replace(/"/g, '\\"');
  99. var js = ' return "' + fmt.replace(/:([-\w]{2,})(?:\[([^\]]+)\])?/g, function(_, name, arg){
  100. return '"\n + (tokens["' + name + '"](req, res, "' + arg + '") || "-") + "';
  101. }) + '";'
  102. return new Function('tokens, req, res', js);
  103. };
  104. /**
  105. * Define a token function with the given `name`,
  106. * and callback `fn(req, res)`.
  107. *
  108. * @param {String} name
  109. * @param {Function} fn
  110. * @return {Object} exports for chaining
  111. * @api public
  112. */
  113. exports.token = function(name, fn) {
  114. exports[name] = fn;
  115. return this;
  116. };
  117. /**
  118. * Define a `fmt` with the given `name`.
  119. *
  120. * @param {String} name
  121. * @param {String|Function} fmt
  122. * @return {Object} exports for chaining
  123. * @api public
  124. */
  125. exports.format = function(name, fmt){
  126. exports[name] = fmt;
  127. return this;
  128. };
  129. /**
  130. * Default format.
  131. */
  132. exports.format('default', ':remote-addr - - [:date] ":method :url HTTP/:http-version" :status :res[content-length] ":referrer" ":user-agent"');
  133. /**
  134. * Short format.
  135. */
  136. exports.format('short', ':remote-addr - :method :url HTTP/:http-version :status :res[content-length] - :response-time ms');
  137. /**
  138. * Tiny format.
  139. */
  140. exports.format('tiny', ':method :url :status :res[content-length] - :response-time ms');
  141. /**
  142. * dev (colored)
  143. */
  144. exports.format('dev', function(tokens, req, res){
  145. var color = 32; // green
  146. var status = res.statusCode;
  147. if (status >= 500) color = 31; // red
  148. else if (status >= 400) color = 33; // yellow
  149. else if (status >= 300) color = 36; // cyan
  150. var fn = compile('\x1b[90m:method :url \x1b[' + color + 'm:status \x1b[90m:response-time ms - :res[content-length]\x1b[0m');
  151. return fn(tokens, req, res);
  152. });
  153. /**
  154. * request url
  155. */
  156. exports.token('url', function(req){
  157. return req.originalUrl || req.url;
  158. });
  159. /**
  160. * request method
  161. */
  162. exports.token('method', function(req){
  163. return req.method;
  164. });
  165. /**
  166. * response time in milliseconds
  167. */
  168. exports.token('response-time', function(req, res){
  169. if (!res._header || !req._startAt) return '';
  170. var diff = process.hrtime(req._startAt);
  171. var ms = diff[0] * 1e3 + diff[1] * 1e-6;
  172. return ms.toFixed(3);
  173. });
  174. /**
  175. * UTC date
  176. */
  177. exports.token('date', function(){
  178. return new Date().toUTCString();
  179. });
  180. /**
  181. * response status code
  182. */
  183. exports.token('status', function(req, res){
  184. return res._header ? res.statusCode : null;
  185. });
  186. /**
  187. * normalized referrer
  188. */
  189. exports.token('referrer', function(req){
  190. return req.headers['referer'] || req.headers['referrer'];
  191. });
  192. /**
  193. * remote address
  194. */
  195. exports.token('remote-addr', function(req){
  196. if (req.ip) return req.ip;
  197. if (req._remoteAddress) return req._remoteAddress;
  198. if (req.connection) return req.connection.remoteAddress;
  199. return undefined;
  200. });
  201. /**
  202. * HTTP version
  203. */
  204. exports.token('http-version', function(req){
  205. return req.httpVersionMajor + '.' + req.httpVersionMinor;
  206. });
  207. /**
  208. * UA string
  209. */
  210. exports.token('user-agent', function(req){
  211. return req.headers['user-agent'];
  212. });
  213. /**
  214. * request header
  215. */
  216. exports.token('req', function(req, res, field){
  217. return req.headers[field.toLowerCase()];
  218. });
  219. /**
  220. * response header
  221. */
  222. exports.token('res', function(req, res, field){
  223. return (res._headers || {})[field.toLowerCase()];
  224. });