query.js 2.0 KB
Newer Older
fxy060608's avatar
fxy060608 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
const encodeReserveRE = /[!'()*]/g
const encodeReserveReplacer = c => '%' + c.charCodeAt(0).toString(16)
const commaRE = /%2C/g

// fixed encodeURIComponent which is more conformant to RFC3986:
// - escapes [!'()*]
// - preserve commas
const encode = str => encodeURIComponent(str)
  .replace(encodeReserveRE, encodeReserveReplacer)
  .replace(commaRE, ',')

const decode = decodeURIComponent

export function parseQuery (query) {
  const res = {}

  query = query.trim().replace(/^(\?|#|&)/, '')

  if (!query) {
    return res
  }

  query.split('&').forEach(param => {
    const parts = param.replace(/\+/g, ' ').split('=')
    const key = decode(parts.shift())
    const val = parts.length > 0
      ? decode(parts.join('='))
      : null

    if (res[key] === undefined) {
      res[key] = val
    } else if (Array.isArray(res[key])) {
      res[key].push(val)
    } else {
      res[key] = [res[key], val]
    }
  })

  return res
}

fxy060608's avatar
fxy060608 已提交
42
export function stringifyQuery (obj, encodeStr = encode) {
fxy060608's avatar
fxy060608 已提交
43 44 45 46 47 48 49 50
  const res = obj ? Object.keys(obj).map(key => {
    const val = obj[key]

    if (val === undefined) {
      return ''
    }

    if (val === null) {
fxy060608's avatar
fxy060608 已提交
51
      return encodeStr(key)
fxy060608's avatar
fxy060608 已提交
52 53 54 55 56 57 58 59 60
    }

    if (Array.isArray(val)) {
      const result = []
      val.forEach(val2 => {
        if (val2 === undefined) {
          return
        }
        if (val2 === null) {
fxy060608's avatar
fxy060608 已提交
61
          result.push(encodeStr(key))
fxy060608's avatar
fxy060608 已提交
62
        } else {
fxy060608's avatar
fxy060608 已提交
63
          result.push(encodeStr(key) + '=' + encodeStr(val2))
fxy060608's avatar
fxy060608 已提交
64 65 66 67 68
        }
      })
      return result.join('&')
    }

fxy060608's avatar
fxy060608 已提交
69
    return encodeStr(key) + '=' + encodeStr(val)
fxy060608's avatar
fxy060608 已提交
70 71
  }).filter(x => x.length > 0).join('&') : null
  return res ? `?${res}` : ''
fxy060608's avatar
fxy060608 已提交
72 73 74 75 76 77 78 79 80 81 82 83
}

export function decodedQuery (query = {}) {
  const decodedQuery = {}
  Object.keys(query).forEach(name => {
    try {
      decodedQuery[name] = decode(query[name])
    } catch (e) {
      decodedQuery[name] = query[name]
    }
  })
  return decodedQuery
fxy060608's avatar
fxy060608 已提交
84
}