You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					206 lines
				
				4.7 KiB
			
		
		
			
		
	
	
					206 lines
				
				4.7 KiB
			| 
								 
											3 years ago
										 
									 | 
							
								'use strict'
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								var fs = require('fs')
							 | 
						||
| 
								 | 
							
								var qs = require('querystring')
							 | 
						||
| 
								 | 
							
								var validate = require('har-validator')
							 | 
						||
| 
								 | 
							
								var extend = require('extend')
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								function Har (request) {
							 | 
						||
| 
								 | 
							
								  this.request = request
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Har.prototype.reducer = function (obj, pair) {
							 | 
						||
| 
								 | 
							
								  // new property ?
							 | 
						||
| 
								 | 
							
								  if (obj[pair.name] === undefined) {
							 | 
						||
| 
								 | 
							
								    obj[pair.name] = pair.value
							 | 
						||
| 
								 | 
							
								    return obj
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // existing? convert to array
							 | 
						||
| 
								 | 
							
								  var arr = [
							 | 
						||
| 
								 | 
							
								    obj[pair.name],
							 | 
						||
| 
								 | 
							
								    pair.value
							 | 
						||
| 
								 | 
							
								  ]
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  obj[pair.name] = arr
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return obj
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Har.prototype.prep = function (data) {
							 | 
						||
| 
								 | 
							
								  // construct utility properties
							 | 
						||
| 
								 | 
							
								  data.queryObj = {}
							 | 
						||
| 
								 | 
							
								  data.headersObj = {}
							 | 
						||
| 
								 | 
							
								  data.postData.jsonObj = false
							 | 
						||
| 
								 | 
							
								  data.postData.paramsObj = false
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // construct query objects
							 | 
						||
| 
								 | 
							
								  if (data.queryString && data.queryString.length) {
							 | 
						||
| 
								 | 
							
								    data.queryObj = data.queryString.reduce(this.reducer, {})
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // construct headers objects
							 | 
						||
| 
								 | 
							
								  if (data.headers && data.headers.length) {
							 | 
						||
| 
								 | 
							
								    // loweCase header keys
							 | 
						||
| 
								 | 
							
								    data.headersObj = data.headers.reduceRight(function (headers, header) {
							 | 
						||
| 
								 | 
							
								      headers[header.name] = header.value
							 | 
						||
| 
								 | 
							
								      return headers
							 | 
						||
| 
								 | 
							
								    }, {})
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // construct Cookie header
							 | 
						||
| 
								 | 
							
								  if (data.cookies && data.cookies.length) {
							 | 
						||
| 
								 | 
							
								    var cookies = data.cookies.map(function (cookie) {
							 | 
						||
| 
								 | 
							
								      return cookie.name + '=' + cookie.value
							 | 
						||
| 
								 | 
							
								    })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    if (cookies.length) {
							 | 
						||
| 
								 | 
							
								      data.headersObj.cookie = cookies.join('; ')
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // prep body
							 | 
						||
| 
								 | 
							
								  function some (arr) {
							 | 
						||
| 
								 | 
							
								    return arr.some(function (type) {
							 | 
						||
| 
								 | 
							
								      return data.postData.mimeType.indexOf(type) === 0
							 | 
						||
| 
								 | 
							
								    })
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (some([
							 | 
						||
| 
								 | 
							
								    'multipart/mixed',
							 | 
						||
| 
								 | 
							
								    'multipart/related',
							 | 
						||
| 
								 | 
							
								    'multipart/form-data',
							 | 
						||
| 
								 | 
							
								    'multipart/alternative'])) {
							 | 
						||
| 
								 | 
							
								    // reset values
							 | 
						||
| 
								 | 
							
								    data.postData.mimeType = 'multipart/form-data'
							 | 
						||
| 
								 | 
							
								  } else if (some([
							 | 
						||
| 
								 | 
							
								    'application/x-www-form-urlencoded'])) {
							 | 
						||
| 
								 | 
							
								    if (!data.postData.params) {
							 | 
						||
| 
								 | 
							
								      data.postData.text = ''
							 | 
						||
| 
								 | 
							
								    } else {
							 | 
						||
| 
								 | 
							
								      data.postData.paramsObj = data.postData.params.reduce(this.reducer, {})
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      // always overwrite
							 | 
						||
| 
								 | 
							
								      data.postData.text = qs.stringify(data.postData.paramsObj)
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  } else if (some([
							 | 
						||
| 
								 | 
							
								    'text/json',
							 | 
						||
| 
								 | 
							
								    'text/x-json',
							 | 
						||
| 
								 | 
							
								    'application/json',
							 | 
						||
| 
								 | 
							
								    'application/x-json'])) {
							 | 
						||
| 
								 | 
							
								    data.postData.mimeType = 'application/json'
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    if (data.postData.text) {
							 | 
						||
| 
								 | 
							
								      try {
							 | 
						||
| 
								 | 
							
								        data.postData.jsonObj = JSON.parse(data.postData.text)
							 | 
						||
| 
								 | 
							
								      } catch (e) {
							 | 
						||
| 
								 | 
							
								        this.request.debug(e)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								        // force back to text/plain
							 | 
						||
| 
								 | 
							
								        data.postData.mimeType = 'text/plain'
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return data
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								Har.prototype.options = function (options) {
							 | 
						||
| 
								 | 
							
								  // skip if no har property defined
							 | 
						||
| 
								 | 
							
								  if (!options.har) {
							 | 
						||
| 
								 | 
							
								    return options
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  var har = {}
							 | 
						||
| 
								 | 
							
								  extend(har, options.har)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // only process the first entry
							 | 
						||
| 
								 | 
							
								  if (har.log && har.log.entries) {
							 | 
						||
| 
								 | 
							
								    har = har.log.entries[0]
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // add optional properties to make validation successful
							 | 
						||
| 
								 | 
							
								  har.url = har.url || options.url || options.uri || options.baseUrl || '/'
							 | 
						||
| 
								 | 
							
								  har.httpVersion = har.httpVersion || 'HTTP/1.1'
							 | 
						||
| 
								 | 
							
								  har.queryString = har.queryString || []
							 | 
						||
| 
								 | 
							
								  har.headers = har.headers || []
							 | 
						||
| 
								 | 
							
								  har.cookies = har.cookies || []
							 | 
						||
| 
								 | 
							
								  har.postData = har.postData || {}
							 | 
						||
| 
								 | 
							
								  har.postData.mimeType = har.postData.mimeType || 'application/octet-stream'
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  har.bodySize = 0
							 | 
						||
| 
								 | 
							
								  har.headersSize = 0
							 | 
						||
| 
								 | 
							
								  har.postData.size = 0
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (!validate.request(har)) {
							 | 
						||
| 
								 | 
							
								    return options
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // clean up and get some utility properties
							 | 
						||
| 
								 | 
							
								  var req = this.prep(har)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  // construct new options
							 | 
						||
| 
								 | 
							
								  if (req.url) {
							 | 
						||
| 
								 | 
							
								    options.url = req.url
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (req.method) {
							 | 
						||
| 
								 | 
							
								    options.method = req.method
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (Object.keys(req.queryObj).length) {
							 | 
						||
| 
								 | 
							
								    options.qs = req.queryObj
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  if (Object.keys(req.headersObj).length) {
							 | 
						||
| 
								 | 
							
								    options.headers = req.headersObj
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  function test (type) {
							 | 
						||
| 
								 | 
							
								    return req.postData.mimeType.indexOf(type) === 0
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								  if (test('application/x-www-form-urlencoded')) {
							 | 
						||
| 
								 | 
							
								    options.form = req.postData.paramsObj
							 | 
						||
| 
								 | 
							
								  } else if (test('application/json')) {
							 | 
						||
| 
								 | 
							
								    if (req.postData.jsonObj) {
							 | 
						||
| 
								 | 
							
								      options.body = req.postData.jsonObj
							 | 
						||
| 
								 | 
							
								      options.json = true
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  } else if (test('multipart/form-data')) {
							 | 
						||
| 
								 | 
							
								    options.formData = {}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    req.postData.params.forEach(function (param) {
							 | 
						||
| 
								 | 
							
								      var attachment = {}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      if (!param.fileName && !param.contentType) {
							 | 
						||
| 
								 | 
							
								        options.formData[param.name] = param.value
							 | 
						||
| 
								 | 
							
								        return
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      // attempt to read from disk!
							 | 
						||
| 
								 | 
							
								      if (param.fileName && !param.value) {
							 | 
						||
| 
								 | 
							
								        attachment.value = fs.createReadStream(param.fileName)
							 | 
						||
| 
								 | 
							
								      } else if (param.value) {
							 | 
						||
| 
								 | 
							
								        attachment.value = param.value
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      if (param.fileName) {
							 | 
						||
| 
								 | 
							
								        attachment.options = {
							 | 
						||
| 
								 | 
							
								          filename: param.fileName,
							 | 
						||
| 
								 | 
							
								          contentType: param.contentType ? param.contentType : null
							 | 
						||
| 
								 | 
							
								        }
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								      options.formData[param.name] = attachment
							 | 
						||
| 
								 | 
							
								    })
							 | 
						||
| 
								 | 
							
								  } else {
							 | 
						||
| 
								 | 
							
								    if (req.postData.text) {
							 | 
						||
| 
								 | 
							
								      options.body = req.postData.text
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  return options
							 | 
						||
| 
								 | 
							
								}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								exports.Har = Har
							 |