solidity-parser-antlr/test/utils.js

const { assert } = require('chai')
const parser = require('../src/index')

function print(obj) {
  console.log(JSON.stringify(obj, null, 2))
}

function parseContract(source, options = {}) {
  var ast = parser.parse(source, options)
  assert.isOk(ast.children[0])
  return ast.children[0]
}

function parseNode(source, options = {}) {
  var contract = parseContract("contract test { " + source + " }", options)
  assert.isOk(contract.subNodes[0])
  return contract.subNodes[0]
}

function parseStatement(source, options = {}) {
  var ast = parseNode("function () { " + source + " }", options)
  assert.isOk(ast.body.statements[0])
  return ast.body.statements[0]
}

function parseExpression(source, options = {}) {
  var ast = parseNode("function () { " + source + "; }", options)
  assert.isOk(ast.body.statements[0].expression)
  return ast.body.statements[0].expression
}

function parseAssembly(source, options = {}) {
  var ast = parseNode("function () { assembly { " + source + " } }", options)
  assert.isOk(ast.body.statements[0].body.operations[0])
  return ast.body.statements[0].body.operations[0]
}

module.exports = {
  parseAssembly,
  parseContract,
  parseExpression,
  parseNode,
  parseStatement,
  print
}