fix: correctly visit by class

diauweb · web-flow · commit 65d7338ba8b1 · 2020-11-01T16:51:42.000+08:00
diff --git a/index.js b/index.js
@@ -25,7 +25,8 @@ function rehypeDetails(options) {
     visit(tree, 'element', visitor)
 
     function visitor(node, index, parent) {
-      if (node.tagName !== 'div' || node.properties.class !== 'details') {
+      const className = node.properties.className || []
+      if (node.tagName !== 'div' || !className.includes('details')) {
         return;
       }
       // console.log(node)

Original file line number	Diff line number	Diff line change
`@@ -25,7 +25,8 @@ function rehypeDetails(options) {`
`25`	`25`	`visit(tree, 'element', visitor)`
`26`	`26`
`27`	`27`	`function visitor(node, index, parent) {`
`28`		`- if (node.tagName !== 'div' \|\| node.properties.class !== 'details') {`
	`28`	`+ const className = node.properties.className \|\| []`
	`29`	`+ if (node.tagName !== 'div' \|\| !className.includes('details')) {`
`29`	`30`	`return;`
`30`	`31`	`}`
`31`	`32`	`// console.log(node)`