first commit

liusuyi

2023-04-24 4737f1e038743ced243c9e52423404d9034d6107

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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
var _ = require('lodash')
var logger = require('./logger').getInstance()
var ERRORS = require('./errors')
 
module.exports = {
  create: createPathRewriter
}
 
/**
 * Create rewrite function, to cache parsed rewrite rules.
 *
 * @param {Object} rewriteConfig
 * @return {Function} Function to rewrite paths; This function should accept `path` (request.url) as parameter
 */
function createPathRewriter(rewriteConfig) {
  var rulesCache
 
  if (!isValidRewriteConfig(rewriteConfig)) {
    return
  }
 
  if (_.isFunction(rewriteConfig)) {
    var customRewriteFn = rewriteConfig
    return customRewriteFn
  } else {
    rulesCache = parsePathRewriteRules(rewriteConfig)
    return rewritePath
  }
 
  function rewritePath(path) {
    var result = path
 
    _.forEach(rulesCache, function(rule) {
      if (rule.regex.test(path)) {
        result = result.replace(rule.regex, rule.value)
        logger.debug('[HPM] Rewriting path from "%s" to "%s"', path, result)
        return false
      }
    })
 
    return result
  }
}
 
function isValidRewriteConfig(rewriteConfig) {
  if (_.isFunction(rewriteConfig)) {
    return true
  } else if (!_.isEmpty(rewriteConfig) && _.isPlainObject(rewriteConfig)) {
    return true
  } else if (
    _.isUndefined(rewriteConfig) ||
    _.isNull(rewriteConfig) ||
    _.isEqual(rewriteConfig, {})
  ) {
    return false
  } else {
    throw new Error(ERRORS.ERR_PATH_REWRITER_CONFIG)
  }
}
 
function parsePathRewriteRules(rewriteConfig) {
  var rules = []
 
  if (_.isPlainObject(rewriteConfig)) {
    _.forIn(rewriteConfig, function(value, key) {
      rules.push({
        regex: new RegExp(key),
        value: rewriteConfig[key]
      })
      logger.info(
        '[HPM] Proxy rewrite rule created: "%s" ~> "%s"',
        key,
        rewriteConfig[key]
      )
    })
  }
 
  return rules
}