solidoc2/builders/modifier-code-builder.js

Solidity Documentation Generator 2

const enumerable = require('linq')

const build = (node) => {
  if (!node) {
    return ''
  }

  const builder = []

  const parameters = node.parameters.parameters

  const parameterList = enumerable.from(parameters).select((x) => {
    const argumentName = x.name
    const dataType = x.typeDescriptions.typeString.replace('contract ', '')
    return dataType + ' ' + argumentName
  }).toArray()

  builder.push('```js')
  builder.push('\n')
  builder.push(`modifier ${node.name}(`)

  builder.push(parameterList.join(', '))

  builder.push(') ')

  builder.push(node.visibility.toLowerCase())
  builder.push('\n')
  builder.push('```')

  return builder.join('')
}

module.exports = { build }