@goodandready/dsh-context-lens 0.1.15 → 0.1.17

package/lib/index.js

@@ -13,6 +13,7 @@ export const Config = z.object({
   tokenSavingsTracking: z.boolean().default(true).description('Track and display token budget savings'),
   autoCompressThreshold: z.number().default(4000).description('Auto-compress threshold in chars (0 to disable)'),
   budgetLimit: z.number().default(100000).description('Session token budget; compressions stop counting after this limit'),
+  budgetAlertPercent: z.number().default(90).description('Budget percentage threshold (50-99%) for warning alert'),
   autoCollapse: z.boolean().default(true).description('Warn in UI when budget is nearly exhausted (does not force-close the settings card)')
 });
 
@@ -37,6 +38,14 @@ function getFocus(key) {
   return focusBySession.get(key);
 }
 
+function clearFocus(key) {
+  if (key) {
+    focusBySession.set(key, { paths: [], updatedAt: new Date().toISOString() });
+  } else {
+    focusBySession.clear();
+  }
+}
+
 const OUTPUT_SCHEMA = { type: 'object', properties: { success: { type: 'boolean' } }, additionalProperties: true };
 const renderOutput = (_args, result) => JSON.stringify(result, null, 2);
 
@@ -53,6 +62,7 @@ export function apply(ctx, config) {
       const cfg = getConfig();
       if (cfg && cfg.tokenSavingsTracking === false) return null;
       if (cfg && typeof cfg.budgetLimit === 'number') tracker.setBudgetLimit(cfg.budgetLimit);
+      if (cfg && typeof cfg.budgetAlertPercent === 'number') tracker.setBudgetAlertPercent(cfg.budgetAlertPercent);
       return tracker.record(original, compressed);
     } catch {
       // #39: never record on config/read errors
@@ -114,12 +124,12 @@ export function apply(ctx, config) {
 
     ctx.tools.register({
       name: 'context_lens_compress_code',
-      description: 'Generate AST skeleton for a large source file (JS/TS/Python/Go/Rust/Java) to save context',
+      description: 'Generate AST skeleton for a large source file (JS/TS/Python/Go/Rust/Java/C/C++/SQL) to save context',
       parameters: {
         type: 'object',
         properties: {
           code: { type: 'string', description: 'Source code to skeletonize' },
-          language: { type: 'string', enum: ['js', 'ts', 'py', 'python', 'go', 'rust', 'java'], description: 'Language hint' },
+          language: { type: 'string', enum: ['js', 'ts', 'py', 'python', 'go', 'rust', 'java', 'c', 'cpp', 'sql'], description: 'Language hint' },
           maxDepth: { type: 'number', description: 'Max depth' },
           filePath: { type: 'string', description: 'File path to check focus (if in focus, returns full code)' },
           sessionId: { type: 'string', description: 'Session id for focus lookup' }
@@ -178,7 +188,8 @@ export function apply(ctx, config) {
       execute: async (params, meta) => {
         const cfg = getConfig();
         if (typeof cfg.budgetLimit === 'number') tracker.setBudgetLimit(cfg.budgetLimit);
-        const stats = tracker.getStats(cfg.budgetLimit);
+        if (typeof cfg.budgetAlertPercent === 'number') tracker.setBudgetAlertPercent(cfg.budgetAlertPercent);
+        const stats = tracker.getStats(cfg.budgetLimit, cfg.budgetAlertPercent);
         const history = tracker.getHistory();
         const key = sessionKey(params || {}, meta);
         return {
@@ -189,7 +200,8 @@ export function apply(ctx, config) {
           autoCollapse: cfg.autoCollapse !== false,
           focus: getFocus(key),
           sessionId: key,
-          autoCompressThreshold: cfg.autoCompressThreshold
+          autoCompressThreshold: cfg.autoCompressThreshold,
+          budgetAlertPercent: cfg.budgetAlertPercent ?? 90
         };
       }
     });
@@ -202,6 +214,7 @@ export function apply(ctx, config) {
       handler: (req, res) => {
         const cfg = getConfig();
         if (typeof cfg.budgetLimit === 'number') tracker.setBudgetLimit(cfg.budgetLimit);
+        if (typeof cfg.budgetAlertPercent === 'number') tracker.setBudgetAlertPercent(cfg.budgetAlertPercent);
         let sId = '__default__';
         try {
           const u = new URL(req.url, 'http://127.0.0.1');
@@ -211,7 +224,7 @@ export function apply(ctx, config) {
         res.end(JSON.stringify({
           ok: true,
           plugin: 'dsh-context-lens',
-          stats: tracker.getStats(cfg.budgetLimit),
+          stats: tracker.getStats(cfg.budgetLimit, cfg.budgetAlertPercent),
           history: tracker.getHistory(),
           focus: getFocus(sId),
           sessionId: sId
@@ -219,6 +232,23 @@ export function apply(ctx, config) {
       }
     }), 'dsh-context-lens status route');
 
+    // Quick clear-focus API route
+    ctx.effect(() => ctx.webServer.register({
+      kind: 'exact',
+      path: '/dsh-context-lens/clear-focus',
+      handler: async (req, res) => {
+        let sId = lastActiveSession || '__default__';
+        try {
+          const u = new URL(req.url, 'http://127.0.0.1');
+          const qId = u.searchParams.get('sessionId') || u.searchParams.get('session_id');
+          if (qId) sId = qId;
+        } catch {}
+        clearFocus(sId);
+        res.setHeader('content-type', 'application/json');
+        res.end(JSON.stringify({ ok: true, sessionId: sId, focus: getFocus(sId) }));
+      }
+    }), 'dsh-context-lens clear-focus route');
+
     // #45: server-backed preview (same compressor as tools)
     ctx.effect(() => ctx.webServer.register({
       kind: 'exact',
@@ -242,6 +272,4 @@ export function apply(ctx, config) {
       }
     }), 'dsh-context-lens compress-preview route');
   }
-}
-
-export { compressLog, skeletonize, shouldAutoCompress };
+}

package/lib/tokens/tracker.js

@@ -7,6 +7,7 @@ let calls = 0;
 const history = []; // last 10
 const MAX_HISTORY = 10;
 let defaultBudgetLimit = 100000; // tokens
+let defaultBudgetAlertPercent = 90; // threshold percent for low budget warning
 
 export { estimateTokens };
 
@@ -16,6 +17,12 @@ export function setBudgetLimit(limit) {
   }
 }
 
+export function setBudgetAlertPercent(percent) {
+  if (typeof percent === 'number' && percent >= 50 && percent <= 99) {
+    defaultBudgetAlertPercent = percent;
+  }
+}
+
 export function record(originalText, compressedText) {
   const o = estimateTokens(originalText);
   const c = estimateTokens(compressedText);
@@ -41,16 +48,19 @@ export function record(originalText, compressedText) {
   return { originalTokens: o, compressedTokens: c, savedTokens: saved };
 }
 
-export function getStats(budgetLimitOverride) {
+export function getStats(budgetLimitOverride, budgetAlertPercentOverride) {
   const limit = (typeof budgetLimitOverride === 'number' && budgetLimitOverride > 0)
     ? budgetLimitOverride
     : defaultBudgetLimit;
+  const alertPercent = (typeof budgetAlertPercentOverride === 'number' && budgetAlertPercentOverride >= 50 && budgetAlertPercentOverride <= 99)
+    ? budgetAlertPercentOverride
+    : defaultBudgetAlertPercent;
   const saved = Math.max(0, totalOriginal - totalCompressed);
   const pct = totalOriginal ? Math.round((1 - totalCompressed / totalOriginal) * 100) : 0;
   const budgetUsed = totalCompressed;
   const budgetRemaining = Math.max(0, limit - budgetUsed);
   const budgetPercent = Math.min(100, Math.round((budgetUsed / limit) * 100));
-  const lowBudget = budgetPercent > 90;
+  const lowBudget = budgetPercent >= alertPercent;
   return {
     totalOriginal,
     totalCompressed,
@@ -62,6 +72,7 @@ export function getStats(budgetLimitOverride) {
     budgetPercent,
     lowBudget,
     budgetLimit: limit,
+    budgetAlertPercent: alertPercent,
     countingStopped: budgetUsed >= limit
   };
 }
@@ -79,7 +90,4 @@ export function reset() {
   totalCompressed = 0;
   calls = 0;
   history.length = 0;
-  defaultBudgetLimit = 100000;
-}
-
-export default { estimateTokens, setBudgetLimit, record, getStats, getHistory, clearHistory, reset };
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@goodandready/dsh-context-lens",
-  "version": "0.1.15",
+  "version": "0.1.17",
   "description": "DSH plugin for AST context compression, test log filtering, and token budget guard",
   "type": "module",
   "main": "lib/index.js",
@@ -14,6 +14,8 @@
     "lib/",
     "cordis.patch.yml",
     "README.md",
+    "README.zh.md",
+    "README.ru.md",
     "LICENSE"
   ],
   "scripts": {