@eduardoeb3/espree-logging-module-eduardoeb3/lib/addLogging.js

[](https://code.visualstudio.com/) [](https://ull-esit-pl-2021.github.io/espree-logging-module-Eduar

/**
 * Universidad de La Laguna
 * Escuela Superior de Ingeniería y Tecnología
 * Grado en Ingeniería Informática
 * Procesadores de Lenguajes
 *
 * @author Eduardo Expósito Barrera
 * @date 7 Mar 2021
 * @brief Programa que se encarga de añadir sentencias al código
 */

const escodegen = require('escodegen');
const esprima = require('espree');
const estraverse = require('estraverse');

/**
 * Función que se encarga de recorrer el AST del código pasado
 * @param {String} code entrada en formato string
 * @param {String} pattern entrada en formato string
 * @returns el código en forma de string que ha sido generado a través del AST
 */
exports.addLogging = function (code, pattern) {
  let ast = esprima.parse(code, { ecmaVersion: 6, loc: true });
  estraverse.traverse(ast, {
    enter: function (node, parent) {
      if (node.type === 'FunctionDeclaration' ||
        node.type === 'FunctionExpression' || node.type === 'ArrowFunctionExpression') {
        if ((node.id && new RegExp(pattern).test(node.id.name)) || pattern === undefined) {
          addBeforeCode(node);
        }
      }
    }
  });
  return escodegen.generate(ast);
}

/**
 * Función que se encarga de añadir al árbol las nuevas sentencias
 * @param {Object} node entrada en formato objeto de parte del AST
 */
function addBeforeCode(node) {
  let name = node.id ? node.id.name : '<anonymous function>';
  let loc = node.loc.start;
  let line = loc.line - 1;
  let temp = node.params.map(function (argumentos) {
    return "${ " + argumentos.name + " }";
  });
  temp = temp.join(', ');
  let beforeCode = `console.log(\`Entering ${name}(${temp}) at line ${line}\`);`;
  let beforeNodes = esprima.parse(beforeCode, { ecmaVersion: 6 }).body; // Is an Array of ASTs
  node.body.body = beforeNodes.concat(node.body.body);
}