Externalise element detection

Author: wooorm

index.js

@@ -1,5 +1,8 @@
 'use strict';
 
+/* Dependencies. */
+var is = require('hast-util-is-element');
+
 /* Expose. */
 module.exports = menuState;
 
@@ -8,7 +11,6 @@ function menuState(nodes) {
   var index;
   var node;
   var type;
-  var name;
 
   if (!nodes || typeof nodes !== 'object' || nodes.length === 0) {
     return null;
@@ -17,19 +19,17 @@ function menuState(nodes) {
   index = nodes.length - 1;
   node = nodes[index];
 
-  if (tagName(node) !== 'menu') {
+  if (!is(node, 'menu')) {
     return null;
   }
 
   while (node) {
-    name = tagName(node);
-
     /* Stop at `template` elements and non-elements. */
-    if (!name || name === 'template') {
+    if (!is(node) || is(node, 'template')) {
       break;
     }
 
-    if (name === 'menu') {
+    if (is(node, 'menu')) {
       type = node.properties && node.properties.type;
 
       if (type === 'context' || type === 'toolbar') {
@@ -42,12 +42,3 @@ function menuState(nodes) {
 
   return 'toolbar';
 }
-
-/* Get the `tagName` of an element. */
-function tagName(value) {
-  if (value && typeof value === 'object' && value.type === 'element') {
-    return value.tagName;
-  }
-
-  return null;
-}

package.json

@@ -21,7 +21,9 @@
   "files": [
     "index.js"
   ],
-  "dependencies": {},
+  "dependencies": {
+    "hast-util-is-element": "^1.0.0"
+  },
   "devDependencies": {
     "browserify": "^13.0.0",
     "esmangle": "^1.0.1",