har-to-k6/src/parse/request.js

Convert LI-HAR to k6 script

const headers = require('./headers')
const postData = require('./postData')
const queryString = require('./queryString')
const state = require('./state/request')
const { emptyObject, getContentTypeValue } = require('../aid')

const EMPTY_BODY_METHODS = ['HEAD', 'GET', 'OPTIONS']

function request(node, spec) {
  spec.method = node.method.toUpperCase()
  spec.address = node.url

  if (node.comment) {
    spec.comment = node.comment
  }

  if (node.queryString) {
    queryString(node.queryString, node.url, spec.query)
  }

  if (node.headers) {
    headers(node.headers, spec.headers)
  }

  if (
    !EMPTY_BODY_METHODS.includes(node.method) &&
    node.postData &&
    !emptyObject(node.postData)
  ) {
    postData(node.postData, spec.post)
    contentType(node.postData.mimeType, spec.headers)
  }

  state(spec)

  if (spec.state.post.boundary) {
    addBoundary(spec.state.post.boundary, spec.headers)
  }
}

// Fallback to content type from postData
// Preserves explicit header which potentially has more information
function contentType(mimeType, headers) {
  if (!headers.has('Content-Type') && !headers.has('content-type')) {
    const item = { value: mimeType }
    const items = new Set([item])
    headers.set('content-type', items)
  }
}

function addBoundary(boundary, headers) {
  const contentType = headers.get('Content-Type') || headers.get('content-type')

  if (contentType) {
    const items = [...contentType.values()]
    const newItems = items.map(item => {
      const value = getContentTypeValue(item.value)
      if (value === 'multipart/form-data') {
        return { value: `${value}; boundary=${boundary}` }
      }

      return item
    })

    headers.delete('Content-Type') // Remove uppercased content-type in case it exists
    headers.set('content-type', new Set(newItems))
  }
}

module.exports = request