From ec09a5cf2bb2d82fe9cc35b9eefe01d04d36d53e Mon Sep 17 00:00:00 2001
From: phukon <hello@rikiphukon.com>
Date: Tue, 17 Sep 2024 03:09:56 +0530
Subject: [PATCH] feat: codemod on multiple files at once and preserve line
 offsets

---
 main/index.cjs | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/main/index.cjs b/main/index.cjs
index eb9c87c..d712598 100644
--- a/main/index.cjs
+++ b/main/index.cjs
@@ -10,8 +10,13 @@
 // **Interpreting the mappings entries (Base 64 VLQ)**
 // - [0]: Column index in the compiled file
 // - [1]: What original source file the location in the compiled source maps to
-// - [2]: Row index in the original source file (i.e. the line number)
+// - [2]: Row index in the original source file (i.e. the line number in offsets ⚠)
 // - [3]: Column index in the original source file
+
+/** ⚠ Regarding offsets
+ * If the previous mapping was on line n in the original source code,
+ * then "AAGA" would map to line n + 3.
+ */
 const prettier = require('prettier');
 const parser = require('@babel/parser');
 const traverse = require('@babel/traverse').default;
@@ -21,7 +26,7 @@ const fs = require('fs');
 const path = require('path');
 const { v4: uuidv4 } = require('uuid');
 const { SourceMapGenerator } = require('source-map');
-const sourceCodeDir = './test';
+const sourceCodeDir = './samples';
 const readmeFilePath = './README.md';
 const sourceMapFilePath = './README.map.json';
 let collectedComments = [];
@@ -276,12 +281,21 @@ class FileProcessor {
   generateSourceMap(sourceMapFilePath) {
     const map = new SourceMapGenerator({ file: readmeFilePath });
     let line = 1;
-    let offset = 0;
+    let fileOffsets = {}; // to store offsets for each file
+
     collectedComments.forEach((comment) => {
       const location = commentLocations.find(
         (loc) => loc.docmapId === comment.docmapId
       );
       if (location) {
+        // Initialize offset for this file if it doesn't exist
+        if (!fileOffsets[location.filePath]) {
+          fileOffsets[location.filePath] = 0;
+        }
+
+        // Use the file-specific offset
+        let offset = fileOffsets[location.filePath];
+
         // Increase offset by 1 if there's a blank line before the comment
         if (location.hasBlankLineBefore) {
           offset += 1;
@@ -290,9 +304,10 @@ class FileProcessor {
         const lines = comment.text.split('\n');
         if (lines.length > 1) {
           offset += lines.length + 3; // the starting chars + the tag +  the ending chars = 3
-        } else if (lines.length === 1 && comment.isMultiline) { // Inline comments and Comment blocks can still have one line after parsing
+        } else if (lines.length === 1 && comment.isMultiline) {
           offset += 3;
         }
+
         lines.forEach((_, i) => {
           let mapping = {
             generated: { line: line + i, column: 0 },
@@ -312,6 +327,10 @@ class FileProcessor {
           // });
         });
         line += lines.length + 1; // +1 for the blank line between comments
+        
+        // Update the file-specific offset
+        fileOffsets[location.filePath] = offset;
+
         const sourceContent = fs.readFileSync(location.filePath, 'utf-8');
         map.setSourceContent(location.filePath, sourceContent);
       }