about summary refs log tree commit diff stats
path: root/src/css/select.nim
blob: 0d142558ba0b4250bdae11e5ed29822082645035 (plain) (blame)
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
import tables
import strutils

import css/cssparser
import css/selectorparser
import css/stylednode
import html/dom
import html/tags

func attrSelectorMatches(elem: Element, sel: Selector): bool =
  case sel.rel
  of ' ': return sel.attr in elem.attributes
  of '=': return elem.attr(sel.attr) == sel.value
  of '~': return sel.value in elem.attr(sel.attr).split(Whitespace)
  of '|':
    let val = elem.attr(sel.attr)
    return val == sel.value or sel.value.startsWith(val & '-')
  of '^': return elem.attr(sel.attr).startsWith(sel.value)
  of '$': return elem.attr(sel.attr).endsWith(sel.value)
  of '*': return elem.attr(sel.attr).contains(sel.value)
  else: return false

func pseudoSelectorMatches[T: Element|StyledNode](elem: T, sel: Selector, felem: T): bool =
  let selem = elem
  when elem is StyledNode:
    let elem = Element(elem.node)
  case sel.pseudo
  of PSEUDO_FIRST_CHILD: return elem.parentNode.firstElementChild == elem
  of PSEUDO_LAST_CHILD: return elem.parentNode.lastElementChild == elem
  of PSEUDO_ONLY_CHILD: return elem.parentNode.firstElementChild == elem and elem.parentNode.lastElementChild == elem
  of PSEUDO_HOVER:
    when selem is StyledNode: felem.addDependency(selem, DEPEND_HOVER)
    return elem.hover
  of PSEUDO_ROOT: return elem == elem.document.html
  of PSEUDO_NTH_CHILD:
    let n = int64(sel.pseudonum - 1)
    var i = 0
    for child in elem.parentNode.children:
      if i == n:
        return child == elem
      inc i
    return false
  of PSEUDO_CHECKED:
    when selem is StyledNode: felem.addDependency(selem, DEPEND_CHECKED)
    if elem.tagType == TAG_INPUT:
      return HTMLInputElement(elem).checked
    elif elem.tagType == TAG_OPTION:
      return HTMLOptionElement(elem).selected
    return false
  of PSEUDO_FOCUS:
    when selem is StyledNode: felem.addDependency(selem, DEPEND_FOCUS)
    return elem.document.focus == elem

func selectorsMatch*[T: Element|StyledNode](elem: T, selectors: SelectorList, felem: T = nil): bool

func funcSelectorMatches[T: Element|StyledNode](elem: T, sel: Selector, felem: T): bool =
  case sel.name
  of "not":
    for slist in sel.fsels:
      if elem.selectorsMatch(slist, felem):
        return false
    return true
  of "is", "where":
    for slist in sel.fsels:
      if elem.selectorsMatch(slist, felem):
        return true
    return false
  else: discard

func combinatorSelectorMatches[T: Element|StyledNode](elem: T, sel: Selector, felem: T): bool =
  let selem = elem
  #combinator without at least two members makes no sense
  assert sel.csels.len > 1
  if selem.selectorsMatch(sel.csels[^1], felem):
    var i = sel.csels.len - 2
    case sel.ct
    of DESCENDANT_COMBINATOR:
      when selem is StyledNode:
        var e = elem.parent
      else:
        var e = elem.parentElement
      while e != nil and i >= 0:
        if e.selectorsMatch(sel.csels[i], felem):
          dec i
        when elem is StyledNode:
          e = e.parent
        else:
          e = e.parentElement
    of CHILD_COMBINATOR:
      when elem is StyledNode:
        var e = elem.parent
      else:
        var e = elem.parentElement
      while e != nil and i >= 0:
        if not e.selectorsMatch(sel.csels[i], felem):
          return false
        dec i
        when elem is StyledNode:
          e = e.parent
        else:
          e = e.parentElement
    of NEXT_SIBLING_COMBINATOR:
      var found = false
      when elem is StyledNode:
        var parent = elem.parent
      else:
        var parent = elem.parentElement
      for child in parent.children_rev:
        when elem is StyledNode:
          if child.t != STYLED_ELEMENT or child.node == nil: continue
        if found:
          if not child.selectorsMatch(sel.csels[i], felem):
            return false
          dec i
          if i < 0:
            return true
        if child == elem:
          found = true
    of SUBSEQ_SIBLING_COMBINATOR:
      var found = false
      when selem is StyledNode:
        var parent = selem.parent
      else:
        var parent = elem.parentElement
      for child in parent.children_rev:
        when selem is StyledNode:
          if child.t != STYLED_ELEMENT or child.node == nil: continue
        if found:
          if child.selectorsMatch(sel.csels[i], felem):
            dec i
          if i < 0:
            return true
        if child == selem:
          found = true
    return i == -1
  return false

func selectorMatches[T: Element|StyledNode](elem: T, sel: Selector, felem: T): bool =
  let selem = elem
  when elem is StyledNode:
    let elem = Element(selem.node)
  case sel.t
  of TYPE_SELECTOR:
    return elem.tagType == sel.tag
  of CLASS_SELECTOR:
    return sel.class in elem.classList
  of ID_SELECTOR:
    return sel.id == elem.id
  of ATTR_SELECTOR:
    return elem.attrSelectorMatches(sel)
  of PSEUDO_SELECTOR:
    return pseudoSelectorMatches(selem, sel, felem)
  of PSELEM_SELECTOR:
    return true
  of UNIVERSAL_SELECTOR:
    return true
  of FUNC_SELECTOR:
    return funcSelectorMatches(selem, sel, felem)
  of COMBINATOR_SELECTOR:
    return combinatorSelectorMatches(selem, sel, felem)

# WARNING for StyledNode, this has the side effect of modifying depends.
#TODO make that an explicit flag or something, also get rid of the Element case
func selectorsMatch*[T: Element|StyledNode](elem: T, selectors: SelectorList, felem: T = nil): bool =
  let felem = if felem != nil:
    felem
  else:
    elem

  for sel in selectors.sels:
    if not selectorMatches(elem, sel, felem):
      return false
  return true

#TODO idk, it's not like we have JS anyways
#func selectElems[T: Element|StyledNode](element: T, sel: Selector, felem: T): seq[T] =
#  case sel.t
#  of TYPE_SELECTOR:
#    return element.filterDescendants((elem) => elem.tagType == sel.tag)
#  of ID_SELECTOR:
#    return element.filterDescendants((elem) => elem.id == sel.id)
#  of CLASS_SELECTOR:
#    return element.filterDescendants((elem) => sel.class in elem.classList)
#  of UNIVERSAL_SELECTOR:
#    return element.all_descendants
#  of ATTR_SELECTOR:
#    return element.filterDescendants((elem) => attrSelectorMatches(elem, sel))
#  of PSEUDO_SELECTOR:
#    return element.filterDescendants((elem) => pseudoSelectorMatches(elem, sel, felem))
#  of PSELEM_SELECTOR:
#    return element.all_descendants
#  of FUNC_SELECTOR:
#    return element.filterDescendants((elem) => selectorMatches(elem, sel))
#  of COMBINATOR_SELECTOR:
#    return element.filterDescendants((elem) => selectorMatches(elem, sel))
#
#func selectElems(element: Element, selectors: SelectorList): seq[Element] =
#  assert(selectors.len > 0)
#  let sellist = optimizeSelectorList(selectors)
#  result = element.selectElems(selectors[0], element)
#  var i = 1
#
#  while i < sellist.len:
#    result = result.filter((elem) => selectorMatches(elem, sellist[i], elem))
#    inc i
#
#proc querySelectorAll*(document: Document, q: string): seq[Element] =
#  let ss = newStringStream(q)
#  let cvals = parseListOfComponentValues(ss)
#  let selectors = parseSelectors(cvals)
#
#  if document.html != nil:
#    for sel in selectors:
#      result.add(document.html.selectElems(sel))
#
#proc querySelector*(document: Document, q: string): Element =
#  let elems = document.querySelectorAll(q)
#  if elems.len > 0:
#    return elems[0]
#  return nil
#
#proc querySelectorAll*(element: Element, q: string): seq[Element] =
#  let ss = newStringStream(q)
#  let cvals = parseListOfComponentValues(ss)
#  let selectors = parseSelectors(cvals)
#
#  for sel in selectors:
#    result.add(element.selectElems(sel))
#
#proc querySelector*(element: Element, q: string): Element =
#  let elems = element.querySelectorAll(q)
#  if elems.len > 0:
#    return elems[0]
#  return nil