wiggum-cli 0.2.6 → 0.3.1

package/src/utils/colors.ts

@@ -72,16 +72,24 @@ export function drawLine(width: number = 50): string {
   return simpson.brown(box.horizontal.repeat(width));
 }
 
+/**
+ * Strip ANSI escape codes from text to get visible length
+ */
+function stripAnsi(text: string): string {
+  // eslint-disable-next-line no-control-regex
+  return text.replace(/\x1b\[[0-9;]*m/g, '');
+}
+
 /**
  * Draw a box around text
  */
 export function drawBox(text: string, padding: number = 1): string {
   const paddedText = ' '.repeat(padding) + text + ' '.repeat(padding);
-  const width = paddedText.length;
+  const visibleWidth = stripAnsi(paddedText).length;
 
-  const top = simpson.brown(box.topLeft + box.horizontal.repeat(width) + box.topRight);
+  const top = simpson.brown(box.topLeft + box.horizontal.repeat(visibleWidth) + box.topRight);
   const middle = simpson.brown(box.vertical) + paddedText + simpson.brown(box.vertical);
-  const bottom = simpson.brown(box.bottomLeft + box.horizontal.repeat(width) + box.bottomRight);
+  const bottom = simpson.brown(box.bottomLeft + box.horizontal.repeat(visibleWidth) + box.bottomRight);
 
   return `${top}\n${middle}\n${bottom}`;
 }

package/src/utils/header.ts

@@ -2,11 +2,11 @@ import cfonts from 'cfonts';
 import { simpson, drawBox, SIMPSON_COLORS } from './colors.js';
 
 /**
- * Display the RALPH ASCII header with welcome box
+ * Display the WIGGUM CLI ASCII header with welcome box
  */
 export function displayHeader(): void {
   // Welcome box like Claude Code
-  const welcomeText = simpson.pink('*') + ' Welcome to ' + simpson.yellow('Ralph') + '🍩';
+  const welcomeText = '🍩 Welcome to ' + simpson.yellow('Wiggum CLI') + ': AI-powered Ralph development loop CLI 🍩';
   console.log('');
   console.log(drawBox(welcomeText, 2));
   console.log('');
@@ -27,6 +27,6 @@ export function displayHeader(): void {
  */
 export function displayMinimalHeader(): void {
   console.log('');
-  console.log(simpson.yellow('Ralph') + simpson.brown(' │ ') + 'AI-powered loop development');
+  console.log(simpson.yellow('Wiggum CLI') + simpson.brown(' │ ') + 'AI-powered Ralph development loop');
   console.log('');
 }

package/src/utils/json-repair.ts

@@ -0,0 +1,113 @@
+/**
+ * JSON Repair Utility
+ * Fixes common JSON syntax errors from AI responses
+ */
+
+/**
+ * Attempt to repair malformed JSON from AI responses
+ */
+export function repairJson(text: string): string {
+  let json = text;
+
+  // Remove any leading/trailing whitespace
+  json = json.trim();
+
+  // Remove trailing commas before ] or }
+  json = json.replace(/,(\s*[\]}])/g, '$1');
+
+  // Fix missing commas between array elements or object properties
+  // Pattern: value followed by newline and another value without comma
+  json = json.replace(/("|\d|true|false|null|\]|\})(\s*\n\s*)("|\[|\{)/g, '$1,$2$3');
+
+  // Fix single quotes to double quotes (but not inside strings)
+  // This is a simplified approach - may not work for all cases
+  json = json.replace(/'/g, '"');
+
+  // Remove JavaScript-style comments
+  json = json.replace(/\/\/.*$/gm, '');
+  json = json.replace(/\/\*[\s\S]*?\*\//g, '');
+
+  // Fix unquoted keys (simple cases)
+  json = json.replace(/(\{|\,)\s*([a-zA-Z_][a-zA-Z0-9_]*)\s*:/g, '$1"$2":');
+
+  // Remove any text before the first { or [
+  const firstBrace = json.indexOf('{');
+  const firstBracket = json.indexOf('[');
+  let startIndex = -1;
+
+  if (firstBrace !== -1 && firstBracket !== -1) {
+    startIndex = Math.min(firstBrace, firstBracket);
+  } else if (firstBrace !== -1) {
+    startIndex = firstBrace;
+  } else if (firstBracket !== -1) {
+    startIndex = firstBracket;
+  }
+
+  if (startIndex > 0) {
+    json = json.substring(startIndex);
+  }
+
+  // Remove any text after the last } or ]
+  const lastBrace = json.lastIndexOf('}');
+  const lastBracket = json.lastIndexOf(']');
+  let endIndex = -1;
+
+  if (lastBrace !== -1 && lastBracket !== -1) {
+    endIndex = Math.max(lastBrace, lastBracket);
+  } else if (lastBrace !== -1) {
+    endIndex = lastBrace;
+  } else if (lastBracket !== -1) {
+    endIndex = lastBracket;
+  }
+
+  if (endIndex !== -1 && endIndex < json.length - 1) {
+    json = json.substring(0, endIndex + 1);
+  }
+
+  return json;
+}
+
+/**
+ * Parse JSON with repair attempts
+ * Tries to fix common issues before parsing
+ */
+export function parseJsonSafe<T>(text: string): T | null {
+  // First try parsing as-is
+  try {
+    return JSON.parse(text) as T;
+  } catch {
+    // Try with repairs
+  }
+
+  // Try repairing the JSON
+  try {
+    const repaired = repairJson(text);
+    return JSON.parse(repaired) as T;
+  } catch {
+    // Repair failed
+  }
+
+  // Last resort: try to extract JSON from markdown code blocks
+  const jsonMatch = text.match(/```(?:json)?\s*([\s\S]*?)```/);
+  if (jsonMatch) {
+    try {
+      const repaired = repairJson(jsonMatch[1]);
+      return JSON.parse(repaired) as T;
+    } catch {
+      // Still failed
+    }
+  }
+
+  // Try finding a JSON object
+  const objectMatch = text.match(/\{[\s\S]*\}/);
+  if (objectMatch) {
+    try {
+      const repaired = repairJson(objectMatch[0]);
+      return JSON.parse(repaired) as T;
+    } catch {
+      // Still failed
+    }
+  }
+
+  return null;
+}

package/src/utils/tracing.ts

@@ -0,0 +1,76 @@
+/**
+ * Braintrust Tracing Utility
+ * Provides AI call tracing for debugging and analysis
+ */
+
+import { initLogger, wrapAISDK } from 'braintrust';
+import * as ai from 'ai';
+
+// Re-export traced utilities
+export { traced, currentSpan, wrapTraced } from 'braintrust';
+
+/**
+ * Initialize Braintrust logger if API key is available
+ */
+let loggerInitialized = false;
+
+export function initTracing(): void {
+  if (loggerInitialized) return;
+
+  const apiKey = process.env.BRAINTRUST_API_KEY;
+  if (!apiKey) {
+    // Silently skip tracing if no API key
+    return;
+  }
+
+  try {
+    initLogger({
+      apiKey,
+      projectName: process.env.BRAINTRUST_PROJECT_NAME || 'wiggum-cli',
+    });
+    loggerInitialized = true;
+  } catch {
+    // Silently fail if tracing can't be initialized
+  }
+}
+
+/**
+ * Check if tracing is enabled
+ */
+export function isTracingEnabled(): boolean {
+  return !!process.env.BRAINTRUST_API_KEY;
+}
+
+/**
+ * Get wrapped AI SDK functions for automatic tracing
+ * Falls back to original functions if tracing not available
+ */
+export function getTracedAI() {
+  initTracing();
+
+  if (isTracingEnabled()) {
+    return wrapAISDK(ai);
+  }
+
+  // Return original AI SDK functions if tracing not enabled
+  return ai;
+}
+
+/**
+ * Wrap a function with tracing
+ * No-op if tracing is not enabled
+ */
+export function maybeTraced<T extends (...args: unknown[]) => unknown>(
+  fn: T,
+  options: { type?: string; name?: string } = {}
+): T {
+  if (!isTracingEnabled()) {
+    return fn;
+  }
+
+  const { wrapTraced } = require('braintrust');
+  return wrapTraced(fn, {
+    type: options.type || 'function',
+    name: options.name || fn.name || 'anonymous',
+  });
+}