150 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
		
		
			
		
	
	
			150 lines
		
	
	
		
			2.9 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
|  | /*! | ||
|  |  * vary | ||
|  |  * Copyright(c) 2014-2017 Douglas Christopher Wilson | ||
|  |  * MIT Licensed | ||
|  |  */ | ||
|  | 
 | ||
|  | 'use strict' | ||
|  | 
 | ||
|  | /** | ||
|  |  * Module exports. | ||
|  |  */ | ||
|  | 
 | ||
|  | module.exports = vary | ||
|  | module.exports.append = append | ||
|  | 
 | ||
|  | /** | ||
|  |  * RegExp to match field-name in RFC 7230 sec 3.2 | ||
|  |  * | ||
|  |  * field-name    = token | ||
|  |  * token         = 1*tchar | ||
|  |  * tchar         = "!" / "#" / "$" / "%" / "&" / "'" / "*" | ||
|  |  *               / "+" / "-" / "." / "^" / "_" / "`" / "|" / "~" | ||
|  |  *               / DIGIT / ALPHA | ||
|  |  *               ; any VCHAR, except delimiters | ||
|  |  */ | ||
|  | 
 | ||
|  | var FIELD_NAME_REGEXP = /^[!#$%&'*+\-.^_`|~0-9A-Za-z]+$/ | ||
|  | 
 | ||
|  | /** | ||
|  |  * Append a field to a vary header. | ||
|  |  * | ||
|  |  * @param {String} header | ||
|  |  * @param {String|Array} field | ||
|  |  * @return {String} | ||
|  |  * @public | ||
|  |  */ | ||
|  | 
 | ||
|  | function append (header, field) { | ||
|  |   if (typeof header !== 'string') { | ||
|  |     throw new TypeError('header argument is required') | ||
|  |   } | ||
|  | 
 | ||
|  |   if (!field) { | ||
|  |     throw new TypeError('field argument is required') | ||
|  |   } | ||
|  | 
 | ||
|  |   // get fields array
 | ||
|  |   var fields = !Array.isArray(field) | ||
|  |     ? parse(String(field)) | ||
|  |     : field | ||
|  | 
 | ||
|  |   // assert on invalid field names
 | ||
|  |   for (var j = 0; j < fields.length; j++) { | ||
|  |     if (!FIELD_NAME_REGEXP.test(fields[j])) { | ||
|  |       throw new TypeError('field argument contains an invalid header name') | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  |   // existing, unspecified vary
 | ||
|  |   if (header === '*') { | ||
|  |     return header | ||
|  |   } | ||
|  | 
 | ||
|  |   // enumerate current values
 | ||
|  |   var val = header | ||
|  |   var vals = parse(header.toLowerCase()) | ||
|  | 
 | ||
|  |   // unspecified vary
 | ||
|  |   if (fields.indexOf('*') !== -1 || vals.indexOf('*') !== -1) { | ||
|  |     return '*' | ||
|  |   } | ||
|  | 
 | ||
|  |   for (var i = 0; i < fields.length; i++) { | ||
|  |     var fld = fields[i].toLowerCase() | ||
|  | 
 | ||
|  |     // append value (case-preserving)
 | ||
|  |     if (vals.indexOf(fld) === -1) { | ||
|  |       vals.push(fld) | ||
|  |       val = val | ||
|  |         ? val + ', ' + fields[i] | ||
|  |         : fields[i] | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  |   return val | ||
|  | } | ||
|  | 
 | ||
|  | /** | ||
|  |  * Parse a vary header into an array. | ||
|  |  * | ||
|  |  * @param {String} header | ||
|  |  * @return {Array} | ||
|  |  * @private | ||
|  |  */ | ||
|  | 
 | ||
|  | function parse (header) { | ||
|  |   var end = 0 | ||
|  |   var list = [] | ||
|  |   var start = 0 | ||
|  | 
 | ||
|  |   // gather tokens
 | ||
|  |   for (var i = 0, len = header.length; i < len; i++) { | ||
|  |     switch (header.charCodeAt(i)) { | ||
|  |       case 0x20: /*   */ | ||
|  |         if (start === end) { | ||
|  |           start = end = i + 1 | ||
|  |         } | ||
|  |         break | ||
|  |       case 0x2c: /* , */ | ||
|  |         list.push(header.substring(start, end)) | ||
|  |         start = end = i + 1 | ||
|  |         break | ||
|  |       default: | ||
|  |         end = i + 1 | ||
|  |         break | ||
|  |     } | ||
|  |   } | ||
|  | 
 | ||
|  |   // final token
 | ||
|  |   list.push(header.substring(start, end)) | ||
|  | 
 | ||
|  |   return list | ||
|  | } | ||
|  | 
 | ||
|  | /** | ||
|  |  * Mark that a request is varied on a header field. | ||
|  |  * | ||
|  |  * @param {Object} res | ||
|  |  * @param {String|Array} field | ||
|  |  * @public | ||
|  |  */ | ||
|  | 
 | ||
|  | function vary (res, field) { | ||
|  |   if (!res || !res.getHeader || !res.setHeader) { | ||
|  |     // quack quack
 | ||
|  |     throw new TypeError('res argument is required') | ||
|  |   } | ||
|  | 
 | ||
|  |   // get existing header
 | ||
|  |   var val = res.getHeader('Vary') || '' | ||
|  |   var header = Array.isArray(val) | ||
|  |     ? val.join(', ') | ||
|  |     : String(val) | ||
|  | 
 | ||
|  |   // set new header
 | ||
|  |   if ((val = append(header, field))) { | ||
|  |     res.setHeader('Vary', val) | ||
|  |   } | ||
|  | } |