You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
117 lines
2.8 KiB
JavaScript
117 lines
2.8 KiB
JavaScript
'use strict'
|
|
var utils = require('./utils')
|
|
|
|
var assign = utils.assign
|
|
var isFunction = utils.isFunction
|
|
var isRegExp = utils.isRegExp
|
|
var isString = utils.isString
|
|
var noop = utils.noop
|
|
|
|
function RouteNode (options) {
|
|
options = options || {}
|
|
this._children = Object.create(null)
|
|
this._regExs = Object.create(null)
|
|
this._funcs = Object.create(null)
|
|
}
|
|
|
|
assign(RouteNode.prototype, {
|
|
handler: null,
|
|
|
|
add: function add(parts, handler, context) {
|
|
parts = parts || []
|
|
handler = handler || noop
|
|
context = context || null
|
|
|
|
if (parts.length === 0) {
|
|
this.handler = handler
|
|
this.context = context
|
|
return this
|
|
}
|
|
var part = parts.shift()
|
|
var node = null
|
|
|
|
if (isString(part)) {
|
|
node = this._children[part]
|
|
} else if (isRegExp(part)) {
|
|
this._regExs[part] = part
|
|
} else if (isFunction(part)) {
|
|
this._funcs[part] = part
|
|
}
|
|
|
|
if (node == null) {
|
|
node = new RouteNode()
|
|
this._children[part] = node
|
|
}
|
|
return node.add(parts, handler, context)
|
|
},
|
|
|
|
get: function get(parts, args, done) {
|
|
parts = parts || []
|
|
args = args || []
|
|
|
|
if (parts.length === 0) {
|
|
return done(null, this.handler, this.context, args)
|
|
}
|
|
var part = parts.shift()
|
|
var childNode = this._children[part]
|
|
if (childNode) {
|
|
return childNode.get(parts, args, done)
|
|
} else {
|
|
var results = this._getRegexChild(part)
|
|
if (results && results.node) {
|
|
return results.node.get(parts, args.concat(results.args), done)
|
|
} else {
|
|
results = this._getFunctionChild(part)
|
|
if (results && results.node) {
|
|
return results.node.get(parts, args.concat(results.args), done)
|
|
} else {
|
|
return done(new Error('not found'))
|
|
}
|
|
}
|
|
}
|
|
},
|
|
|
|
_getRegexChild: function _getRegexChild(part) {
|
|
var self = this
|
|
|
|
return Object.keys(this._regExs).reduce(function checkRegex(memo, key) {
|
|
if (memo) {
|
|
return memo
|
|
}
|
|
var regex = self._regExs[key]
|
|
var results = regex.exec(part)
|
|
if (!results) {
|
|
return null
|
|
}
|
|
|
|
var lastIndex = null
|
|
var args = []
|
|
while (results && results.index !== lastIndex) {
|
|
args = args.concat(results[0])
|
|
lastIndex = results.index
|
|
results = regex.exec(part)
|
|
}
|
|
return {args: args, key: key, node: self._children[key]}
|
|
}, null)
|
|
},
|
|
|
|
_getFunctionChild: function _getFunctionChild(part) {
|
|
var self = this
|
|
return Object.keys(self._funcs).reduce(function checkFunc(memo, key) {
|
|
if (memo) {
|
|
return memo
|
|
}
|
|
var func = self._funcs[key]
|
|
var result = func(part)
|
|
if (result) {
|
|
return {args: [result], key: key, node: self._children[key]}
|
|
} else {
|
|
return null
|
|
}
|
|
}, null)
|
|
}
|
|
|
|
})
|
|
|
|
module.exports = RouteNode
|