solidoc2/node_utils/function.js

Solidity Documentation Generator 2

const enumerable = require('linq')

const getFunctions = (contract) => {
  let functions = []

  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 === 'FunctionDefinition') {
      functions.push(node)
    }

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

    functions = functions.concat(candidates)
  }

  return functions
}

module.exports = {
  getFunctions
}