commentme 1.0.7 → 1.0.9

package/bin/commentme.js

@@ -9,14 +9,26 @@ import { deleteComment } from "../deletecoms.js";
 import { removeCommentsFromFile as skim } from "../skimcoms.js";
 import { unskimComments as unskim } from "../unskimcoms.js";
 import { generateCommentsPerFunc, generateCommentsPerClass, generateCommentsPerLine, generateExplanation } from "../generate.js";
+import { sanitizeFile } from "../sanitize.js";
 // import { connectDB, disconnectDB } from "../config/db.js";
 import { ensureAuth } from "../auth/authGuard.js";
 import { logout } from "../auth/logout.js";
 import { saveApiKey, clearApiKey } from "../utils/apiKeyManager.js";
 import { promptPassword } from "../utils/passwordPrompt.js";
 import dotenv from "dotenv";
+import { readFileSync } from "fs";
+import { dirname, join } from "path";
+import { fileURLToPath } from "url";
+
 dotenv.config();
 
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+// Read version from package.json
+const packageJson = JSON.parse(readFileSync(join(__dirname, "../package.json"), "utf8"));
+const version = packageJson.version;
+
 
 function promptInput(defaultValue = "") {
   const rl = readline.createInterface({
@@ -39,10 +51,15 @@ async function main() {
   console.log("Command:", command);
 
   try {
-    // Show help without connecting to DB or requiring auth
+    // Show help or version without connecting to DB or requiring auth
+    if (command === "--version" || command === "-v") {
+      console.log(`commentme version ${version}`);
+      return;
+    }
+
     if (command === "--help" || command === "-h" || !command) {
       console.log(`
-commentme CLI
+commentme CLI (v${version})
 
 Commands:
   commentme --get line-7-7 <file>      Get a specific comment by line range
@@ -53,9 +70,13 @@ Commands:
   commentme --unskim <file>      Restore comments to a file
   commentme --generate <file>    Generate AI comments and docs for a file
   commentme --explain <file>     Generate a full markdown explanation of a code file
+  commentme --sanitize <file>    Sanitize file for production (removes noisy comments)
   commentme --set-key            Set your own OpenRouter API key (stored securely)
   commentme --clear-key          Remove your saved API key
+  commentme --login              Log in to your account
+  commentme --signup             Create a new account
   commentme --logout             Log out from your session
+  commentme --version            Show version number
   commentme --help               Show this help message
 `);
       return;
@@ -153,6 +174,11 @@ Select generation type:
         await generateExplanation(args[1]);
         break;
 
+      case "--sanitize":
+        if (!args[1]) throw new Error("Usage: commentme --sanitize <file>");
+        await sanitizeFile(args[1]);
+        break;
+
       case "--set-key": {
         console.log("Paste your OpenRouter API key (input is hidden):");
         const key = await promptPassword("🔑 API Key: ");
@@ -174,7 +200,7 @@ Select generation type:
 
       default:
         console.log(`
-commentme CLI
+commentme CLI (v${version})
 
 Commands:
   commentme --login
@@ -187,9 +213,12 @@ Commands:
   commentme --unskim <file>
   commentme --generate <file>
   commentme --explain <file>
+  commentme --sanitize <file>
   commentme --set-key
   commentme --clear-key
   commentme --logout
+  commentme --version
+  commentme --help
 `);
     }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "commentme",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "description": "A CLI tool to manage, redact and unredact your comments keeping the codebase clutterfree.",
   "type": "module",
   "bin": {

package/sanitize-test.js

@@ -0,0 +1,21 @@
+/**
+ * This is a JSDoc block.
+ * It should be PRESERVED.
+ * @param {string} msg 
+ */
+function testDoc(msg) {
+    
+    
+    console.log("This console.log should be PRESERVED.");
+
+    
+    const x = 10; 
+
+    /**
+     * Another JSDoc.
+     * Should be PRESERVED.
+     */
+    return msg + x;
+}
+
+testDoc("Hello");

package/sanitize.js

@@ -0,0 +1,62 @@
+import * as acorn from "acorn";
+import fs from "fs";
+import path from "path";
+import { getCurrentUserId } from "./utils/currentUser.js";
+
+/**
+ * Sanitizes a file for production by removing:
+ * 1. Single-line comments (// ...)
+ * 2. Block comments (/* ... * /) EXCEPT JSDocs (/** ... * /)
+ * 
+ * Note: console.log statements are preserved per user request.
+ */
+export async function sanitizeFile(filePath) {
+    if (!fs.existsSync(filePath)) {
+        throw new Error(`File not found: ${filePath}`);
+    }
+
+    // Check authentication
+    try {
+        getCurrentUserId();
+    } catch (e) {
+        console.error(e.message);
+        return;
+    }
+
+    const code = fs.readFileSync(filePath, "utf8");
+    const comments = [];
+
+    try {
+        acorn.parse(code, {
+            ecmaVersion: 2020,
+            sourceType: "module",
+            locations: true,
+            onComment: (isBlock, text, start, end) => {
+                const isJSDoc = isBlock && text.startsWith("*");
+
+                // If it's a single-line comment OR a non-JSDoc block comment, mark for removal
+                if (!isBlock || !isJSDoc) {
+                    comments.push({ start, end });
+                }
+            }
+        });
+    } catch (e) {
+        console.error(`❌ Error parsing ${filePath}: ${e.message}`);
+        console.log("⚠️  Falling back to manual sanitization might be risky. Skipping this file.");
+        return;
+    }
+
+    // Sort comments in descending order to avoid offset issues while slicing
+    comments.sort((a, b) => b.start - a.start);
+
+    let result = code;
+    for (const comment of comments) {
+        result = result.slice(0, comment.start) + result.slice(comment.end);
+    }
+
+    // Optional: Clean up empty lines that only contained comments
+    // result = result.replace(/^\s*[\r\n]/gm, "");
+
+    fs.writeFileSync(filePath, result, "utf8");
+    console.log(`✅ ${filePath} sanitized (comments removed, documentation preserved)`);
+}