index.js

'use strict';

const PEG = require('peggy');
const fs = require('fs');
const path = require('path');

import { generate } from 'peggy';
import { readFileSync } from 'fs';
import { resolve } from 'path';

const builtParsers = {
  solidity: require('./build/parser'),
  imports: require('./build/imports_parser'),
};

function parseComments(sourceCode) {
  // for Line comment regexp, the "." doesn't cover line termination chars so we're good :)
  const comments = [],
    commentParser = /(\/\*(\*(?!\/)|[^*])*\*\/)|(\/\/.*)/g;
  let nextComment;

  // eslint-disable-next-line no-cond-assign
  while ((nextComment = commentParser.exec(sourceCode))) {
    const text = nextComment[0],
      types = { '//': 'Line', '/*': 'Block' };

    comments.push({
      text,
      type: types[text.slice(0, 2)],
      start: nextComment.index,
      end: nextComment.index + text.length,
    });
  }

  return comments;
}

// TODO: Make all this async.
module.exports = {
  getParser(parser_name, rebuild) {
    if (rebuild == true) {
      let parserfile = fs.readFileSync(path.resolve(`./${parser_name}.pegjs`), {
        encoding: 'utf8',
      });
      return PEG.generate(parserfile);
    } else {
      return builtParsers[parser_name];
    }
  },
  parse(source, options, parser_name, rebuild) {
    if (typeof parser_name == 'boolean') {
      rebuild = parser_name;
      parser_name = null;
    }

    if (parser_name == null) {
      parser_name = 'solidity';
    }

    let parser = this.getParser(parser_name, rebuild);
    let result;

    try {
      result = parser.parse(source);
    } catch (e) {
      if (e instanceof parser.SyntaxError) {
        e.message +=
          ' Line: ' +
          e.location.start.line +
          ', Column: ' +
          e.location.start.column;
      }
      throw e;
    }

    if (typeof options === 'object' && options.comment === true) {
      result.comments = parseComments(source);
    }

    return result;
  },
  parseFile(file, parser_name, rebuild) {
    return this.parse(
      fs.readFileSync(path.resolve(file), { encoding: 'utf8' }),
      parser_name,
      rebuild,
    );
  },
  parseComments,
};