zames_pro 1.0.3 → 1.1.1

package/src/index.js

@@ -1,9 +1,8 @@
 #!/usr/bin/env node
 import path from 'path'
 import fs from 'fs/promises'
-import chalk from 'chalk'
 import { fileURLToPath } from 'url'
-import * as readlinePromises from 'readline/promises'
+import { theme } from './theme.js'
 
 import { DeepSeekBrowser } from './browser.js'
 import { createTools } from './tools.js'
@@ -61,8 +60,10 @@ const chatIdArg = getArg('--chat', null)
 // заставляет дослать его заново — например, если промпт обновился.
 const resendPrompt = hasFlag('--resend-prompt')
 // --new-chat: намеренно начать с чистого чата, игнорируя сохранённый
-// last-chat (полезно в watch-режиме, где по умолчанию чат восстанавливается).
-const newChatFlag = hasFlag('--new-chat')
+// --resume-last: при старте вернуться в последний сохранённый чат.
+// По умолчанию НЕ восстанавливаем — открывается новый чат.
+// (Флаг --new-chat сохранён для совместимости и ничего не меняет.)
+const resumeLastFlag = hasFlag('--resume-last')
 
 // ---------- last chat persistence ----------
 // Запоминаем последний открытый chat id, чтобы после перезапуска процесса
@@ -171,9 +172,9 @@ async function autoReload() {
   const { errors } = await reloadModules()
   if (errors.length) {
     console.error(
-      chalk.yellow('⚠ авто-reload: часть модулей не загрузилась, работаю на прежней версии:'),
+      theme.warn('⚠ авто-reload: часть модулей не загрузилась, работаю на прежней версии:'),
     )
-    for (const e of errors) console.error(chalk.yellow('  ' + e))
+    for (const e of errors) console.error(theme.warn('  ' + e))
   }
 }
 
@@ -182,13 +183,14 @@ async function autoReload() {
 
 function printHelp() {
   console.log(`
-${chalk.bold('zames')} — агент поверх chat.deepseek.com через Playwright
+${theme.bold('zames')} — агент поверх chat.deepseek.com через Playwright
 
-${chalk.bold('Опции CLI:')}
+${theme.bold('Опции CLI:')}
   --dir <path>       рабочая директория агента
   --task <text>      задача одной строкой
   --chat <id>        продолжить существующий чат по id
-  --new-chat         начать новый чат (игнорировать сохранённый)
+  --resume-last      вернуться в последний сохранённый чат
+  --new-chat         начать новый чат (по умолчанию и так новый)
   --resend-prompt    дослать system-prompt в существующий чат
   --max-iter <n>     лимит итераций (по умолчанию ${config.maxIterations})
   --headless         браузер без UI
@@ -198,7 +200,7 @@ ${chalk.bold('Опции CLI:')}
   --version, -v      показать версию
   --help, -h         эта справка
 
-${chalk.bold('Обычные команды:')}
+${theme.bold('Обычные команды:')}
   /new, /clear             новый чат (сброс контекста)
   /chats                   список последних чатов DeepSeek
   /resume <n>              открыть чат №n из /chats
@@ -215,7 +217,7 @@ ${chalk.bold('Обычные команды:')}
   /help, help              справка
   /exit, /quit, exit       выход
 
-${chalk.bold('Самообзор (отладка агента):')}
+${theme.bold('Самообзор (отладка агента):')}
   /self-review [фокус]     снять снапшот src/ и запустить ревью
                             после этой команды ты остаёшься В СНАПШОТЕ
                             и можешь писать «исправь ошибки» и т.п.
@@ -225,11 +227,12 @@ ${chalk.bold('Самообзор (отладка агента):')}
   /self-diff <name>        различия между текущим src/ и снапшотом
   /self-apply <name>       применить снапшот к src/ (с бэкапом)
 
-${chalk.bold('Файлы:')}
+${theme.bold('Файлы:')}
   Логи:        ${config.transcript.dir}
   Undo:        ~/.zames/undo
   Профиль:     ~/.zames/profile
   Снапшоты:    ~/.zames/snapshots
+  Временные:   <проект>/tmp (в .gitignore, чистится при запуске)
   Конфиг:      ${CONFIG_PATHS.HOME_CONFIG}
                ${CONFIG_PATHS.PROJECT_CONFIG}
 `)
@@ -239,19 +242,244 @@ function dirLabel(p) {
   return path.basename(p) || p
 }
 
+// Временные файлы агента (одноразовые скрипты и т.п.) складываем в
+// <проект>/tmp — эта папка в .gitignore и очищается при каждом запуске.
+const TMP_DIR = path.join(__dirname, '..', 'tmp')
+
+async function cleanTmpDir() {
+  try {
+    await fs.rm(TMP_DIR, { recursive: true, force: true })
+    await fs.mkdir(TMP_DIR, { recursive: true })
+  } catch (e) {
+    if (debug) console.error('tmp: не удалось очистить:', e.message)
+  }
+}
+
+// Ввод строки в терминале с корректной обработкой вставки (Shift+Insert,
+// Ctrl+Shift+V, правая кнопка мыши и т.п.).
+//
+// Зачем свой ридер вместо readline:
+//   1) readline отправляет строку на ПЕРВОМ переводе строки. При вставке
+//      многострочного текста это приводило к немедленной отправке и к тому,
+//      что в чат уходила только первая строка. Здесь переводы строк внутри
+//      вставки заменяются на пробелы, а отправка происходит только по
+//      одиночному нажатию Enter.
+//   2) Включаем bracketed paste mode (\x1b[?2004h): терминал оборачивает
+//      вставленный текст в маркеры \x1b[200~ … \x1b[201~, поэтому мы точно
+//      знаем, что это вставка, а не набор с клавиатуры, и Enter внутри неё
+//      не считается отправкой.
 async function promptOnce(question) {
-  process.stdin.resume()
-  if (process.stdin.isTTY && process.stdin.setRawMode) {
-    process.stdin.setRawMode(false)
+  const stdin = process.stdin
+  const stdout = process.stdout
+
+  // Не-TTY (пайп, редирект): читаем всё до EOF одной строкой.
+  if (!stdin.isTTY || !stdin.setRawMode) {
+    const chunks = []
+    return await new Promise((resolve) => {
+      const onData = (b) => chunks.push(b)
+      const onEnd = () => {
+        stdin.removeListener('data', onData)
+        stdin.removeListener('end', onEnd)
+        resolve(Buffer.concat(chunks).toString('utf-8'))
+      }
+      stdin.on('data', onData)
+      stdin.on('end', onEnd)
+      stdin.resume()
+    })
+  }
+
+  const wasRaw = stdin.isRaw
+  stdin.setRawMode(true)
+  stdin.resume()
+
+  // Bracketed paste включаем/выключаем парно.
+  stdout.write('\x1b[?2004h')
+  stdout.write(question)
+
+  let line = ''
+  let cursor = 0
+  let inPaste = false
+  const PASTE_START = '\x1b[200~'
+  const PASTE_END = '\x1b[201~'
+
+  const redraw = () => {
+    // Возвращаемся в начало строки, стираем и печатаем заново.
+    stdout.write(String.fromCharCode(13))
+    stdout.write('\x1b[K')
+    stdout.write(question + line)
+    // Ставим курсор в нужную позицию.
+    const back = line.length - cursor
+    if (back > 0) stdout.write('\x1b[' + back + 'D')
   }
-  const rl = readlinePromises.createInterface({
-    input: process.stdin,
-    output: process.stdout,
+
+  return await new Promise((resolve) => {
+    const finish = (value, submit) => {
+      stdin.removeListener('data', onData)
+      stdout.write('\x1b[?2004l')
+      if (stdin.setRawMode) stdin.setRawMode(wasRaw || false)
+      if (submit) stdout.write(String.fromCharCode(10))
+      resolve(value)
+    }
+
+    const insertText = (text) => {
+      // Нормализуем переводы строк: они приходят от многострочной вставки,
+      // но означают «отправить». Внутри сообщения заменяем на пробел, чтобы
+      // вся вставка ушла ОДНИМ сообщением.
+      const clean = text
+        .replace(/\r\n/g, ' ')
+        .replace(/\r/g, ' ')
+        .replace(/\n/g, ' ')
+      line = line.slice(0, cursor) + clean + line.slice(cursor)
+      cursor += clean.length
+    }
+
+    const onData = (buf) => {
+      let s = buf.toString('utf-8')
+
+      // Fallback для терминалов без bracketed paste: если весь чанк — это
+      // «голый» перевод строки (один байт), значит нажали Enter → отправляем.
+      // Если переводы строк пришли ВМЕСТЕ с другим текстом в одном чанке —
+      // это вставка; такие переводы строк не отправляют сообщение, а
+      // заменяются на пробелы (см. insertText).
+      if (!inPaste && (s === '\r' || s === '\n')) {
+        return finish(line, true)
+      }
+
+      while (s.length) {
+        if (inPaste) {
+          const end = s.indexOf(PASTE_END)
+          if (end === -1) {
+            insertText(s)
+            s = ''
+          } else {
+            insertText(s.slice(0, end))
+            s = s.slice(end + PASTE_END.length)
+            inPaste = false
+          }
+          redraw()
+          continue
+        }
+
+        const start = s.indexOf(PASTE_START)
+        if (start !== -1) {
+          // Всё до маркера обрабатываем как обычный ввод.
+          const before = s.slice(0, start)
+          s = s.slice(start + PASTE_START.length)
+          inPaste = true
+          if (before) {
+            for (const ch of before) {
+              if (ch === '\r' || ch === '\n') { /* внутри вставки — пропускаем */ }
+              else insertText(ch)
+            }
+            redraw()
+          }
+          continue
+        }
+
+        const ch = s[0]
+        const code = s.charCodeAt(0)
+        s = s.slice(1)
+
+        if (ch === '\r' || ch === '\n') {
+          // Перевод строки внутри чанка с другим текстом (вставка без
+          // bracketed paste): не отправляем, а вставляем пробел.
+          insertText(' ')
+          redraw()
+          continue
+        }
+        if (code === 3) {
+          // Ctrl+C — прерываем ввод.
+          return finish('', true)
+        }
+        if (code === 4) {
+          // Ctrl+D — как отправка пустой строки.
+          return finish(line, true)
+        }
+        if (code === 21) {
+          // Ctrl+U — стереть строку.
+          line = ''
+          cursor = 0
+          redraw()
+          continue
+        }
+        if (code === 127 || code === 8) {
+          // Backspace.
+          if (cursor > 0) {
+            line = line.slice(0, cursor - 1) + line.slice(cursor)
+            cursor--
+            redraw()
+          }
+          continue
+        }
+        if (ch === '\x1b') {
+          // Escape-последовательности (стрелки, Home/End, Delete…).
+          const rest = s
+          if (rest.startsWith('[D')) {
+            if (cursor > 0) cursor--
+            s = s.slice(2)
+            redraw()
+            continue
+          }
+          if (rest.startsWith('[C')) {
+            if (cursor < line.length) cursor++
+            s = s.slice(2)
+            redraw()
+            continue
+          }
+          if (rest.startsWith('[H') || rest.startsWith('[1~')) {
+            cursor = 0
+            s = s.slice(rest.startsWith('[1~') ? 3 : 2)
+            redraw()
+            continue
+          }
+          if (rest.startsWith('[F') || rest.startsWith('[4~')) {
+            cursor = line.length
+            s = s.slice(rest.startsWith('[4~') ? 3 : 2)
+            redraw()
+            continue
+          }
+          if (rest.startsWith('[3~')) {
+            // Delete.
+            if (cursor < line.length) {
+              line = line.slice(0, cursor) + line.slice(cursor + 1)
+              redraw()
+            }
+            s = s.slice(3)
+            continue
+          }
+          // Прочие ESC-последовательности пропускаем до буквы/тильды.
+          const m = s.match(/^\[[0-9;]*[A-Za-z~]/)
+          if (m) s = s.slice(m[0].length)
+          continue
+        }
+        if (code < 32) continue // прочие управляющие символы игнорируем
+
+        insertText(ch)
+        redraw()
+      }
+    }
+
+    stdin.on('data', onData)
   })
-  try {
-    return await rl.question(question)
-  } finally {
-    rl.close()
+}
+
+// Слежение за клавиатурой во время работы агента.
+// Esc — прервать текущую генерацию (клик Stop в браузере).
+function watchEscape(onEscape) {
+  const stdin = process.stdin
+  if (!stdin.isTTY) return () => {}
+  const wasRaw = stdin.isRaw
+  if (stdin.setRawMode) stdin.setRawMode(true)
+  stdin.resume()
+  const handler = (buf) => {
+    if (buf.length === 1 && buf[0] === 27) {
+      onEscape()
+    }
+  }
+  stdin.on('data', handler)
+  return () => {
+    stdin.removeListener('data', handler)
+    if (stdin.setRawMode) stdin.setRawMode(wasRaw || false)
   }
 }
 
@@ -277,9 +505,15 @@ async function runTask(browser, tools, taskText, workdir, opts) {
   const { transcript, freshChat, sendSystemPrompt } = opts
 
   const ui = mod.createSpinner()
-  ui.taskHeader(taskText)
   ui.thinking()
 
+  // Esc во время работы — прервать генерацию.
+  const stopWatching = watchEscape(() => {
+    ui.stop()
+    console.error(theme.warn('⏹ Esc — прерываю генерацию...'))
+    browser.stopGeneration().catch(() => {})
+  })
+
   let finished = false
 
   try {
@@ -299,13 +533,15 @@ async function runTask(browser, tools, taskText, workdir, opts) {
         ui.assistant(msg)
         finished = true
       },
+      debugLog: debug,
     })
   } catch (e) {
     ui.stop()
-    console.error(chalk.red('\n✖ Ошибка агента:'), e.message)
+    console.error(theme.error('\n✖ Ошибка агента:'), e.message)
     if (debug) console.error(e.stack)
     transcript?.log('agent_error', { error: e.message })
   } finally {
+    stopWatching()
     if (!finished) ui.stop()
   }
 }
@@ -327,9 +563,11 @@ async function main() {
   }
 
   if (calibrate) {
-    console.log(chalk.yellow('\n🔧 Режим калибровки селекторов\n'))
+    console.log(theme.warn('\n🔧 Режим калибровки селекторов\n'))
   }
 
+  await cleanTmpDir()
+
   let currentWorkdir
   try {
     currentWorkdir = await resolveWorkdir()
@@ -337,13 +575,13 @@ async function main() {
   const sandboxRoot = currentWorkdir
   } catch (e) {
     console.error(
-      chalk.red('Не удалось определить рабочую директорию:'),
+      theme.error('Не удалось определить рабочую директорию:'),
       e.message,
     )
     process.exit(1)
   }
 
-  console.log(chalk.gray(`Рабочая директория: ${currentWorkdir}`))
+  console.log(theme.system(`Рабочая директория: ${currentWorkdir}`))
 
   const transcript = new Transcript({
     dir: config.transcript.dir,
@@ -351,7 +589,7 @@ async function main() {
     sessionName: dirLabel(currentWorkdir),
   })
   if (transcript.file) {
-    console.log(chalk.gray(`Транскрипт: ${transcript.file}`))
+    console.log(theme.system(`Транскрипт: ${transcript.file}`))
   }
 
   const undo = new UndoStore(config.undo)
@@ -372,7 +610,7 @@ async function main() {
     await browser.waitForLogin()
   } catch (e) {
     bootSpinner.stop()
-    console.error(chalk.red('Не удалось запустить браузер:'), e.message)
+    console.error(theme.error('Не удалось запустить браузер:'), e.message)
     if (debug) console.error(e.stack)
     await browser.close().catch(() => {})
     transcript.close()
@@ -386,21 +624,23 @@ async function main() {
     let freshChat = true
     let sendSystemPrompt = true
 
-    // Приоритет: явный --chat, иначе сохранённый last-chat (если не --new-chat).
+    // По умолчанию — новый чат. Продолжить прошлый можно явно:
+    //   --chat <id>     открыть конкретный чат
+    //   --resume-last   вернуться в последний сохранённый чат
     let resumeId = chatIdArg
-    if (!resumeId && !newChatFlag) {
+    if (!resumeId && resumeLastFlag) {
       resumeId = await loadLastChat()
-      if (resumeId) console.log(chalk.gray(`Восстанавливаю чат ${resumeId}...`))
+      if (resumeId) console.log(theme.system(`Восстанавливаю чат ${resumeId}...`))
     }
 
     if (resumeId) {
       try {
-        console.log(chalk.gray(`Открываю чат ${resumeId}...`))
+        console.log(theme.system(`Открываю чат ${resumeId}...`))
         await browser.openChat(resumeId)
         freshChat = false
         sendSystemPrompt = resendPrompt
       } catch (e) {
-        console.error(chalk.red(`Не удалось открыть чат: ${e.message}`))
+        console.error(theme.error(`Не удалось открыть чат: ${e.message}`))
       }
     }
 
@@ -418,7 +658,7 @@ async function main() {
   }
 
   console.log(
-    chalk.gray(
+    theme.system(
       'Интерактивный режим. Введите задачу. Команды — /help. Выход — /exit.\n',
     ),
   )
@@ -442,42 +682,45 @@ async function main() {
     // висеть отдельным процессом после выхода агента.
     await mod.closeWeb().catch(() => {})
     transcript.close()
-    console.log(chalk.gray('\nВыход.'))
+    console.log(theme.system('\nВыход.'))
     process.exit(0)
   })
 
-  // Приоритет: явный --chat, иначе сохранённый last-chat (если не --new-chat).
+  // По умолчанию — новый чат. Продолжить прошлый можно явно:
+  //   --chat <id>     открыть конкретный чат
+  //   --resume-last   вернуться в последний сохранённый чат
   let resumeId = chatIdArg
-  if (!resumeId && !newChatFlag) {
+  if (!resumeId && resumeLastFlag) {
     resumeId = await loadLastChat()
+    if (resumeId) console.log(theme.system(`Восстанавливаю чат ${resumeId}...`))
   }
 
   if (resumeId) {
     try {
-      console.log(chalk.gray(`Открываю чат ${resumeId}...`))
+      console.log(theme.system(`Открываю чат ${resumeId}...`))
       await browser.openChat(resumeId)
       currentChatId = resumeId
       freshChatNext = false
       sendSystemPromptNext = resendPrompt
       await saveLastChat(resumeId)
-      console.log(chalk.gray(`Чат открыт: ${resumeId}\n`))
+      console.log(theme.system(`Чат открыт: ${resumeId}\n`))
     } catch (e) {
-      console.error(chalk.red(`Не удалось открыть чат: ${e.message}`))
+      console.error(theme.error(`Не удалось открыть чат: ${e.message}`))
     }
   }
 
   while (running) {
     let input
     try {
-      let label
+      // Компактное приглашение: золотая стрелка + светло-голубая директория.
+      let tail
       if (reviewMode) {
-        label = `zames[REVIEW:${reviewMode.snapName}]> `
-      } else if (currentChatId) {
-        label = `zames[${dirLabel(currentWorkdir)}|${currentChatId.slice(0, 6)}]> `
+        tail =
+          theme.warn('REVIEW') + theme.dim(':') + theme.dir(reviewMode.snapName)
       } else {
-        label = `zames[${dirLabel(currentWorkdir)}]> `
+        tail = theme.dir(dirLabel(currentWorkdir))
       }
-      input = await promptOnce(chalk.cyan(label))
+      input = await promptOnce(theme.prompt('❯ ') + tail + theme.dim(' › '))
     } catch {
       break
     }
@@ -495,7 +738,7 @@ async function main() {
     }
 
     if (['/new', '/clear', 'new'].includes(lower)) {
-      console.log(chalk.gray('Создаю новый чат...'))
+      console.log(theme.system('Создаю новый чат...'))
       try {
         await browser.newChat()
         freshChatNext = false
@@ -503,9 +746,9 @@ async function main() {
         currentChatId = await browser.getCurrentChatId()
         await saveLastChat(currentChatId)
         transcript.log('new_chat')
-        console.log(chalk.gray('Новый чат.\n'))
+        console.log(theme.system('Новый чат.\n'))
       } catch (e) {
-        console.error(chalk.red('Не удалось создать новый чат:'), e.message)
+        console.error(theme.error('Не удалось создать новый чат:'), e.message)
       }
       continue
     }
@@ -544,7 +787,7 @@ async function main() {
         await saveLastChat(currentChatId)
 
         console.log(
-          chalk.cyan(
+          theme.user(
             '\n💡 Теперь ты в режиме ревью. Просто пиши агенту, например:\n' +
               '   «исправь ошибки»\n' +
               '   «доработай обработку ошибок в ask()»\n' +
@@ -555,7 +798,7 @@ async function main() {
           ),
         )
       } catch (e) {
-        console.error(chalk.red('Самообзор провалился:'), e.message)
+        console.error(theme.error('Самообзор провалился:'), e.message)
         if (debug) console.error(e.stack)
       }
       continue
@@ -564,7 +807,7 @@ async function main() {
     if (lower === '/self-fix' || lower.startsWith('/self-fix ')) {
       const rest = trimmed.slice('/self-fix'.length).trim()
       if (!rest) {
-        console.error(chalk.red('Использование: /self-fix <name> [фокус]'))
+        console.error(theme.error('Использование: /self-fix <name> [фокус]'))
         continue
       }
       const sp = rest.indexOf(' ')
@@ -574,7 +817,7 @@ async function main() {
       const snapRoot = path.join(ZAMES_HOME, 'snapshots', name)
       const stat = await fs.stat(snapRoot).catch(() => null)
       if (!stat || !stat.isDirectory()) {
-        console.error(chalk.red(`Снапшот не найден: ${snapRoot}`))
+        console.error(theme.error(`Снапшот не найден: ${snapRoot}`))
         continue
       }
 
@@ -593,7 +836,7 @@ async function main() {
           workdir: snapRoot,
           tools,
         })
-        console.log(chalk.gray('Инициализирую review-чат для снапшота...'))
+        console.log(theme.system('Инициализирую review-чат для снапшота...'))
         await browser.ask(sysPrompt, { timeout: 60_000 })
 
         if (focus) {
@@ -611,7 +854,7 @@ async function main() {
             onThinking: () => ui.thinking(),
             onAssistantThought: (text) => {
               ui.stop()
-              console.log(chalk.gray('\n💭 ' + text.slice(0, 1200) + '\n'))
+              console.log(theme.system('\n💭 ' + text.slice(0, 1200) + '\n'))
             },
             onToolCall: (name, toolArgs) => ui.toolCall(name, toolArgs),
             onToolResult: (r) => ui.toolResult(r),
@@ -633,19 +876,19 @@ async function main() {
         await saveLastChat(currentChatId)
 
         console.log(
-          chalk.cyan(
+          theme.user(
             `\n💡 Режим ревью по снапшоту ${name}. Пиши агенту задачу или /self-done.\n`,
           ),
         )
       } catch (e) {
-        console.error(chalk.red('Не удалось войти в снапшот:'), e.message)
+        console.error(theme.error('Не удалось войти в снапшот:'), e.message)
       }
       continue
     }
 
     if (lower === '/self-done') {
       if (!reviewMode) {
-        console.log(chalk.gray('Ты и так не в режиме ревью.'))
+        console.log(theme.system('Ты и так не в режиме ревью.'))
         continue
       }
       const back = reviewMode.originalWorkdir
@@ -655,7 +898,7 @@ async function main() {
       freshChatNext = true
       sendSystemPromptNext = true
       console.log(
-        chalk.gray(`Вернулся в ${back}. Следующая задача начнёт новый чат.\n`),
+        theme.system(`Вернулся в ${back}. Следующая задача начнёт новый чат.\n`),
       )
       continue
     }
@@ -664,7 +907,7 @@ async function main() {
       try {
         await mod.selfList({ config })
       } catch (e) {
-        console.error(chalk.red('Ошибка:'), e.message)
+        console.error(theme.error('Ошибка:'), e.message)
       }
       continue
     }
@@ -672,13 +915,13 @@ async function main() {
     if (lower === '/self-diff' || lower.startsWith('/self-diff ')) {
       const name = trimmed.slice('/self-diff'.length).trim()
       if (!name) {
-        console.error(chalk.red('Использование: /self-diff <name>'))
+        console.error(theme.error('Использование: /self-diff <name>'))
         continue
       }
       try {
         await mod.selfDiff({ config, name })
       } catch (e) {
-        console.error(chalk.red('Ошибка:'), e.message)
+        console.error(theme.error('Ошибка:'), e.message)
       }
       continue
     }
@@ -686,13 +929,13 @@ async function main() {
     if (lower === '/self-apply' || lower.startsWith('/self-apply ')) {
       const name = trimmed.slice('/self-apply'.length).trim()
       if (!name) {
-        console.error(chalk.red('Использование: /self-apply <name>'))
+        console.error(theme.error('Использование: /self-apply <name>'))
         continue
       }
       try {
         await mod.selfApply({ config, name })
       } catch (e) {
-        console.error(chalk.red('Ошибка:'), e.message)
+        console.error(theme.error('Ошибка:'), e.message)
       }
       continue
     }
@@ -707,23 +950,23 @@ async function main() {
         spin.stop()
         if (!lastChats.length) {
           console.log(
-            chalk.gray(
+            theme.system(
               'Чатов не найдено. Возможно, сайдбар свёрнут или селекторы устарели.',
             ),
           )
         } else {
-          console.log(chalk.gray('Последние чаты DeepSeek:'))
+          console.log(theme.system('Последние чаты DeepSeek:'))
           lastChats.forEach((c, i) => {
             const n = String(i + 1).padStart(2, ' ')
             console.log(
-              `  ${chalk.cyan(n)}. ${c.title}  ${chalk.gray('(' + c.id.slice(0, 8) + '…)')}`,
+              `  ${theme.user(n)}. ${c.title}  ${theme.system('(' + c.id.slice(0, 8) + '…)')}`,
             )
           })
-          console.log(chalk.gray('\nИспользуй /resume <n> для продолжения.\n'))
+          console.log(theme.system('\nИспользуй /resume <n> для продолжения.\n'))
         }
       } catch (e) {
         spin.stop()
-        console.error(chalk.red('Не удалось получить список:'), e.message)
+        console.error(theme.error('Не удалось получить список:'), e.message)
       }
       continue
     }
@@ -732,26 +975,26 @@ async function main() {
       const arg = trimmed.slice(7).trim()
       if (!arg) {
         console.error(
-          chalk.red('Использование: /resume <n>  (или /chats для списка)'),
+          theme.error('Использование: /resume <n>  (или /chats для списка)'),
         )
         continue
       }
       const n = Number(arg)
       if (!Number.isFinite(n) || n < 1) {
-        console.error(chalk.red('Нужен номер из /chats.'))
+        console.error(theme.error('Нужен номер из /chats.'))
         continue
       }
       if (!lastChats.length) {
-        console.error(chalk.red('Сначала выполни /chats.'))
+        console.error(theme.error('Сначала выполни /chats.'))
         continue
       }
       const pick = lastChats[n - 1]
       if (!pick) {
-        console.error(chalk.red(`Нет чата №${n}. Всего: ${lastChats.length}.`))
+        console.error(theme.error(`Нет чата №${n}. Всего: ${lastChats.length}.`))
         continue
       }
 
-      console.log(chalk.gray(`Открываю: ${pick.title}`))
+      console.log(theme.system(`Открываю: ${pick.title}`))
       try {
         await browser.openChat(pick.id)
         currentChatId = pick.id
@@ -760,94 +1003,94 @@ async function main() {
         await saveLastChat(pick.id)
         transcript.log('resume_chat', { id: pick.id, title: pick.title })
         console.log(
-          chalk.green(`Чат открыт.`) +
-            chalk.gray(
+          theme.assistant(`Чат открыт.`) +
+            theme.system(
               resendPrompt ? ' Системный промпт будет переслан на следующей задаче.\n' : ' Контекст чата сохранён. Системный промпт не пересылается (--resend-prompt чтобы дослать).\n',
             ),
         )
       } catch (e) {
-        console.error(chalk.red('Не удалось открыть чат:'), e.message)
+        console.error(theme.error('Не удалось открыть чат:'), e.message)
       }
       continue
     }
 
     if (lower === '/chat') {
       if (currentChatId) {
-        console.log(chalk.gray(`Текущий chat id: ${currentChatId}`))
+        console.log(theme.system(`Текущий chat id: ${currentChatId}`))
         console.log(
-          chalk.gray(
+          theme.system(
             `URL: https://chat.deepseek.com/a/chat/s/${currentChatId}`,
           ),
         )
       } else {
         const id = await browser.getCurrentChatId()
         console.log(
-          chalk.gray(id ? `Текущий chat id: ${id}` : 'Чат ещё не создан.'),
+          theme.system(id ? `Текущий chat id: ${id}` : 'Чат ещё не создан.'),
         )
       }
       continue
     }
 
     if (lower === '/pwd') {
-      console.log(chalk.gray(currentWorkdir))
+      console.log(theme.system(currentWorkdir))
       continue
     }
 
     if (lower === '/reload') {
-      console.log(chalk.gray('Перечитываю модули логики...'))
+      console.log(theme.system('Перечитываю модули логики...'))
       try {
         const { count, errors } = await reloadModules()
         if (errors.length) {
-          console.error(chalk.red('Часть модулей не перезагрузилась:'))
-          for (const e of errors) console.error(chalk.red('  ' + e))
+          console.error(theme.error('Часть модулей не перезагрузилась:'))
+          for (const e of errors) console.error(theme.error('  ' + e))
         } else {
           console.log(
-            chalk.green(
+            theme.assistant(
               `Перезагружено модулей: ${count}. Браузер и чат не тронуты.`,
             ),
           )
         }
       } catch (e) {
-        console.error(chalk.red('Ошибка reload:'), e.message)
+        console.error(theme.error('Ошибка reload:'), e.message)
       }
       continue
     }
 
     if (lower === '/status') {
-      console.log(chalk.gray(`Рабочая директория: ${currentWorkdir}`))
+      console.log(theme.system(`Рабочая директория: ${currentWorkdir}`))
       console.log(
-        chalk.gray(`Режим ревью: ${reviewMode ? reviewMode.snapName : 'нет'}`),
+        theme.system(`Режим ревью: ${reviewMode ? reviewMode.snapName : 'нет'}`),
       )
       if (reviewMode) {
         console.log(
-          chalk.gray(`Исходная директория: ${reviewMode.originalWorkdir}`),
+          theme.system(`Исходная директория: ${reviewMode.originalWorkdir}`),
         )
       }
-      console.log(chalk.gray(`Текущий чат: ${currentChatId || '(нет)'}`))
+      console.log(theme.system(`Текущий чат: ${currentChatId || '(нет)'}`))
       console.log(
-        chalk.gray(
+        theme.system(
           `Fresh chat на след. задаче: ${freshChatNext ? 'да' : 'нет'}`,
         ),
       )
       console.log(
-        chalk.gray(
+        theme.system(
           `System prompt на след. задаче: ${sendSystemPromptNext ? 'да' : 'нет'}`,
         ),
       )
       console.log(
-        chalk.gray(`Resend prompt (--resend-prompt): ${resendPrompt ? 'да' : 'нет'}`),
+        theme.system(`Resend prompt (--resend-prompt): ${resendPrompt ? 'да' : 'нет'}`),
       )
       console.log(
-        chalk.gray(`Last chat: ${(await loadLastChat()) || 'нет'}`),
+        theme.system(`Last chat: ${(await loadLastChat()) || 'нет'}`),
       )
       console.log(
-        chalk.gray(`Dev mode (auto-reload): ${devMode ? 'вкл' : 'выкл'}`),
+        theme.system(`Dev mode (auto-reload): ${devMode ? 'вкл' : 'выкл'}`),
       )
-      console.log(chalk.gray(`Лимит итераций: ${maxIter}`))
-      console.log(chalk.gray(`Headless: ${headless ? 'да' : 'нет'}`))
-      console.log(chalk.gray(`Debug: ${debug ? 'да' : 'нет'}`))
-      console.log(chalk.gray(`Undo: ${config.undo.enabled ? 'вкл' : 'выкл'}`))
-      console.log(chalk.gray(`Транскрипт: ${transcript.file || 'выкл'}`))
+      console.log(theme.system(`Лимит итераций: ${maxIter}`))
+      console.log(theme.system(`Headless: ${headless ? 'да' : 'нет'}`))
+      console.log(theme.system(`Debug: ${debug ? 'да' : 'нет'}`))
+      console.log(theme.system(`Undo: ${config.undo.enabled ? 'вкл' : 'выкл'}`))
+      console.log(theme.system(`Транскрипт: ${transcript.file || 'выкл'}`))
       continue
     }
 
@@ -857,7 +1100,7 @@ async function main() {
     }
 
     if (lower === '/transcript') {
-      console.log(chalk.gray(transcript.file || '(выключен)'))
+      console.log(theme.system(transcript.file || '(выключен)'))
       continue
     }
 
@@ -865,14 +1108,14 @@ async function main() {
       const result = await undo.undoLast()
       if (result.ok) {
         console.log(
-          chalk.green(`↶ Откатили: ${result.record.originalPath}`) +
-            chalk.gray(
+          theme.assistant(`↶ Откатили: ${result.record.originalPath}`) +
+            theme.system(
               result.record.existed ? ' (восстановлено)' : ' (удалено)',
             ),
         )
         transcript.log('undo', { path: result.record.originalPath })
       } else {
-        console.error(chalk.red(`Не удалось откатить: ${result.reason}`))
+        console.error(theme.error(`Не удалось откатить: ${result.reason}`))
       }
       continue
     }
@@ -880,12 +1123,12 @@ async function main() {
     if (lower === '/undo-list' || lower === '/history') {
       const list = await undo.list(10)
       if (!list.length) {
-        console.log(chalk.gray('История пуста.'))
+        console.log(theme.system('История пуста.'))
       } else {
         for (const r of list) {
           const stamp = new Date(r.stamp).toLocaleString()
           const flag = r.existed ? 'изменён' : 'создан'
-          console.log(chalk.gray(`${stamp}  [${flag}]  ${r.originalPath}`))
+          console.log(theme.system(`${stamp}  [${flag}]  ${r.originalPath}`))
         }
       }
       continue
@@ -896,11 +1139,11 @@ async function main() {
       const file = path.join(config.transcript.dir, `dom-${stamp}.html`)
       try {
         const result = await browser.dumpDom(file)
-        console.log(chalk.green(`HTML сохранён: ${result.file}`))
-        console.log(chalk.gray('Селекторы:'))
+        console.log(theme.assistant(`HTML сохранён: ${result.file}`))
+        console.log(theme.system('Селекторы:'))
         console.log(JSON.stringify(result.selectors, null, 2))
       } catch (e) {
-        console.error(chalk.red('Не удалось сохранить DOM:'), e.message)
+        console.error(theme.error('Не удалось сохранить DOM:'), e.message)
       }
       continue
     }
@@ -913,34 +1156,34 @@ async function main() {
           : sandboxRoot
         const rel = path.relative(sandboxRoot, newDir)
         if (rel.startsWith('..') || path.isAbsolute(rel)) {
-          console.error(chalk.red('Нельзя выйти за пределы: ' + sandboxRoot))
+          console.error(theme.error('Нельзя выйти за пределы: ' + sandboxRoot))
           continue
         }
         const stat = await fs.stat(newDir).catch(() => null)
         if (!stat || !stat.isDirectory()) {
-          console.error(chalk.red('Не директория: ' + newDir))
+          console.error(theme.error('Не директория: ' + newDir))
           continue
         }
         if (newDir === currentWorkdir) {
-          console.log(chalk.gray('Уже здесь.'))
+          console.log(theme.system('Уже здесь.'))
           continue
         }
         if (reviewMode) {
-          console.log(chalk.gray('Вышел из режима ревью (/cd).'))
+          console.log(theme.system('Вышел из режима ревью (/cd).'))
           reviewMode = null
         }
         currentWorkdir = newDir
         freshChatNext = true
         sendSystemPromptNext = true
-        console.log(chalk.gray('Рабочая директория: ' + newDir))
+        console.log(theme.system('Рабочая директория: ' + newDir))
       } catch (e) {
-        console.error(chalk.red('Не удалось перейти: ' + e.message))
+        console.error(theme.error('Не удалось перейти: ' + e.message))
       }
       continue
     }
 
     if (lower.startsWith('/')) {
-      console.error(chalk.red(`Неизвестная команда: ${trimmed}. Набери /help.`))
+      console.error(theme.error(`Неизвестная команда: ${trimmed}. Набери /help.`))
       continue
     }
 
@@ -972,7 +1215,7 @@ async function main() {
 }
 
 main().catch((e) => {
-  console.error(chalk.red('Критическая ошибка:'), e.message)
+  console.error(theme.error('Критическая ошибка:'), e.message)
   if (debug) console.error(e.stack)
   process.exit(1)
 })