secufusion-mcp 1.0.7 → 1.0.9

package/index.js

@@ -49,6 +49,36 @@ function getCurrentBranch(cwd) {
     }
 }
 // ─────────────────────────────────────────────
+//  Telemetry & Compression helpers
+// ─────────────────────────────────────────────
+function compressCode(code) {
+    return code
+        .replace(/\/\*[\s\S]*?\*\//g, "") // Remove multi-line comments
+        .replace(/\/\/.*$/gm, "") // Remove single-line comments
+        .replace(/^\s*[\r\n]/gm, "") // Remove empty lines
+        .replace(/[ \t]+$/gm, ""); // Remove trailing whitespace
+}
+function appendTelemetry(response, inputChars) {
+    const inputTokens = Math.ceil(inputChars / 4);
+    let outputChars = 0;
+    if (response.content && Array.isArray(response.content)) {
+        for (const c of response.content) {
+            if (c.type === "text" && c.text) {
+                outputChars += c.text.length;
+            }
+        }
+    }
+    const outputTokens = Math.ceil(outputChars / 4);
+    const inputCost = (inputTokens / 1000000) * 3.0;
+    const outputCost = (outputTokens / 1000000) * 15.0;
+    const totalCost = (inputCost + outputCost).toFixed(4);
+    const telemetry = `\n\n> 📊 Telemetry\n> Input Tokens: ~${inputTokens}\n> Output Tokens: ~${outputTokens}\n> Estimated Cost: ${totalCost}`;
+    if (response.content && response.content.length > 0 && response.content[0].type === "text") {
+        response.content[0].text += telemetry;
+    }
+    return response;
+}
+// ─────────────────────────────────────────────
 //  run_pre_pr_checks helpers
 // ─────────────────────────────────────────────
 const CONSOLE_LOG_REGEX = [
@@ -282,6 +312,7 @@ server.tool("manage_branch_state", "Manage the structured JSON state for the cur
         .optional()
         .describe("Explicit actionable instruction for resuming. Required for 'update'."),
 }, async ({ action, task_description, reference_file_path, pending_acs, completed_acs, next_step }) => {
+    const inputChars = JSON.stringify({ action, task_description, reference_file_path, pending_acs, completed_acs, next_step }).length;
     const cwd = process.cwd();
     const branch = getCurrentBranch(cwd);
     const statePath = path.resolve(STATE_FILE);
@@ -297,18 +328,18 @@ server.tool("manage_branch_state", "Manage the structured JSON state for the cur
     if (action === "read") {
         const branchState = state[branch];
         if (!branchState) {
-            return {
+            return appendTelemetry({
                 content: [
                     {
                         type: "text",
                         text: `No state found for branch '${branch}'. Please initialize first.`,
                     },
                 ],
-            };
+            }, inputChars);
         }
-        return {
+        return appendTelemetry({
             content: [{ type: "text", text: JSON.stringify(branchState, null, 2) }],
-        };
+        }, inputChars);
     }
     if (action === "initialize") {
         state[branch] = {
@@ -320,31 +351,42 @@ server.tool("manage_branch_state", "Manage the structured JSON state for the cur
             last_updated: new Date().toISOString()
         };
         writeFile(statePath, JSON.stringify(state, null, 2));
-        return {
+        let refText = "";
+        if (reference_file_path) {
+            const refPath = resolve(reference_file_path);
+            const content = readFileSafe(refPath);
+            if (content) {
+                refText = `\n\n**Reference Code (Compressed):**\n\`\`\`\n${compressCode(content)}\n\`\`\``;
+            }
+            else {
+                refText = `\n\n*(Warning: reference file '${reference_file_path}' not found)*`;
+            }
+        }
+        return appendTelemetry({
             content: [
                 {
                     type: "text",
-                    text: `✅ Branch state initialized for '${branch}'.\n\nState:\n${JSON.stringify(state[branch], null, 2)}`,
+                    text: `✅ Branch state initialized for '${branch}'.\n\nState:\n${JSON.stringify(state[branch], null, 2)}${refText}`,
                 },
             ],
-        };
+        }, inputChars);
     }
     if (action === "update") {
         if (!state[branch]) {
-            return {
+            return appendTelemetry({
                 isError: true,
                 content: [
                     { type: "text", text: `No state found for branch '${branch}'. Initialize first.` },
                 ],
-            };
+            }, inputChars);
         }
         if (!next_step) {
-            return {
+            return appendTelemetry({
                 isError: true,
                 content: [
                     { type: "text", text: "ERROR: 'next_step' is required when updating." },
                 ],
-            };
+            }, inputChars);
         }
         if (pending_acs)
             state[branch].pending_acs = pending_acs;
@@ -353,16 +395,16 @@ server.tool("manage_branch_state", "Manage the structured JSON state for the cur
         state[branch].next_step = next_step;
         state[branch].last_updated = new Date().toISOString();
         writeFile(statePath, JSON.stringify(state, null, 2));
-        return {
+        return appendTelemetry({
             content: [
                 {
                     type: "text",
                     text: `✅ Branch state updated for '${branch}'. Next step recorded:\n> ${next_step}`,
                 },
             ],
-        };
+        }, inputChars);
     }
-    return { content: [{ type: "text", text: "Invalid action." }] };
+    return appendTelemetry({ content: [{ type: "text", text: "Invalid action." }] }, inputChars);
 });
 // ─── Tool 2: log_rejected_pattern ────────────────────────────────────────────
 server.tool("log_rejected_pattern", "Record a coding pattern that was rejected by the team so it is never repeated. " +
@@ -390,6 +432,7 @@ server.tool("log_rejected_pattern", "Record a coding pattern that was rejected b
         .optional()
         .describe("Optional: the file or code area where this was observed."),
 }, async ({ pattern, reason, category, file_context }) => {
+    const inputChars = JSON.stringify({ pattern, reason, category, file_context }).length;
     const rejectedPath = resolve(REJECTED_FILE);
     let records = [];
     const existing = readFileSafe(rejectedPath);
@@ -411,7 +454,7 @@ server.tool("log_rejected_pattern", "Record a coding pattern that was rejected b
     };
     records.push(newEntry);
     writeFile(rejectedPath, JSON.stringify(records, null, 2));
-    return {
+    return appendTelemetry({
         content: [
             {
                 type: "text",
@@ -422,7 +465,7 @@ server.tool("log_rejected_pattern", "Record a coding pattern that was rejected b
                     `This will be checked automatically in all future architectural suggestions.`,
             },
         ],
-    };
+    }, inputChars);
 });
 // ─── Tool 3: run_pre_pr_checks ───────────────────────────────────────────────
 server.tool("run_pre_pr_checks", "Run all SecuFusion pre-PR guardrail checks across the workspace. " +
@@ -448,6 +491,7 @@ server.tool("run_pre_pr_checks", "Run all SecuFusion pre-PR guardrail checks acr
         .optional()
         .describe("Explicitly skip specific checks. Use sparingly — document the reason in your PR description."),
 }, async ({ work_item_id, root_dir, skip_checks = [] }) => {
+    const inputChars = JSON.stringify({ work_item_id, root_dir, skip_checks }).length;
     const scanRoot = root_dir ? path.resolve(root_dir) : process.cwd();
     const errors = [];
     const warnings = [];
@@ -613,10 +657,10 @@ server.tool("run_pre_pr_checks", "Run all SecuFusion pre-PR guardrail checks acr
                 `- **Ready to raise PR** 🚀\n` +
                 rejectedReminder;
     }
-    return {
+    return appendTelemetry({
         content: [{ type: "text", text: report }],
         isError: hasErrors,
-    };
+    }, inputChars);
 });
 // ─────────────────────────────────────────────
 //  Start transport

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "secufusion-mcp",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "type": "module",
   "description": "SecuFusion MCP server - developer workflow tooling with guardrails",
   "main": "index.js",