Fix some block box bugs

1 files changed, 5 insertions, 6 deletions

diff --git a/src/css/select.nim b/src/css/select.nim
index 3a2e0e27..1bd45dcd 100644
--- a/src/css/select.nim
+++ b/src/css/select.nim
@@ -32,15 +32,12 @@ func attrSelectorMatches(elem: Element, sel: Selector): bool =
   of '*': return elem.attr(sel.attr).contains(sel.value)
   else: return false
 
-func pseudoElemSelectorMatches(elem: Element, s: string): SelectResult =
-  case s
+func pseudoElemSelectorMatches(elem: Element, sel: Selector): SelectResult =
+  case sel.elem
   of "before": return selectres(true, PSEUDO_BEFORE)
   of "after": return selectres(true, PSEUDO_AFTER)
   else: return selectres(false)
 
-func pseudoElemSelectorMatches(elem: Element, sel: Selector): SelectResult =
-  return elem.pseudoElemSelectorMatches(sel.elem)
-
 func pseudoSelectorMatches(elem: Element, sel: Selector): SelectResult =
   case sel.pseudo
   of "first-child": return selectres(elem.parentNode.firstElementChild == elem)
@@ -48,7 +45,9 @@ func pseudoSelectorMatches(elem: Element, sel: Selector): SelectResult =
   of "only-child": return selectres(elem.parentNode.firstElementChild == elem and elem.parentNode.lastElementChild == elem)
   of "hover": return selectres(elem.hover)
   of "root": return selectres(elem == elem.ownerDocument.root)
-  else: return elem.pseudoElemSelectorMatches(sel.pseudo)
+  of "before": return selectres(true, PSEUDO_BEFORE)
+  of "after": return selectres(true, PSEUDO_AFTER)
+  else: return selectres(false)
 
 func selectorsMatch*(elem: Element, selectors: SelectorList): SelectResult