读Zepto源码之Selector模块

Selector 模块是对 Zepto 选择器的扩展，使得 Zepto 选择器也可以支持部分 CSS3 选择器和 eq 等 Zepto 定义的选择器。

在阅读本篇文章之前，最好先阅读《读Zepto源码之神奇的$》。

读 Zepto 源码系列文章已经放到了github上，欢迎star: reading-zepto

源码版本

本文阅读的源码为 zepto1.2.0

GitBook

《reading-zepto》

辅助方法

visible

function visible(elem){
  elem = $(elem)
  return !!(elem.width() || elem.height()) && elem.css("display") !== "none"
}

判断元素是否可见。

可见的标准是元素有宽或者高，并且 display 值不为 none。

filters

var filters = $.expr[':'] = {
  visible:  function(){ if (visible(this)) return this },
  hidden:   function(){ if (!visible(this)) return this },
  selected: function(){ if (this.selected) return this },
  checked:  function(){ if (this.checked) return this },
  parent:   function(){ return this.parentNode },
  first:    function(idx){ if (idx === 0) return this },
  last:     function(idx, nodes){ if (idx === nodes.length - 1) return this },
  eq:       function(idx, _, value){ if (idx === value) return this },
  contains: function(idx, _, text){ if ($(this).text().indexOf(text) > -1) return this },
  has:      function(idx, _, sel){ if (zepto.qsa(this, sel).length) return this }
}

定义了一系列的过滤函数，返回符合条件的元素。这些过滤函数会将集合中符合条件的元素过滤出来，是实现相关选择器的核心。

visible: 过滤可见元素，匹配 el:visible 选择器
hidden: 过滤不可见元素，匹配 el:hidden 选择器
selected: 过滤选中的元素，匹配 el:selected 选择器
checked: 过滤勾选中的元素，匹配 el:checked 选择器
parent: 返回至少包含一个子元素的元素，匹配 el:parent 选择器
first: 返回第一个元素，匹配 el:first 选择器
last: 返回最后一个元素，匹配 el:last 选择器
eq: 返回指定索引的元素，匹配 el:eq(index) 选择器
contains: 返回包含指定文本的元素，匹配 el:contains(text)
has: 返回匹配指定选择器的元素，匹配 el:has(sel)

process

var filterRe = new RegExp('(.*):(\\w+)(?:\\(([^)]+)\\))?$\\s*'),
function process(sel, fn) {
  sel = sel.replace(/=#\]/g, '="#"]')
  var filter, arg, match = filterRe.exec(sel)
  if (match && match[2] in filters) {
    filter = filters[match[2]], arg = match[3]
    sel = match[1]
    if (arg) {
      var num = Number(arg)
      if (isNaN(num)) arg = arg.replace(/^["']|["']$/g, '')
      else arg = num
    }
  }
  return fn(sel, filter, arg)
}

process 方法是根据参数 sel，分解出选择器、伪类名和伪类参数（如 eq 、 has 的参数），根据伪类来选择对应的 filter ，传递给回调函数 fn 。

分解参数最主要靠的是 filterRe 这条正则，正则太过复杂，很难解释，不过用正则可视化网站 regexper.com，可以很清晰地看到，正则分成三大组，第一组匹配的是 : 前面的选择器，第二组匹配的是伪类名，第三组匹配的是伪类参数。

sel = sel.replace(/=#\]/g, '="#"]')

这段是处理 a[href^=#] 的情况，其实就是将 # 包在 "" 里面，以符合标准的属性选择器，这是 Zepto 的容错能力。这个选择器不会匹配到 filters 上的过滤函数，最后调用的是 querySelectorAll 方法，具体见《读Zepto源码之神奇的$》对 qsa 函数的分析。

if (match && match[2] in filters) {
  filter = filters[match[2]], arg = match[3]
  sel = match[1]
  ...
}

match[2] 也即第二组匹配的是伪类名，也是对应 filters 中的 key 值，伪类名存在于 filters 中时，则将选择器，伪类名和伪类参数存入对应的变量。

if (arg) {
  var num = Number(arg)
  if (isNaN(num)) arg = arg.replace(/^["']|["']$/g, '')
  else arg = num
}

如果伪类的参数不可以用 Number 转换，则参数为字符串，用正则将字符串前后的 " 或 ' 去掉，再赋值给 arg.

return fn(sel, filter, arg)

最后执行回调，将解释出来的参数传入回调函数中，将执行结果返回。

重写的方法

qsa

var zepto = $.zepto, oldQsa = zepto.qsa, oldMatches = zepto.matches，
    childRe  = /^\s*>/,
    classTag = 'Zepto' + (+new Date())

zepto.qsa = function(node, selector) {
  return process(selector, function(sel, filter, arg){
    try {
      var taggedParent
      if (!sel && filter) sel = '*'
      else if (childRe.test(sel))
        taggedParent = $(node).addClass(classTag), sel = '.'+classTag+' '+sel

      var nodes = oldQsa(node, sel)
      } catch(e) {
        console.error('error performing selector: %o', selector)
        throw e
      } finally {
        if (taggedParent) taggedParent.removeClass(classTag)
      }
    return !filter ? nodes :
    zepto.uniq($.map(nodes, function(n, i){ return filter.call(n, i, nodes, arg) }))
  })
}

改过的 qsa 调用的是 process 方法，在回调函数中处理大部分逻辑。

思路是通过选择器获取到所有节点，然后再调用对应伪类的对应方法来过滤出符合条件的节点。

处理选择器，根据选择器获取节点

var taggedParent
if (!sel && filter) sel = '*'
else if (childRe.test(sel))
  taggedParent = $(node).addClass(classTag), sel = '.'+classTag+' '+sel

var nodes = oldQsa(node, sel)

如果选择器和过滤器都不存在，则将 sel 设置 * ，即获取所有元素。

如果 >sel 形式的选择器，查找所有子元素。

这里的做法是，向元素 node 中添加唯一的样式名 classTag，然后用唯一样式名和选择器拼接成子元素选择器。

最后调用原有的 qsa 函数 oldQsa 来获取符合选择器的所有元素。

清理所添加的样式

if (taggedParent) taggedParent.removeClass(classTag)

如果存在 taggedParent ，则将元素上的 classTag 清理掉。

调用对应的过滤器，过滤元素

return !filter ? nodes :
    zepto.uniq($.map(nodes, function(n, i){ return filter.call(n, i, nodes, arg) }))

如果没有过滤器，则将所有元素返回，如果存在过滤器，则遍历集合，调用对应的过滤器获取元素，并将新集合的元素去重。

matches

zepto.matches = function(node, selector){
  return process(selector, function(sel, filter, arg){
    return (!sel || oldMatches(node, sel)) &&
      (!filter || filter.call(node, null, arg) === node)
  })
}

matches 也是调用 process 方法，这里很巧妙地用了 || 和 && 的短路操作。

其实要做的事情就是，如果可以用 oldMatches 匹配，则使用 oldMatches 匹配的结果，否则使用过滤器过滤出来的结果。

系列文章

参考

try-catch语句的“伪块作用域”

License

署名-非商业性使用-禁止演绎 4.0 国际 (CC BY-NC-ND 4.0)

最后，所有文章都会同步发送到微信公众号上，欢迎关注,欢迎提意见：读Zepto源码之Selector模块

作者：对角另一面