solidoc2/node_utils/modifier.js

Solidity Documentation Generator 2

const enumerable = require('linq')

const getModifiers = (contract) => {
  let modifiers = []

  const nodes = enumerable.from(contract.ast.nodes).where(function (x) {
    return x.nodes
  }).toArray()

  for (let i = 0; i < nodes.length; i++) {
    const node = nodes[i]

    if (node.nodeType === 'ModifierDefinition') {
      modifiers.push(node)
    }

    const candidates = enumerable.from(node.nodes).where(function (x) {
      return x.nodeType === 'ModifierDefinition'
    }).toArray()

    modifiers = modifiers.concat(candidates)
  }

  return modifiers
}

module.exports = {
  getModifiers
}