@specs-feup/clava/src-api/clava/util/CodeInserter.ts

A C/C++ source-to-source compiler written in Typescript

import Io from "@specs-feup/lara/api/lara/Io.js";
import LineInserter from "@specs-feup/lara/api/lara/util/LineInserter.js";
import { FileJp } from "../../Joinpoints.js";
import Clava from "../Clava.js";

/**
 * Writes the original code of the application, with the possibility of inserting new lines of code.
 */
export default class CodeInserter {
  /**
   * Maps $file AST ids to an object that maps line numbers to strings to insert.
   */
  protected linesToInserts: Record<string, Record<number, string[]>>;

  constructor() {
    this.linesToInserts = {};
  }

  add($file: FileJp, line: number, content: string) {
    const astId = $file.astId;
    const fileLines = this.getFileLines(astId);
    this.addContent(fileLines, line, content);
  }

  /**
   * Writes the code of the current tree, plus the lines to insert, to the given folder.
   */
  write(outputFolder: string) {
    const lineInserter = new LineInserter();

    // Write each file, inserting lines if needed
    for (const $jp of Clava.getProgram().getDescendants("file")) {
      const $file = $jp as FileJp;

      if (!Io.isFile($file.filepath)) {
        console.log(
          "CodeInserter.write: skipping file, could not find path '" +
            $file.filepath +
            "'"
        );
        continue;
      }

      // Original code
      let fileCode = Io.readFile($file.filepath);

      // Intrument code, if needed
      const fileLines = this.linesToInserts[$file.astId];
      if (fileLines !== undefined) {
        fileCode = lineInserter.add(fileCode, fileLines);
      }

      // Get path for writing file
      const outputFilepath = $file.getDestinationFilepath(outputFolder);

      Io.writeFile(outputFilepath, fileCode);
    }
  }

  private getFileLines(astId: string): Record<number, string[]> {
    let fileLines = this.linesToInserts[astId];

    if (fileLines === undefined) {
      fileLines = {};
      this.linesToInserts[astId] = fileLines;
    }

    return fileLines;
  }

  private addContent(
    fileLines: Record<number, string[]>,
    line: number,
    content: string
  ) {
    const lineStrings = fileLines[line];
    if (lineStrings === undefined) {
      fileLines[line] = [];
    }

    fileLines[line].push(content);
  }
}