node-attributes.js

// Return attributes and values of a node in a convenient way:
/* example: 
    <ExampleElement attr1="15" attr2>
    {   attr1: {
            hasValue: true,
            value: 15
        },
        attr2: {
            hasValue: false
        }
Inclusion of hasValue is in case an eslint rule cares about boolean values
explicitely assigned to attribute vs the attribute being used as a flag
*/
class NodeAttributes {
  constructor(ASTnode) {
    this.attributes = {}
    ASTnode.attributes.forEach((attribute) => {
      if (!attribute.type || attribute.type !== 'JSXAttribute') {
        return
      }
      this.attributes[attribute.name.name] = {
        hasValue: !!attribute.value,
      }
      if (attribute.value) {
        if (attribute.value.value) {
          this.attributes[attribute.name.name].value = attribute.value.value
        } else if (attribute.value.expression) {
          this.attributes[attribute.name.name].value =
            attribute.value.expression.value
        }
      }
    })
  }
  hasAny() {
    return !!Object.keys(this.attributes).length
  }
  has(attrName) {
    return !!this.attributes[attrName]
  }
  hasValue(attrName) {
    return !!this.attributes[attrName].hasValue
  }
  value(attrName) {
    if (!this.attributes[attrName]) {
      return true
    }

    return this.attributes[attrName].value
  }
}

module.exports = NodeAttributes