har-to-k6/src/render/imports.js

Convert LI-HAR to k6 script

function imports(spec) {
  if (any(spec)) {
    const lines = []
    k6(spec, lines)
    http(spec, lines)
    k6JsLibs(spec, lines)
    return lines.join(`\n`)
  } else {
    return null
  }
}

function any(spec) {
  return Object.values(spec).find((value) => value)
}

function k6(spec, lines) {
  const items = []

  if (spec.check || spec.group || spec.sleep) {
    if (spec.sleep) {
      items.push('sleep')
    }
    if (spec.check) {
      items.push('check')
    }
    if (spec.group) {
      items.push('group')
    }
  }

  const content = items.join(`, `)
  if (items.length > 0) {
    lines.push(`import { ${content} } from "k6";`)
  }
}

function http(spec, lines) {
  if (spec.http) {
    lines.push(`import http from "k6/http";`)
  }
}

const K6_JS_LIBS = (() => {
  const BASE_URL = 'https://jslib.k6.io'
  return {
    jsonpath: `${BASE_URL}/jsonpath/1.0.2/index.js`,
    url: `${BASE_URL}/url/1.0.0/index.js`,
    formData: `${BASE_URL}/formdata/0.0.2/index.js`,
  }
})()

function k6JsLibs(spec, lines) {
  if (
    spec.URL ||
    spec.URLSearchParams ||
    spec.jsonpath ||
    spec.MimeBuilder ||
    spec.formData
  ) {
    if (lines.length > 0) {
      lines.push('\n')
    }

    if (spec.URLSearchParams || spec.URL) {
      const modules = [
        spec.URL && 'URL',
        spec.URLSearchParams && 'URLSearchParams',
      ]
        .filter(Boolean)
        .join(', ')

      lines.push(`import { ${modules} } from "${K6_JS_LIBS.url}"`)
    }

    if (spec.jsonpath) {
      lines.push(`import jsonpath from "${K6_JS_LIBS.jsonpath}"`)
    }

    if (spec.formData) {
      lines.push(`import { FormData } from "${K6_JS_LIBS.formData}"`)
    }
  }
}

module.exports = imports