@semalt-ai/code 1.7.0 → 1.8.1

package/lib/ui/ansi.js

@@ -0,0 +1,65 @@
+'use strict';
+
+const RST  = '\x1b[0m';
+const BOLD = '\x1b[1m';
+const DIM  = '\x1b[2m';
+
+const THEME = {
+  user:  '\x1b[36m',
+  agent: '\x1b[32m',
+  sys:   '\x1b[33m',
+  error: '\x1b[31;1m',
+  warn:  '\x1b[33;1m',
+  tool:  '\x1b[35m',
+  dim:   '\x1b[2m',
+  reset: '\x1b[0m',
+};
+
+const FG_GRAY   = '\x1b[38;5;245m';
+const FG_DARK   = '\x1b[38;5;240m';
+const FG_BLUE   = '\x1b[38;5;75m';
+const FG_CYAN   = '\x1b[38;5;116m';
+const FG_GREEN  = '\x1b[38;5;114m';
+const FG_YELLOW = '\x1b[38;5;222m';
+const FG_RED    = '\x1b[38;5;203m';
+const FG_TEAL   = '\x1b[38;5;73m';
+
+const BOX_H  = '─';
+const BOX_V  = '│';
+const BOX_TL = '╭';
+const BOX_TR = '╮';
+const BOX_BL = '╰';
+const BOX_BR = '╯';
+
+const FG_CODE_BG     = '\x1b[48;5;236m';
+const BG_SELECTED    = '\x1b[48;5;237m';
+const FG_CODE_BORDER = '\x1b[38;5;240m';
+const FG_CODE_LANG   = '\x1b[38;5;75m';
+const FG_TAG         = '\x1b[38;5;176m';
+const FG_FILEPATH    = '\x1b[38;5;222m';
+
+const KEYWORDS = new Set([
+  'def', 'class', 'import', 'from', 'return', 'if', 'else', 'elif',
+  'for', 'while', 'try', 'except', 'finally', 'with', 'as', 'in',
+  'not', 'and', 'or', 'is', 'None', 'True', 'False', 'async', 'await',
+  'function', 'const', 'let', 'var', 'export', 'require', 'func',
+  'type', 'struct', 'interface', 'package', 'fn', 'pub', 'use',
+  'mod', 'impl', 'match', 'enum', 'self', 'print', 'len', 'range',
+  'yield', 'lambda', 'raise', 'pass', 'break', 'continue', 'del',
+  'global', 'nonlocal', 'assert',
+]);
+
+const SPINNER_DEFS = {
+  thinking:          { frames: ['⠋','⠙','⠹','⠸','⠼','⠴','⠦','⠧','⠇','⠏'], color: '\x1b[36m' },
+  streaming:         { frames: ['▁','▂','▃','▄','▅','▆','▇','█','▇','▆','▅','▄','▃','▂'], color: '\x1b[32m' },
+  tool:              { frames: ['⣾','⣽','⣻','⢿','⡿','⣟','⣯','⣷'], color: '\x1b[33m' },
+  waiting_download:  { frames: ['⬇ ','⬇⠂','⬇⠆','⬇⠇','⬇⠧','⬇⠷','⬇⠿','⬇⠾','⬇⠼','⬇⠸','⬇⠰','⬇⠠'], color: '\x1b[38;5;75m' },
+};
+
+module.exports = {
+  RST, BOLD, DIM, THEME,
+  FG_GRAY, FG_DARK, FG_BLUE, FG_CYAN, FG_GREEN, FG_YELLOW, FG_RED, FG_TEAL,
+  BOX_H, BOX_V, BOX_TL, BOX_TR, BOX_BL, BOX_BR,
+  FG_CODE_BG, BG_SELECTED, FG_CODE_BORDER, FG_CODE_LANG, FG_TAG, FG_FILEPATH,
+  KEYWORDS, SPINNER_DEFS,
+};

package/lib/ui/chat-history.js

@@ -0,0 +1,217 @@
+'use strict';
+
+const { RST, DIM, BOLD, FG_CYAN, FG_GREEN, FG_YELLOW, FG_RED, FG_DARK, FG_GRAY } = require('./ansi');
+const { getCols, stripAnsi } = require('./utils');
+
+
+function safeContent(text) {
+  if (!text) return '';
+  return text.replace(/<\/?[a-zA-Z_][a-zA-Z0-9_]*(\s[^>]*)?>/g, '');
+}
+
+const TOOL_ICONS = {
+  exec: '▶', shell: '▶', run_command: '▶', run: '▶',
+  read_file: '[R]', write_file: '✎',
+  list_dir: '[D]', search_files: '[D]',
+  http_get: '↓', download: '↓',
+  ask_user: '?',
+  store_memory: '◆', recall_memory: '◆', list_memories: '◆',
+};
+
+function _fmtTime(ts) {
+  const d = ts instanceof Date ? ts : (ts ? new Date(ts) : new Date());
+  return `${String(d.getHours()).padStart(2, '0')}:${String(d.getMinutes()).padStart(2, '0')}`;
+}
+
+const BG_USER = '\x1b[48;5;237m';
+
+function _printUser(content, ts) {
+  const time = _fmtTime(ts);
+  process.stdout.write(`\n${FG_CYAN}▸ You${RST}  ${DIM}${time}${RST}\n`);
+  for (const line of (content || '').split('\n')) {
+    process.stdout.write(`${BG_USER}  ${line}\x1b[K${RST}\n`);
+  }
+  process.stdout.write('\n');
+}
+
+function _printAI(content, ts) {
+  const time = _fmtTime(ts);
+  process.stdout.write(`\n${FG_GREEN}▸ AI-agent${RST}  ${DIM}${time}${RST}\n`);
+  for (const line of (content || '').split('\n')) {
+    process.stdout.write(`  ${line}\n`);
+  }
+  process.stdout.write('\n');
+}
+
+const MAX_TOOL_DISPLAY = 15;
+
+class ChatHistory {
+  constructor() {
+    this._streamWritten = false;
+    this._streamStart = null;
+    this._msgById = {};
+    this._msgLineCount = {};
+    this._pendingWs = '';
+    this._didStream = false;
+    this._toolExpanded = {};
+    this._lastExpandableId = null;
+    this._toolIdCounter = 0;
+    this._messages = [];
+  }
+
+  _flush() {
+    this._pendingWs = '';
+    if (this._streamWritten) {
+      process.stdout.write('\n\n');
+      this._streamWritten = false;
+      this._streamStart = null;
+      this._didStream = true;
+    }
+  }
+
+  addMessage(msg) {
+    if (msg.role === 'think') return;
+    const content = safeContent(msg.content || '');
+    if ((msg.role === 'assistant' || msg.role === 'system') && !content.trim()) return;
+
+    this._flush();
+
+    if (msg.id) this._msgById[msg.id] = msg;
+
+    if (msg.role === 'user') {
+      _printUser(content, msg.ts);
+    } else if (msg.role === 'assistant') {
+      _printAI(content, msg.ts);
+    } else if (msg.role === 'shell') {
+      const cmd = msg.cmd || '';
+      const out = safeContent(msg.content || '');
+      process.stdout.write(`\n${DIM}  $ ${cmd}${RST}\n`);
+      if (out.trim()) {
+        for (const line of out.split('\n')) {
+          if (line.trim()) process.stdout.write(`${DIM}    ${line}${RST}\n`);
+        }
+      }
+      process.stdout.write('\n');
+    } else if (msg.role === 'tool') {
+      const icon = TOOL_ICONS[msg.tag] || '⚙';
+      process.stdout.write(`${DIM}  ${icon}  ${content}${RST}\n`);
+      let lc = 1;
+      if (msg.output) {
+        // Wrap long lines so char-heavy single-line output (e.g. minified HTML)
+        // is treated the same as multi-line output for truncation purposes.
+        const wrapAt = Math.max(60, getCols() - 8);
+        const outLines = [];
+        for (const raw of msg.output.split('\n')) {
+          if (!raw.trim()) continue;
+          if (raw.length <= wrapAt) {
+            outLines.push(raw);
+          } else {
+            for (let i = 0; i < raw.length; i += wrapAt) {
+              outLines.push(raw.slice(i, i + wrapAt));
+            }
+          }
+        }
+        const truncatable = outLines.length > MAX_TOOL_DISPLAY;
+
+        if (truncatable && !msg.id) {
+          msg.id = `tool-${++this._toolIdCounter}`;
+        }
+        if (truncatable && msg.id) {
+          this._msgById[msg.id] = msg;
+          this._lastExpandableId = msg.id;
+        }
+
+        const expanded = truncatable && msg.id && this._toolExpanded[msg.id];
+        const visible = (truncatable && !expanded) ? outLines.slice(0, MAX_TOOL_DISPLAY) : outLines;
+        const remaining = outLines.length - visible.length;
+
+        for (const ol of visible) { process.stdout.write(`${DIM}     ${ol}${RST}\n`); lc++; }
+        if (remaining > 0) {
+          process.stdout.write(`${DIM}     … ${remaining} more lines  ${FG_DARK}(ctrl+o to expand ${remaining} rows)${RST}\n`);
+          lc++;
+        } else if (truncatable && expanded) {
+          process.stdout.write(`${DIM}     ${FG_DARK}(ctrl+o to collapse)${RST}\n`);
+          lc++;
+        }
+      }
+      if (msg.id) this._msgLineCount[msg.id] = lc;
+    } else if (msg.role === 'permission') {
+      process.stdout.write(`\n${FG_YELLOW}  ⚠ Permission required: ${content}${RST}\n`);
+    } else {
+      const isErr = !!msg.isError && !msg.isWarning;
+      const color = isErr ? FG_RED : FG_YELLOW;
+      const prefix = isErr ? '✕' : '⚠';
+      const lines = content.split('\n');
+      process.stdout.write(`${color}  ${prefix}  ${lines[0]}${RST}\n`);
+      for (let i = 1; i < lines.length; i++) {
+        process.stdout.write(`${color}     ${lines[i]}${RST}\n`);
+      }
+      if (msg.id) this._msgLineCount[msg.id] = lines.length;
+    }
+  }
+
+  streamToken(token) {
+    if (!this._streamWritten) {
+      if (!token.trim()) {
+        this._pendingWs += token;
+        return;
+      }
+      const time = _fmtTime(new Date());
+      process.stdout.write(`\n${FG_GREEN}▸ AI-agent${RST}  ${DIM}${time}${RST}\n  `);
+      this._streamWritten = true;
+      this._streamStart = new Date();
+      this._pendingWs = '';
+    }
+    process.stdout.write(token.replace(/\n/g, '\n  '));
+  }
+
+  clearStreamingContent() {
+    this._flush();
+  }
+
+  finalizeLastMessage(cleanContent) {
+    if (this._streamWritten) {
+      this._flush();
+    } else if (!this._didStream && cleanContent && cleanContent.trim()) {
+      _printAI(cleanContent, new Date());
+    }
+    this._didStream = false;
+  }
+
+  clearMessages() {
+    this._streamWritten = false;
+    this._pendingWs = '';
+    this._didStream = false;
+    this._msgById = {};
+    this._msgLineCount = {};
+    this._toolExpanded = {};
+    this._lastExpandableId = null;
+    this._messages = [];
+  }
+
+  toggleLastExpand() {
+    const id = this._lastExpandableId;
+    if (!id) return;
+    const msg = this._msgById[id];
+    if (!msg) return;
+    this._toolExpanded[id] = !this._toolExpanded[id];
+    this.addMessage(msg);
+  }
+
+  rerenderById(id) {
+    const msg = this._msgById[id];
+    if (!msg) return;
+    this.addMessage(msg);
+  }
+
+  removeById(id) {
+    delete this._msgById[id];
+    delete this._msgLineCount[id];
+  }
+
+  invalidateCache() {}
+  scrollUp() {}
+  scrollDown() {}
+}
+
+module.exports = { ChatHistory, safeContent, TOOL_ICONS };