composto-ai 0.2.0 → 0.2.1

package/dist/{chunk-MFNGSHTZ.js → chunk-T2SFR7QS.js}

@@ -499,36 +499,89 @@ function detectLanguage(filePath) {
 
 // src/ir/ast-walker.ts
 var TIER_MAP = {
-  // Tier 1 — structural declarations
+  // Tier 1 — structural declarations (JS/TS)
   import_statement: "T1_KEEP",
   function_declaration: "T1_KEEP",
   class_declaration: "T1_KEEP",
   interface_declaration: "T1_KEEP",
   type_alias_declaration: "T1_KEEP",
   enum_declaration: "T1_KEEP",
-  // Tier 2 — control flow
+  // Tier 1 — Python
+  function_definition: "T1_KEEP",
+  class_definition: "T1_KEEP",
+  import_from_statement: "T1_KEEP",
+  decorated_definition: "T1_KEEP",
+  // Tier 1 — Go
+  function_item: "T1_KEEP",
+  // Rust
+  method_declaration: "T1_KEEP",
+  // Go
+  type_declaration: "T1_KEEP",
+  // Go
+  import_declaration: "T1_KEEP",
+  // Go
+  use_declaration: "T1_KEEP",
+  // Rust
+  struct_item: "T1_KEEP",
+  // Rust
+  enum_item: "T1_KEEP",
+  // Rust
+  trait_item: "T1_KEEP",
+  // Rust
+  impl_item: "T1_KEEP",
+  // Rust
+  // Tier 2 — control flow (universal)
   if_statement: "T2_CONTROL",
+  if_expression: "T2_CONTROL",
+  // Rust
   else_clause: "WALK_ONLY",
+  elif_clause: "T2_CONTROL",
+  // Python
   for_statement: "T2_CONTROL",
   for_in_statement: "T2_CONTROL",
+  for_expression: "T2_CONTROL",
+  // Rust
   while_statement: "T2_CONTROL",
   do_statement: "T2_CONTROL",
   switch_statement: "T2_CONTROL",
   switch_case: "T2_CONTROL",
   switch_default: "T2_CONTROL",
+  match_expression: "T2_CONTROL",
+  // Rust
   return_statement: "T2_CONTROL",
+  return_expression: "T2_CONTROL",
+  // Rust
   throw_statement: "T2_CONTROL",
+  raise_statement: "T2_CONTROL",
+  // Python
   try_statement: "T2_CONTROL",
   catch_clause: "T2_CONTROL",
+  except_clause: "T2_CONTROL",
+  // Python
+  with_statement: "T2_CONTROL",
+  // Python
+  defer_statement: "T2_CONTROL",
+  // Go
   // Tier 3 — compressible expressions
   lexical_declaration: "T3_COMPRESS",
   expression_statement: "T3_COMPRESS",
-  // Walk-only — containers that need traversal but no emission
+  assignment: "T3_COMPRESS",
+  // Python
+  short_var_declaration: "T3_COMPRESS",
+  // Go
+  // Walk-only — containers
   program: "WALK_ONLY",
+  module: "WALK_ONLY",
+  // Python
   statement_block: "WALK_ONLY",
+  block: "WALK_ONLY",
+  // Python/Go/Rust
   class_body: "WALK_ONLY",
+  // JS/TS
   switch_body: "WALK_ONLY",
-  export_statement: "WALK_ONLY"
+  export_statement: "WALK_ONLY",
+  source_file: "WALK_ONLY"
+  // Go/Rust
 };
 function tierOf(nodeType) {
   return TIER_MAP[nodeType] ?? "T4_DROP";
@@ -644,6 +697,35 @@ function emitTier2(node) {
       const paramText = param ? param.text : "...";
       return `CATCH:${paramText}`;
     }
+    // Python
+    case "raise_statement": {
+      const val = node.childCount > 1 ? node.child(1)?.text ?? "" : "";
+      return `RAISE:${val.length > 50 ? val.slice(0, 47) + "..." : val}`;
+    }
+    case "except_clause":
+      return "EXCEPT";
+    case "elif_clause": {
+      const cond = extractCondition(node);
+      return `ELIF:${cond}`;
+    }
+    case "with_statement":
+      return "WITH";
+    // Rust
+    case "if_expression": {
+      const cond = extractCondition(node);
+      return `IF:${cond}`;
+    }
+    case "for_expression":
+      return "LOOP";
+    case "match_expression":
+      return "MATCH";
+    case "return_expression": {
+      const val = node.childCount > 1 ? node.child(1)?.text ?? "" : "";
+      return `RET ${val.length > 60 ? val.slice(0, 57) + "..." : val}`.trimEnd();
+    }
+    // Go
+    case "defer_statement":
+      return "DEFER";
     default:
       return null;
   }
@@ -681,6 +763,46 @@ function emitTier1(node) {
       const name = node.childForFieldName("name")?.text ?? "Anonymous";
       return `${outPrefix}ENUM:${name}`;
     }
+    // Python
+    case "function_definition": {
+      const name = node.childForFieldName("name")?.text ?? "anonymous";
+      const params = node.childForFieldName("parameters")?.text ?? "()";
+      const returnType = node.childForFieldName("return_type")?.text ?? "";
+      const rt = returnType ? ` -> ${returnType}` : "";
+      return `FN:${name}${collapseText(params, 60)}${rt}`;
+    }
+    case "class_definition": {
+      const name = node.childForFieldName("name")?.text ?? "Anonymous";
+      const superclass = node.childForFieldName("superclasses")?.text ?? "";
+      const sc = superclass ? `(${collapseText(superclass, 40)})` : "";
+      return `CLASS:${name}${sc}`;
+    }
+    case "import_from_statement": {
+      return `USE:${collapseText(node.text, 80)}`;
+    }
+    case "decorated_definition": {
+      return null;
+    }
+    // Go
+    case "method_declaration":
+    case "type_declaration":
+    case "import_declaration": {
+      return `${collapseText(node.text, 80)}`;
+    }
+    // Rust
+    case "function_item": {
+      const name = node.childForFieldName("name")?.text ?? "anonymous";
+      const params = node.childForFieldName("parameters")?.text ?? "()";
+      return `FN:${name}${collapseText(params, 60)}`;
+    }
+    case "struct_item":
+    case "enum_item":
+    case "trait_item":
+    case "impl_item":
+    case "use_declaration": {
+      const firstLine = node.text.split("\n")[0];
+      return collapseText(firstLine, 80);
+    }
     default:
       return null;
   }
@@ -726,6 +848,17 @@ function emitTier3(node) {
         return null;
       }
       if (expr.type === "call_expression") {
+        const callee = expr.child(0)?.text ?? "";
+        if (callee === "ObjectSetPrototypeOf" || callee === "Object.setPrototypeOf") {
+          const args = expr.child(1);
+          if (args && args.childCount >= 4) {
+            const child = args.child(1)?.text ?? "?";
+            const parent = args.child(3)?.text ?? "?";
+            const shortChild = child.length > 30 ? child.slice(0, 27) + "..." : child;
+            const shortParent = parent.length > 30 ? parent.slice(0, 27) + "..." : parent;
+            return `EXTENDS:${shortChild} < ${shortParent}`;
+          }
+        }
         return null;
       }
       return null;
@@ -743,7 +876,9 @@ function walkNode(node, depth, lines) {
       for (let i = 0; i < node.childCount; i++) {
         const child = node.child(i);
         const childType = child.type;
-        if (childType === "statement_block" || childType === "class_body") {
+        if (childType === "statement_block" || childType === "class_body" || childType === "block" || // Python/Go/Rust
+        childType === "body" || // Python class/function body
+        childType === "declaration_list") {
           walkNode(child, depth + 1, lines);
         }
       }
@@ -834,7 +969,7 @@ async function astWalkIR(code, filePath) {
   const lines = [];
   walkNode(root, 0, lines);
   if (lines.length === 0) return null;
-  const merged = [];
+  const pass1 = [];
   let useBlock = [];
   for (const line of lines) {
     if (line.startsWith("USE:")) {
@@ -843,20 +978,45 @@ async function astWalkIR(code, filePath) {
     } else {
       if (useBlock.length > 0) {
         if (useBlock.length <= 3) {
-          for (const mod of useBlock) merged.push(`USE:${mod}`);
+          for (const mod of useBlock) pass1.push(`USE:${mod}`);
         } else {
-          merged.push(`USE:[${useBlock.join(", ")}]`);
+          pass1.push(`USE:[${useBlock.join(", ")}]`);
         }
         useBlock = [];
       }
-      merged.push(line);
+      pass1.push(line);
     }
   }
   if (useBlock.length > 0) {
     if (useBlock.length <= 3) {
-      for (const mod of useBlock) merged.push(`USE:${mod}`);
+      for (const mod of useBlock) pass1.push(`USE:${mod}`);
     } else {
-      merged.push(`USE:[${useBlock.join(", ")}]`);
+      pass1.push(`USE:[${useBlock.join(", ")}]`);
+    }
+  }
+  const merged = [];
+  let guardBlock = [];
+  for (const line of pass1) {
+    const guardMatch = line.match(/^(\s*)IF:(.+?) \u2192 RET/);
+    if (guardMatch) {
+      guardBlock.push(guardMatch[2].trim());
+      continue;
+    }
+    if (guardBlock.length > 0) {
+      if (guardBlock.length < 3) {
+        for (const g of guardBlock) merged.push(`  IF:${g} \u2192 RET`);
+      } else {
+        merged.push(`  GUARD:[${guardBlock.join(", ")}]`);
+      }
+      guardBlock = [];
+    }
+    merged.push(line);
+  }
+  if (guardBlock.length > 0) {
+    if (guardBlock.length < 3) {
+      for (const g of guardBlock) merged.push(`  IF:${g} \u2192 RET`);
+    } else {
+      merged.push(`  GUARD:[${guardBlock.join(", ")}]`);
     }
   }
   return merged.join("\n");
@@ -879,10 +1039,11 @@ ${declarations.join("\n")}`;
 }
 async function generateL1(code, filePath, health) {
   const ir = await astWalkIR(code, filePath) ?? fingerprintFile(code, 0.75);
+  const result = ir.length < code.length ? ir : code;
   if (health) {
-    return annotateIR(ir, health);
+    return annotateIR(result, health);
   }
-  return ir;
+  return result;
 }
 function generateL2(delta, health) {
   const parts = [`FILE: ${delta.file}`];

package/dist/index.js

@@ -12,7 +12,7 @@ import {
   packContext,
   runDetector,
   summarize
-} from "./chunk-MFNGSHTZ.js";
+} from "./chunk-T2SFR7QS.js";
 
 // src/cli/commands.ts
 import { readFileSync, readdirSync } from "fs";

package/dist/mcp/server.js

@@ -12,7 +12,7 @@ import {
   packContext,
   runDetector,
   summarize
-} from "../chunk-MFNGSHTZ.js";
+} from "../chunk-T2SFR7QS.js";
 
 // src/mcp/server.ts
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "composto-ai",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Proactive AI team companion — less tokens, more insight",
   "type": "module",
   "bin": {