Blame view

node_modules/http-proxy-middleware/lib/path-rewriter.js 1.77 KB
ce4c83ff   wxy   初始提交
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
  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
  }