@0xdevabir/enhance 0.1.1 → 0.1.2

package/dist/index.js

@@ -602,7 +602,7 @@ function truncateLines(content, maxLines) {
 }
 
 // src/enhancer/injectors.ts
-function getInjections(stack, intent, config) {
+function getFallbackInjections(stack, intent, config) {
   const rules = [];
   if (stack.language === "typescript") {
     rules.push("Use TypeScript throughout \u2014 no `any`, prefer `unknown` with type guards");
@@ -680,6 +680,31 @@ function getInjections(stack, intent, config) {
   return rules;
 }
 
+// src/enhancer/angles.ts
+var ANGLE_LABELS = {
+  completeness: "Completeness",
+  production: "Production Safety",
+  dx: "Developer Experience",
+  alternative: "Alternative Approach"
+};
+var ANGLE_ORDER = ["completeness", "production", "dx", "alternative"];
+function getAngle(iteration) {
+  const idx = Math.min(iteration, ANGLE_ORDER.length - 1);
+  return ANGLE_ORDER[idx];
+}
+function getAngleFocus(angle) {
+  switch (angle) {
+    case "completeness":
+      return 'Cover every feature, sub-case, and edge case. Assume the reviewer will ask "what about X?" for every possible X. Nothing should be missing.';
+    case "production":
+      return "Focus on what breaks at 3am: failure modes, error handling, observability, security vulnerabilities, race conditions, data consistency, and retry semantics.";
+    case "dx":
+      return "Focus on implementation guidance: the correct step-by-step order, what confuses junior developers, integration pitfalls, what to verify first, and common mistakes.";
+    case "alternative":
+      return "Propose a meaningfully different architecture or pattern than the obvious approach. Explain the trade-off. Challenge the default solution.";
+  }
+}
+
 // src/enhancer/template.ts
 function buildEnhancedPrompt(opts) {
   const { intent } = opts;
@@ -692,9 +717,30 @@ function buildEnhancedPrompt(opts) {
       return buildFeaturePrompt(opts);
   }
 }
+function buildVersionHeader(opts) {
+  const { angle, iteration } = opts;
+  if (angle === void 0) return null;
+  const v = (iteration ?? 0) + 1;
+  const label = ANGLE_LABELS[angle];
+  return `\u{1F4CD} v${v} \u2014 ${label} angle`;
+}
+function buildAssumptionsSection(assumptions) {
+  if (!assumptions.length) return null;
+  return `## Assumptions
+
+${assumptions.map((a) => `- ${a}`).join("\n")}`;
+}
+function buildGotchasSection(gotchas) {
+  if (!gotchas.length) return null;
+  return `## Watch out for
+
+${gotchas.map((g) => `- ${g}`).join("\n")}`;
+}
 function buildSimplePrompt(opts) {
-  const { stack, intent, contextFiles, gitContext, injections, projectInstructions } = opts;
+  const { stack, intent, contextFiles, gitContext, injections, projectInstructions, assumptions = [], gotchas = [] } = opts;
   const sections = [];
+  const versionHeader = buildVersionHeader(opts);
+  if (versionHeader) sections.push(versionHeader);
   if (projectInstructions) {
     sections.push(`## Project Instructions
 
@@ -710,6 +756,10 @@ ${buildTaskDescription(intent)}`);
   sections.push(`## Requirements
 
 ${rules.map((r) => `- ${r}`).join("\n")}`);
+  const assumptionsSection = buildAssumptionsSection(assumptions);
+  if (assumptionsSection) sections.push(assumptionsSection);
+  const gotchasSection = buildGotchasSection(gotchas);
+  if (gotchasSection) sections.push(gotchasSection);
   if (gitContext) {
     sections.push(`## Recent Changes
 
@@ -726,8 +776,10 @@ ${buildOutputFormat(intent)}`);
   return sections.join("\n\n---\n\n");
 }
 function buildFeaturePrompt(opts) {
-  const { stack, structure, intent, contextFiles, gitContext, injections, projectInstructions } = opts;
+  const { stack, structure, intent, contextFiles, gitContext, injections, projectInstructions, assumptions = [], gotchas = [] } = opts;
   const sections = [];
+  const versionHeader = buildVersionHeader(opts);
+  if (versionHeader) sections.push(versionHeader);
   if (projectInstructions) {
     sections.push(`## Project Instructions
 
@@ -752,6 +804,10 @@ ${buildTaskDescription(intent)}`);
   sections.push(`## Requirements
 
 ${requirementLines}`);
+  const assumptionsSection = buildAssumptionsSection(assumptions);
+  if (assumptionsSection) sections.push(assumptionsSection);
+  const gotchasSection = buildGotchasSection(gotchas);
+  if (gotchasSection) sections.push(gotchasSection);
   if (gitContext) {
     sections.push(`## Recent Changes
 
@@ -768,8 +824,10 @@ ${buildOutputFormat(intent)}`);
   return sections.join("\n\n---\n\n");
 }
 function buildSystemPrompt(opts) {
-  const { stack, structure, intent, contextFiles, gitContext, injections, projectInstructions } = opts;
+  const { stack, structure, intent, contextFiles, gitContext, injections, projectInstructions, assumptions = [], gotchas = [] } = opts;
   const sections = [];
+  const versionHeader = buildVersionHeader(opts);
+  if (versionHeader) sections.push(versionHeader);
   if (projectInstructions) {
     sections.push(`## Project Instructions
 
@@ -794,6 +852,10 @@ ${buildTaskDescription(intent)}
   sections.push(`## Requirements
 
 ${requirementLines}`);
+  const assumptionsSection = buildAssumptionsSection(assumptions);
+  if (assumptionsSection) sections.push(assumptionsSection);
+  const gotchasSection = buildGotchasSection(gotchas);
+  if (gotchasSection) sections.push(gotchasSection);
   if (gitContext) {
     sections.push(`## Recent Changes
 
@@ -886,10 +948,97 @@ function capitalize(s) {
   return map[s] ?? s.charAt(0).toUpperCase() + s.slice(1);
 }
 
+// src/enhancer/domain-requirements.ts
+import Anthropic from "@anthropic-ai/sdk";
+async function generateDomainEnhancements(rawPrompt, intent, stack, angle, apiKey) {
+  const client = new Anthropic({ apiKey });
+  const stackSummary = [
+    stack.language === "typescript" ? "TypeScript" : "JavaScript",
+    ...stack.frameworks,
+    stack.orm ? `ORM:${stack.orm}` : null,
+    stack.uiLibrary ? `UI:${stack.uiLibrary}` : null
+  ].filter(Boolean).join(", ");
+  const system = `You are a senior software engineer specializing in ${intent.feature} systems.
+Generate implementation requirements, assumptions, and gotchas for a specific task.
+
+CRITICAL RULES for requirements:
+1. Every requirement must be SPECIFIC to this exact task \u2014 not generic boilerplate
+2. Ask: "Would this sentence appear in a prompt about a completely different feature?" If yes \u2192 discard
+3. Angle focus: ${getAngleFocus(angle)}
+
+REJECTED examples (too generic \u2014 never include):
+- "Use TypeScript throughout \u2014 no any"
+- "Handle errors explicitly"
+- "Keep components small and focused"
+- "Follow best practices"
+
+GOOD examples (specific, expert-level):
+- "Stripe webhooks fire multiple times \u2014 use idempotency keys keyed on event.id to prevent double-processing"
+- "Store payment_intent_id not charge_id \u2014 charge IDs change on retries"
+- "JWT refresh tokens must be rotated on every use to prevent token theft via stolen refresh token reuse"
+
+Return ONLY valid JSON \u2014 no markdown, no prose:
+{
+  "requirements": ["...", "..."],  // 10-13 specific items
+  "assumptions": ["Assumed ...", "Assumed ..."],  // 3-5 items starting with "Assumed"
+  "gotchas": ["gotcha description", "..."]  // 2-3 non-obvious things that WILL bite
+}`;
+  const user = `Task: "${rawPrompt}"
+Domain: ${intent.feature}
+Stack: ${stackSummary}
+Action: ${intent.action} | Entity: ${intent.entity}
+
+Generate requirements, assumptions, and gotchas. JSON only.`;
+  const response = await client.messages.create({
+    model: "claude-haiku-4-5-20251001",
+    max_tokens: 1536,
+    system,
+    messages: [{ role: "user", content: user }]
+  });
+  const text = response.content[0]?.type === "text" ? response.content[0].text : "{}";
+  try {
+    const parsed = JSON.parse(text);
+    if (Array.isArray(parsed.requirements) && parsed.requirements.length > 0) {
+      return {
+        requirements: parsed.requirements,
+        assumptions: Array.isArray(parsed.assumptions) ? parsed.assumptions : [],
+        gotchas: Array.isArray(parsed.gotchas) ? parsed.gotchas : []
+      };
+    }
+    throw new Error("empty requirements");
+  } catch {
+    const jsonMatch = text.match(/\{[\s\S]*\}/);
+    if (jsonMatch) {
+      const parsed = JSON.parse(jsonMatch[0]);
+      return {
+        requirements: Array.isArray(parsed.requirements) ? parsed.requirements : [],
+        assumptions: Array.isArray(parsed.assumptions) ? parsed.assumptions : [],
+        gotchas: Array.isArray(parsed.gotchas) ? parsed.gotchas : []
+      };
+    }
+    throw new Error("Failed to parse domain enhancements from AI response");
+  }
+}
+
 // src/enhancer/index.ts
-function enhance(rawPrompt, scan, contextFiles, projectInstructions = "", gitContext = "", intentOverride, config) {
+async function enhance(rawPrompt, scan, contextFiles, projectInstructions = "", gitContext = "", intentOverride, config, iteration = 0, _previousPrompt) {
   const intent = intentOverride ?? analyzeIntent(rawPrompt);
-  const injections = getInjections(scan.stack, intent, config);
+  const angle = getAngle(iteration);
+  let injections;
+  let assumptions = [];
+  let gotchas = [];
+  if (config?.apiKey) {
+    try {
+      const domain = await generateDomainEnhancements(rawPrompt, intent, scan.stack, angle, config.apiKey);
+      injections = domain.requirements;
+      assumptions = domain.assumptions;
+      gotchas = domain.gotchas;
+    } catch {
+      injections = getFallbackInjections(scan.stack, intent, config);
+    }
+  } else {
+    injections = getFallbackInjections(scan.stack, intent, config);
+  }
   const enhanced = buildEnhancedPrompt({
     stack: scan.stack,
     structure: scan.structure,
@@ -897,15 +1046,19 @@ function enhance(rawPrompt, scan, contextFiles, projectInstructions = "", gitCon
     contextFiles,
     injections,
     projectInstructions,
-    gitContext
+    gitContext,
+    angle,
+    assumptions,
+    gotchas,
+    iteration
   });
-  return { original: rawPrompt, enhanced, intent, stack: scan.stack };
+  return { original: rawPrompt, enhanced, intent, stack: scan.stack, iteration, angle: ANGLE_LABELS[angle] };
 }
 
 // src/enhancer/planner.ts
-import Anthropic from "@anthropic-ai/sdk";
+import Anthropic2 from "@anthropic-ai/sdk";
 async function decomposeToPlan(rawPrompt, intent, stack, apiKey) {
-  const client = new Anthropic({ apiKey });
+  const client = new Anthropic2({ apiKey });
   const stackSummary = `${stack.frameworks.join(", ")} / ${stack.language}`;
   const systemPrompt = `You are a software architect. Break complex development tasks into sequential, focused implementation steps.
 Each step should be small enough to implement in a single AI session.
@@ -952,7 +1105,7 @@ Rules:
 }
 
 // src/providers/claude.ts
-import Anthropic2 from "@anthropic-ai/sdk";
+import Anthropic3 from "@anthropic-ai/sdk";
 var MODEL_BY_COMPLEXITY = {
   simple: "claude-haiku-4-5-20251001",
   feature: "claude-sonnet-4-6",
@@ -969,7 +1122,7 @@ var ClaudeProvider = class {
         "ANTHROPIC_API_KEY environment variable is not set."
       );
     }
-    this.client = new Anthropic2({ apiKey: config.apiKey });
+    this.client = new Anthropic3({ apiKey: config.apiKey });
     this.model = config.model;
   }
   async send(prompt, options) {
@@ -1230,7 +1383,7 @@ Examples:
   enhance "fix the auth bug" --dry-run
   enhance "create dashboard" --confirm --verbose
   enhance "refactor user service" --provider codex
-  enhance "add auth" --action add --feature auth`).option("-p, --provider <name>", "AI provider: claude | codex | opencode").option("--dry-run", "Print enhanced prompt without sending to AI").option("--print-prompt", "Print enhanced prompt before sending to AI").option("--preview", "Color-coded prompt preview with per-section token counts").option("--verbose", "Show detected stack, intent, and context stats").option("--confirm", "Show detected intent and ask for approval before enhancing").option("--plan", "Decompose complex task into sequential sub-prompts (uses AI)").option("--action <action>", "Override detected action (create|fix|refactor|explain|add|delete)").option("--entity <entity>", "Override detected entity (page|component|api|hook|util|config|style)").option("--feature <feature>", "Override detected feature (auth|payment|user|upload|...)").action(async (rawPrompt, opts) => {
+  enhance "add auth" --action add --feature auth`).option("-p, --provider <name>", "AI provider: claude | codex | opencode").option("--dry-run", "Print enhanced prompt without sending to AI").option("--print-prompt", "Print enhanced prompt before sending to AI").option("--preview", "Color-coded prompt preview with per-section token counts").option("--verbose", "Show detected stack, intent, and context stats").option("--confirm", "Show detected intent and ask for approval before enhancing").option("--plan", "Decompose complex task into sequential sub-prompts (uses AI)").option("--iteration <n>", "Angle iteration: 0=completeness, 1=production, 2=dx, 3=alternative", "0").option("--action <action>", "Override detected action (create|fix|refactor|explain|add|delete)").option("--entity <entity>", "Override detected entity (page|component|api|hook|util|config|style)").option("--feature <feature>", "Override detected feature (auth|payment|user|upload|...)").action(async (rawPrompt, opts) => {
   if (!rawPrompt) {
     program.help();
     return;
@@ -1301,8 +1454,14 @@ Examples:
       spinner.start("Enhancing prompt...");
     }
     spinner.text = "Enhancing prompt...";
-    const enhanced = enhance(rawPrompt, scan, contextFiles, projectInstructions, gitContext, intent, config);
+    const iteration = Math.max(0, parseInt(opts.iteration ?? "0", 10) || 0);
+    const enhanced = await enhance(rawPrompt, scan, contextFiles, projectInstructions, gitContext, intent, config, iteration);
     spinner.stop();
+    if (enhanced.angle) {
+      console.log(chalk3.dim(`
+  \u{1F4CD} v${iteration + 1} \u2014 ${enhanced.angle} angle
+`));
+    }
     if (opts.preview) {
       printColoredPreview(enhanced.enhanced);
       if (opts.dryRun) process.exit(0);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@0xdevabir/enhance",
-  "version": "0.1.1",
+  "version": "0.1.2",
   "description": "AI prompt middleware — intelligently rewrites developer prompts before sending to coding agents",
   "bin": {
     "enhance": "./bin/enhance.js"