liuwill/linz-log-generator

var yargsParser = require('yargs-parser')
var querystring = require('querystring')

var trimString = require('./utils').trimString

function parseCurlToRequest(curlStr) {
  var argvObj = yargsParser(curlStr)
  if (!argvObj._.length || argvObj._[0] !== 'curl') {
    throw new Error('curl')
  }

  var curlData = {
    url: argvObj._[1],
    headers: parseHeadersFromCurl(argvObj.H)
  }

  for (var key in argvObj) {
    if (key !== '_' && key !== 'H') {
      curlData[key] = argvObj[key]
    }
  }

  return curlData
}

function parseRequestParams(requestUrl) {
  var queryParamStr = requestUrl
  var paramPos = requestUrl.indexOf('?')
  if (paramPos >= 0) {
    queryParamStr = requestUrl.substr(paramPos + 1)
  }

  return querystring.parse(queryParamStr)
}

function parseRequestRoute(requestUrl) {
  var paramPos = requestUrl.indexOf('?')
  if (paramPos < 0) {
    return requestUrl
  }

  return requestUrl.substr(0, paramPos)
}

function parseHeadersFromCurl(headerStrList) {
  var headers = {}

  for (var i in headerStrList) {
    var headerStr = headerStrList[i]
    var markPos = headerStr.indexOf(':')
    var key = headerStr.substr(0, markPos).toLowerCase()

    var contentStr = headerStr.substr(markPos + 1)
    headers[key] = trimString(contentStr)
  }
  return headers
}

function generateRequest(curlStr) {
  var curlData = parseCurlToRequest(curlStr)
  var queryParam = parseRequestParams(curlData.url)

  curlData['query'] = queryParam
  curlData['path'] = parseRequestRoute(curlData.url)

  curlData['request'] = {
    query: curlData['query'],
    headers: curlData['headers']
  }
  return curlData
}

exports.parseHeadersFromCurl = parseHeadersFromCurl
exports.generateRequest = generateRequest
exports.parseRequestParams = parseRequestParams
exports.parseCurlToRequest = parseCurlToRequest
exports.parseRequestRoute = parseRequestRoute