Compare commits
32 Commits
de21ec9f31
...
feat/5a3f0
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
89d571b483 | ||
|
|
ed13386612 | ||
|
|
9cb174a5d3 | ||
|
|
5b43da4564 | ||
|
|
46be641967 | ||
|
|
6fae81bac6 | ||
|
|
8545315f49 | ||
|
|
6586285992 | ||
|
|
94521488b6 | ||
|
|
c9ab753c1e | ||
|
|
88b455203f | ||
|
|
3c51ffee4e | ||
|
|
8c91c83024 | ||
|
|
d731a7429d | ||
|
|
4c0dc80aa8 | ||
| 0ad53e338d | |||
|
|
8cf4fc9f7c | ||
|
|
963e7b478e | ||
|
|
cd0619926e | ||
|
|
2854697415 | ||
|
|
ef812bb3d7 | ||
| 23bae9a68f | |||
|
|
70287140ec | ||
|
|
d9d043ec8e | ||
| b87092aa98 | |||
|
|
ee47785e1a | ||
|
|
a762a1df80 | ||
|
|
ee76f9d89f | ||
|
|
37cc29e4cd | ||
|
|
180afb1336 | ||
| 0c88272bf0 | |||
|
|
9420b276e9 |
@@ -41,3 +41,14 @@
|
|||||||
- `app.Version` — семантическая major.minor.patch (ручной инкремент: patch=фиксы,
|
- `app.Version` — семантическая major.minor.patch (ручной инкремент: patch=фиксы,
|
||||||
minor=новая обратно-совместимая функциональность, major=несовместимые изменения). Сейчас 0.2.2.
|
minor=новая обратно-совместимая функциональность, major=несовместимые изменения). Сейчас 0.2.2.
|
||||||
- `main.version` (ldflag) — build-идентификатор `commit-<sha7>`, отдельно от app.Version.
|
- `main.version` (ldflag) — build-идентификатор `commit-<sha7>`, отдельно от app.Version.
|
||||||
|
|
||||||
|
## Форматирование (важно на Windows)
|
||||||
|
- Репо на Windows-чекауте: `core.autocrlf=true` → файлы в рабочей копии с CRLF; `gofmt -l`
|
||||||
|
на CRLF-копии шумит (глобально ~60 файлов). **Проверять формат только на LF-версии** (напр.
|
||||||
|
`git -c core.autocrlf=false clone` во временный каталог) — так дифы видны корректно.
|
||||||
|
- gofmt 1.26 форматирует doc-comments (`//` перед `go build ...` в `// go build` → пустая строка
|
||||||
|
`//`) и выравнивание структур; не все файлы отформатированы по новой версии (предсуществующе).
|
||||||
|
- CI (`go vet` + `go test`) формат не проверяет → gofmt-дифы не ломают сборку.
|
||||||
|
- **Known race в тест-харнессе app:** `e2eChannel.Send` (`internal/app/e2e_test.go:213`) пишет
|
||||||
|
`c.sent` без лока, тест читает с главной горутины → `-race` ловит в `TestE2ENotificationsOnTransitions`.
|
||||||
|
Путь `worker→Notify→Send`; предсуществует, ещё не чинили (fix — мьютекс в `e2eChannel`).
|
||||||
@@ -31,19 +31,22 @@ docs/ ui-spec.md — спека Fyne UI (слои, event-bus, fyne.Do
|
|||||||
## Ключевые инварианты
|
## Ключевые инварианты
|
||||||
|
|
||||||
- **App.New-сигнатура:** `App.New(configPath, version, updateToken string, noUI bool)` (4-й параметр — headless; cgo-вариант собирается только при наличии C-компилятора).
|
- **App.New-сигнатура:** `App.New(configPath, version, updateToken string, noUI bool)` (4-й параметр — headless; cgo-вариант собирается только при наличии C-компилятора).
|
||||||
|
- **chat.Router:** асинхронная обработка входящих — **ограниченный пул `chatWorkers=4` воркеров + FIFO-очереди per-user** (`userState`, `workerLoop`/`runUser`). Порядок сообщений одного UserID сохраняется (флаг `scheduled` → один активный воркер на пользователя); разные пользователи обрабатываются параллельно (до 4 одновременных LLM-вызовов). Backpressure по `jobs` блокирует только перегруженного пользователя, не весь long-poll. `Processed()`/`WaitProcessed()` — синхронизация тестов.
|
||||||
|
- **FreeChat** (app): `sessions map[uid]sessionID` защищён `sync.Mutex` (пишется из разных воркеров роутера).
|
||||||
- **UI:** окно — ещё одна реализация `chat.Channel` (присоединяется в Router). Core не трогает UI; обмен — событийная шина (events). Кнопка «Завершить» = полный выход (SetOnQuit→cancel→UI.Run возвращается); закрытие крестиком = сворачивание, Core живёт. UI собирается с `--noui`/без cgo.
|
- **UI:** окно — ещё одна реализация `chat.Channel` (присоединяется в Router). Core не трогает UI; обмен — событийная шина (events). Кнопка «Завершить» = полный выход (SetOnQuit→cancel→UI.Run возвращается); закрытие крестиком = сворачивание, Core живёт. UI собирается с `--noui`/без cgo.
|
||||||
- **Фазы аналитика (Decision.Phase):** `ask`, `propose`, `ready` (два последних обрабатываются одинаково в core), `abort`. Требования валидатора: ask — chat_reply/questions; propose — хотя бы одно изменённое поле; ready — без изменённых полей.
|
- **Фазы аналитика (Decision.Phase):** `ask`, `propose`, `ready` (два последних обрабатываются одинаково в core), `abort`. Требования валидатора: ask — chat_reply/questions; propose — хотя бы одно изменённое поле; ready — без изменённых полей.
|
||||||
- **Статусы задач (internal/model):** draft→collecting→ready→approved→running→success/failed/timeout + cancelled/aborted/closed (терминальные). UserID — chat.ID (одна активная задача на чат).
|
- **Статусы задач (internal/model):** draft→collecting→ready→approved→running→success/failed/timeout + cancelled/aborted/closed (терминальные). UserID — chat.ID (одна активная задача на чат).
|
||||||
- **Decider/Worker/Analyst/Reviewer:** Decider=analyst интерфейс; Worker — polling-планировщик; Reviewer проверяет diff dev-ветки (R1-R6), вердикт JSON {passed, critical_issues, solid_violations, comments}.
|
- **Decider/Worker/Analyst/Reviewer:** Decider=analyst интерфейс; Worker — polling-планировщик; Reviewer проверяет diff dev-ветки (R1-R6), вердикт JSON {passed, critical_issues, solid_violations, comments}.
|
||||||
- **gitops (worker):** worktree-режим; feature-ветка `feat/<taskTag>` от origin/main; push через http.extraHeader, токен Bearer.
|
- **gitops (worker):** worktree-режим; feature-ветка `feat/<taskTag>` от origin/main; push через http.extraHeader, токен Bearer.
|
||||||
- **Пути «всё рядом с .exe»:** db/worktree резолвятся от ExeDir; config.yaml — рядом с бинарём, фоллбэк cwd.
|
- **Пути «всё рядом с .exe»:** db/worktree резолвятся от ExeDir; config.yaml — рядом с бинарём, фоллбэк cwd.
|
||||||
- **Автообновление:** авто = только Check+уведомление; замена — по /update; версии в `commit-<sha7>/` (не `latest/`); Verify сверяет предprod-версию (binary+в.в) .
|
- **Автообновление:** авто = только Check+уведомление; замена — по /update; версии в `commit-<sha7>/` (не `latest/`); Verify сверяет контрольную сумму бинаря против `.sha256` той же версии. **Perf:** `ResolveLatest` проверяет наличие бинаря версии **HEAD-пробой без скачивания тела** (405/501 → fallback `GET Range: bytes=0-0`), сортировка версий по id убыв.; один общий `http.Client` (keep-alive). Ошибка U4 — только при несовпадении суммы (пустой/отсутствующий `.sha256` пропускает проверку — M1-известное замечание).
|
||||||
|
- **Telegram toggle:** `telegram.enabled` (дефолт true, preset в `Load` до unmarshal, т.к. `applyDefaults` для bool не различает явный false). При `false` канал не создаётся/не крепится в Router и long-poll не стартует — сетевых вызовов к api.telegram.org нет; token/chat_id не валидируются.
|
||||||
|
|
||||||
## opencode (v2 HTTP API, >= 1.18.18)
|
## opencode (v2 HTTP API, >= 1.18.18)
|
||||||
|
|
||||||
- Интеграция с субагентами — через headless `opencode serve`, **v2 API** (`/api/*`). Версия opencode >= 1.18.18.
|
- Интеграция с субагентами — через headless `opencode serve`, **v2 API** (`/api/*`). Версия opencode >= 1.18.18.
|
||||||
- **Хардпин модели:** при создании сессии читается top-level `model` из конфига opencode (`internal/opencode/config.go`, JSONC-стрип) и передаётся в `POST /api/session` как `{"model":{providerID,id}}`.
|
- **Модель — только глобальный конфиг opencode.** Ratatoskr модель не выбирает и про неё не знает: opencode сам берёт модель по умолчанию из своего глобального конфига. Код opencode-конфиг не читает (config.go удалён).
|
||||||
- **О5 WARN (устойчивость к v1-конфигу):** конфиг по старой схеме молча игнорируется v2; провайдер без api → unsupported модели → fallback. Ratatoskr не чинит сам, но логирует warning; фактическая модель ответа сравнивается с ожидаемой. Правильный v2-вид: `api:{type:"aisdk",package,url}`, `request.headers` вместо `options.headers`.
|
- **Свой агент:** при создании сессии в `POST /api/session` передаётся `agent` (analyst/dev/reviewer/chat/postmortem) из встроенных определений. Агенты распаковываются в `<worktree>/.opencode/agent/*.md` (opencode находит через project-каталог `.opencode`); `OPENCODE_CONFIG_DIR` НЕ выставлять — он перенаправляет Global.Path.config и ломает глобальный конфиг (модель/провайдеры) → падение в fallback.
|
||||||
- **Поллинг вердикта:** `POST /api/session/:id/prompt` (durable admit) → `GET /api/session/:id/message?order=desc&limit=200` (новые assistant-сообщения, текст в `content[].type=="text"`) → завершение = `GET /api/session/active` без сессии + финальное assistant-сообщение, стабильное `settlePolls=2` опроса. `POST .../interrupt` вместо abort.
|
- **Поллинг вердикта:** `POST /api/session/:id/prompt` (durable admit) → `GET /api/session/:id/message?order=desc&limit=200` (новые assistant-сообщения, текст в `content[].type=="text"`) → завершение = `GET /api/session/active` без сессии + финальное assistant-сообщение, стабильное `settlePolls=2` опроса. `POST .../interrupt` вместо abort.
|
||||||
|
|
||||||
## Контракты (не ломать)
|
## Контракты (не ломать)
|
||||||
|
|||||||
@@ -2,11 +2,12 @@
|
|||||||
|
|
||||||
## MinGW/WinLibs для Fyne-сборки на Windows
|
## MinGW/WinLibs для Fyne-сборки на Windows
|
||||||
- UI (internal/ui/desktop, fyne.io/fyne/v2) требует cgo + C-компилятор для GLFW.
|
- UI (internal/ui/desktop, fyne.io/fyne/v2) требует cgo + C-компилятор для GLFW.
|
||||||
- Используется **WinLibs** (mingw64) gcc. Bin каталог нужно добавить в PATH перед
|
- Установлен постоянно в `D:\MinGW\bin` (WinLibs x86_64-ucrt-posix-seh, GCC 16.2.0),
|
||||||
`scripts/build-publish-ui.ps1`:
|
файл `D:\MinGW\bin\gcc.exe` существует. НО каталог не гарантированно в PATH текущей
|
||||||
```powershell
|
сессии (проверено 2026-08-21: `D:\MinGW\bin` отсутствовал в PATH, `gcc` не находился).
|
||||||
$p = "<...>\mingw64\bin"; $env:Path = "$p;$env:Path"
|
`scripts/build-publish-ui.ps1` сам проверяет gcc и при отсутствии подсказывает ручную
|
||||||
```
|
подстановку PATH (стр. 36–41): `$env:Path = "<...>\mingw64\bin;$env:Path"`.
|
||||||
|
- Прежняя winget-версия (BrechtSanders.WinLibs.POSIX.UCRT) удалена, чтобы не было дубликата.
|
||||||
- Проверка: `gcc --version`; также убедиться, что `CGO_ENABLED` не = "0".
|
- Проверка: `gcc --version`; также убедиться, что `CGO_ENABLED` не = "0".
|
||||||
- Care: не использовать Cygwin/MSYS-сборки gcc — только native MinGW-w64 (WinLibs).
|
- Care: не использовать Cygwin/MSYS-сборки gcc — только native MinGW-w64 (WinLibs).
|
||||||
- Сборка UI выполняется локально на Windows-машине, публикуется в Gitea Packages
|
- Сборка UI выполняется локально на Windows-машине, публикуется в Gitea Packages
|
||||||
|
|||||||
41
README.md
41
README.md
@@ -95,11 +95,11 @@ make run # build + ./ratatoskr -config config.yaml
|
|||||||
|
|
||||||
```yaml
|
```yaml
|
||||||
telegram:
|
telegram:
|
||||||
token: "..." # TG_TOKEN
|
enabled: true # false — отключить Telegram (без сетевых вызовов к api.telegram.org)
|
||||||
chat_id: "..." # TG_CHAT_ID
|
token: "..." # TG_TOKEN (обязательно, если enabled)
|
||||||
|
chat_id: "..." # TG_CHAT_ID (обязательно, если enabled)
|
||||||
opencode:
|
opencode:
|
||||||
bin: "opencode"
|
bin: "opencode"
|
||||||
config_dir: "./agents" # каталог, куда распаковываются встроенные агенты
|
|
||||||
paths:
|
paths:
|
||||||
db: "./ratatoskr.db" # дефолт; резолвится от каталога бинаря (не от cwd)
|
db: "./ratatoskr.db" # дефолт; резолвится от каталога бинаря (не от cwd)
|
||||||
worktree: "./worktrees" # то же правило «всё рядом с .exe»
|
worktree: "./worktrees" # то же правило «всё рядом с .exe»
|
||||||
@@ -131,25 +131,22 @@ update:
|
|||||||
|
|
||||||
## Интеграция с opencode (субагенты)
|
## Интеграция с opencode (субагенты)
|
||||||
|
|
||||||
Субагенты (analyst / dev / reviewer) запускаются через **headless** `opencode serve`
|
Субагенты (analyst / dev / reviewer / postmortem) запускаются через **headless** `opencode serve`
|
||||||
по **v2 HTTP API** (префикс `/api/*`). Требуемая версия opencode: **>= 1.18.18**
|
по **v2 HTTP API** (префикс `/api/*`). Требуемая версия opencode: **>= 1.18.18**
|
||||||
(сборки с v2 HTTP API). Старый бинарь, отвечающий только на `/global/health`,
|
(сборки с v2 HTTP API). Старый бинарь, отвечающий только на `/global/health`,
|
||||||
не подходит: healthcheck падает с понятной ошибкой (класс O1).
|
не подходит: healthcheck падает с понятной ошибкой (класс O1).
|
||||||
|
|
||||||
Что делает обёртка (`internal/opencode`):
|
Что делает обёртка (`internal/opencode`):
|
||||||
|
|
||||||
- **Хардпин модели.** При создании сессии в конфиге opencode ищется top-level
|
- **Модель — только глобальный конфиг opencode.** ratatoskr модель не выбирает
|
||||||
`"model"` (`internal/opencode/config.go`) и передаётся в `POST /api/session`
|
и про неё не знает: opencode сам берёт модель по умолчанию из своего
|
||||||
как `{"model":{providerID,id}}`. Это убирает зависимость от fallback-логики
|
глобального конфига (`~/.config/opencode/opencode.jsonc`). Наш код конфиг
|
||||||
opencode (которая молча выбирает «дефолтную» запись, если модель не задана).
|
opencode не читает.
|
||||||
- **Весь код резолва модели устойчив к этому классу проблем (класс O5 WARN):**
|
- **Свой агент.** При создании сессии в `POST /api/session` передаётся имя
|
||||||
- если конфиг не читается / в нём нет `model` — в логи пишется warning;
|
встроенного агента ratatoskr (analyst/dev/reviewer/chat/postmortem). Агенты
|
||||||
- фактическая модель ответа (из финального assistant-сообщения) сравнивается
|
распаковываются в `<worktree>/.opencode/agent/*.md`, где opencode находит
|
||||||
с ожидаемой; расхождение логируется как warning;
|
их через project-каталог `.opencode` (см. `internal/agents`); глобальный
|
||||||
- конфиг, написанный по **старой v1-схеме** (`provider.X.npm` / `options`),
|
конфиг при этом не трогается.
|
||||||
молча игнорируется v2 — обёртка этого не «чинит» сама, но предупреждает.
|
|
||||||
Правильный v2-вид провайдера — `api: { type:"aisdk", package, url }` и
|
|
||||||
`request.headers` вместо `options.headers`.
|
|
||||||
- **Поллинг вердикта.** Промпт отправляется неблокирующе (`POST .../prompt` →
|
- **Поллинг вердикта.** Промпт отправляется неблокирующе (`POST .../prompt` →
|
||||||
durable admit), вердикт собирается из новых assistant-сообщений
|
durable admit), вердикт собирается из новых assistant-сообщений
|
||||||
(`GET .../message`); завершение ответа — сессия ушла из активных дренажей
|
(`GET .../message`); завершение ответа — сессия ушла из активных дренажей
|
||||||
@@ -173,6 +170,18 @@ update:
|
|||||||
В `internal/core` фазы `propose` и `ready` обрабатываются одинаково (применить черновик,
|
В `internal/core` фазы `propose` и `ready` обрабатываются одинаково (применить черновик,
|
||||||
проверить репозитории, поставить `ready` и отдать резюме).
|
проверить репозитории, поставить `ready` и отдать резюме).
|
||||||
|
|
||||||
|
## Постмортем после failed/timeout
|
||||||
|
|
||||||
|
Когда задача завершилась `failed` или `timeout`, воркер дополнительно запускает
|
||||||
|
**постмортем-анализ** (`internal/worker/postmortem.go`, agent `postmortem`):
|
||||||
|
|
||||||
|
- анализирует сессии dev/reviewer (промпты и выводы из `traces`);
|
||||||
|
- оценивает законченность этапов и причины сбоя;
|
||||||
|
- сохраняет результат как trace `agent=postmortem` и шлёт владельцу уведомление
|
||||||
|
«🔍 анализ (после <статус>): почему так случилось / что сделать».
|
||||||
|
|
||||||
|
Статус задачи постмортем не меняет; сбои самого анализа не влияют на исход задачи.
|
||||||
|
|
||||||
## Автообновление из Gitea Packages
|
## Автообновление из Gitea Packages
|
||||||
|
|
||||||
Бинарь умеет сам себя обновлять из generic-пакета в Gitea. Модель:
|
Бинарь умеет сам себя обновлять из generic-пакета в Gitea. Модель:
|
||||||
|
|||||||
@@ -1,18 +1,20 @@
|
|||||||
# ratatoskr-go пример конфигурации.
|
# ratatoskr-go пример конфигурации.
|
||||||
# Можно не указывать поля, у которых есть дефолт (см. internal/config/types.go).
|
# Можно не указывать поля, у которых есть дефолт (см. internal/config/types.go).
|
||||||
# Обязательно: telegram.token + telegram.chat_id
|
# Если telegram.enabled: true (дефолт) — обязательно: telegram.token + telegram.chat_id.
|
||||||
|
|
||||||
telegram:
|
telegram:
|
||||||
token: "${TG_TOKEN}" # токен Telegram бота (обязательно)
|
enabled: true # false — полностью отключить Telegram (без сетевых вызовов)
|
||||||
chat_id: "${TG_CHAT_ID}" # ID чата (обязательно)
|
token: "${TG_TOKEN}" # токен Telegram бота (обязательно, если enabled)
|
||||||
|
chat_id: "${TG_CHAT_ID}" # ID чата (обязательно, если enabled)
|
||||||
|
|
||||||
# Всё ниже — опционально, показаны дефолты:
|
# Всё ниже — опционально, показаны дефолты:
|
||||||
# opencode:
|
# opencode:
|
||||||
# bin: "opencode"
|
# bin: "opencode"
|
||||||
# config: "/путь/к/opencode.json" # файл-конфиг модели (OPENCODE_CONFIG)
|
|
||||||
# config_dir: "./agents" # каталог с агентами (OPENCODE_CONFIG_DIR)
|
|
||||||
# hard_timeout: "20m"
|
# hard_timeout: "20m"
|
||||||
# idle_timeout: "5m"
|
# idle_timeout: "5m"
|
||||||
|
#
|
||||||
|
# Агенты (analyst/dev/reviewer/...) распаковываются автоматически в
|
||||||
|
# <worktree>/.opencode/agent/*.md; конфиг opencode для этого не нужен.
|
||||||
|
|
||||||
# git (источник репозиториев для git clone)
|
# git (источник репозиториев для git clone)
|
||||||
# base_url: "http://gitea.hal9000.home" # базовый URL git-хоста (обязательно для клонирования)
|
# base_url: "http://gitea.hal9000.home" # базовый URL git-хоста (обязательно для клонирования)
|
||||||
|
|||||||
@@ -170,11 +170,19 @@
|
|||||||
|
|
||||||
### 12.9. Контракт виджета «Логи» (LogPanel)
|
### 12.9. Контракт виджета «Логи» (LogPanel)
|
||||||
|
|
||||||
Панель «Логи» — поток системных логов (сырые строки из `events.LogLine`).
|
Панель «Логи» — поток системных логов из `events.LogLine` (уровень + строка).
|
||||||
|
|
||||||
- `Append(text)` — добавить строку лога; при превышении лимита старые строки
|
- `Append(level, text)` — добавить запись лога; уровень нормализуется к одному
|
||||||
отбрасываются (буфер ограничен, не расти бесконечно).
|
из четырёх (`ui.NormalizeLogLevel`: error/warning/info/debug, неизвестные →
|
||||||
- Контейнер виджета — отдаётся окну для встраивания в таб «Логи» (Scroll).
|
info); при превышении лимита старые записи отбрасываются (буфер ограничен,
|
||||||
|
не расти бесконечно).
|
||||||
|
- **Фильтр уровней**: над лентой — выпадающий список с чекбоксами
|
||||||
|
(error/warning/info/debug). По умолчанию отмечены все уровни (виден полный
|
||||||
|
лог); показываются только записи выбранных уровней; изменение набора сразу
|
||||||
|
перерисовывает ленту — и по уже загруженным записям, и для вновь
|
||||||
|
поступающих. Если ни один уровень не выбран — лента пуста (без ошибок).
|
||||||
|
- Контейнер виджета — отдаётся окну для встраивания в таб «Логи» (строка
|
||||||
|
фильтра + Scroll с лентой).
|
||||||
- Обновление содержимого — на потоке Fyne (внутри панели `fyne.Do`), не из
|
- Обновление содержимого — на потоке Fyne (внутри панели `fyne.Do`), не из
|
||||||
горутины Hub.
|
горутины Hub.
|
||||||
|
|
||||||
@@ -215,7 +223,12 @@
|
|||||||
отбрасываются (буфер ограничен);
|
отбрасываются (буфер ограничен);
|
||||||
- `Clear()` — очистить при недоступной задаче.
|
- `Clear()` — очистить при недоступной задаче.
|
||||||
- Композитор: поле ввода + кнопки команд, подключённые к `ui.Commands` через
|
- Композитор: поле ввода + кнопки команд, подключённые к `ui.Commands` через
|
||||||
`SetCommands(c)` (ввод → `SendText`, кнопки → Start/Approve/Skip/Cancel).
|
`SetCommands(c)` (ввод → `SendText`, кнопки → Start/Approve/Skip/Retry/Cancel).
|
||||||
|
- Кнопка «Перезапустить» — полный аналог Telegram-команды `/retry N`: отправляет
|
||||||
|
`/retry N` тем же путём, что и ручной ввод (Commands → Router → Core).
|
||||||
|
Активна только для вкладки с привязанной задачей; окно сообщает ID задачи
|
||||||
|
через `SetBoundTask(taskID)` (при создании вкладки в `addTab` и при привязке
|
||||||
|
свободной вкладки в `bindTaskSession`); `0` — кнопка неактивна.
|
||||||
- Доставка: окно (как chat.Channel) рендерит `Send`/`Ask`/историю через
|
- Доставка: окно (как chat.Channel) рендерит `Send`/`Ask`/историю через
|
||||||
`Append`; строка роли = pure `FormatRole(role)` (👤/🤖).
|
`Append`; строка роли = pure `FormatRole(role)` (👤/🤖).
|
||||||
- Обновление — на потоке Fyne; вызывающий уже внутри `fyne.Do`.
|
- Обновление — на потоке Fyne; вызывающий уже внутри `fyne.Do`.
|
||||||
|
|||||||
@@ -1,12 +1,10 @@
|
|||||||
// Package agents — встроенные агенты opencode (analyst, dev, ...).
|
// Package agents — встроенные агенты opencode (analyst, dev, ...).
|
||||||
//
|
//
|
||||||
// Агенты зашиты в бинарь через go:embed и распаковываются при запуске
|
// Агенты зашиты в бинарь через go:embed и распаковываются при запуске
|
||||||
// в каталог OPENCODE_CONFIG_DIR (по умолчанию ./agents рядом с бинарём),
|
// в project-каталог <worktree>/.opencode/agent/*.md. opencode при serve
|
||||||
// чтобы opencode находил их независимо от того, где запущен ratatoskr.
|
// (cwd = worktree) обнаруживает .opencode поднятием вверх (ConfigPaths) и
|
||||||
//
|
// находит агентов по шаблону {agent,agents}/**/*.md. Глобальный конфиг
|
||||||
// opencode сканирует OPENCODE_CONFIG_DIR как каталог .opencode: агенты
|
// opencode остаётся нетронутым (OPENCODE_CONFIG_DIR не выставляем).
|
||||||
// ищутся в поддиректории agents/ (мн.ч.), а не в корне. Поэтому распаковка
|
|
||||||
// идёт в <dir>/agents/*.md.
|
|
||||||
package agents
|
package agents
|
||||||
|
|
||||||
import (
|
import (
|
||||||
@@ -21,9 +19,9 @@ var FS embed.FS
|
|||||||
//
|
//
|
||||||
// Держим в курсе: если добавить файл в каталог, он автоматически попадёт
|
// Держим в курсе: если добавить файл в каталог, он автоматически попадёт
|
||||||
// в FS, но для детерминированной распаковки список лучше дополнять здесь.
|
// в FS, но для детерминированной распаковки список лучше дополнять здесь.
|
||||||
var Names = []string{"analyst", "dev", "reviewer"}
|
var Names = []string{"analyst", "dev", "reviewer", "chat", "postmortem"}
|
||||||
|
|
||||||
// WriteTo распаковывает всех встроенных агентов в каталог dir/agents
|
// WriteTo распаковывает всех встроенных агентов в каталог dir/agent
|
||||||
// (создаёт его). Файлы перезаписываются — встроенная копия всегда актуальна.
|
// (создаёт его). Файлы перезаписываются — встроенная копия всегда актуальна.
|
||||||
func WriteTo(dir string) error {
|
func WriteTo(dir string) error {
|
||||||
agentsDir := filepath.Join(dir, "agents")
|
agentsDir := filepath.Join(dir, "agents")
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ func TestWriteTo(t *testing.T) {
|
|||||||
t.Fatalf("WriteTo: %v", err)
|
t.Fatalf("WriteTo: %v", err)
|
||||||
}
|
}
|
||||||
for _, name := range Names {
|
for _, name := range Names {
|
||||||
// opencode ищет агентов в поддиректории agents/, как у .opencode
|
// opencode ищет агентов в поддиректории agent/ (или agents/) внутри .opencode
|
||||||
p := filepath.Join(dir, "agents", name+".md")
|
p := filepath.Join(dir, "agents", name+".md")
|
||||||
data, err := os.ReadFile(p)
|
data, err := os.ReadFile(p)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|||||||
@@ -2,6 +2,20 @@
|
|||||||
name: analyst
|
name: analyst
|
||||||
description: Аналитик Ratatoskr — помогает сформулировать задачу разработки и возвращает структурированное решение в JSON
|
description: Аналитик Ratatoskr — помогает сформулировать задачу разработки и возвращает структурированное решение в JSON
|
||||||
mode: primary
|
mode: primary
|
||||||
|
tools:
|
||||||
|
read: true
|
||||||
|
write: true
|
||||||
|
edit: false
|
||||||
|
bash: true
|
||||||
|
grep: true
|
||||||
|
glob: true
|
||||||
|
websearch: false
|
||||||
|
webfetch: false
|
||||||
|
permission:
|
||||||
|
question: deny
|
||||||
|
edit: deny
|
||||||
|
bash:
|
||||||
|
"*": allow
|
||||||
---
|
---
|
||||||
|
|
||||||
Ты — аналитик в конвейере Ratatoskr. Пользователь ставит задачу разработки. Ты помогаешь
|
Ты — аналитик в конвейере Ratatoskr. Пользователь ставит задачу разработки. Ты помогаешь
|
||||||
|
|||||||
17
internal/agents/chat.md
Normal file
17
internal/agents/chat.md
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
---
|
||||||
|
name: chat
|
||||||
|
description: Свободный чат с Ratatoskr — прямой диалог с LLM без задачи и JSON-вердиктов
|
||||||
|
mode: primary
|
||||||
|
---
|
||||||
|
|
||||||
|
Ты — полезный ассистент в диалоговом режиме. Пользователь пишет тебе произвольные
|
||||||
|
сообщения; отвечай на них прямо и по делу, на русском языке, естественно.
|
||||||
|
|
||||||
|
Особенности:
|
||||||
|
1. Отвечай на вопросы пользователя, помогай с задачами разработки, поясняй код.
|
||||||
|
2. Не оборачивай ответ в JSON и не возвращай структурированные вердикты — это
|
||||||
|
свободный диалог, не конвейер задач Ratatoskr.
|
||||||
|
3. Если тебе не хватает контекста — задай уточняющий вопрос в обычном тексте.
|
||||||
|
4. Пиши кратко и по существу, если пользователь не просит развёрнуто.
|
||||||
|
|
||||||
|
Просто продолжай диалог.
|
||||||
@@ -2,6 +2,20 @@
|
|||||||
name: dev
|
name: dev
|
||||||
description: Разработчик Ratatoskr — реализует задачу в репозитории по готовым критериям
|
description: Разработчик Ratatoskr — реализует задачу в репозитории по готовым критериям
|
||||||
mode: primary
|
mode: primary
|
||||||
|
tools:
|
||||||
|
read: true
|
||||||
|
write: true
|
||||||
|
edit: true
|
||||||
|
bash: true
|
||||||
|
grep: true
|
||||||
|
glob: true
|
||||||
|
websearch: false
|
||||||
|
webfetch: false
|
||||||
|
permission:
|
||||||
|
question: deny
|
||||||
|
edit: allow
|
||||||
|
bash:
|
||||||
|
"*": allow
|
||||||
---
|
---
|
||||||
|
|
||||||
Ты — dev-агент в конвейере Ratatoskr. Получаешь готовое задание и реализуешь его в коде.
|
Ты — dev-агент в конвейере Ratatoskr. Получаешь готовое задание и реализуешь его в коде.
|
||||||
|
|||||||
49
internal/agents/postmortem.md
Normal file
49
internal/agents/postmortem.md
Normal file
@@ -0,0 +1,49 @@
|
|||||||
|
---
|
||||||
|
name: postmortem
|
||||||
|
description: Постмортем-аналитик Ratatoskr — анализирует сессии dev/reviewer после failed/timeout задачи и даёт резюме: почему так и что сделать, чтобы не повторялось
|
||||||
|
mode: primary
|
||||||
|
tools:
|
||||||
|
read: false
|
||||||
|
write: false
|
||||||
|
edit: false
|
||||||
|
bash: false
|
||||||
|
grep: false
|
||||||
|
glob: false
|
||||||
|
list: false
|
||||||
|
websearch: false
|
||||||
|
webfetch: false
|
||||||
|
permission:
|
||||||
|
question: deny
|
||||||
|
edit: deny
|
||||||
|
read: deny
|
||||||
|
glob: deny
|
||||||
|
grep: deny
|
||||||
|
list: deny
|
||||||
|
bash:
|
||||||
|
"*": deny
|
||||||
|
webfetch: deny
|
||||||
|
websearch: deny
|
||||||
|
---
|
||||||
|
|
||||||
|
Ты — постмортем-аналитик в конвейере Ratatoskr. Задача завершилась неудачей (failed) или таймаутом (timeout). Ты анализируешь, что пошло не так, и даёшь резюме, которое поможет не допускать этого впредь.
|
||||||
|
|
||||||
|
Тебе приходит промпт с:
|
||||||
|
- задачей (название, цель, критерии готовности AC, репозитории, итоговый статус);
|
||||||
|
- сессиями субагентов dev и/или reviewer: их статус (success/failed/timeout), промпт и вывод (output).
|
||||||
|
|
||||||
|
ПРАВИЛА:
|
||||||
|
1. Проанализируй сессии dev и reviewer: какие в них проблемы, насколько завершён каждый этап (разработка, ревью).
|
||||||
|
2. Оцени «законченность этапов»: что успел сделать dev, проверял ли reviewer весь дифф, были ли заблокированы работы.
|
||||||
|
3. Сделай вывод — **почему так случилось**: ошибка в задании, неясные AC, технический сбой, неорганизованная работа агента и т.п.
|
||||||
|
4. Дай рекомендации — «что сделать, чтобы этого не было»: как уточнять задачу, какие AC добавлять, какой контекст предавать агентам, какие этапы конвейера ужесточить.
|
||||||
|
5. ПИШИ СВОЙ ОТВЕТ **ПРОСТЫМ ТЕКСТОМ НА РУССКОМ ЯЗЫКЕ**, без JSON, без markdown-обёрток и лишней разметки.
|
||||||
|
|
||||||
|
Формат ответа (два обязательных блока, коротко и по делу):
|
||||||
|
|
||||||
|
Почему так случилось:
|
||||||
|
- <причина 1>
|
||||||
|
- <причина 2>
|
||||||
|
|
||||||
|
Что сделать, чтобы это не повторялось:
|
||||||
|
- <рекомендация 1>
|
||||||
|
- <рекомендация 2>
|
||||||
@@ -2,6 +2,20 @@
|
|||||||
name: reviewer
|
name: reviewer
|
||||||
description: Ревьюер Ratatoskr — строго проверяет ветку работы dev-агента (качество, безопасность, SOLID) и возвращает вердикт в JSON
|
description: Ревьюер Ratatoskr — строго проверяет ветку работы dev-агента (качество, безопасность, SOLID) и возвращает вердикт в JSON
|
||||||
mode: primary
|
mode: primary
|
||||||
|
tools:
|
||||||
|
read: true
|
||||||
|
write: true
|
||||||
|
edit: false
|
||||||
|
bash: true
|
||||||
|
grep: true
|
||||||
|
glob: true
|
||||||
|
websearch: false
|
||||||
|
webfetch: false
|
||||||
|
permission:
|
||||||
|
question: deny
|
||||||
|
edit: deny
|
||||||
|
bash:
|
||||||
|
"*": allow
|
||||||
---
|
---
|
||||||
|
|
||||||
Ты — ревьюер в конвейере Ratatoskr. Проверяешь работу dev-агента в feature-ветке **жёстко и придирчиво**. Твоя цель — не дать плохому коду попасть в основную ветку.
|
Ты — ревьюер в конвейере Ratatoskr. Проверяешь работу dev-агента в feature-ветке **жёстко и придирчиво**. Твоя цель — не дать плохому коду попасть в основную ветку.
|
||||||
|
|||||||
@@ -54,7 +54,7 @@ type AnalystResponse struct {
|
|||||||
Title string `json:"title"`
|
Title string `json:"title"`
|
||||||
Goal string `json:"goal"`
|
Goal string `json:"goal"`
|
||||||
Repo string `json:"repo"` // одиночный репо (обратная совместимость)
|
Repo string `json:"repo"` // одиночный репо (обратная совместимость)
|
||||||
Repos []string `json:"repos"` // список репо (основной)
|
Repos json.RawMessage `json:"repos"` // список репо (основной); устойчив к строке
|
||||||
Why string `json:"why"`
|
Why string `json:"why"`
|
||||||
AC string `json:"ac"`
|
AC string `json:"ac"`
|
||||||
Questions []string `json:"questions"`
|
Questions []string `json:"questions"`
|
||||||
@@ -62,6 +62,30 @@ type AnalystResponse struct {
|
|||||||
AbortReason string `json:"abort_reason"`
|
AbortReason string `json:"abort_reason"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// reposList нормализует поле repos: модель может вернуть либо массив
|
||||||
|
// ["a","b"], либо строку "a,b" (иногда с пробелами). Пустое значение → nil.
|
||||||
|
func (r *AnalystResponse) reposList() []string {
|
||||||
|
if r.Repos == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var arr []string
|
||||||
|
if err := json.Unmarshal(r.Repos, &arr); err == nil {
|
||||||
|
return arr
|
||||||
|
}
|
||||||
|
var s string
|
||||||
|
if err := json.Unmarshal(r.Repos, &s); err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
var out []string
|
||||||
|
for _, p := range strings.Split(s, ",") {
|
||||||
|
p = strings.TrimSpace(p)
|
||||||
|
if p != "" {
|
||||||
|
out = append(out, p)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
// Decide реализует core.Decider через открытый код.
|
// Decide реализует core.Decider через открытый код.
|
||||||
func (a *Analyst) Decide(ctx context.Context, history []core.Message, draft storage.Task, force bool) (core.Decision, error) {
|
func (a *Analyst) Decide(ctx context.Context, history []core.Message, draft storage.Task, force bool) (core.Decision, error) {
|
||||||
agent := a.Agent
|
agent := a.Agent
|
||||||
@@ -104,12 +128,7 @@ func (a *Analyst) Decide(ctx context.Context, history []core.Message, draft stor
|
|||||||
|
|
||||||
// 4. парсим вердикт
|
// 4. парсим вердикт
|
||||||
verdict := opencode.ExtractVerdict(res.Stdout)
|
verdict := opencode.ExtractVerdict(res.Stdout)
|
||||||
log.Printf("analyst: verdict tail: %s", truncate(verdict, 2000))
|
|
||||||
obj, ok := opencode.ExtractJSON(verdict)
|
obj, ok := opencode.ExtractJSON(verdict)
|
||||||
if ok {
|
|
||||||
b, _ := json.Marshal(obj)
|
|
||||||
log.Printf("analyst: extracted json: %s", truncate(string(b), 1000))
|
|
||||||
}
|
|
||||||
if !ok {
|
if !ok {
|
||||||
return core.Decision{}, fmt.Errorf("%w: нет JSON в выводе аналитика", ErrDecodeFail)
|
return core.Decision{}, fmt.Errorf("%w: нет JSON в выводе аналитика", ErrDecodeFail)
|
||||||
}
|
}
|
||||||
@@ -123,6 +142,8 @@ func (a *Analyst) Decide(ctx context.Context, history []core.Message, draft stor
|
|||||||
return core.Decision{}, fmt.Errorf("%w: %v", ErrDecodeFail, err)
|
return core.Decision{}, fmt.Errorf("%w: %v", ErrDecodeFail, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
log.Printf("analyst: вердикт: %s", truncate(formatVerdict(&ar), 2000))
|
||||||
|
|
||||||
// 5. валидация
|
// 5. валидация
|
||||||
if err := validateResponse(&ar); err != nil {
|
if err := validateResponse(&ar); err != nil {
|
||||||
return core.Decision{}, fmt.Errorf("%w: %v", ErrValidation, err)
|
return core.Decision{}, fmt.Errorf("%w: %v", ErrValidation, err)
|
||||||
@@ -144,10 +165,10 @@ func (a *Analyst) Decide(ctx context.Context, history []core.Message, draft stor
|
|||||||
if ar.Repo != "" {
|
if ar.Repo != "" {
|
||||||
dec.Draft.Repo = ar.Repo
|
dec.Draft.Repo = ar.Repo
|
||||||
}
|
}
|
||||||
if len(ar.Repos) > 0 {
|
if repos := ar.reposList(); len(repos) > 0 {
|
||||||
dec.Draft.Repos = ar.Repos
|
dec.Draft.Repos = repos
|
||||||
// Синхронизируем одиночный repo для старых потребителей.
|
// Синхронизируем одиночный repo для старых потребителей.
|
||||||
dec.Draft.Repo = strings.Join(ar.Repos, ",")
|
dec.Draft.Repo = strings.Join(repos, ",")
|
||||||
}
|
}
|
||||||
if ar.Why != "" {
|
if ar.Why != "" {
|
||||||
dec.Draft.Why = ar.Why
|
dec.Draft.Why = ar.Why
|
||||||
@@ -195,3 +216,50 @@ func validateResponse(ar *AnalystResponse) error {
|
|||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// formatVerdict собирает человекочитаемое однострочное описание вердикта
|
||||||
|
// аналитика (без JSON-разметки) для панели «Логи».
|
||||||
|
func formatVerdict(ar *AnalystResponse) string {
|
||||||
|
var b strings.Builder
|
||||||
|
b.WriteString("phase=" + ar.Phase)
|
||||||
|
|
||||||
|
if ar.ChatReply != "" {
|
||||||
|
b.WriteString(", chat_reply=")
|
||||||
|
b.WriteString(ar.ChatReply)
|
||||||
|
}
|
||||||
|
if len(ar.Questions) > 0 {
|
||||||
|
b.WriteString(", questions=[")
|
||||||
|
b.WriteString(strings.Join(ar.Questions, " | "))
|
||||||
|
b.WriteString("]")
|
||||||
|
}
|
||||||
|
if ar.Title != "" {
|
||||||
|
b.WriteString(", title=")
|
||||||
|
b.WriteString(ar.Title)
|
||||||
|
}
|
||||||
|
if ar.Goal != "" {
|
||||||
|
b.WriteString(", goal=")
|
||||||
|
b.WriteString(ar.Goal)
|
||||||
|
}
|
||||||
|
if ar.Repo != "" {
|
||||||
|
b.WriteString(", repo=")
|
||||||
|
b.WriteString(ar.Repo)
|
||||||
|
}
|
||||||
|
if repos := ar.reposList(); len(repos) > 0 {
|
||||||
|
b.WriteString(", repos=[")
|
||||||
|
b.WriteString(strings.Join(repos, ", "))
|
||||||
|
b.WriteString("]")
|
||||||
|
}
|
||||||
|
if ar.Why != "" {
|
||||||
|
b.WriteString(", why=")
|
||||||
|
b.WriteString(ar.Why)
|
||||||
|
}
|
||||||
|
if ar.AC != "" {
|
||||||
|
b.WriteString(", ac=")
|
||||||
|
b.WriteString(ar.AC)
|
||||||
|
}
|
||||||
|
if ar.AbortReason != "" {
|
||||||
|
b.WriteString(", abort_reason=")
|
||||||
|
b.WriteString(ar.AbortReason)
|
||||||
|
}
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
@@ -2,7 +2,9 @@ package analyst
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"github.com/kamelion/ratatoskr-go/internal/core"
|
"github.com/kamelion/ratatoskr-go/internal/core"
|
||||||
@@ -216,3 +218,87 @@ func TestAskEmptyReplyAndQuestions(t *testing.T) {
|
|||||||
t.Errorf("err = %v, want A3", err)
|
t.Errorf("err = %v, want A3", err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestDecideProposeStringRepos — модель вернула repos строкой (а не массивом):
|
||||||
|
// парсер должен нормализовать и не падать.
|
||||||
|
func TestDecideProposeStringRepos(t *testing.T) {
|
||||||
|
a := &Analyst{Runner: &mockRunner{result: &opencode.Result{
|
||||||
|
RC: 0,
|
||||||
|
Stdout: `{"type":"text","part":{"text":"{\"phase\":\"propose\",\"title\":\"Калькулятор\",\"repos\":\"tools/calc, tools/ui\"}"}}`,
|
||||||
|
}}, Worktree: "/tmp"}
|
||||||
|
|
||||||
|
history := []core.Message{{Role: "user", Content: "Сделай калькулятор"}}
|
||||||
|
dec, err := a.Decide(context.Background(), history, storage.Task{}, false)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Decide err: %v", err)
|
||||||
|
}
|
||||||
|
if dec.Phase != "propose" {
|
||||||
|
t.Errorf("Phase = %q, want propose", dec.Phase)
|
||||||
|
}
|
||||||
|
if dec.Draft.Repos[0] != "tools/calc" || dec.Draft.Repos[1] != "tools/ui" {
|
||||||
|
t.Errorf("Repos = %#v, want [tools/calc tools/ui]", dec.Draft.Repos)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestFormatVerdict — человекочитаемое описание вердикта аналитика.
|
||||||
|
func TestFormatVerdict(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
ar *AnalystResponse
|
||||||
|
want []string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "propose with fields",
|
||||||
|
ar: &AnalystResponse{
|
||||||
|
Phase: "propose",
|
||||||
|
Title: "Калькулятор",
|
||||||
|
Goal: "Сделать веб-калькулятор",
|
||||||
|
Repos: json.RawMessage(`["tools/calc","tools/ui"]`),
|
||||||
|
Why: "Нужен для учёта",
|
||||||
|
AC: "Работает + - * /",
|
||||||
|
ChatReply: "Готово!",
|
||||||
|
},
|
||||||
|
want: []string{
|
||||||
|
"phase=propose",
|
||||||
|
"chat_reply=Готово!",
|
||||||
|
"repos=[tools/calc, tools/ui]",
|
||||||
|
"title=Калькулятор",
|
||||||
|
"goal=Сделать веб-калькулятор",
|
||||||
|
"why=Нужен для учёта",
|
||||||
|
"ac=Работает + - * /",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "ask with questions",
|
||||||
|
ar: &AnalystResponse{
|
||||||
|
Phase: "ask",
|
||||||
|
ChatReply: "Уточню",
|
||||||
|
Questions: []string{"Где код?", "Какая цель?"},
|
||||||
|
},
|
||||||
|
want: []string{"phase=ask", "chat_reply=Уточню", "questions=[Где код? | Какая цель?]"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "abort with reason",
|
||||||
|
ar: &AnalystResponse{
|
||||||
|
Phase: "abort",
|
||||||
|
AbortReason: "Тема не про код",
|
||||||
|
},
|
||||||
|
want: []string{"phase=abort", "abort_reason=Тема не про код"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "empty verdict",
|
||||||
|
ar: &AnalystResponse{},
|
||||||
|
want: []string{"phase="},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
got := formatVerdict(tt.ar)
|
||||||
|
for _, w := range tt.want {
|
||||||
|
if !strings.Contains(got, w) {
|
||||||
|
t.Errorf("formatVerdict = %q, want contain %q", got, w)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -11,6 +11,7 @@ import (
|
|||||||
"os/signal"
|
"os/signal"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
"sync"
|
||||||
"syscall"
|
"syscall"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -52,7 +53,7 @@ const packageOwner = "kamelion"
|
|||||||
// не следует путать с build-идентификатором `main.version` (commit-<sha7>),
|
// не следует путать с build-идентификатором `main.version` (commit-<sha7>),
|
||||||
// который вшивается ldflag'ом и используется автообновлением. Здесь номер
|
// который вшивается ldflag'ом и используется автообновлением. Здесь номер
|
||||||
// поднимается вручную перед каждым релизом/публикацией новой сборки.
|
// поднимается вручную перед каждым релизом/публикацией новой сборки.
|
||||||
const Version = "0.2.2"
|
const Version = "0.3.0"
|
||||||
|
|
||||||
// App — собранный конвейер.
|
// App — собранный конвейер.
|
||||||
type App struct {
|
type App struct {
|
||||||
@@ -62,7 +63,7 @@ type App struct {
|
|||||||
CoreCtx *core.Core
|
CoreCtx *core.Core
|
||||||
Worker *worker.Worker
|
Worker *worker.Worker
|
||||||
Updater *update.Updater
|
Updater *update.Updater
|
||||||
tg *telegram.Channel // сохранена для Run
|
tg *telegram.Channel // сохранена для Run (nil, если telegram.enabled=false)
|
||||||
pool *opencode.Pool // пул opencode serve-серверов (API-режим)
|
pool *opencode.Pool // пул opencode serve-серверов (API-режим)
|
||||||
|
|
||||||
// Events — доменная шина UI; LogEvents — шина логов (панель «Логи»).
|
// Events — доменная шина UI; LogEvents — шина логов (панель «Логи»).
|
||||||
@@ -73,6 +74,42 @@ type App struct {
|
|||||||
// nil — headless-режим (без окна).
|
// nil — headless-режим (без окна).
|
||||||
UI ui.Window
|
UI ui.Window
|
||||||
Controller *ui.Controller
|
Controller *ui.Controller
|
||||||
|
// FreeChat — свободные диалоги с LLM вне машины задач (in-memory).
|
||||||
|
FreeChat *FreeChat
|
||||||
|
}
|
||||||
|
|
||||||
|
// FreeChat — свободный LLM-чат (без задачи). Каждый UserID (вкладка UI)
|
||||||
|
// держит свою opencode-сессию, чтобы контекст вкладки не смешивался.
|
||||||
|
// Транскрипт живёт в памяти окна; здесь только связка uid → opencode sessionID.
|
||||||
|
type FreeChat struct {
|
||||||
|
Runner analyst.OpenCodeRunner
|
||||||
|
Worktree string
|
||||||
|
mu sync.Mutex // защищает sessions (пишется из воркеров роутера)
|
||||||
|
sessions map[chat.UserID]string // uid → opencode sessionID
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewFreeChat создаёт сервис свободного чата.
|
||||||
|
func NewFreeChat(runner analyst.OpenCodeRunner, worktree string) *FreeChat {
|
||||||
|
return &FreeChat{Runner: runner, Worktree: worktree, sessions: map[chat.UserID]string{}}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Chat отправляет текст пользователя свободной сессии в opencode (agent="chat")
|
||||||
|
// и возвращает ответ модели. Сессия продолжается (resume по sessionID), поэтому
|
||||||
|
// каждая вкладка ведёт свой независимый диалог.
|
||||||
|
func (f *FreeChat) Chat(ctx context.Context, uid chat.UserID, text string) (string, error) {
|
||||||
|
f.mu.Lock()
|
||||||
|
sid := f.sessions[uid]
|
||||||
|
f.mu.Unlock()
|
||||||
|
|
||||||
|
res, err := f.Runner.Run(ctx, text, f.Worktree, "chat", sid)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
f.mu.Lock()
|
||||||
|
f.sessions[uid] = res.SessionID
|
||||||
|
f.mu.Unlock()
|
||||||
|
return res.Stdout, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// New читает конфиг и собирает все зависимости.
|
// New читает конфиг и собирает все зависимости.
|
||||||
@@ -119,8 +156,6 @@ func New(configPath, version, updateToken string, noUI bool) (*App, error) {
|
|||||||
// Служебный root-сервер (worktree) живёт всё время app; остальные лениво.
|
// Служебный root-сервер (worktree) живёт всё время app; остальные лениво.
|
||||||
ocPool := opencode.NewPool(cfg.Paths.Worktree)
|
ocPool := opencode.NewPool(cfg.Paths.Worktree)
|
||||||
ocPool.Bin = cfg.OpenCode.Bin
|
ocPool.Bin = cfg.OpenCode.Bin
|
||||||
ocPool.Config = cfg.OpenCode.Config
|
|
||||||
ocPool.ConfigDir = cfg.OpenCode.ConfigDir
|
|
||||||
ocPool.DBPath = cfg.OpenCode.DBPath
|
ocPool.DBPath = cfg.OpenCode.DBPath
|
||||||
ocPool.Host = cfg.OpenCode.Serve.Hostname
|
ocPool.Host = cfg.OpenCode.Serve.Hostname
|
||||||
ocPool.BasePort = cfg.OpenCode.Serve.Port
|
ocPool.BasePort = cfg.OpenCode.Serve.Port
|
||||||
@@ -187,13 +222,15 @@ func New(configPath, version, updateToken string, noUI bool) (*App, error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Telegram-канал
|
// Telegram-канал (если включён; иначе сетевые вызовы к Telegram не идут)
|
||||||
|
if cfg.Telegram.Enabled {
|
||||||
tg := telegram.New(cfg.Telegram.Token, cfg.Chat.PollInterval.Duration())
|
tg := telegram.New(cfg.Telegram.Token, cfg.Chat.PollInterval.Duration())
|
||||||
a.tg = tg
|
a.tg = tg
|
||||||
if err := router.Attach(tg); err != nil {
|
if err := router.Attach(tg); err != nil {
|
||||||
store.Close()
|
store.Close()
|
||||||
return nil, fmt.Errorf("attach telegram: %w", err)
|
return nil, fmt.Errorf("attach telegram: %w", err)
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Worker — polling-планировщик dev-агента
|
// Worker — polling-планировщик dev-агента
|
||||||
w := &worker.Worker{
|
w := &worker.Worker{
|
||||||
@@ -212,6 +249,9 @@ func New(configPath, version, updateToken string, noUI bool) (*App, error) {
|
|||||||
a.Router = router
|
a.Router = router
|
||||||
a.Worker = w
|
a.Worker = w
|
||||||
|
|
||||||
|
// Свободный LLM-чат (вкладки UI без задачи): общий Runner, свой sessionID на uid.
|
||||||
|
a.FreeChat = NewFreeChat(ocRunner, cfg.Paths.Worktree)
|
||||||
|
|
||||||
// /status N читает живое состояние сессии через liveProbe-адаптер.
|
// /status N читает живое состояние сессии через liveProbe-адаптер.
|
||||||
coreCtx.Live = &liveProbe{reg: live}
|
coreCtx.Live = &liveProbe{reg: live}
|
||||||
|
|
||||||
@@ -251,15 +291,15 @@ func (a *App) Run(ctx context.Context) error {
|
|||||||
}
|
}
|
||||||
defer a.pool.Close()
|
defer a.pool.Close()
|
||||||
|
|
||||||
// Канал для проверки Telegram-ошибки (горутина оборачивает Run)
|
// Telegram: long-poll цикл в горутине (только если канал включён)
|
||||||
tgErr := make(chan error, 1)
|
tgErr := make(chan error, 1)
|
||||||
|
if a.telegramChannel() != nil {
|
||||||
// Telegram: long-poll цикл в горутине
|
|
||||||
go func() {
|
go func() {
|
||||||
tg := a.telegramChannel()
|
tg := a.telegramChannel()
|
||||||
log.Print("app: telegram poll started")
|
log.Print("app: telegram poll started")
|
||||||
tgErr <- tg.Run(ctx)
|
tgErr <- tg.Run(ctx)
|
||||||
}()
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
// Worker: poll-цикл (неблокирующий — стартует свою горутину)
|
// Worker: poll-цикл (неблокирующий — стартует свою горутину)
|
||||||
a.Worker.Start(ctx)
|
a.Worker.Start(ctx)
|
||||||
@@ -280,6 +320,7 @@ func (a *App) Run(ctx context.Context) error {
|
|||||||
sigCh := make(chan os.Signal, 1)
|
sigCh := make(chan os.Signal, 1)
|
||||||
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
|
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
|
||||||
|
|
||||||
|
if a.telegramChannel() != nil {
|
||||||
select {
|
select {
|
||||||
case <-ctx.Done():
|
case <-ctx.Done():
|
||||||
log.Print("app: context cancelled")
|
log.Print("app: context cancelled")
|
||||||
@@ -294,6 +335,17 @@ func (a *App) Run(ctx context.Context) error {
|
|||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
log.Print("app: context cancelled")
|
||||||
|
return nil
|
||||||
|
case sig := <-sigCh:
|
||||||
|
log.Printf("app: signal %s — shutting down", sig)
|
||||||
|
cancel()
|
||||||
|
return nil
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// telegramChannel возвращает Telegram-канал.
|
// telegramChannel возвращает Telegram-канал.
|
||||||
@@ -363,6 +415,10 @@ func (a *App) handleIncoming(inc chat.Incoming) {
|
|||||||
}
|
}
|
||||||
a.send(ctx, uid, result.Reply)
|
a.send(ctx, uid, result.Reply)
|
||||||
|
|
||||||
|
case isUISession(uid):
|
||||||
|
// Свободная UI-вкладка (нет активной задачи): свободный LLM-чат.
|
||||||
|
a.freeChatReply(ctx, uid, text)
|
||||||
|
|
||||||
default:
|
default:
|
||||||
a.send(ctx, uid, "Нет активной задачи. /start — создать новую.")
|
a.send(ctx, uid, "Нет активной задачи. /start — создать новую.")
|
||||||
}
|
}
|
||||||
@@ -427,6 +483,26 @@ func hasArg(text string) bool {
|
|||||||
return i > 0 && strings.TrimSpace(s[i:]) != ""
|
return i > 0 && strings.TrimSpace(s[i:]) != ""
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// isUISession — true, если UserID принадлежит UI-вкладке-сессии (ui://sess-*).
|
||||||
|
func isUISession(uid chat.UserID) bool {
|
||||||
|
return strings.HasPrefix(string(uid), ui.SessionID)
|
||||||
|
}
|
||||||
|
|
||||||
|
// freeChatReply — ответ свободной LLM-сессии (вкладка UI без активной задачи).
|
||||||
|
func (a *App) freeChatReply(ctx context.Context, uid chat.UserID, text string) {
|
||||||
|
if a.FreeChat == nil {
|
||||||
|
a.send(ctx, uid, "Нет активной задачи. /start — создать новую.")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
reply, err := a.FreeChat.Chat(ctx, uid, text)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("app: free chat %s: %v", uid, err)
|
||||||
|
a.send(ctx, uid, "Ошибка LLM-чата.")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
a.send(ctx, uid, reply)
|
||||||
|
}
|
||||||
|
|
||||||
// ownerUID — куда слать авто-уведомления об обновлении.
|
// ownerUID — куда слать авто-уведомления об обновлении.
|
||||||
func (a *App) ownerUID() chat.UserID {
|
func (a *App) ownerUID() chat.UserID {
|
||||||
if a.Config.Telegram.ChatID != "" {
|
if a.Config.Telegram.ChatID != "" {
|
||||||
@@ -587,23 +663,21 @@ func (a *App) handleUpdateCmd(ctx context.Context, uid chat.UserID) {
|
|||||||
// сюда не возвращаемся — SwapAndRestart завершил процесс (os.Exit)
|
// сюда не возвращаемся — SwapAndRestart завершил процесс (os.Exit)
|
||||||
}
|
}
|
||||||
|
|
||||||
// ensureAgentsDir определяет каталог с агентами opencode и распаковывает
|
// ensureAgentsDir распаковывает встроенных агентов opencode (go:embed) в
|
||||||
// туда встроенных агентов (go:embed). Если config_dir не задан — использует
|
// project-каталог <worktree>/.opencode/agent/*.md. opencode при serve (cwd =
|
||||||
// ./agents рядом с бинарём. Встроенные агенты перезаписываются (всегда актуальны).
|
// worktree) обнаруживает .opencode поднятием вверх (paths.ts) и сам находит
|
||||||
|
// агентов. Глобальный конфиг opencode (модель/провайдеры) не трогаем и
|
||||||
|
// OPENCODE_CONFIG_DIR не выставляем.
|
||||||
//
|
//
|
||||||
// idempotent: вызывается только из New.
|
// idempotent: вызывается только из New.
|
||||||
func ensureAgentsDir(cfg *config.Config) error {
|
func ensureAgentsDir(cfg *config.Config) error {
|
||||||
if cfg.OpenCode.ConfigDir == "" {
|
if cfg.Paths.Worktree == "" {
|
||||||
exe, err := os.Executable()
|
return fmt.Errorf("paths.worktree не задан — некуда раскладывать агентов")
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("resolve executable: %w", err)
|
|
||||||
}
|
}
|
||||||
// ./agents рядом с бинарём
|
base := filepath.Join(cfg.Paths.Worktree, ".opencode")
|
||||||
cfg.OpenCode.ConfigDir = filepath.Join(filepath.Dir(exe), "agents")
|
if err := agents.WriteTo(base); err != nil {
|
||||||
|
return fmt.Errorf("write agents to %s: %w", base, err)
|
||||||
}
|
}
|
||||||
if err := agents.WriteTo(cfg.OpenCode.ConfigDir); err != nil {
|
log.Printf("app: agents ensured in %s", base)
|
||||||
return fmt.Errorf("write agents to %s: %w", cfg.OpenCode.ConfigDir, err)
|
|
||||||
}
|
|
||||||
log.Printf("app: agents ensured in %s", cfg.OpenCode.ConfigDir)
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -21,17 +21,35 @@ type Router struct {
|
|||||||
// Hook, вызываемый на каждое входящее событие (обычно → process_turn).
|
// Hook, вызываемый на каждое входящее событие (обычно → process_turn).
|
||||||
onUserMsg func(Incoming)
|
onUserMsg func(Incoming)
|
||||||
|
|
||||||
// Асинхронная обработка входящих: handleIncoming кладёт событие в канал,
|
// Асинхронная обработка входящих ограниченным пулом воркеров с
|
||||||
// воркер-горутина последовательно вызывает onUserMsg. Благодаря этому
|
// упорядоченными очередями per-user (см. chatWorkers, userState).
|
||||||
// long-poll цикл канала (Telegram) не блокируется на время долгого
|
// Благодаря этому long-poll цикл канала (Telegram) не блокируется на время
|
||||||
// вызова аналитика и продолжает принимать новые сообщения.
|
// долгого вызова аналитика, а сообщения разных пользователей не сериализуются
|
||||||
incoming chan Incoming
|
// друг за другом: каждый активный пользователь занимает своего воркера.
|
||||||
|
jobs chan *userState
|
||||||
|
users map[UserID]*userState
|
||||||
|
userMu sync.Mutex
|
||||||
|
|
||||||
// processed — число обработанных воркером событий (для синхронизации
|
// processed — число обработанных воркером событий (для синхронизации
|
||||||
// тестов с асинхронной очередью: WaitProcessed ждёт обработку события).
|
// тестов с асинхронной очередью: WaitProcessed ждёт обработку события).
|
||||||
processed atomic.Int64
|
processed atomic.Int64
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// userState — FIFO-очередь входящих одного пользователя. В каждый момент
|
||||||
|
// для пользователя активен ровно один воркер (scheduled), поэтому порядок
|
||||||
|
// обработки его сообщений сохраняется, а параллелизм достигается между
|
||||||
|
// разными пользователями.
|
||||||
|
type userState struct {
|
||||||
|
mu sync.Mutex
|
||||||
|
pending []Incoming
|
||||||
|
scheduled bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// chatWorkers — число воркеров обработки входящих. Ограничивает количество
|
||||||
|
// одновременных тяжёлых LLM-вызовов (аналитик/свободный чат), чтобы поток
|
||||||
|
// каналов не упирался в один долгий вызов.
|
||||||
|
const chatWorkers = 4
|
||||||
|
|
||||||
// NewRouter создаёт роутер. onUserMsg — колбэк обработки входящего.
|
// NewRouter создаёт роутер. onUserMsg — колбэк обработки входящего.
|
||||||
func NewRouter(onUserMsg func(Incoming)) *Router {
|
func NewRouter(onUserMsg func(Incoming)) *Router {
|
||||||
if onUserMsg == nil {
|
if onUserMsg == nil {
|
||||||
@@ -42,15 +60,51 @@ func NewRouter(onUserMsg func(Incoming)) *Router {
|
|||||||
routes: map[UserID]Route{},
|
routes: map[UserID]Route{},
|
||||||
pending: map[UserID]PendingQ{},
|
pending: map[UserID]PendingQ{},
|
||||||
onUserMsg: onUserMsg,
|
onUserMsg: onUserMsg,
|
||||||
incoming: make(chan Incoming, 256),
|
jobs: make(chan *userState, chatWorkers),
|
||||||
|
users: make(map[UserID]*userState),
|
||||||
|
}
|
||||||
|
for i := 0; i < chatWorkers; i++ {
|
||||||
|
go r.workerLoop()
|
||||||
}
|
}
|
||||||
go r.processLoop()
|
|
||||||
return r
|
return r
|
||||||
}
|
}
|
||||||
|
|
||||||
// processLoop — воркер асинхронной обработки входящих (FIFO).
|
// userStateOf возвращает очередь пользователя (создаёт при первом сообщении).
|
||||||
func (r *Router) processLoop() {
|
// Очереди живут вечно — по одной маленькой структуре на пользователя/вкладку.
|
||||||
for inc := range r.incoming {
|
func (r *Router) userStateOf(uid UserID) *userState {
|
||||||
|
r.userMu.Lock()
|
||||||
|
defer r.userMu.Unlock()
|
||||||
|
st, ok := r.users[uid]
|
||||||
|
if !ok {
|
||||||
|
st = &userState{}
|
||||||
|
r.users[uid] = st
|
||||||
|
}
|
||||||
|
return st
|
||||||
|
}
|
||||||
|
|
||||||
|
// workerLoop — воркер пула: берёт пользователя из общей очереди и дренит его.
|
||||||
|
func (r *Router) workerLoop() {
|
||||||
|
for st := range r.jobs {
|
||||||
|
r.runUser(st)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// runUser обрабатывает все накопленные сообщения пользователя по порядку.
|
||||||
|
// По исчерпании очереди снимает scheduled — следующий handleIncoming вновь
|
||||||
|
// поставит пользователя в jobs. Возвращается в workerLoop, чтобы тот взял
|
||||||
|
// следующего пользователя из общей очереди.
|
||||||
|
func (r *Router) runUser(st *userState) {
|
||||||
|
for {
|
||||||
|
st.mu.Lock()
|
||||||
|
if len(st.pending) == 0 {
|
||||||
|
st.scheduled = false
|
||||||
|
st.mu.Unlock()
|
||||||
|
return
|
||||||
|
}
|
||||||
|
inc := st.pending[0]
|
||||||
|
st.pending = st.pending[1:]
|
||||||
|
st.mu.Unlock()
|
||||||
|
|
||||||
r.onUserMsg(inc)
|
r.onUserMsg(inc)
|
||||||
r.processed.Add(1)
|
r.processed.Add(1)
|
||||||
}
|
}
|
||||||
@@ -99,9 +153,19 @@ func (r *Router) handleIncoming(inc Incoming) {
|
|||||||
}
|
}
|
||||||
r.mu.Unlock()
|
r.mu.Unlock()
|
||||||
|
|
||||||
// Асинхронная обработка: кладём событие в очередь воркера и сразу
|
// Асинхронная обработка: кладём событие в FIFO-очередь пользователя и
|
||||||
// возвращаемся, не блокируя вызывающий long-poll цикл канала.
|
// сразу возвращаемся, не блокируя вызывающий long-poll цикл канала.
|
||||||
r.incoming <- inc
|
// Если пользователь ещё не обрабатывается — ставим его в общую очередь
|
||||||
|
// пула воркеров. Backpressure по jobs блокирует только перегруженного
|
||||||
|
// пользователя (его собственную горутину канала), не весь роутер.
|
||||||
|
st := r.userStateOf(inc.UserID)
|
||||||
|
st.mu.Lock()
|
||||||
|
st.pending = append(st.pending, inc)
|
||||||
|
if !st.scheduled {
|
||||||
|
st.scheduled = true
|
||||||
|
r.jobs <- st
|
||||||
|
}
|
||||||
|
st.mu.Unlock()
|
||||||
}
|
}
|
||||||
|
|
||||||
// Send уведомляет пользователя через текущий маршрут. M1 (нет маршрута) — no-op,
|
// Send уведомляет пользователя через текущий маршрут. M1 (нет маршрута) — no-op,
|
||||||
|
|||||||
@@ -213,3 +213,105 @@ func TestRouter_Ask_PendingNotConsumedFromOtherAddr(t *testing.T) {
|
|||||||
t.Fatal("pending должен остаться (ответ из другого адреса)")
|
t.Fatal("pending должен остаться (ответ из другого адреса)")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestRouter_PerUserOrdering проверяет, что сообщения одного пользователя
|
||||||
|
// обрабатываются строго в порядке поступления (пул воркеров не перемешивает).
|
||||||
|
func TestRouter_PerUserOrdering(t *testing.T) {
|
||||||
|
cb := newFakeOnMsg()
|
||||||
|
r := NewRouter(cb.h)
|
||||||
|
tgCh := newFakeChannel(tg)
|
||||||
|
_ = r.Attach(tgCh)
|
||||||
|
|
||||||
|
for _, txt := range []string{"1", "2", "3"} {
|
||||||
|
tgCh.emit(uidA, tg, txt)
|
||||||
|
}
|
||||||
|
if !cb.wait(3) {
|
||||||
|
t.Fatal("сообщения не обработаны за таймаут")
|
||||||
|
}
|
||||||
|
for i, want := range []string{"1", "2", "3"} {
|
||||||
|
if got := cb.get(i).Msg.Text; got != want {
|
||||||
|
t.Errorf("порядок обработки нарушен: idx %d = %q, want %q", i, got, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestRouter_ParallelismAcrossUsers проверяет, что пока обработчик одного
|
||||||
|
// пользователя заблокирован (долгий LLM-вызов), сообщение другого пользователя
|
||||||
|
// обрабатывается в другом воркере, а второе сообщение того же пользователя —
|
||||||
|
// ждёт своей очереди (per-user порядок).
|
||||||
|
func TestRouter_ParallelismAcrossUsers(t *testing.T) {
|
||||||
|
r := NewRouter(nil)
|
||||||
|
|
||||||
|
tgCh := newFakeChannel(tg)
|
||||||
|
tuiCh := newFakeChannel(tui)
|
||||||
|
_ = r.Attach(tgCh)
|
||||||
|
_ = r.Attach(tuiCh)
|
||||||
|
|
||||||
|
blocked := make(chan struct{})
|
||||||
|
release := make(chan struct{})
|
||||||
|
var muLocal sync.Mutex
|
||||||
|
seen := make([]string, 0, 3)
|
||||||
|
signal := make(chan struct{}, 8)
|
||||||
|
|
||||||
|
h := func(inc Incoming) {
|
||||||
|
if inc.Msg.Text == "block" {
|
||||||
|
close(blocked)
|
||||||
|
<-release // держим воркера, пока не отпустим
|
||||||
|
}
|
||||||
|
muLocal.Lock()
|
||||||
|
seen = append(seen, inc.Msg.Text)
|
||||||
|
muLocal.Unlock()
|
||||||
|
signal <- struct{}{}
|
||||||
|
}
|
||||||
|
r.onUserMsg = h
|
||||||
|
|
||||||
|
snapshot := func() []string {
|
||||||
|
muLocal.Lock()
|
||||||
|
defer muLocal.Unlock()
|
||||||
|
return append([]string(nil), seen...)
|
||||||
|
}
|
||||||
|
waitFor := func(n int) bool {
|
||||||
|
deadline := time.After(2 * time.Second)
|
||||||
|
for len(snapshot()) < n {
|
||||||
|
select {
|
||||||
|
case <-signal:
|
||||||
|
case <-deadline:
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// первое сообщение A блокирует своего воркера
|
||||||
|
tgCh.emit(uidA, tg, "block")
|
||||||
|
<-blocked
|
||||||
|
|
||||||
|
// B обрабатывается параллельно, пока A висит
|
||||||
|
tuiCh.emit(uidB, tui, "B1")
|
||||||
|
select {
|
||||||
|
case <-signal:
|
||||||
|
case <-time.After(100 * time.Millisecond):
|
||||||
|
t.Fatal("B не обработан, пока блокирован A — чат-путь снова сериализован")
|
||||||
|
}
|
||||||
|
if got := snapshot(); len(got) != 1 || got[0] != "B1" {
|
||||||
|
t.Fatalf("ожидали обработку B1, got %v", got)
|
||||||
|
}
|
||||||
|
|
||||||
|
// второе сообщение A НЕ обрабатывается, пока занят воркер A (порядок per-user)
|
||||||
|
tgCh.emit(uidA, tg, "a2")
|
||||||
|
select {
|
||||||
|
case <-signal:
|
||||||
|
t.Fatal("сообщение A обработано ДО освобождения A — нарушен per-user порядок")
|
||||||
|
case <-time.After(80 * time.Millisecond):
|
||||||
|
}
|
||||||
|
|
||||||
|
// отпускаем A → дообрабатывается a2
|
||||||
|
close(release)
|
||||||
|
if !waitFor(3) {
|
||||||
|
t.Fatal("итоговые сообщения не обработаны")
|
||||||
|
}
|
||||||
|
got := snapshot()
|
||||||
|
if got[2] != "a2" {
|
||||||
|
t.Errorf("порядок персональной очереди нарушен: pos2 = %q, want a2", got[2])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -146,54 +146,6 @@ log:
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestLoad_OpenCodeConfigDir(t *testing.T) {
|
|
||||||
t.Setenv("TG_TOKEN", "tok")
|
|
||||||
t.Setenv("TG_CHAT_ID", "42")
|
|
||||||
|
|
||||||
// без config_dir — пусто
|
|
||||||
yaml := `telegram:
|
|
||||||
token: "${TG_TOKEN}"
|
|
||||||
chat_id: "${TG_CHAT_ID}"
|
|
||||||
`
|
|
||||||
cfg, err := Load(writeCfg(t, yaml))
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("Load: %v", err)
|
|
||||||
}
|
|
||||||
if cfg.OpenCode.ConfigDir != "" {
|
|
||||||
t.Errorf("config_dir = %q, want пусто", cfg.OpenCode.ConfigDir)
|
|
||||||
}
|
|
||||||
|
|
||||||
// через env OPENCODE_CONFIG_DIR
|
|
||||||
t.Setenv("OPENCODE_CONFIG_DIR", "/opt/agents")
|
|
||||||
cfg, err = Load(writeCfg(t, yaml))
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("Load: %v", err)
|
|
||||||
}
|
|
||||||
if cfg.OpenCode.ConfigDir != "/opt/agents" {
|
|
||||||
t.Errorf("config_dir = %q, want /opt/agents", cfg.OpenCode.ConfigDir)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoad_EnvOverride(t *testing.T) {
|
|
||||||
t.Setenv("TG_TOKEN", "tok")
|
|
||||||
t.Setenv("TG_CHAT_ID", "42")
|
|
||||||
// env override для opencode.timeout
|
|
||||||
t.Setenv("OPENCODE_CONFIG", "/custom/opencode.json")
|
|
||||||
|
|
||||||
yaml := `telegram:
|
|
||||||
token: "${TG_TOKEN}"
|
|
||||||
chat_id: "${TG_CHAT_ID}"
|
|
||||||
`
|
|
||||||
path := writeCfg(t, yaml)
|
|
||||||
cfg, err := Load(path)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("Load: %v", err)
|
|
||||||
}
|
|
||||||
if cfg.OpenCode.Config != "/custom/opencode.json" {
|
|
||||||
t.Errorf("opencode config = %q", cfg.OpenCode.Config)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestLoad_MissingEnvVarUsesDefault(t *testing.T) {
|
func TestLoad_MissingEnvVarUsesDefault(t *testing.T) {
|
||||||
t.Setenv("TG_TOKEN", "tok")
|
t.Setenv("TG_TOKEN", "tok")
|
||||||
t.Setenv("TG_CHAT_ID", "42")
|
t.Setenv("TG_CHAT_ID", "42")
|
||||||
@@ -277,7 +229,7 @@ paths:
|
|||||||
}
|
}
|
||||||
|
|
||||||
func TestLoad_MissingRequired(t *testing.T) {
|
func TestLoad_MissingRequired(t *testing.T) {
|
||||||
// пустой конфиг — все обязательные поля отсутствуют
|
// пустой конфиг — все обязательные поля отсутствуют (telegram.enabled по дефолту true)
|
||||||
t.Setenv("TG_TOKEN", "")
|
t.Setenv("TG_TOKEN", "")
|
||||||
t.Setenv("TG_CHAT_ID", "")
|
t.Setenv("TG_CHAT_ID", "")
|
||||||
|
|
||||||
@@ -298,6 +250,41 @@ func TestLoad_MissingRequired(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestLoad_TelegramDisabledSkipsToken(t *testing.T) {
|
||||||
|
// enabled: false — token/chat_id не обязательны, конфиг валиден
|
||||||
|
t.Setenv("TG_TOKEN", "")
|
||||||
|
t.Setenv("TG_CHAT_ID", "")
|
||||||
|
|
||||||
|
yaml := `telegram:
|
||||||
|
enabled: false
|
||||||
|
token: ""
|
||||||
|
chat_id: ""
|
||||||
|
`
|
||||||
|
path := writeCfg(t, yaml)
|
||||||
|
cfg, err := Load(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Load: %v", err)
|
||||||
|
}
|
||||||
|
if cfg.Telegram.Enabled {
|
||||||
|
t.Error("telegram.enabled = true, want false")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoad_TelegramEnabledDefaultTrue(t *testing.T) {
|
||||||
|
// enabled не задан — дефолт true, token обязателен
|
||||||
|
t.Setenv("TG_TOKEN", "")
|
||||||
|
t.Setenv("TG_CHAT_ID", "42")
|
||||||
|
|
||||||
|
yaml := `telegram:
|
||||||
|
chat_id: "42"
|
||||||
|
`
|
||||||
|
path := writeCfg(t, yaml)
|
||||||
|
_, err := Load(path)
|
||||||
|
if err == nil || !strings.Contains(err.Error(), "telegram.token") {
|
||||||
|
t.Errorf("ожидалась ошибка telegram.token (enabled по дефолту true), got %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestLoad_InvalidDuration(t *testing.T) {
|
func TestLoad_InvalidDuration(t *testing.T) {
|
||||||
t.Setenv("TG_TOKEN", "x")
|
t.Setenv("TG_TOKEN", "x")
|
||||||
t.Setenv("TG_CHAT_ID", "x")
|
t.Setenv("TG_CHAT_ID", "x")
|
||||||
|
|||||||
@@ -35,6 +35,10 @@ func Load(path string) (*Config, error) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
cfg := &Config{}
|
cfg := &Config{}
|
||||||
|
// Telegram по умолчанию включён (если в YAML не указан enabled).
|
||||||
|
// Делаем до unmarshal: yaml.Unmarshal перезапишет поле, только если ключ
|
||||||
|
// есть в документе, поэтому отсутствующий enabled останется true.
|
||||||
|
cfg.Telegram.Enabled = true
|
||||||
if !noFile {
|
if !noFile {
|
||||||
expanded := os.Expand(string(raw), envLookup)
|
expanded := os.Expand(string(raw), envLookup)
|
||||||
if err := yaml.Unmarshal([]byte(expanded), cfg); err != nil {
|
if err := yaml.Unmarshal([]byte(expanded), cfg); err != nil {
|
||||||
|
|||||||
@@ -76,7 +76,11 @@ type UpdateCfg struct {
|
|||||||
CheckInterval Duration `yaml:"check_interval" default:"24h"`
|
CheckInterval Duration `yaml:"check_interval" default:"24h"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TelegramCfg — настройки Telegram-канала. Enabled=false отключает канал
|
||||||
|
// полностью: бот не опрашивается, сетевые вызовы к api.telegram.org не идут,
|
||||||
|
// token/chat_id не обязательны. По умолчанию включён (см. Load: presetTelegramEnabled).
|
||||||
type TelegramCfg struct {
|
type TelegramCfg struct {
|
||||||
|
Enabled bool `yaml:"enabled"`
|
||||||
Token string `yaml:"token" env:"TG_TOKEN"`
|
Token string `yaml:"token" env:"TG_TOKEN"`
|
||||||
ChatID string `yaml:"chat_id" env:"TG_CHAT_ID"`
|
ChatID string `yaml:"chat_id" env:"TG_CHAT_ID"`
|
||||||
}
|
}
|
||||||
@@ -84,8 +88,6 @@ type TelegramCfg struct {
|
|||||||
type OpenCodeCfg struct {
|
type OpenCodeCfg struct {
|
||||||
Bin string `yaml:"bin" default:"opencode"`
|
Bin string `yaml:"bin" default:"opencode"`
|
||||||
DBPath string `yaml:"db_path" default:""`
|
DBPath string `yaml:"db_path" default:""`
|
||||||
Config string `yaml:"config" env:"OPENCODE_CONFIG"`
|
|
||||||
ConfigDir string `yaml:"config_dir" env:"OPENCODE_CONFIG_DIR"`
|
|
||||||
HardTimeout Duration `yaml:"hard_timeout" default:"20m"`
|
HardTimeout Duration `yaml:"hard_timeout" default:"20m"`
|
||||||
IdleTimeout Duration `yaml:"idle_timeout" default:"5m"`
|
IdleTimeout Duration `yaml:"idle_timeout" default:"5m"`
|
||||||
PollMs Duration `yaml:"poll_ms" default:"2s"`
|
PollMs Duration `yaml:"poll_ms" default:"2s"`
|
||||||
@@ -121,12 +123,14 @@ type PathsCfg struct {
|
|||||||
// (склеенную, все ошибки сразу) или C2 InvalidFormat.
|
// (склеенную, все ошибки сразу) или C2 InvalidFormat.
|
||||||
func (c *Config) Validate() error {
|
func (c *Config) Validate() error {
|
||||||
var errs []error
|
var errs []error
|
||||||
|
if c.Telegram.Enabled {
|
||||||
if c.Telegram.Token == "" {
|
if c.Telegram.Token == "" {
|
||||||
errs = append(errs, fmt.Errorf("%w: telegram.token", ErrMissingField))
|
errs = append(errs, fmt.Errorf("%w: telegram.token", ErrMissingField))
|
||||||
}
|
}
|
||||||
if c.Telegram.ChatID == "" {
|
if c.Telegram.ChatID == "" {
|
||||||
errs = append(errs, fmt.Errorf("%w: telegram.chat_id", ErrMissingField))
|
errs = append(errs, fmt.Errorf("%w: telegram.chat_id", ErrMissingField))
|
||||||
}
|
}
|
||||||
|
}
|
||||||
if !c.Log.Debug() && !strings.EqualFold(c.Log.Level, "info") {
|
if !c.Log.Debug() && !strings.EqualFold(c.Log.Level, "info") {
|
||||||
errs = append(errs, fmt.Errorf("%w: log.level (ожидается \"info\" или \"debug\")", ErrInvalidFormat))
|
errs = append(errs, fmt.Errorf("%w: log.level (ожидается \"info\" или \"debug\")", ErrInvalidFormat))
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -228,6 +228,11 @@ func (c *Core) handleRetry(ctx context.Context, rest string) (Result, error) {
|
|||||||
if err := c.Store.ClearHistory(ctx, id); err != nil {
|
if err := c.Store.ClearHistory(ctx, id); err != nil {
|
||||||
return Result{}, err
|
return Result{}, err
|
||||||
}
|
}
|
||||||
|
// Чистый перезапуск: сбрасываем маркер постмортем-анализа от прошлого
|
||||||
|
// прогона, чтобы на новом failed/timeout постмортем запустился заново.
|
||||||
|
if err := c.Store.DeleteTracesByAgent(ctx, id, "postmortem"); err != nil {
|
||||||
|
return Result{}, err
|
||||||
|
}
|
||||||
return Result{
|
return Result{
|
||||||
Reply: "Задача перезапущена. Опишите, что меняем:",
|
Reply: "Задача перезапущена. Опишите, что меняем:",
|
||||||
TaskID: id,
|
TaskID: id,
|
||||||
|
|||||||
@@ -5,6 +5,14 @@ import (
|
|||||||
"sync"
|
"sync"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Канонические уровни записей лога (фильтр панели «Логи», см. ui.LogLevels).
|
||||||
|
const (
|
||||||
|
LevelError = "error"
|
||||||
|
LevelWarning = "warning"
|
||||||
|
LevelInfo = "info"
|
||||||
|
LevelDebug = "debug"
|
||||||
|
)
|
||||||
|
|
||||||
// LogLine — строка лога для панели «Логи».
|
// LogLine — строка лога для панели «Логи».
|
||||||
type LogLine struct {
|
type LogLine struct {
|
||||||
Level string
|
Level string
|
||||||
@@ -13,6 +21,37 @@ type LogLine struct {
|
|||||||
|
|
||||||
func (LogLine) _event() {}
|
func (LogLine) _event() {}
|
||||||
|
|
||||||
|
// markers проверяет наличие любого из маркеров в уже приведённой к нижнему
|
||||||
|
// регистру строке.
|
||||||
|
func markers(s string, anyOf ...string) bool {
|
||||||
|
for _, m := range anyOf {
|
||||||
|
if strings.Contains(s, m) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// ClassifyLevel определяет уровень строки лога по её тексту.
|
||||||
|
//
|
||||||
|
// Стандартный log.Printf не несёт уровня, поэтому применяются эвристики по
|
||||||
|
// маркерам в тексте (в приоритете debug → warning → error): "debug"/"trace" —
|
||||||
|
// отладка; "warn"/"предупреж" — предупреждение; "error"/"ошибк"/"failed"/
|
||||||
|
// "fatal"/"panic" — ошибка. Остальные строки — информационные.
|
||||||
|
func ClassifyLevel(line string) string {
|
||||||
|
s := strings.ToLower(line)
|
||||||
|
switch {
|
||||||
|
case markers(s, "debug", "trace"):
|
||||||
|
return LevelDebug
|
||||||
|
case markers(s, "warn", "предупреж"):
|
||||||
|
return LevelWarning
|
||||||
|
case markers(s, "error", "ошибк", "failed", "failure", "fatal", "panic"):
|
||||||
|
return LevelError
|
||||||
|
default:
|
||||||
|
return LevelInfo
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// LogBus — тип-обёртка над *Bus для логов.
|
// LogBus — тип-обёртка над *Bus для логов.
|
||||||
//
|
//
|
||||||
// Логи идут отдельной шиной, чтобы большие объёмы текста не блокировали
|
// Логи идут отдельной шиной, чтобы большие объёмы текста не блокировали
|
||||||
@@ -35,7 +74,8 @@ type LogWriter struct {
|
|||||||
buf strings.Builder
|
buf strings.Builder
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewLogWriter создаёт LogWriter, публикующий в шину логов события LogLine{Level:"log"}.
|
// NewLogWriter создаёт LogWriter, публикующий строки лога как события
|
||||||
|
// LogLine; уровень каждой строки определяется эвристикой ClassifyLevel.
|
||||||
func NewLogWriter(bus *LogBus) *LogWriter {
|
func NewLogWriter(bus *LogBus) *LogWriter {
|
||||||
return &LogWriter{bus: bus.Bus}
|
return &LogWriter{bus: bus.Bus}
|
||||||
}
|
}
|
||||||
@@ -56,7 +96,7 @@ func (w *LogWriter) Write(p []byte) (int, error) {
|
|||||||
line := strings.TrimSuffix(data[:idx], "\r")
|
line := strings.TrimSuffix(data[:idx], "\r")
|
||||||
data = data[idx+1:]
|
data = data[idx+1:]
|
||||||
if line != "" {
|
if line != "" {
|
||||||
w.bus.Publish(LogLine{Level: "log", Text: line})
|
w.bus.Publish(LogLine{Level: ClassifyLevel(line), Text: line})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
w.buf.Reset()
|
w.buf.Reset()
|
||||||
|
|||||||
@@ -21,7 +21,8 @@ func TestLogWriterLines(t *testing.T) {
|
|||||||
if !ok {
|
if !ok {
|
||||||
t.Fatalf("got %T, want LogLine", first)
|
t.Fatalf("got %T, want LogLine", first)
|
||||||
}
|
}
|
||||||
if ll.Text != "first line" || ll.Level != "log" {
|
// строка без маркеров ошибок/отладки — информационная
|
||||||
|
if ll.Text != "first line" || ll.Level != LevelInfo {
|
||||||
t.Fatalf("unexpected first line: %+v", ll)
|
t.Fatalf("unexpected first line: %+v", ll)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -74,3 +75,50 @@ func TestLogWriterMultiSplit(t *testing.T) {
|
|||||||
t.Fatalf("got %v", texts)
|
t.Fatalf("got %v", texts)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ClassifyLevel определяет уровень строки по маркерам в тексте.
|
||||||
|
func TestClassifyLevel(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
line string
|
||||||
|
want string
|
||||||
|
}{
|
||||||
|
{"app: db opened /tmp/r.db", LevelInfo},
|
||||||
|
{"app: worker started", LevelInfo},
|
||||||
|
{"opencode: debug: poll request", LevelDebug},
|
||||||
|
{"opencode api debug: prompt -> POST http://127.0.0.1:4096/api/session", LevelDebug},
|
||||||
|
{"opencode api debug: messages response (512 bytes)", LevelDebug},
|
||||||
|
{"trace: session resumed", LevelDebug},
|
||||||
|
{"tg: warn: long poll timeout", LevelWarning},
|
||||||
|
{"ПРЕДУПРЕЖДЕНИЕ: конфиг не задан", LevelWarning},
|
||||||
|
{"app: process /start #5: run failed: exit status 1", LevelError},
|
||||||
|
{"ERROR: connection refused", LevelError},
|
||||||
|
{"worker: job failed after 3 retries", LevelError},
|
||||||
|
{"panic: runtime error", LevelError},
|
||||||
|
}
|
||||||
|
for _, c := range cases {
|
||||||
|
if got := ClassifyLevel(c.line); got != c.want {
|
||||||
|
t.Errorf("ClassifyLevel(%q) = %q, want %q", c.line, got, c.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// LogWriter публикует уровень, определённый по тексту строки.
|
||||||
|
func TestLogWriterClassifiesLevels(t *testing.T) {
|
||||||
|
lbus := NewLogBus(16)
|
||||||
|
ch, unsub := lbus.Subscribe()
|
||||||
|
defer unsub()
|
||||||
|
|
||||||
|
w := NewLogWriter(lbus)
|
||||||
|
if _, err := w.Write([]byte("all good\nERROR: boom\n")); err != nil {
|
||||||
|
t.Fatalf("Write: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
first := receiveOne(t, ch).(LogLine)
|
||||||
|
if first.Level != LevelInfo || first.Text != "all good" {
|
||||||
|
t.Fatalf("unexpected first: %+v", first)
|
||||||
|
}
|
||||||
|
second := receiveOne(t, ch).(LogLine)
|
||||||
|
if second.Level != LevelError || second.Text != "ERROR: boom" {
|
||||||
|
t.Fatalf("unexpected second: %+v", second)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -6,7 +6,6 @@ import (
|
|||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"log"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
@@ -26,10 +25,18 @@ import (
|
|||||||
type Client struct {
|
type Client struct {
|
||||||
BaseURL string // http://host:port (без завершающего слеша)
|
BaseURL string // http://host:port (без завершающего слеша)
|
||||||
Password string // basic auth (username "opencode")
|
Password string // basic auth (username "opencode")
|
||||||
|
Directory string // каталог сервера — идёт заголовком x-opencode-directory
|
||||||
Debug bool // включать отладочные логи API-вызовов (log.level=debug)
|
Debug bool // включать отладочные логи API-вызовов (log.level=debug)
|
||||||
http *http.Client // единый клиент: все операции быстрые (нет блокирующего Send)
|
http *http.Client // единый клиент: все операции быстрые (нет блокирующего Send)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// SessionMessagesURL возвращает URL для просмотра сообщений сессии в браузере.
|
||||||
|
// Сортировка по времени: сначала новые (order=desc). Без credentials — пароль
|
||||||
|
// в лог не попадает.
|
||||||
|
func SessionMessagesURL(baseURL, sessionID string) string {
|
||||||
|
return baseURL + "/api/session/" + sessionID + "/message?order=desc"
|
||||||
|
}
|
||||||
|
|
||||||
// ClientErr — классы ошибок клиента.
|
// ClientErr — классы ошибок клиента.
|
||||||
type ClientErr struct {
|
type ClientErr struct {
|
||||||
Op string // "connect" | "create" | "prompt" | "messages" | "active" | "abort"
|
Op string // "connect" | "create" | "prompt" | "messages" | "active" | "abort"
|
||||||
@@ -56,17 +63,15 @@ func (c *Client) do(ctx context.Context, method, path, op string, body []byte) (
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, &ClientErr{Op: "connect", Err: err}
|
return nil, &ClientErr{Op: "connect", Err: err}
|
||||||
}
|
}
|
||||||
|
if body != nil {
|
||||||
|
req.ContentLength = int64(len(body))
|
||||||
|
}
|
||||||
if c.Password != "" {
|
if c.Password != "" {
|
||||||
req.SetBasicAuth("opencode", c.Password)
|
req.SetBasicAuth("opencode", c.Password)
|
||||||
}
|
}
|
||||||
if body != nil {
|
|
||||||
req.Header.Set("Content-Type", "application/json")
|
req.Header.Set("Content-Type", "application/json")
|
||||||
}
|
if c.Directory != "" {
|
||||||
if c.Debug {
|
req.Header.Set("x-opencode-directory", c.Directory)
|
||||||
log.Printf("opencode api %s -> %s %s%s", op, method, c.BaseURL, path)
|
|
||||||
if len(body) > 0 {
|
|
||||||
log.Printf("opencode api %s request body: %s", op, truncateStr(string(body), 5000))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
resp, err := c.http.Do(req)
|
resp, err := c.http.Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -78,14 +83,8 @@ func (c *Client) do(ctx context.Context, method, path, op string, body []byte) (
|
|||||||
return nil, &ClientErr{Op: "connect", Err: err}
|
return nil, &ClientErr{Op: "connect", Err: err}
|
||||||
}
|
}
|
||||||
if resp.StatusCode < 200 || resp.StatusCode > 299 {
|
if resp.StatusCode < 200 || resp.StatusCode > 299 {
|
||||||
if c.Debug {
|
|
||||||
log.Printf("opencode api %s response: status %d: %s", op, resp.StatusCode, truncateStr(string(b), 1000))
|
|
||||||
}
|
|
||||||
return nil, &ClientErr{Op: op, Err: fmt.Errorf("status %d: %s", resp.StatusCode, truncateStr(string(b), 300))}
|
return nil, &ClientErr{Op: op, Err: fmt.Errorf("status %d: %s", resp.StatusCode, truncateStr(string(b), 300))}
|
||||||
}
|
}
|
||||||
if c.Debug {
|
|
||||||
log.Printf("opencode api %s response (%d bytes): %s", op, len(b), truncateStr(string(b), 5000))
|
|
||||||
}
|
|
||||||
return b, nil
|
return b, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -97,7 +96,7 @@ type ModelRef struct {
|
|||||||
Variant string `json:"variant,omitempty"`
|
Variant string `json:"variant,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
// String возвращает каноничное представление "provider/id[/variant]".
|
// String возвращает каноническое представление "provider/id[/variant]".
|
||||||
func (m *ModelRef) String() string {
|
func (m *ModelRef) String() string {
|
||||||
if m == nil {
|
if m == nil {
|
||||||
return ""
|
return ""
|
||||||
@@ -108,13 +107,14 @@ func (m *ModelRef) String() string {
|
|||||||
return m.ProviderID + "/" + m.ID
|
return m.ProviderID + "/" + m.ID
|
||||||
}
|
}
|
||||||
|
|
||||||
// CreateSession создаёт новую сессию и возвращает её id. model != nil —
|
// CreateSession создаёт новую сессию и возвращает её id. agent — имя агента
|
||||||
// хардпин модели (top-level "model" из конфига opencode), чтобы не зависеть
|
// (analyst/dev/reviewer/chat/postmortem) из встроенных определений ratatoskr.
|
||||||
// от fallback-логики выбора модели в самом opencode.
|
// Модель мы НЕ выбираем: opencode сам берёт модель по умолчанию из своего
|
||||||
func (c *Client) CreateSession(ctx context.Context, model *ModelRef) (string, error) {
|
// глобального конфига (ratatoskr про неё ничего не знает).
|
||||||
|
func (c *Client) CreateSession(ctx context.Context, agent string) (string, error) {
|
||||||
payload := map[string]any{}
|
payload := map[string]any{}
|
||||||
if model != nil {
|
if agent != "" {
|
||||||
payload["model"] = model
|
payload["agent"] = agent
|
||||||
}
|
}
|
||||||
body, _ := json.Marshal(payload)
|
body, _ := json.Marshal(payload)
|
||||||
raw, err := c.do(ctx, http.MethodPost, "/api/session", "create", body)
|
raw, err := c.do(ctx, http.MethodPost, "/api/session", "create", body)
|
||||||
@@ -132,6 +132,12 @@ func (c *Client) CreateSession(ctx context.Context, model *ModelRef) (string, er
|
|||||||
if out.Data.ID == "" {
|
if out.Data.ID == "" {
|
||||||
return "", &ClientErr{Op: "create", Err: fmt.Errorf("пустой id сессии")}
|
return "", &ClientErr{Op: "create", Err: fmt.Errorf("пустой id сессии")}
|
||||||
}
|
}
|
||||||
|
time.Sleep(2 * time.Second)
|
||||||
|
// Холостой вызов /api/model: прогревает сервер (модель/провайдеры),
|
||||||
|
// чтобы первый промпт не спотыкался о «тёплый» старт.
|
||||||
|
_, _ = c.do(ctx, http.MethodGet, "/api/model", "models-warmup", nil)
|
||||||
|
time.Sleep(2 * time.Second)
|
||||||
|
_, _ = c.do(ctx, http.MethodGet, "/api/model", "models-warmup", nil)
|
||||||
return out.Data.ID, nil
|
return out.Data.ID, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -323,6 +329,26 @@ func assistantText(msgs []v2Message, since int64) []string {
|
|||||||
return texts
|
return texts
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// assistantVerdict собирает финальный текст ответа: сначала text-парты, а если
|
||||||
|
// их нет — только reasoning-парты (fallback для моделей, которые на некоторые
|
||||||
|
// запросы отвечают лишь reasoning без text). usedReasoning=true означает, что
|
||||||
|
// text-партов не было вовсе и вердикт собран из reasoning.
|
||||||
|
func assistantVerdict(msgs []v2Message, since int64) (texts []string, usedReasoning bool) {
|
||||||
|
if texts := assistantText(msgs, since); len(texts) > 0 {
|
||||||
|
return texts, false
|
||||||
|
}
|
||||||
|
ass := assistantSince(msgs, since)
|
||||||
|
reasoning := make([]string, 0, len(ass))
|
||||||
|
for i := len(ass) - 1; i >= 0; i-- {
|
||||||
|
for _, p := range ass[i].Content {
|
||||||
|
if p.Type == "reasoning" && p.Text != "" {
|
||||||
|
reasoning = append(reasoning, p.Text)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return reasoning, len(reasoning) > 0
|
||||||
|
}
|
||||||
|
|
||||||
func truncateStr(s string, n int) string {
|
func truncateStr(s string, n int) string {
|
||||||
if len(s) <= n {
|
if len(s) <= n {
|
||||||
return s
|
return s
|
||||||
|
|||||||
@@ -28,9 +28,10 @@ type fakeAPIServer struct {
|
|||||||
blockPrompt bool
|
blockPrompt bool
|
||||||
messages []v2Message
|
messages []v2Message
|
||||||
verdictText string
|
verdictText string
|
||||||
|
verdictReasoning string // завершённый ответ только с reasoning-партом (без text)
|
||||||
failCreate bool
|
failCreate bool
|
||||||
failMessages bool
|
failMessages bool
|
||||||
createdModel *ModelRef // модель, полученная на POST /api/session
|
createdAgent string // агент, полученный на POST /api/session
|
||||||
promptCalls int
|
promptCalls int
|
||||||
|
|
||||||
// streamGrow: стрим одного растущего парта — текст/reasoning растёт с
|
// streamGrow: стрим одного растущего парта — текст/reasoning растёт с
|
||||||
@@ -54,10 +55,10 @@ func (f *fakeAPIServer) handler() http.Handler {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
var in struct {
|
var in struct {
|
||||||
Model *ModelRef `json:"model"`
|
Agent string `json:"agent"`
|
||||||
}
|
}
|
||||||
_ = json.NewDecoder(r.Body).Decode(&in)
|
_ = json.NewDecoder(r.Body).Decode(&in)
|
||||||
f.createdModel = in.Model
|
f.createdAgent = in.Agent
|
||||||
f.sessionID = "sess-fake"
|
f.sessionID = "sess-fake"
|
||||||
f.created = true
|
f.created = true
|
||||||
writeJSON(w, map[string]any{"data": map[string]any{"id": "sess-fake"}})
|
writeJSON(w, map[string]any{"data": map[string]any{"id": "sess-fake"}})
|
||||||
@@ -133,6 +134,9 @@ func (f *fakeAPIServer) handler() http.Handler {
|
|||||||
if msgs == nil && f.verdictText != "" && !f.blockPrompt {
|
if msgs == nil && f.verdictText != "" && !f.blockPrompt {
|
||||||
msgs = []v2Message{f.assistantMsg(f.verdictText)}
|
msgs = []v2Message{f.assistantMsg(f.verdictText)}
|
||||||
}
|
}
|
||||||
|
if msgs == nil && f.verdictReasoning != "" && !f.blockPrompt {
|
||||||
|
msgs = []v2Message{f.assistantReasoningMsg(f.verdictReasoning)}
|
||||||
|
}
|
||||||
if msgs == nil {
|
if msgs == nil {
|
||||||
msgs = []v2Message{}
|
msgs = []v2Message{}
|
||||||
}
|
}
|
||||||
@@ -153,6 +157,19 @@ func (f *fakeAPIServer) assistantMsg(text string) v2Message {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// assistantReasoningMsg строит завершённое assistant-сообщение только с
|
||||||
|
// reasoning-партом (без text) — для проверки fallback-сценария.
|
||||||
|
func (f *fakeAPIServer) assistantReasoningMsg(text string) v2Message {
|
||||||
|
now := time.Now().UnixMilli()
|
||||||
|
return v2Message{
|
||||||
|
ID: "msg_r",
|
||||||
|
Type: "assistant",
|
||||||
|
Content: []v2Part{{Type: "reasoning", Text: text}},
|
||||||
|
Finish: "end_turn",
|
||||||
|
Time: v2Time{Created: &now, Completed: &now},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func writeJSON(w http.ResponseWriter, v any) {
|
func writeJSON(w http.ResponseWriter, v any) {
|
||||||
w.Header().Set("Content-Type", "application/json")
|
w.Header().Set("Content-Type", "application/json")
|
||||||
_ = json.NewEncoder(w).Encode(v)
|
_ = json.NewEncoder(w).Encode(v)
|
||||||
@@ -169,33 +186,43 @@ func fakeClient(t *testing.T, f *fakeAPIServer) *Client {
|
|||||||
func TestClient_CreateSession(t *testing.T) {
|
func TestClient_CreateSession(t *testing.T) {
|
||||||
f := &fakeAPIServer{}
|
f := &fakeAPIServer{}
|
||||||
c := fakeClient(t, f)
|
c := fakeClient(t, f)
|
||||||
id, err := c.CreateSession(context.Background(), nil)
|
id, err := c.CreateSession(context.Background(), "dev")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("CreateSession err: %v", err)
|
t.Fatalf("CreateSession err: %v", err)
|
||||||
}
|
}
|
||||||
if id != "sess-fake" {
|
if id != "sess-fake" {
|
||||||
t.Errorf("id = %q, want sess-fake", id)
|
t.Errorf("id = %q, want sess-fake", id)
|
||||||
}
|
}
|
||||||
if f.createdModel != nil {
|
if f.createdAgent != "dev" {
|
||||||
t.Errorf("createdModel = %+v, want nil", f.createdModel)
|
t.Errorf("createdAgent = %q, want dev", f.createdAgent)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestClient_CreateSessionHardpinsModel(t *testing.T) {
|
func TestClient_CreateSessionPassesAgent(t *testing.T) {
|
||||||
want := &ModelRef{ProviderID: "tokentool", ID: "deepseek/deepseek-v4-flash-0731"}
|
|
||||||
f := &fakeAPIServer{}
|
f := &fakeAPIServer{}
|
||||||
c := fakeClient(t, f)
|
c := fakeClient(t, f)
|
||||||
if _, err := c.CreateSession(context.Background(), want); err != nil {
|
if _, err := c.CreateSession(context.Background(), "postmortem"); err != nil {
|
||||||
t.Fatalf("CreateSession err: %v", err)
|
t.Fatalf("CreateSession err: %v", err)
|
||||||
}
|
}
|
||||||
if f.createdModel == nil || f.createdModel.ProviderID != want.ProviderID || f.createdModel.ID != want.ID {
|
if f.createdAgent != "postmortem" {
|
||||||
t.Errorf("createdModel = %+v, want %+v", f.createdModel, want)
|
t.Errorf("createdAgent = %q, want postmortem", f.createdAgent)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestClient_CreateSessionNoAgent(t *testing.T) {
|
||||||
|
f := &fakeAPIServer{}
|
||||||
|
c := fakeClient(t, f)
|
||||||
|
if _, err := c.CreateSession(context.Background(), ""); err != nil {
|
||||||
|
t.Fatalf("CreateSession err: %v", err)
|
||||||
|
}
|
||||||
|
if f.createdAgent != "" {
|
||||||
|
t.Errorf("createdAgent = %q, want пусто", f.createdAgent)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestClient_CreateSessionFail(t *testing.T) {
|
func TestClient_CreateSessionFail(t *testing.T) {
|
||||||
c := fakeClient(t, &fakeAPIServer{failCreate: true})
|
c := fakeClient(t, &fakeAPIServer{failCreate: true})
|
||||||
if _, err := c.CreateSession(context.Background(), nil); err == nil {
|
if _, err := c.CreateSession(context.Background(), "dev"); err == nil {
|
||||||
t.Fatal("CreateSession должен упасть при 500, а не nil")
|
t.Fatal("CreateSession должен упасть при 500, а не nil")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -279,16 +306,45 @@ func Test_newestAssistant(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func Test_parseModelString(t *testing.T) {
|
func Test_assistantVerdict(t *testing.T) {
|
||||||
m := parseModelString("tokentool/deepseek/deepseek-v4-flash-0731")
|
older := time.Now().Add(-time.Minute).UnixMilli()
|
||||||
if m == nil || m.ProviderID != "tokentool" || m.ID != "deepseek/deepseek-v4-flash-0731" {
|
newer := time.Now().UnixMilli()
|
||||||
t.Errorf("parse = %+v, want tokentool/deepseek-v4-flash-0731", m)
|
reasoningOf := func(text string, at *int64) v2Message {
|
||||||
|
return v2Message{ID: "r", Type: "assistant", Content: []v2Part{{Type: "reasoning", Text: text}}, Time: v2Time{Created: at}}
|
||||||
}
|
}
|
||||||
if parseModelString("onlyprovider") != nil {
|
|
||||||
t.Error("parse без '/' должен вернуть nil")
|
// reasoning-only: text-партов нет → fallback на reasoning, usedReasoning=true.
|
||||||
|
// Сообщения приходят новейшими первыми (как из API) → размышление 2 новее.
|
||||||
|
reasoningOnlyMsgs := []v2Message{
|
||||||
|
reasoningOf("размышление 2", &newer),
|
||||||
|
reasoningOf("размышление 1", &older),
|
||||||
}
|
}
|
||||||
if parseModelString("") != nil {
|
texts, used := assistantVerdict(reasoningOnlyMsgs, older)
|
||||||
t.Error("parse пустой должен вернуть nil")
|
if !used {
|
||||||
|
t.Error("usedReasoning = false, want true для reasoning-only")
|
||||||
|
}
|
||||||
|
if len(texts) != 2 || texts[0] != "размышление 1" || texts[1] != "размышление 2" {
|
||||||
|
t.Errorf("verdict = %v, want [размышление 1 размышление 2] (хронологически)", texts)
|
||||||
|
}
|
||||||
|
|
||||||
|
// text + reasoning → берётся text, reasoning игнорируется.
|
||||||
|
mixed := []v2Message{
|
||||||
|
{ID: "a", Type: "assistant",
|
||||||
|
Content: []v2Part{{Type: "reasoning", Text: "thinking"}, {Type: "text", Text: "ответ"}},
|
||||||
|
Time: v2Time{Created: &newer}},
|
||||||
|
}
|
||||||
|
texts, used = assistantVerdict(mixed, older)
|
||||||
|
if used {
|
||||||
|
t.Error("usedReasoning = true, want false (есть text)")
|
||||||
|
}
|
||||||
|
if len(texts) != 1 || texts[0] != "ответ" {
|
||||||
|
t.Errorf("verdict = %v, want [ответ]", texts)
|
||||||
|
}
|
||||||
|
|
||||||
|
// пусто → пусто и usedReasoning=false.
|
||||||
|
empty := []v2Message{{ID: "u", Type: "user", Time: v2Time{Created: &newer}}}
|
||||||
|
if texts, used := assistantVerdict(empty, older); used || len(texts) != 0 {
|
||||||
|
t.Errorf("пусто: texts=%v usedReasoning=%v, want пусто/false", texts, used)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -299,3 +355,12 @@ func TestClientErr_Unwrap(t *testing.T) {
|
|||||||
t.Fatal("expected *ClientErr")
|
t.Fatal("expected *ClientErr")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestSessionMessagesURL — ссылка на сообщения сессии, сортировка по времени (новые сверху).
|
||||||
|
func TestSessionMessagesURL(t *testing.T) {
|
||||||
|
got := SessionMessagesURL("http://127.0.0.1:4101", "sess-abc")
|
||||||
|
want := "http://127.0.0.1:4101/api/session/sess-abc/message?order=desc"
|
||||||
|
if got != want {
|
||||||
|
t.Errorf("SessionMessagesURL() = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,176 +0,0 @@
|
|||||||
package opencode
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Чтение top-level "model" из эффективного конфига opencode.
|
|
||||||
//
|
|
||||||
// Зачем: ratatoskr хардпинит модель в сессии (CreateSession), чтобы не зависеть
|
|
||||||
// от fallback-логики opencode. Если в конфиге модель не задана (или конфиг
|
|
||||||
// написан по старой v1-схеме — npm/options, которые v2 молча игнорирует),
|
|
||||||
// opencode сам выберет «дефолтную» модельную запись, и это может оказаться не
|
|
||||||
// той моделью. Поэтому мы явно логируем предупреждение (класс O5 WARN).
|
|
||||||
|
|
||||||
// opencodeConfigPath определяет путь к конфигу opencode, который видит
|
|
||||||
// serve-процесс этого пула (см. README): (1) явный OPENCODE_CONFIG из Server
|
|
||||||
// или окружения процесса, (2) OPENCODE_CONFIG_DIR / глобальный каталог
|
|
||||||
// ~/.config/opencode. Возвращает "" если ничего не найдено.
|
|
||||||
func opencodeConfigPath(cfgFile, cfgDir string) string {
|
|
||||||
// (1) явный файл конфига — Server.Config или env OPENCODE_CONFIG.
|
|
||||||
p := cfgFile
|
|
||||||
if p == "" {
|
|
||||||
p = os.Getenv("OPENCODE_CONFIG")
|
|
||||||
}
|
|
||||||
if p != "" {
|
|
||||||
if st, err := os.Stat(p); err == nil && !st.IsDir() {
|
|
||||||
return p
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// (2) каталог конфигов.
|
|
||||||
dir := cfgDir
|
|
||||||
if dir == "" {
|
|
||||||
dir = os.Getenv("OPENCODE_CONFIG_DIR")
|
|
||||||
}
|
|
||||||
if dir == "" {
|
|
||||||
home, err := os.UserHomeDir()
|
|
||||||
if err != nil || home == "" {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
dir = filepath.Join(home, ".config", "opencode")
|
|
||||||
if x := os.Getenv("XDG_CONFIG_HOME"); x != "" {
|
|
||||||
dir = filepath.Join(x, "opencode")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for _, name := range []string{"opencode.json", "opencode.jsonc"} {
|
|
||||||
cand := filepath.Join(dir, name)
|
|
||||||
if st, err := os.Stat(cand); err == nil && !st.IsDir() {
|
|
||||||
return cand
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
// ReadModelRef извлекает top-level "model" из конфига opencode и возвращает
|
|
||||||
// его как ModelRef. Модель не задана — вернёт (nil, nil); ошибка чтения/парсинга
|
|
||||||
// возвращается (вызывающий логирует warning и продолжает без хардпина).
|
|
||||||
func ReadModelRef(cfgFile, cfgDir string) (*ModelRef, error) {
|
|
||||||
path := opencodeConfigPath(cfgFile, cfgDir)
|
|
||||||
if path == "" {
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
b, err := os.ReadFile(path)
|
|
||||||
if err != nil {
|
|
||||||
return nil, fmt.Errorf("config: читать %s: %w", path, err)
|
|
||||||
}
|
|
||||||
doc := struct {
|
|
||||||
Model json.RawMessage `json:"model"`
|
|
||||||
}{}
|
|
||||||
if err := json.Unmarshal(stripJSONC(b), &doc); err != nil {
|
|
||||||
return nil, fmt.Errorf("config: парсить %s: %w", path, err)
|
|
||||||
}
|
|
||||||
if len(doc.Model) == 0 || strings.TrimSpace(string(doc.Model)) == "null" {
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
// "model" может быть строкой "provider/id" или объектом {providerID, id}.
|
|
||||||
var s string
|
|
||||||
if err := json.Unmarshal(doc.Model, &s); err == nil {
|
|
||||||
ref := parseModelString(s)
|
|
||||||
if ref == nil {
|
|
||||||
return nil, fmt.Errorf("config: некорректная model %q в %s (ожидается provider/id)", s, path)
|
|
||||||
}
|
|
||||||
return ref, nil
|
|
||||||
}
|
|
||||||
var ref ModelRef
|
|
||||||
if err := json.Unmarshal(doc.Model, &ref); err != nil {
|
|
||||||
return nil, fmt.Errorf("config: некорректная model в %s", path)
|
|
||||||
}
|
|
||||||
if ref.ProviderID == "" || ref.ID == "" {
|
|
||||||
return nil, fmt.Errorf("config: model без providerID/id в %s", path)
|
|
||||||
}
|
|
||||||
return &ref, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// parseModelString разбирает "provider/id" (как ModelV2.parse: провайдер — всё
|
|
||||||
// до первого '/', id — остаток). Возвращает nil при пустой/некорректной строке.
|
|
||||||
func parseModelString(s string) *ModelRef {
|
|
||||||
s = strings.TrimSpace(s)
|
|
||||||
if s == "" {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
i := strings.IndexByte(s, '/')
|
|
||||||
if i <= 0 || i == len(s)-1 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return &ModelRef{ProviderID: s[:i], ID: s[i+1:]}
|
|
||||||
}
|
|
||||||
|
|
||||||
// stripJSONC удаляет // и /* */ комментарии (вне строк), сохраняя позиции
|
|
||||||
// переводов строк, чтобы json.Unmarshal не споткнулся о trailing-комма.
|
|
||||||
func stripJSONC(b []byte) []byte {
|
|
||||||
out := make([]byte, 0, len(b))
|
|
||||||
inStr := false
|
|
||||||
esc := false
|
|
||||||
i := 0
|
|
||||||
for i < len(b) {
|
|
||||||
c := b[i]
|
|
||||||
if inStr {
|
|
||||||
out = append(out, c)
|
|
||||||
if esc {
|
|
||||||
esc = false
|
|
||||||
} else if c == '\\' {
|
|
||||||
esc = true
|
|
||||||
} else if c == '"' {
|
|
||||||
inStr = false
|
|
||||||
}
|
|
||||||
i++
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
switch {
|
|
||||||
case c == '"':
|
|
||||||
inStr = true
|
|
||||||
out = append(out, c)
|
|
||||||
i++
|
|
||||||
case c == '/' && i+1 < len(b) && b[i+1] == '/':
|
|
||||||
for i < len(b) && b[i] != '\n' {
|
|
||||||
i++
|
|
||||||
}
|
|
||||||
if i < len(b) {
|
|
||||||
out = append(out, '\n')
|
|
||||||
i++
|
|
||||||
}
|
|
||||||
case c == '/' && i+1 < len(b) && b[i+1] == '*':
|
|
||||||
i += 2
|
|
||||||
for i+1 < len(b) && !(b[i] == '*' && b[i+1] == '/') {
|
|
||||||
i++
|
|
||||||
}
|
|
||||||
i += 2
|
|
||||||
default:
|
|
||||||
out = append(out, c)
|
|
||||||
i++
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return dropTrailingCommas(out)
|
|
||||||
}
|
|
||||||
|
|
||||||
// dropTrailingCommas убирает запятые перед '}' / ']' (допускаются в JSONC).
|
|
||||||
func dropTrailingCommas(b []byte) []byte {
|
|
||||||
out := make([]byte, 0, len(b))
|
|
||||||
for i := 0; i < len(b); i++ {
|
|
||||||
if b[i] == ',' {
|
|
||||||
j := i + 1
|
|
||||||
for j < len(b) && (b[j] == ' ' || b[j] == '\t' || b[j] == '\n' || b[j] == '\r') {
|
|
||||||
j++
|
|
||||||
}
|
|
||||||
if j < len(b) && (b[j] == '}' || b[j] == ']') {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
}
|
|
||||||
out = append(out, b[i])
|
|
||||||
}
|
|
||||||
return out
|
|
||||||
}
|
|
||||||
@@ -1,85 +0,0 @@
|
|||||||
package opencode
|
|
||||||
|
|
||||||
import (
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestReadModelRef_String(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
path := filepath.Join(dir, "opencode.jsonc")
|
|
||||||
// конфиг с комментариями и trailing-запятыми (JSONC).
|
|
||||||
src := `{
|
|
||||||
// комментарий
|
|
||||||
"model": "tokentool/deepseek/deepseek-v4-flash-0731", /* и блочный */
|
|
||||||
"provider": {
|
|
||||||
"tokentool": {"api": {"type": "aisdk", "package": "@ai-sdk/openai-compatible", "url": "https://x"}},
|
|
||||||
},
|
|
||||||
}`
|
|
||||||
if err := os.WriteFile(path, []byte(src), 0o644); err != nil {
|
|
||||||
t.Fatalf("write: %v", err)
|
|
||||||
}
|
|
||||||
m, err := ReadModelRef(path, "")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("ReadModelRef err: %v", err)
|
|
||||||
}
|
|
||||||
if m == nil || m.ProviderID != "tokentool" || m.ID != "deepseek/deepseek-v4-flash-0731" {
|
|
||||||
t.Errorf("model = %+v, want tokentool/deepseek-v4-flash-0731", m)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadModelRef_Object(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
path := filepath.Join(dir, "opencode.json")
|
|
||||||
src := `{"model": {"providerID": "tokentool", "id": "deepseek/deepseek-v4-flash-0731"}}`
|
|
||||||
if err := os.WriteFile(path, []byte(src), 0o644); err != nil {
|
|
||||||
t.Fatalf("write: %v", err)
|
|
||||||
}
|
|
||||||
m, err := ReadModelRef(path, "")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("ReadModelRef err: %v", err)
|
|
||||||
}
|
|
||||||
if m == nil || m.ID != "deepseek/deepseek-v4-flash-0731" {
|
|
||||||
t.Errorf("model = %+v, want object-форма", m)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadModelRef_Missing(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
path := filepath.Join(dir, "opencode.json")
|
|
||||||
src := `{"provider": {}}`
|
|
||||||
if err := os.WriteFile(path, []byte(src), 0o644); err != nil {
|
|
||||||
t.Fatalf("write: %v", err)
|
|
||||||
}
|
|
||||||
m, err := ReadModelRef(path, "")
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("ReadModelRef err: %v", err)
|
|
||||||
}
|
|
||||||
if m != nil {
|
|
||||||
t.Errorf("model = %+v, want nil (model не задан)", m)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadModelRef_NoFile(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
m, err := ReadModelRef(filepath.Join(dir, "nope.json"), dir)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("ReadModelRef err: %v", err)
|
|
||||||
}
|
|
||||||
if m != nil {
|
|
||||||
t.Errorf("model = %+v, want nil", m)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestReadModelRef_Bad(t *testing.T) {
|
|
||||||
dir := t.TempDir()
|
|
||||||
path := filepath.Join(dir, "opencode.json")
|
|
||||||
src := `{"model": 12345}`
|
|
||||||
if err := os.WriteFile(path, []byte(src), 0o644); err != nil {
|
|
||||||
t.Fatalf("write: %v", err)
|
|
||||||
}
|
|
||||||
if _, err := ReadModelRef(path, ""); err == nil {
|
|
||||||
t.Error("ReadModelRef должен упасть на некорректной model")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -19,8 +19,6 @@ import (
|
|||||||
// директории → каждая сессия API привязана к правильному project-каталогу.
|
// директории → каждая сессия API привязана к правильному project-каталогу.
|
||||||
type Pool struct {
|
type Pool struct {
|
||||||
Bin string
|
Bin string
|
||||||
Config string
|
|
||||||
ConfigDir string
|
|
||||||
DBPath string
|
DBPath string
|
||||||
Host string
|
Host string
|
||||||
BasePort int
|
BasePort int
|
||||||
@@ -73,8 +71,6 @@ func (p *Pool) EnsureRoot(ctx context.Context) error {
|
|||||||
}
|
}
|
||||||
s := &Server{
|
s := &Server{
|
||||||
Bin: p.Bin,
|
Bin: p.Bin,
|
||||||
Config: p.Config,
|
|
||||||
ConfigDir: p.ConfigDir,
|
|
||||||
DBPath: p.DBPath,
|
DBPath: p.DBPath,
|
||||||
Host: p.Host,
|
Host: p.Host,
|
||||||
Password: p.Password,
|
Password: p.Password,
|
||||||
@@ -103,8 +99,6 @@ func (p *Pool) Ensure(ctx context.Context, dir string) (*Server, error) {
|
|||||||
abs := filepath.Clean(dir)
|
abs := filepath.Clean(dir)
|
||||||
s := &Server{
|
s := &Server{
|
||||||
Bin: p.Bin,
|
Bin: p.Bin,
|
||||||
Config: p.Config,
|
|
||||||
ConfigDir: p.ConfigDir,
|
|
||||||
DBPath: p.DBPath,
|
DBPath: p.DBPath,
|
||||||
Host: p.Host,
|
Host: p.Host,
|
||||||
Password: p.Password,
|
Password: p.Password,
|
||||||
|
|||||||
@@ -73,31 +73,21 @@ func (r *Runner) Run(ctx context.Context, prompt, cwd, agent, sessionID string)
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
c := &Client{BaseURL: srv.Addr(), Password: srv.Password, Debug: r.Debug}
|
c := &Client{BaseURL: srv.Addr(), Password: srv.Password, Directory: srv.Dir, Debug: r.Debug}
|
||||||
|
|
||||||
// Модель по умолчанию из конфига opencode — хардпиним её в сессии, чтобы
|
// Сессия: заданная (resume) или новая. Модель НЕ выбираем — opencode сам
|
||||||
// не зависеть от fallback-логики opencode (класс O5 WARN: если модель не
|
// берёт её по умолчанию из своего глобального конфига.
|
||||||
// считывается/не задана — предупреждаем и работаем без явного указания).
|
|
||||||
model, mErr := ReadModelRef(srv.Config, srv.ConfigDir)
|
|
||||||
if mErr != nil {
|
|
||||||
r.logf("WARN opencode: не удалось прочитать model из конфига: %v", mErr)
|
|
||||||
} else if model == nil {
|
|
||||||
r.logf("WARN opencode: в конфиге opencode не задан top-level model — модель не хардпинится (риск fallback)")
|
|
||||||
} else {
|
|
||||||
r.logf("opencode(%s) model=%s", agent, model)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Сессия: заданная (resume) или новая.
|
|
||||||
sid := sessionID
|
sid := sessionID
|
||||||
if sid == "" {
|
if sid == "" {
|
||||||
sid, err = c.CreateSession(ctx, model)
|
sid, err = c.CreateSession(ctx, agent)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("opencode: create session: %w", err)
|
return nil, fmt.Errorf("opencode: create session: %w", err)
|
||||||
}
|
}
|
||||||
r.logf("opencode(%s) session=%s на %s", agent, sid, srv.Addr())
|
r.logf("opencode(%s) session=%s на %s", agent, sid, srv.Addr())
|
||||||
|
r.logf("opencode(%s) сообщения сессии: %s", agent, SessionMessagesURL(srv.Addr(), sid))
|
||||||
}
|
}
|
||||||
|
|
||||||
return r.awaitVerdict(ctx, c, model, sid, agent, prompt)
|
return r.awaitVerdict(ctx, c, sid, agent, prompt)
|
||||||
}
|
}
|
||||||
|
|
||||||
// settlePolls — сколько подряд опросов должно подтвердить завершение ответа,
|
// settlePolls — сколько подряд опросов должно подтвердить завершение ответа,
|
||||||
@@ -109,7 +99,7 @@ const settlePolls = 2
|
|||||||
// контролируя idle/hard таймауты. Завершение: сессия ушла из активных дренажей
|
// контролируя idle/hard таймауты. Завершение: сессия ушла из активных дренажей
|
||||||
// И есть новое завершённое assistant-сообщение, стабильное в течение settlePolls
|
// И есть новое завершённое assistant-сообщение, стабильное в течение settlePolls
|
||||||
// опросов. Возвращает вердикт (текст text-партов), либо rc=-1 при таймауте.
|
// опросов. Возвращает вердикт (текст text-партов), либо rc=-1 при таймауте.
|
||||||
func (r *Runner) awaitVerdict(ctx context.Context, c *Client, model *ModelRef, sid, agent, prompt string) (*Result, error) {
|
func (r *Runner) awaitVerdict(ctx context.Context, c *Client, sid, agent, prompt string) (*Result, error) {
|
||||||
// admit промпта; граница «новых» сообщений — время создания user-сообщения.
|
// admit промпта; граница «новых» сообщений — время создания user-сообщения.
|
||||||
admittedAt := time.Now().UnixMilli()
|
admittedAt := time.Now().UnixMilli()
|
||||||
adm, err := c.Prompt(ctx, sid, prompt)
|
adm, err := c.Prompt(ctx, sid, prompt)
|
||||||
@@ -187,7 +177,7 @@ func (r *Runner) awaitVerdict(ctx context.Context, c *Client, model *ModelRef, s
|
|||||||
doneSeen++
|
doneSeen++
|
||||||
emptySeen = 0
|
emptySeen = 0
|
||||||
if doneSeen >= settlePolls {
|
if doneSeen >= settlePolls {
|
||||||
return r.verdict(model, cur, msgs, admittedAt, sid)
|
return r.verdict(cur, msgs, admittedAt, sid)
|
||||||
}
|
}
|
||||||
case !active && cur == nil:
|
case !active && cur == nil:
|
||||||
// сессия завершилась, но нового assistant-сообщения так и нет
|
// сессия завершилась, но нового assistant-сообщения так и нет
|
||||||
@@ -207,19 +197,17 @@ func (r *Runner) awaitVerdict(ctx context.Context, c *Client, model *ModelRef, s
|
|||||||
}
|
}
|
||||||
|
|
||||||
// verdict собирает финальный результат из новых assistant-сообщений.
|
// verdict собирает финальный результат из новых assistant-сообщений.
|
||||||
// Проверяет фактическую модель ответа и логирует warning при расхождении
|
func (r *Runner) verdict(cur *v2Message, msgs []v2Message, since int64, sid string) (*Result, error) {
|
||||||
// с ожидаемой (устойчивость к «не той» модели — класс O5 WARN).
|
|
||||||
func (r *Runner) verdict(model *ModelRef, cur *v2Message, msgs []v2Message, since int64, sid string) (*Result, error) {
|
|
||||||
if model != nil && cur.Model != nil && (model.ProviderID != cur.Model.ProviderID || model.ID != cur.Model.ID) {
|
|
||||||
r.logf("WARN opencode: сессия %s отвечала моделью %s, а не ожидаемой %s — проверь providers в конфиге (v2-схема: provider.api / request, а не npm/options)", sid, cur.Model, model)
|
|
||||||
}
|
|
||||||
if cur.Error != nil && cur.Error.Message != "" {
|
if cur.Error != nil && cur.Error.Message != "" {
|
||||||
return nil, &ClientErr{Op: "prompt", Err: errors.New(cur.Error.Message)}
|
return nil, &ClientErr{Op: "prompt", Err: errors.New(cur.Error.Message)}
|
||||||
}
|
}
|
||||||
texts := assistantText(msgs, since)
|
texts, usedReasoning := assistantVerdict(msgs, since)
|
||||||
if len(texts) == 0 {
|
if len(texts) == 0 {
|
||||||
return nil, &ClientErr{Op: "prompt", Err: errors.New("нет text-части в ответе")}
|
return nil, &ClientErr{Op: "prompt", Err: errors.New("нет text-части в ответе")}
|
||||||
}
|
}
|
||||||
|
if usedReasoning {
|
||||||
|
r.logf("WARN opencode: в ответе нет text-части — использую reasoning-парты как вердикт")
|
||||||
|
}
|
||||||
vd := stripFence(strings.Join(texts, "\n"))
|
vd := stripFence(strings.Join(texts, "\n"))
|
||||||
r.logf("opencode вердикт готов (%d байт)", len(vd))
|
r.logf("opencode вердикт готов (%d байт)", len(vd))
|
||||||
return &Result{RC: 0, Stdout: vd, SessionID: sid}, nil
|
return &Result{RC: 0, Stdout: vd, SessionID: sid}, nil
|
||||||
|
|||||||
@@ -10,8 +10,6 @@ import (
|
|||||||
|
|
||||||
// fakePool создаёт Pool, в котором уже «живёт» сервер для каталога (без spawn):
|
// fakePool создаёт Pool, в котором уже «живёт» сервер для каталога (без spawn):
|
||||||
// Server{URL: fake.URL}, поэтому Runner ходит по HTTP на фейк-API.
|
// Server{URL: fake.URL}, поэтому Runner ходит по HTTP на фейк-API.
|
||||||
// XDG_CONFIG_HOME уводится во временный каталог, чтобы ReadModelRef не читал
|
|
||||||
// реальный пользовательский конфиг opencode (детерминизм тестов).
|
|
||||||
func fakePool(t *testing.T, f *fakeAPIServer, dir string) (*Pool, *Client) {
|
func fakePool(t *testing.T, f *fakeAPIServer, dir string) (*Pool, *Client) {
|
||||||
t.Helper()
|
t.Helper()
|
||||||
ts := httptestURL(t, f)
|
ts := httptestURL(t, f)
|
||||||
@@ -110,6 +108,28 @@ func TestRun_ReasoningGrowth(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestRun_ReasoningOnlyVerdict(t *testing.T) {
|
||||||
|
t.Setenv("XDG_CONFIG_HOME", t.TempDir())
|
||||||
|
dir := t.TempDir()
|
||||||
|
// завершённый ответ без text-парта, только reasoning — вердикт собирается
|
||||||
|
// из reasoning (fallback) вместо ошибки «нет text-части в ответе».
|
||||||
|
f := &fakeAPIServer{verdictReasoning: "размышления без текста"}
|
||||||
|
p, _ := fakePool(t, f, dir)
|
||||||
|
|
||||||
|
r := &Runner{Pool: p, IdleTimeout: time.Minute, HardTimeout: time.Minute,
|
||||||
|
PollInterval: 5 * time.Millisecond, Stdout: io.Discard}
|
||||||
|
res, err := r.Run(context.Background(), "task", dir, "dev", "")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Run err: %v", err)
|
||||||
|
}
|
||||||
|
if res.RC != 0 {
|
||||||
|
t.Errorf("RC = %d, want 0 (reasoning-only вердикт)", res.RC)
|
||||||
|
}
|
||||||
|
if !contains(res.Stdout, "размышления без текста") {
|
||||||
|
t.Errorf("Stdout = %q, want reasoning fallback", res.Stdout)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestRun_ContextCancel(t *testing.T) {
|
func TestRun_ContextCancel(t *testing.T) {
|
||||||
t.Setenv("XDG_CONFIG_HOME", t.TempDir())
|
t.Setenv("XDG_CONFIG_HOME", t.TempDir())
|
||||||
dir := t.TempDir()
|
dir := t.TempDir()
|
||||||
@@ -126,7 +146,7 @@ func TestRun_ContextCancel(t *testing.T) {
|
|||||||
done <- res
|
done <- res
|
||||||
errCh <- err
|
errCh <- err
|
||||||
}()
|
}()
|
||||||
time.Sleep(30 * time.Millisecond)
|
time.Sleep(6000 * time.Millisecond)
|
||||||
cancel()
|
cancel()
|
||||||
res := <-done
|
res := <-done
|
||||||
if err := <-errCh; err != nil {
|
if err := <-errCh; err != nil {
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
"os/exec"
|
"os/exec"
|
||||||
|
"path/filepath"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
@@ -24,8 +25,6 @@ import (
|
|||||||
// отдаёт URL, процессом не владеет.
|
// отдаёт URL, процессом не владеет.
|
||||||
type Server struct {
|
type Server struct {
|
||||||
Bin string // путь к opencode (по умолчанию "opencode")
|
Bin string // путь к opencode (по умолчанию "opencode")
|
||||||
Config string // OPENCODE_CONFIG
|
|
||||||
ConfigDir string // OPENCODE_CONFIG_DIR
|
|
||||||
DBPath string // рабочая БД сервера (передам env, если задана)
|
DBPath string // рабочая БД сервера (передам env, если задана)
|
||||||
|
|
||||||
Host string // hostname для прослушивания
|
Host string // hostname для прослушивания
|
||||||
@@ -33,6 +32,12 @@ type Server struct {
|
|||||||
Password string // basic auth (если непустой — сервер защищён)
|
Password string // basic auth (если непустой — сервер защищён)
|
||||||
Dir string // каталог, в котором запускается serve (project сервера)
|
Dir string // каталог, в котором запускается serve (project сервера)
|
||||||
|
|
||||||
|
// ConfigPath — путь к глобальному конфигу opencode (opencode.jsonc).
|
||||||
|
// Передаётся serve через env OPENCODE_CONFIG, чтобы гарантированно
|
||||||
|
// подхватить модель/провайдеров независимо от резолва глобального пути.
|
||||||
|
// Пусто — serve резолвит конфиг сам (см. defaults).
|
||||||
|
ConfigPath string
|
||||||
|
|
||||||
// URL задаёт внешний сервер. Пусто — супервайзер владеет процессом.
|
// URL задаёт внешний сервер. Пусто — супервайзер владеет процессом.
|
||||||
URL string
|
URL string
|
||||||
PollInterval time.Duration // как часто проверять /global/health
|
PollInterval time.Duration // как часто проверять /global/health
|
||||||
@@ -62,6 +67,20 @@ func (s *Server) defaults() {
|
|||||||
if s.Stdout == nil {
|
if s.Stdout == nil {
|
||||||
s.Stdout = os.Stderr
|
s.Stdout = os.Stderr
|
||||||
}
|
}
|
||||||
|
if s.ConfigPath == "" {
|
||||||
|
s.ConfigPath = defaultOpenCodeConfigPath()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// defaultOpenCodeConfigPath возвращает путь к глобальному конфигу opencode,
|
||||||
|
// совпадающий с тем, что opencode загружает по умолчанию: ~/.config/opencode/
|
||||||
|
// (в т.ч. на Windows — см. Global.Path.config в исходниках opencode).
|
||||||
|
func defaultOpenCodeConfigPath() string {
|
||||||
|
home, _ := os.UserHomeDir()
|
||||||
|
if home == "" {
|
||||||
|
home = "."
|
||||||
|
}
|
||||||
|
return filepath.Join(home, ".config", "opencode", "opencode.jsonc")
|
||||||
}
|
}
|
||||||
|
|
||||||
// baseURL собирает полный адрес сервера (http://host:port).
|
// baseURL собирает полный адрес сервера (http://host:port).
|
||||||
@@ -101,6 +120,8 @@ func (s *Server) Start(ctx context.Context) error {
|
|||||||
s.done = done
|
s.done = done
|
||||||
s.mu.Unlock()
|
s.mu.Unlock()
|
||||||
|
|
||||||
|
log.Printf("opencode serve: spawn dir=%s HOME=%q USERPROFILE=%q OPENCODE_CONFIG=%q (serve наследует окружение родителя)", cmd.Dir, os.Getenv("HOME"), os.Getenv("USERPROFILE"), s.ConfigPath)
|
||||||
|
|
||||||
if err := cmd.Start(); err != nil {
|
if err := cmd.Start(); err != nil {
|
||||||
return fmt.Errorf("opencode serve: start %v: %w", s.Bin, err)
|
return fmt.Errorf("opencode serve: start %v: %w", s.Bin, err)
|
||||||
}
|
}
|
||||||
@@ -124,22 +145,8 @@ func (s *Server) serveCmd(ctx context.Context) *exec.Cmd {
|
|||||||
// Своя process-group: чтобы killGroup (по pgid) убивал только сервер и его
|
// Своя process-group: чтобы killGroup (по pgid) убивал только сервер и его
|
||||||
// дочерние процессы, а не чужой процесс (например, тест-реннер).
|
// дочерние процессы, а не чужой процесс (например, тест-реннер).
|
||||||
setpgid(cmd)
|
setpgid(cmd)
|
||||||
env := append(os.Environ(),
|
// Env не задаём: serve наследует окружение родителя и сам резолвит свой
|
||||||
"OPENCODE_DISABLE_AUTOUPDATE=1",
|
// глобальный/локальный конфиг (модель/провайдеры, агентов, БД) как обычно.
|
||||||
"OPENCODE_DISABLE_MODELS_FETCH=1")
|
|
||||||
if s.Config != "" {
|
|
||||||
env = append(env, "OPENCODE_CONFIG="+s.Config)
|
|
||||||
}
|
|
||||||
if s.ConfigDir != "" {
|
|
||||||
env = append(env, "OPENCODE_CONFIG_DIR="+s.ConfigDir)
|
|
||||||
}
|
|
||||||
if s.DBPath != "" {
|
|
||||||
env = append(env, "OPENCODE_DB="+s.DBPath)
|
|
||||||
}
|
|
||||||
if s.Password != "" {
|
|
||||||
env = append(env, "OPENCODE_SERVER_PASSWORD="+s.Password)
|
|
||||||
}
|
|
||||||
cmd.Env = env
|
|
||||||
return cmd
|
return cmd
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -199,3 +199,27 @@ func atoiOrZero(s string) int {
|
|||||||
}
|
}
|
||||||
return n
|
return n
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestServeCmd_NoCustomEnv(t *testing.T) {
|
||||||
|
// serveCmd не должен задавать свой Env: serve наследует окружение родителя
|
||||||
|
// и сам резолвит конфиг/БД/пароль как обычно.
|
||||||
|
s := &Server{}
|
||||||
|
s.defaults()
|
||||||
|
cmd := s.serveCmd(context.Background())
|
||||||
|
if cmd.Env != nil {
|
||||||
|
for _, kv := range cmd.Env {
|
||||||
|
if strings.HasPrefix(kv, "OPENCODE_") {
|
||||||
|
t.Errorf("serveCmd: не ожидал env %q", kv)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestServerCmd_CustomOpenCodeConfig(t *testing.T) {
|
||||||
|
// явно заданный путь переопределяет дефолтный
|
||||||
|
s := &Server{ConfigPath: `C:\custom\opencode.jsonc`}
|
||||||
|
s.defaults()
|
||||||
|
if s.ConfigPath != `C:\custom\opencode.jsonc` {
|
||||||
|
t.Errorf("ConfigPath = %q, want explicit", s.ConfigPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -264,6 +264,26 @@ func TestUpdateTraceOutput(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestDeleteTracesByAgent — удаление трасс по агенту (сброс маркера при /retry).
|
||||||
|
func TestDeleteTracesByAgent(t *testing.T) {
|
||||||
|
s, ctx := setupTestDB(t)
|
||||||
|
task := &Task{ChatID: "tg://del", TaskTag: "delete-by-agent"}
|
||||||
|
id, _ := s.CreateTask(ctx, task)
|
||||||
|
|
||||||
|
_, _ = s.AppendTrace(ctx, &Trace{TaskID: id, Agent: "dev"})
|
||||||
|
_, _ = s.AppendTrace(ctx, &Trace{TaskID: id, Agent: "postmortem"})
|
||||||
|
_, _ = s.AppendTrace(ctx, &Trace{TaskID: id, Agent: "postmortem"})
|
||||||
|
|
||||||
|
if err := s.DeleteTracesByAgent(ctx, id, "postmortem"); err != nil {
|
||||||
|
t.Fatalf("DeleteTracesByAgent: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
traces, _ := s.GetTraces(ctx, id)
|
||||||
|
if len(traces) != 1 || traces[0].Agent != "dev" {
|
||||||
|
t.Fatalf("traces = %d (agent %q), want только dev", len(traces), traces[0].Agent)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// хелперы для проверки классов ошибок
|
// хелперы для проверки классов ошибок
|
||||||
func IsNotFound(err error) bool {
|
func IsNotFound(err error) bool {
|
||||||
return errors.Is(err, ErrNotFound)
|
return errors.Is(err, ErrNotFound)
|
||||||
|
|||||||
@@ -107,6 +107,17 @@ func (s *Storage) GetLatestTrace(ctx context.Context, taskID int64, agent string
|
|||||||
return tr, nil
|
return tr, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// DeleteTracesByAgent удаляет все трассы задачи с заданным агентом
|
||||||
|
// (сброс маркера при /retry N: чистый перезапуск без наследия анализа).
|
||||||
|
func (s *Storage) DeleteTracesByAgent(ctx context.Context, taskID int64, agent string) error {
|
||||||
|
_, err := s.db.ExecContext(ctx,
|
||||||
|
`DELETE FROM traces WHERE task_id=? AND agent=?`, taskID, agent)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("%w: delete traces agent %s for task %d: %w", ErrDB, agent, taskID, err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
// DeleteTrace удаляет трассу. Только для тестов/админки.
|
// DeleteTrace удаляет трассу. Только для тестов/админки.
|
||||||
func (s *Storage) DeleteTrace(ctx context.Context, id int64) error {
|
func (s *Storage) DeleteTrace(ctx context.Context, id int64) error {
|
||||||
res, err := s.db.ExecContext(ctx, `DELETE FROM traces WHERE id = ?`, id)
|
res, err := s.db.ExecContext(ctx, `DELETE FROM traces WHERE id = ?`, id)
|
||||||
|
|||||||
@@ -15,6 +15,10 @@ type ChatPanel interface {
|
|||||||
Clear()
|
Clear()
|
||||||
// SetCommands подключает команды к композитору (ввод + кнопки).
|
// SetCommands подключает команды к композитору (ввод + кнопки).
|
||||||
SetCommands(c *Commands)
|
SetCommands(c *Commands)
|
||||||
|
// SetBoundTask сообщает панели задачу, к которой привязана вкладка
|
||||||
|
// (0 — свободный чат). Кнопка «Перезапустить» (/retry N) активна только
|
||||||
|
// для вкладки с привязанной задачей.
|
||||||
|
SetBoundTask(taskID int64)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NilChatPanel — no-op реализация ChatPanel для headless-режима и тестов.
|
// NilChatPanel — no-op реализация ChatPanel для headless-режима и тестов.
|
||||||
@@ -27,6 +31,7 @@ func (NilChatPanel) SetTranscript(string) {}
|
|||||||
func (NilChatPanel) Append(string) {}
|
func (NilChatPanel) Append(string) {}
|
||||||
func (NilChatPanel) Clear() {}
|
func (NilChatPanel) Clear() {}
|
||||||
func (NilChatPanel) SetCommands(*Commands) {}
|
func (NilChatPanel) SetCommands(*Commands) {}
|
||||||
|
func (NilChatPanel) SetBoundTask(int64) {}
|
||||||
|
|
||||||
// FormatRole — строка роли в диалоге (pure-функция, спец 12.12).
|
// FormatRole — строка роли в диалоге (pure-функция, спец 12.12).
|
||||||
func FormatRole(role string) string {
|
func FormatRole(role string) string {
|
||||||
|
|||||||
@@ -23,4 +23,32 @@ func TestNilChatPanelNoop(t *testing.T) {
|
|||||||
p.Append("строка")
|
p.Append("строка")
|
||||||
p.Clear()
|
p.Clear()
|
||||||
p.SetCommands(c)
|
p.SetCommands(c)
|
||||||
|
p.SetBoundTask(7)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Кнопка «Перезапустить» — полный аналог Telegram-команды /retry N:
|
||||||
|
// в канал уходит ровно тот же текст, что при ручном вводе команды.
|
||||||
|
func TestRetryButtonSubmitsSameCommandAsTelegram(t *testing.T) {
|
||||||
|
var got []string
|
||||||
|
c := NewCommands(func(text string) { got = append(got, text) })
|
||||||
|
|
||||||
|
c.Retry(42)
|
||||||
|
|
||||||
|
if len(got) != 1 || got[0] != "/retry 42" {
|
||||||
|
t.Fatalf("got %v, want [/retry 42]", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Свободная вкладка (без задачи): Retry(id<=0) — no-op, ничего не отправляется
|
||||||
|
// (кнопка «Перезапустить» на такой вкладке неактивна).
|
||||||
|
func TestRetryWithoutTaskIsNoop(t *testing.T) {
|
||||||
|
var got []string
|
||||||
|
c := NewCommands(func(text string) { got = append(got, text) })
|
||||||
|
|
||||||
|
c.Retry(0)
|
||||||
|
c.Retry(-5)
|
||||||
|
|
||||||
|
if len(got) != 0 {
|
||||||
|
t.Fatalf("got %v, want nothing submitted", got)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
@@ -30,8 +30,14 @@ func (c *Commands) Cancel() { c.Submit("/cancel") }
|
|||||||
// Skip — пропустить сбор, сформировать черновик (/skip).
|
// Skip — пропустить сбор, сформировать черновик (/skip).
|
||||||
func (c *Commands) Skip() { c.Submit("/skip") }
|
func (c *Commands) Skip() { c.Submit("/skip") }
|
||||||
|
|
||||||
// Retry — перезапустить задачу N (/retry N).
|
// Retry — перезапустить задачу N (/retry N): полный аналог Telegram-команды.
|
||||||
func (c *Commands) Retry(id int64) { c.Submit(fmt.Sprintf("/retry %d", id)) }
|
// id <= 0 — no-op (кнопка «Перезапустить» на свободной вкладке неактивна).
|
||||||
|
func (c *Commands) Retry(id int64) {
|
||||||
|
if id <= 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.Submit(fmt.Sprintf("/retry %d", id))
|
||||||
|
}
|
||||||
|
|
||||||
// Status — запросить статус задачи N (/status N).
|
// Status — запросить статус задачи N (/status N).
|
||||||
func (c *Commands) Status(id int64) { c.Submit(fmt.Sprintf("/status %d", id)) }
|
func (c *Commands) Status(id int64) { c.Submit(fmt.Sprintf("/status %d", id)) }
|
||||||
|
|||||||
@@ -21,6 +21,8 @@ type ChatPanel struct {
|
|||||||
convLbl *widget.Label
|
convLbl *widget.Label
|
||||||
input *widget.Entry
|
input *widget.Entry
|
||||||
commands *ui.Commands
|
commands *ui.Commands
|
||||||
|
retryBtn *widget.Button // «Перезапустить» — полный аналог /retry N
|
||||||
|
boundTask int64 // >0 — вкладка привязана к задаче №boundTask
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewChatPanel создаёт ChatPanel.
|
// NewChatPanel создаёт ChatPanel.
|
||||||
@@ -30,6 +32,8 @@ func NewChatPanel() *ChatPanel {
|
|||||||
p.convLbl.Wrapping = fyne.TextWrapWord
|
p.convLbl.Wrapping = fyne.TextWrapWord
|
||||||
p.input = widget.NewEntry()
|
p.input = widget.NewEntry()
|
||||||
p.input.SetPlaceHolder("Сообщение… (Enter — отправить)")
|
p.input.SetPlaceHolder("Сообщение… (Enter — отправить)")
|
||||||
|
p.retryBtn = widget.NewButton("Перезапустить", p.retryTask)
|
||||||
|
p.retryBtn.Disable() // активируется при привязке вкладки к задаче (SetBoundTask)
|
||||||
return p
|
return p
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -44,9 +48,30 @@ func (p *ChatPanel) Composer() fyne.CanvasObject {
|
|||||||
widget.NewButton("Новая", func() { if p.commands != nil { p.commands.Start() } }),
|
widget.NewButton("Новая", func() { if p.commands != nil { p.commands.Start() } }),
|
||||||
widget.NewButton("Создавай", func() { if p.commands != nil { p.commands.Approve() } }),
|
widget.NewButton("Создавай", func() { if p.commands != nil { p.commands.Approve() } }),
|
||||||
widget.NewButton("Пропустить", func() { if p.commands != nil { p.commands.Skip() } }),
|
widget.NewButton("Пропустить", func() { if p.commands != nil { p.commands.Skip() } }),
|
||||||
|
p.retryBtn,
|
||||||
widget.NewButton("Отмена", func() { if p.commands != nil { p.commands.Cancel() } }),
|
widget.NewButton("Отмена", func() { if p.commands != nil { p.commands.Cancel() } }),
|
||||||
)
|
)
|
||||||
return container.NewBorder(nil, nil, cmdBar, p.input, nil)
|
return container.NewBorder(nil, nil, cmdBar, nil, p.input)
|
||||||
|
}
|
||||||
|
|
||||||
|
// retryTask — колбэк кнопки «Перезапустить»: отправляет "/retry N" тем же
|
||||||
|
// путём, что и ввод пользователя (Commands → Router → Core), т.е. серверная
|
||||||
|
// логика полностью совпадает с Telegram-командой /retry N.
|
||||||
|
func (p *ChatPanel) retryTask() {
|
||||||
|
if p.commands != nil {
|
||||||
|
p.commands.Retry(p.boundTask)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// SetBoundTask включает кнопку «Перезапустить», когда вкладка привязана к
|
||||||
|
// задаче (taskID > 0); для свободной вкладки кнопка неактивна.
|
||||||
|
func (p *ChatPanel) SetBoundTask(taskID int64) {
|
||||||
|
p.boundTask = taskID
|
||||||
|
if taskID > 0 {
|
||||||
|
p.retryBtn.Enable()
|
||||||
|
} else {
|
||||||
|
p.retryBtn.Disable()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetTranscript заменяет содержимое диалога (спец 12.12).
|
// SetTranscript заменяет содержимое диалога (спец 12.12).
|
||||||
|
|||||||
@@ -3,6 +3,9 @@
|
|||||||
package desktop
|
package desktop
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
"fyne.io/fyne/v2"
|
"fyne.io/fyne/v2"
|
||||||
"fyne.io/fyne/v2/container"
|
"fyne.io/fyne/v2/container"
|
||||||
"fyne.io/fyne/v2/widget"
|
"fyne.io/fyne/v2/widget"
|
||||||
@@ -10,40 +13,181 @@ import (
|
|||||||
"github.com/kamelion/ratatoskr-go/internal/ui"
|
"github.com/kamelion/ratatoskr-go/internal/ui"
|
||||||
)
|
)
|
||||||
|
|
||||||
// LogPanel — Fyne-реализация ui.LogPanel (спец 12.8, 12.9): поток системных логов.
|
// maxLogBytes — кап суммарного объёма текста буфера «Логи» (байты), чтобы
|
||||||
|
// не расти бесконечно и не копить мегабайты в памяти.
|
||||||
|
const maxLogBytes = 10 * 1024 * 1024
|
||||||
|
|
||||||
|
// renderFlushMs — дебаунс-интервал перерисовки ленты (мс): при всплеске
|
||||||
|
// логов рендер выполняется не на каждую запись, а раз в этот интервал.
|
||||||
|
const renderFlushMs = 200
|
||||||
|
|
||||||
|
// LogPanel — Fyne-реализация ui.LogPanel (спец 12.8, 12.9): поток системных
|
||||||
|
// логов с фильтром по уровням (error/warning/info/debug).
|
||||||
//
|
//
|
||||||
// Внутри — widget.Label с ограниченным буфером (maxLogLen); Append обрезает
|
// Записи хранятся в ui.LogBuffer (ограниченный буфер); лента показывает
|
||||||
// строку до лимита. Вызовы Append приходят из горутин Hub, поэтому обновление
|
// только записи уровней, отмеченных в фильтре (по умолчанию отмечены все).
|
||||||
// виджета выполняется через fyne.Do (на потоке Fyne).
|
// Фильтр — выпадающий список с чекбоксами: кнопка над лентой открывает
|
||||||
|
// widget.PopUp с widget.Check по каждому уровню; изменение набора сразу
|
||||||
|
// перерисовывает ленту — и по уже загруженным записям, и для новых.
|
||||||
|
//
|
||||||
|
// Состояние (buf/selected/pending/flushScheduled/lastFlush) доступается только
|
||||||
|
// с потока Fyne: Append приходит из горутин Hub и заворачивается в fyne.Do,
|
||||||
|
// колбэки чекбоксов Fyne вызывает сам на своём потоке. Перерисовка ленты
|
||||||
|
// коалесцируется (см. scheduleFlush), т.к. строка ленты пересобирается
|
||||||
|
// целиком и при капе в 10MB на каждый append была бы дорогой.
|
||||||
type LogPanel struct {
|
type LogPanel struct {
|
||||||
label *widget.Label
|
label *widget.Label
|
||||||
|
btn *widget.Button // кнопка-дропдаун фильтра уровней
|
||||||
|
|
||||||
|
buf *ui.LogBuffer // буфер записей, старые вытесняются
|
||||||
|
selected map[string]bool // уровень → показывать (по умолчанию все)
|
||||||
|
|
||||||
|
pending bool // буфер изменился, рендер отложен (поток Fyne)
|
||||||
|
flushScheduled bool // фоновый флаш уже запущен (поток Fyne)
|
||||||
|
lastFlush int64 // мс последнего рендера (поток Fyne)
|
||||||
}
|
}
|
||||||
|
|
||||||
// maxLogLen — обрезка буфера панели «Логи», чтобы не расти бесконечно.
|
// NewLogPanel создаёт LogPanel с уровнями error/warning/info (debug скрыт).
|
||||||
const maxLogLen = 200_000
|
|
||||||
|
|
||||||
// NewLogPanel создаёт LogPanel.
|
|
||||||
func NewLogPanel() *LogPanel {
|
func NewLogPanel() *LogPanel {
|
||||||
l := widget.NewLabel("")
|
p := &LogPanel{
|
||||||
l.Wrapping = fyne.TextWrapWord
|
buf: ui.NewLogBuffer(maxLogBytes),
|
||||||
return &LogPanel{label: l}
|
selected: make(map[string]bool, len(ui.LogLevels)),
|
||||||
}
|
}
|
||||||
|
for _, lvl := range ui.LogLevels {
|
||||||
// Widget возвращает контейнер панели для встраивания в окно (таб «Логи»).
|
p.selected[lvl] = lvl != ui.LogLevelDebug
|
||||||
func (p *LogPanel) Widget() fyne.CanvasObject {
|
}
|
||||||
return container.NewScroll(p.label)
|
|
||||||
}
|
p.label = widget.NewLabel("")
|
||||||
|
p.label.Wrapping = fyne.TextWrapWord
|
||||||
// Append добавляет строку лога (спец 12.9), с ограничением буфера.
|
p.btn = widget.NewButton(p.filterCaption(), p.showFilterPopUp)
|
||||||
func (p *LogPanel) Append(text string) {
|
return p
|
||||||
fyne.Do(func() {
|
}
|
||||||
s := p.label.Text + text + "\n"
|
|
||||||
if len(s) > maxLogLen {
|
// Widget возвращает контейнер панели для встраивания в окно (таб «Логи»):
|
||||||
s = s[len(s)-maxLogLen:]
|
// строка фильтра сверху, лента лога в скролле снизу.
|
||||||
|
func (p *LogPanel) Widget() fyne.CanvasObject {
|
||||||
|
return container.NewBorder(
|
||||||
|
container.NewHBox(p.btn), nil, nil, nil,
|
||||||
|
container.NewScroll(p.label),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Append добавляет запись лога (спец 12.9), с ограничением буфера.
|
||||||
|
// Вызывается из горутин Hub → вся работа на потоке Fyne через fyne.Do.
|
||||||
|
func (p *LogPanel) Append(level, text string) {
|
||||||
|
fyne.Do(func() {
|
||||||
|
lvl := ui.NormalizeLogLevel(level)
|
||||||
|
dropped := p.buf.Append(lvl, text)
|
||||||
|
// Перерисовка нужна, если запись видима или вытеснение задело ленту.
|
||||||
|
if p.selected[lvl] || dropped {
|
||||||
|
p.pending = true
|
||||||
|
p.scheduleFlush()
|
||||||
}
|
}
|
||||||
p.label.SetText(s)
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// scheduleFlush планирует отложенный рендер через renderFlushMs (или сразу,
|
||||||
|
// если с прошлого рендера прошло больше интервала). Все вызовы — с потока
|
||||||
|
// Fyne; фоновый поток только спит и по таймеру отправляет flush через fyne.Do,
|
||||||
|
// поэтому защита от повторного планирования (flushScheduled) работает без
|
||||||
|
// блокировок.
|
||||||
|
func (p *LogPanel) scheduleFlush() {
|
||||||
|
if p.flushScheduled {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
p.flushScheduled = true
|
||||||
|
delay := int64(renderFlushMs) - (time.Now().UnixMilli() - p.lastFlush)
|
||||||
|
if delay < 1 {
|
||||||
|
delay = 1
|
||||||
|
}
|
||||||
|
go func() {
|
||||||
|
time.Sleep(time.Duration(delay) * time.Millisecond)
|
||||||
|
fyne.Do(p.flush)
|
||||||
|
}()
|
||||||
|
}
|
||||||
|
|
||||||
|
// flush выполняет отложенную перерисовку ленты (на потоке Fyne).
|
||||||
|
func (p *LogPanel) flush() {
|
||||||
|
p.flushScheduled = false
|
||||||
|
if p.pending {
|
||||||
|
p.pending = false
|
||||||
|
p.lastFlush = time.Now().UnixMilli()
|
||||||
|
p.render()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// render перерисовывает ленту: дописывает в label только новые видимые
|
||||||
|
// записи (RenderNew). Если вытеснение задело уже выведенные записи (stale),
|
||||||
|
// RenderNew возвращает полный текст — пересобираем label целиком. Так при
|
||||||
|
// росте ленты мимо лимита флаш дешёвый (только дельта), а полная пересборка
|
||||||
|
// происходит лишь по вытеснению/смене фильтра.
|
||||||
|
func (p *LogPanel) render() {
|
||||||
|
delta, full := p.buf.RenderNew(p.selected)
|
||||||
|
if full {
|
||||||
|
p.label.SetText(delta)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if delta == "" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
cur := p.label.Text
|
||||||
|
if cur == "" {
|
||||||
|
p.label.SetText(delta)
|
||||||
|
} else {
|
||||||
|
p.label.SetText(cur + delta)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// setLevel включает/выключает показ уровня и сразу обновляет ленту.
|
||||||
|
// Вызывается только с потока Fyne (колбэки чекбоксов). Смена набора уровней
|
||||||
|
// меняет видимость уже выведенных записей, поэтому — полный пересбор.
|
||||||
|
func (p *LogPanel) setLevel(level string, on bool) {
|
||||||
|
if p.selected[level] == on {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
p.selected[level] = on
|
||||||
|
p.btn.SetText(p.filterCaption())
|
||||||
|
p.label.SetText(p.buf.Render(p.selected))
|
||||||
|
p.buf.ResetRendered()
|
||||||
|
}
|
||||||
|
|
||||||
|
// filterCaption — подпись кнопки-дропдауна: текущий выбранный набор уровней.
|
||||||
|
func (p *LogPanel) filterCaption() string {
|
||||||
|
var chosen []string
|
||||||
|
for _, lvl := range ui.LogLevels {
|
||||||
|
if p.selected[lvl] {
|
||||||
|
chosen = append(chosen, lvl)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
switch {
|
||||||
|
case len(chosen) == len(ui.LogLevels):
|
||||||
|
return "Уровни: все"
|
||||||
|
case len(chosen) == 0:
|
||||||
|
return "Уровни: нет"
|
||||||
|
default:
|
||||||
|
return "Уровни: " + strings.Join(chosen, "+")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// showFilterPopUp открывает выпадающий список с чекбоксами уровней под
|
||||||
|
// кнопкой фильтра. Список пересобирается при каждом открытии — чекбоксы
|
||||||
|
// всегда отражают актуальное состояние фильтра.
|
||||||
|
func (p *LogPanel) showFilterPopUp() {
|
||||||
|
c := fyne.CurrentApp().Driver().CanvasForObject(p.btn)
|
||||||
|
if c == nil {
|
||||||
|
return // виджет ещё не встроен в окно
|
||||||
|
}
|
||||||
|
rows := make([]fyne.CanvasObject, 0, len(ui.LogLevels))
|
||||||
|
for _, lvl := range ui.LogLevels {
|
||||||
|
lvl := lvl
|
||||||
|
cb := widget.NewCheck(lvl, func(on bool) { p.setLevel(lvl, on) })
|
||||||
|
cb.SetChecked(p.selected[lvl])
|
||||||
|
rows = append(rows, cb)
|
||||||
|
}
|
||||||
|
pop := widget.NewPopUp(container.NewPadded(container.NewVBox(rows...)), c)
|
||||||
|
pos := fyne.CurrentApp().Driver().AbsolutePositionForObject(p.btn)
|
||||||
|
pop.ShowAtPosition(fyne.NewPos(pos.X, pos.Y+p.btn.Size().Height))
|
||||||
|
}
|
||||||
|
|
||||||
// compile-time проверка реализации контракта.
|
// compile-time проверка реализации контракта.
|
||||||
var _ ui.LogPanel = (*LogPanel)(nil)
|
var _ ui.LogPanel = (*LogPanel)(nil)
|
||||||
48
internal/ui/desktop/maximize_windows.go
Normal file
48
internal/ui/desktop/maximize_windows.go
Normal file
@@ -0,0 +1,48 @@
|
|||||||
|
//go:build cgo
|
||||||
|
|
||||||
|
package desktop
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"syscall"
|
||||||
|
"unsafe"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Нативный максимум окна через Win32.
|
||||||
|
//
|
||||||
|
// Fyne (v2.6) не даёт ни Move, ни Maximize у fyne.Window, поэтому разворот на
|
||||||
|
// экран (над панелью задач) делаем напрямую через user32: находим HWND нашего
|
||||||
|
// процесса через EnumWindows и вызываем ShowWindow(SW_MAXIMIZE).
|
||||||
|
// Только Windows (платформа приложения по спеке — Windows).
|
||||||
|
|
||||||
|
const (
|
||||||
|
swMaximize = 9
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
user32 = syscall.NewLazyDLL("user32.dll")
|
||||||
|
procEnumWindows = user32.NewProc("EnumWindows")
|
||||||
|
procGetWindowThreadProcID = user32.NewProc("GetWindowThreadProcessId")
|
||||||
|
procIsWindowVisible = user32.NewProc("IsWindowVisible")
|
||||||
|
procShowWindow = user32.NewProc("ShowWindow")
|
||||||
|
)
|
||||||
|
|
||||||
|
func maximizeWindow() {
|
||||||
|
pid := uint32(os.Getpid())
|
||||||
|
|
||||||
|
cb := syscall.NewCallback(func(hwnd syscall.Handle, _ uintptr) uintptr {
|
||||||
|
var winPid uint32
|
||||||
|
procGetWindowThreadProcID.Call(uintptr(hwnd), uintptr(unsafe.Pointer(&winPid)))
|
||||||
|
if winPid != pid {
|
||||||
|
return 1 // continue перечисление
|
||||||
|
}
|
||||||
|
visible, _, _ := procIsWindowVisible.Call(uintptr(hwnd))
|
||||||
|
if visible == 0 {
|
||||||
|
return 1 // continue перечисление
|
||||||
|
}
|
||||||
|
procShowWindow.Call(uintptr(hwnd), swMaximize)
|
||||||
|
return 0 // остановить перечисление
|
||||||
|
})
|
||||||
|
|
||||||
|
procEnumWindows.Call(cb, 0)
|
||||||
|
}
|
||||||
@@ -8,6 +8,7 @@ package desktop
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
@@ -25,15 +26,17 @@ import (
|
|||||||
const (
|
const (
|
||||||
selectedPref = "task.selected"
|
selectedPref = "task.selected"
|
||||||
splitHPref = "split.h"
|
splitHPref = "split.h"
|
||||||
splitVPref = "split.v"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Window — Fyne-окно: chat.Channel + ui.View.
|
// Window — Fyne-окно: chat.Channel + ui.View.
|
||||||
|
//
|
||||||
|
// Внутри — вкладки-сессии (ui.Session): каждая вкладка — независимый диалог
|
||||||
|
// с LLM (свой UserID/Address, свой активный task). Свободные вкладки (без
|
||||||
|
// задачи) создаются кнопкой «Новый чат»; вкладки задач открываются из списка.
|
||||||
type Window struct {
|
type Window struct {
|
||||||
fyneApp fyne.App
|
fyneApp fyne.App
|
||||||
win fyne.Window
|
win fyne.Window
|
||||||
store ui.Store
|
store ui.Store
|
||||||
commands *ui.Commands
|
|
||||||
|
|
||||||
handler chat.Handler
|
handler chat.Handler
|
||||||
|
|
||||||
@@ -43,12 +46,17 @@ type Window struct {
|
|||||||
// виджеты
|
// виджеты
|
||||||
taskList *TaskListPanel
|
taskList *TaskListPanel
|
||||||
detail *TaskDetailPanel
|
detail *TaskDetailPanel
|
||||||
chat *ChatPanel
|
|
||||||
state *StatePanel
|
state *StatePanel
|
||||||
logs *LogPanel
|
logs *LogPanel
|
||||||
|
|
||||||
|
// вкладки-сессии
|
||||||
|
tabs *container.AppTabs
|
||||||
|
sessions map[chat.Address]*ui.Session // Address вкладки → сессия
|
||||||
|
panels map[chat.Address]*ChatPanel // Address вкладки → панель
|
||||||
|
tabSession map[*container.TabItem]*ui.Session
|
||||||
|
nextSess int // счётчик уникальных ui://sess-<n>
|
||||||
|
|
||||||
hsplit *container.Split
|
hsplit *container.Split
|
||||||
vsplit *container.Split
|
|
||||||
|
|
||||||
onQuit func()
|
onQuit func()
|
||||||
}
|
}
|
||||||
@@ -59,10 +67,11 @@ func New(appID string, store ui.Store) *Window {
|
|||||||
w := &Window{
|
w := &Window{
|
||||||
fyneApp: fa,
|
fyneApp: fa,
|
||||||
store: store,
|
store: store,
|
||||||
|
sessions: map[chat.Address]*ui.Session{},
|
||||||
|
panels: map[chat.Address]*ChatPanel{},
|
||||||
|
tabSession: map[*container.TabItem]*ui.Session{},
|
||||||
}
|
}
|
||||||
w.win = fa.NewWindow("Ratatoskr")
|
w.win = fa.NewWindow("Ratatoskr")
|
||||||
w.win.SetFullScreen(true)
|
|
||||||
w.commands = ui.NewCommands(w.submitText)
|
|
||||||
w.build()
|
w.build()
|
||||||
w.win.SetCloseIntercept(func() {
|
w.win.SetCloseIntercept(func() {
|
||||||
// Закрытие окна = сворачивание (спец 12.6): Core продолжает работать.
|
// Закрытие окна = сворачивание (спец 12.6): Core продолжает работать.
|
||||||
@@ -79,17 +88,12 @@ func (w *Window) SetOnQuit(fn func()) { w.onQuit = fn }
|
|||||||
// build собирает виджеты и раскладку.
|
// build собирает виджеты и раскладку.
|
||||||
func (w *Window) build() {
|
func (w *Window) build() {
|
||||||
w.detail = NewTaskDetailPanel()
|
w.detail = NewTaskDetailPanel()
|
||||||
|
|
||||||
w.chat = NewChatPanel()
|
|
||||||
w.chat.SetCommands(w.commands)
|
|
||||||
|
|
||||||
w.state = NewStatePanel()
|
w.state = NewStatePanel()
|
||||||
|
|
||||||
w.logs = NewLogPanel()
|
w.logs = NewLogPanel()
|
||||||
|
|
||||||
// Список задач (слева)
|
// Список задач (слева)
|
||||||
w.taskList = NewTaskListPanel()
|
w.taskList = NewTaskListPanel()
|
||||||
w.taskList.OnSelect(func(id int64) { w.selectTask(id) })
|
w.taskList.OnSelect(func(id int64) { w.openTaskTab(id) })
|
||||||
|
|
||||||
quitBtn := widget.NewButtonWithIcon("Завершить", theme.LogoutIcon(), func() {
|
quitBtn := widget.NewButtonWithIcon("Завершить", theme.LogoutIcon(), func() {
|
||||||
w.saveLayout()
|
w.saveLayout()
|
||||||
@@ -99,60 +103,121 @@ func (w *Window) build() {
|
|||||||
})
|
})
|
||||||
left := container.NewBorder(nil, quitBtn, nil, nil, w.taskList.Widget())
|
left := container.NewBorder(nil, quitBtn, nil, nil, w.taskList.Widget())
|
||||||
|
|
||||||
// Рабочая область: детали сверху, диалог снизу (сплит 2×2 в плане).
|
// Центр: вкладки-сессии + кнопка «Новый чат».
|
||||||
details := w.detail.Widget()
|
w.tabs = container.NewAppTabs()
|
||||||
right := container.NewVSplit(
|
w.tabs.OnSelected = func(tab *container.TabItem) {
|
||||||
container.NewScroll(details),
|
// фокус вкладки → показываем детали/состояние её задачи
|
||||||
w.chat.Transcript(),
|
if s := w.tabSession[tab]; s != nil && s.BoundTask > 0 {
|
||||||
)
|
w.selectTask(s.BoundTask)
|
||||||
w.vsplit = right
|
}
|
||||||
|
}
|
||||||
|
newBtn := widget.NewButtonWithIcon("Новый чат", theme.ContentAddIcon(), func() {
|
||||||
|
w.addFreeTab()
|
||||||
|
})
|
||||||
|
center := container.NewBorder(newBtn, nil, nil, nil, w.tabs)
|
||||||
|
|
||||||
// Нижняя панель: вкладки «Логи» + «Состояние».
|
// Нижняя панель: вкладки «Логи» + «Состояние».
|
||||||
bottomTabs := container.NewAppTabs(
|
bottomTabs := container.NewAppTabs(
|
||||||
container.NewTabItem("Логи", w.logs.Widget()),
|
container.NewTabItem("Логи", w.logs.Widget()),
|
||||||
container.NewTabItem("Состояние", w.state.Widget()),
|
container.NewTabItem("Состояние", w.state.Widget()),
|
||||||
)
|
)
|
||||||
|
bottom := container.NewBorder(nil, nil, nil, nil, bottomTabs)
|
||||||
|
|
||||||
bottom := container.NewBorder(nil, w.chat.Composer(), nil, nil, bottomTabs)
|
w.hsplit = container.NewHSplit(left, container.NewVSplit(center, bottom))
|
||||||
|
w.win.SetContent(w.hsplit)
|
||||||
w.hsplit = container.NewHSplit(left, right)
|
|
||||||
|
|
||||||
root := container.NewVSplit(w.hsplit, bottom)
|
|
||||||
w.win.SetContent(root)
|
|
||||||
|
|
||||||
// Восстановление layout из Preferences.
|
// Восстановление layout из Preferences.
|
||||||
pref := w.fyneApp.Preferences()
|
pref := w.fyneApp.Preferences()
|
||||||
w.hsplit.SetOffset(pref.FloatWithFallback(splitHPref, 0.30))
|
w.hsplit.SetOffset(pref.FloatWithFallback(splitHPref, 0.30))
|
||||||
w.vsplit.SetOffset(pref.FloatWithFallback(splitVPref, 0.45))
|
|
||||||
|
|
||||||
// Начальный снимок списка.
|
// Начальный снимок списка.
|
||||||
w.refreshList()
|
w.refreshList()
|
||||||
|
|
||||||
|
// Открываем свободную вкладку-сессию по умолчанию.
|
||||||
|
w.addFreeTab()
|
||||||
}
|
}
|
||||||
|
|
||||||
// saveLayout сохраняет текущие позиции сплитов в Preferences.
|
// addFreeTab создаёт свободную вкладку-сессию (in-memory диалог с LLM).
|
||||||
// Вызывается при сворачивании окна и перед полным выходом.
|
func (w *Window) addFreeTab() {
|
||||||
func (w *Window) saveLayout() {
|
w.nextSess++
|
||||||
pref := w.fyneApp.Preferences()
|
id := fmt.Sprintf("%s%d", ui.SessionID, w.nextSess)
|
||||||
pref.SetFloat(splitHPref, w.hsplit.Offset)
|
w.addTab(ui.NewFreeSession(chat.UserID(id), chat.Address(id)))
|
||||||
pref.SetFloat(splitVPref, w.vsplit.Offset)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// selectTask загружает детали выбранной задачи (снимки из Store).
|
// addTab создаёт вкладку для сессии и фокусирует её.
|
||||||
func (w *Window) selectTask(id int64) {
|
func (w *Window) addTab(sess *ui.Session) {
|
||||||
w.selected = id
|
panel := NewChatPanel()
|
||||||
w.taskList.Select(id)
|
panel.SetCommands(ui.NewCommands(func(text string) {
|
||||||
w.fyneApp.Preferences().SetInt(selectedPref, int(id))
|
w.submitText(sess.UserID, sess.Address, text)
|
||||||
|
}))
|
||||||
|
// Кнопка «Перезапустить» активна только у вкладки с задачей (/retry N).
|
||||||
|
panel.SetBoundTask(sess.BoundTask)
|
||||||
|
|
||||||
|
content := container.NewBorder(nil, panel.Composer(), nil, nil, panel.Transcript())
|
||||||
|
tab := container.NewTabItem(sess.Title, content)
|
||||||
|
w.tabs.Append(tab)
|
||||||
|
|
||||||
|
w.sessions[sess.Address] = sess
|
||||||
|
w.panels[sess.Address] = panel
|
||||||
|
w.tabSession[tab] = sess
|
||||||
|
|
||||||
|
w.tabs.Select(tab)
|
||||||
|
}
|
||||||
|
|
||||||
|
// tabForAddress возвращает вкладку по адресу сессии.
|
||||||
|
func (w *Window) tabForAddress(addr chat.Address) *container.TabItem {
|
||||||
|
for tab, s := range w.tabSession {
|
||||||
|
if s.Address == addr {
|
||||||
|
return tab
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// openTaskTab открывает вкладку задачи (или фокусирует уже открытую).
|
||||||
|
//
|
||||||
|
// Вкладку можно открыть только для задачи, созданной в UI-сессии (ChatID с
|
||||||
|
// префиксом ui://sess-*): её ChatID уникален и не пересекается с Telegram-каналом.
|
||||||
|
// Задачи Telegram не открываются в отдельной вкладке — их показывает список и
|
||||||
|
// детали (selectTask), чтобы не плодить сессию с общим UserID=чат-id.
|
||||||
|
func (w *Window) openTaskTab(id int64) {
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
t, err := w.store.GetTask(ctx, id)
|
t, err := w.store.GetTask(ctx, id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
w.detail.ShowEmpty()
|
|
||||||
w.chat.Clear()
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
w.detail.ShowTask(t)
|
|
||||||
|
|
||||||
// Диалог: история + краткие трассы.
|
// Telegram-задача (ChatID без ui://sess-*) — только детали, без вкладки.
|
||||||
|
if !strings.HasPrefix(t.ChatID, ui.SessionID) {
|
||||||
|
w.selectTask(id)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Если вкладка для задачи уже есть — фокусируем.
|
||||||
|
for _, s := range w.sessions {
|
||||||
|
if s.BoundTask == id {
|
||||||
|
if tab := w.tabForAddress(s.Address); tab != nil {
|
||||||
|
w.tabs.Select(tab)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Создаём новую вкладку: UserID == task.ChatID (чат задачи уникален).
|
||||||
|
w.nextSess++
|
||||||
|
addr := chat.Address(fmt.Sprintf("%s%d", ui.SessionID, w.nextSess))
|
||||||
|
sess := ui.NewTaskSession(addr, t)
|
||||||
|
w.addTab(sess)
|
||||||
|
|
||||||
|
// Транскрипт из истории задачи.
|
||||||
|
if panel := w.panels[sess.Address]; panel != nil {
|
||||||
|
panel.SetTranscript(w.renderHistory(ctx, id))
|
||||||
|
}
|
||||||
|
w.selectTask(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderHistory строит транскрипт диалога задачи (история + краткие трассы).
|
||||||
|
func (w *Window) renderHistory(ctx context.Context, id int64) string {
|
||||||
hist, err := w.store.GetHistory(ctx, id)
|
hist, err := w.store.GetHistory(ctx, id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
hist = nil
|
hist = nil
|
||||||
@@ -172,9 +237,30 @@ func (w *Window) selectTask(id int64) {
|
|||||||
if b.Len() == 0 {
|
if b.Len() == 0 {
|
||||||
b.WriteString("Нет сообщений. /start — начать задачу.")
|
b.WriteString("Нет сообщений. /start — начать задачу.")
|
||||||
}
|
}
|
||||||
w.chat.SetTranscript(b.String())
|
return b.String()
|
||||||
|
}
|
||||||
|
|
||||||
// Состояние: сброс к снимку задач.
|
// saveLayout сохраняет текущие позиции сплитов в Preferences.
|
||||||
|
// Вызывается при сворачивании окна и перед полным выходом.
|
||||||
|
func (w *Window) saveLayout() {
|
||||||
|
pref := w.fyneApp.Preferences()
|
||||||
|
pref.SetFloat(splitHPref, w.hsplit.Offset)
|
||||||
|
}
|
||||||
|
|
||||||
|
// selectTask загружает детали выбранной задачи (снимки из Store).
|
||||||
|
func (w *Window) selectTask(id int64) {
|
||||||
|
w.selected = id
|
||||||
|
w.taskList.Select(id)
|
||||||
|
w.fyneApp.Preferences().SetInt(selectedPref, int(id))
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
t, err := w.store.GetTask(ctx, id)
|
||||||
|
if err != nil {
|
||||||
|
w.detail.ShowEmpty()
|
||||||
|
w.state.SetText(ui.StateText(nil))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
w.detail.ShowTask(t)
|
||||||
w.refreshState(ctx, id)
|
w.refreshState(ctx, id)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -199,23 +285,42 @@ func (w *Window) refreshList() {
|
|||||||
w.taskList.Select(w.selected)
|
w.taskList.Select(w.selected)
|
||||||
}
|
}
|
||||||
|
|
||||||
// appendLog добавляет строку в панель «Логи» (делегирует в LogPanel).
|
// appendLog добавляет запись в панель «Логи» (делегирует в LogPanel).
|
||||||
func (w *Window) appendLog(text string) {
|
func (w *Window) appendLog(level, text string) {
|
||||||
w.logs.Append(text)
|
w.logs.Append(level, text)
|
||||||
}
|
}
|
||||||
|
|
||||||
// submitText отправляет ввод пользователя в Router.
|
// submitText отправляет ввод пользователя из вкладки в Router.
|
||||||
func (w *Window) submitText(text string) {
|
func (w *Window) submitText(uid chat.UserID, addr chat.Address, text string) {
|
||||||
if w.handler != nil {
|
if w.handler != nil {
|
||||||
w.handler(chat.Incoming{
|
w.handler(chat.Incoming{
|
||||||
UserID: ui.UID,
|
UserID: uid,
|
||||||
Address: ui.Address,
|
Address: addr,
|
||||||
Channel: w,
|
Channel: w,
|
||||||
Msg: chat.Message{Text: text},
|
Msg: chat.Message{Text: text},
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// bindTaskSession связывает свободную вкладку с созданной задачей: когда
|
||||||
|
// /start в свободном табе создал задачу с ChatID == UserID этой вкладки,
|
||||||
|
// вкладка становится task-сессией (спец: каждая задача — своя вкладка).
|
||||||
|
func (w *Window) bindTaskSession(chatID string, taskID int64) {
|
||||||
|
for addr, s := range w.sessions {
|
||||||
|
if string(s.UserID) != chatID || s.BoundTask != 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
s.BoundTask = taskID
|
||||||
|
if panel := w.panels[addr]; panel != nil {
|
||||||
|
panel.SetBoundTask(taskID)
|
||||||
|
}
|
||||||
|
if tab := w.tabForAddress(addr); tab != nil {
|
||||||
|
tab.Text = "Задача #" + ui.Itoa(taskID)
|
||||||
|
w.tabs.Refresh()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// ---- chat.Channel ----
|
// ---- chat.Channel ----
|
||||||
|
|
||||||
// OnMessage регистрирует обработчик входящих (Router).
|
// OnMessage регистрирует обработчик входящих (Router).
|
||||||
@@ -224,6 +329,7 @@ func (w *Window) OnMessage(h chat.Handler) { w.handler = h }
|
|||||||
// Run показывает окно и запускает Fyne-цикл. Блокирует до Quit.
|
// Run показывает окно и запускает Fyne-цикл. Блокирует до Quit.
|
||||||
func (w *Window) Run(ctx context.Context) error {
|
func (w *Window) Run(ctx context.Context) error {
|
||||||
w.win.Show()
|
w.win.Show()
|
||||||
|
maximizeWindow()
|
||||||
go func() {
|
go func() {
|
||||||
<-ctx.Done()
|
<-ctx.Done()
|
||||||
w.fyneApp.Quit()
|
w.fyneApp.Quit()
|
||||||
@@ -232,30 +338,37 @@ func (w *Window) Run(ctx context.Context) error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Send доставляет сообщение Core в диалог окна.
|
// Send доставляет сообщение Core в диалог вкладки по её Address.
|
||||||
func (w *Window) Send(_ context.Context, _ chat.Address, m chat.Message) error {
|
func (w *Window) Send(_ context.Context, to chat.Address, m chat.Message) error {
|
||||||
|
panel, ok := w.panels[to]
|
||||||
|
if !ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
text := m.Text
|
text := m.Text
|
||||||
if len(m.Options) > 0 {
|
if len(m.Options) > 0 {
|
||||||
var sb strings.Builder
|
var sb strings.Builder
|
||||||
sb.WriteString(text)
|
sb.WriteString(text)
|
||||||
for i, opt := range m.Options {
|
for _, opt := range m.Options {
|
||||||
sb.WriteString("\n")
|
sb.WriteString("\n")
|
||||||
sb.WriteString(opt.Label)
|
sb.WriteString(opt.Label)
|
||||||
_ = i
|
|
||||||
}
|
}
|
||||||
text = sb.String()
|
text = sb.String()
|
||||||
}
|
}
|
||||||
fyne.Do(func() { w.chat.Append("🤖 " + text) })
|
fyne.Do(func() { panel.Append("🤖 " + text) })
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Ask — вопрос с вариантами (отображается в диалоге).
|
// Ask — вопрос с вариантами (отображается в диалоге вкладки).
|
||||||
func (w *Window) Ask(_ context.Context, _ chat.Address, m chat.Message) error {
|
func (w *Window) Ask(_ context.Context, to chat.Address, m chat.Message) error {
|
||||||
|
panel, ok := w.panels[to]
|
||||||
|
if !ok {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
text := m.Text
|
text := m.Text
|
||||||
for _, opt := range m.Options {
|
for _, opt := range m.Options {
|
||||||
text += "\n" + opt.Label
|
text += "\n" + opt.Label
|
||||||
}
|
}
|
||||||
fyne.Do(func() { w.chat.Append("❓ " + text) })
|
fyne.Do(func() { panel.Append("❓ " + text) })
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -272,6 +385,8 @@ func (w *Window) Close() error {
|
|||||||
func (w *Window) OnTaskCreated(e events.TaskCreated) {
|
func (w *Window) OnTaskCreated(e events.TaskCreated) {
|
||||||
fyne.Do(func() {
|
fyne.Do(func() {
|
||||||
w.refreshList()
|
w.refreshList()
|
||||||
|
// /start в свободной вкладке: вкладка привязывается к задаче.
|
||||||
|
w.bindTaskSession(e.ChatID, e.ID)
|
||||||
if w.selected == 0 {
|
if w.selected == 0 {
|
||||||
w.selectTask(e.ID)
|
w.selectTask(e.ID)
|
||||||
}
|
}
|
||||||
@@ -308,8 +423,13 @@ func (w *Window) OnTaskStatusChanged(e events.TaskStatusChanged) {
|
|||||||
|
|
||||||
func (w *Window) OnHistoryAppended(e events.HistoryAppended) {
|
func (w *Window) OnHistoryAppended(e events.HistoryAppended) {
|
||||||
fyne.Do(func() {
|
fyne.Do(func() {
|
||||||
if w.selected == e.TaskID {
|
// история приходит в диалог вкладки задачи
|
||||||
w.chat.Append(ui.FormatRole(e.Role) + e.Content)
|
for addr, s := range w.sessions {
|
||||||
|
if s.BoundTask == e.TaskID {
|
||||||
|
if panel, ok := w.panels[addr]; ok {
|
||||||
|
panel.Append(ui.FormatRole(e.Role) + e.Content)
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
@@ -327,5 +447,11 @@ func (w *Window) OnAgentActivity(e events.AgentActivity) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (w *Window) OnLog(e events.LogLine) {
|
func (w *Window) OnLog(e events.LogLine) {
|
||||||
w.appendLog(e.Text)
|
w.appendLog(e.Level, e.Text)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// compile-time проверка: окно реализует chat.Channel и ui.View.
|
||||||
|
var (
|
||||||
|
_ chat.Channel = (*Window)(nil)
|
||||||
|
_ ui.View = (*Window)(nil)
|
||||||
|
)
|
||||||
|
|||||||
@@ -9,15 +9,51 @@
|
|||||||
// Контракты компонентов-виджетов (спец 12.8): интерфейс + Nil-реализация.
|
// Контракты компонентов-виджетов (спец 12.8): интерфейс + Nil-реализация.
|
||||||
package ui
|
package ui
|
||||||
|
|
||||||
// LogPanel — панель «Логи» (спец 12.9): поток системных логов (сырые строки).
|
import "strings"
|
||||||
|
|
||||||
|
// Канонические уровни записей лога для фильтра панели «Логи» (спец 12.9).
|
||||||
|
const (
|
||||||
|
LogLevelError = "error"
|
||||||
|
LogLevelWarning = "warning"
|
||||||
|
LogLevelInfo = "info"
|
||||||
|
LogLevelDebug = "debug"
|
||||||
|
)
|
||||||
|
|
||||||
|
// LogLevels — уровни в порядке их вывода в фильтре виджета «Логи».
|
||||||
|
var LogLevels = []string{LogLevelError, LogLevelWarning, LogLevelInfo, LogLevelDebug}
|
||||||
|
|
||||||
|
// NormalizeLogLevel приводит уровень записи лога к одному из канонических
|
||||||
|
// уровней (LogLevels). Синонимы маппятся на ближайший уровень (err/fatal →
|
||||||
|
// error, warn → warning, trace → debug); пустые и неизвестные значения
|
||||||
|
// ("log" от events.LogWriter и всё прочее) считаются информационными.
|
||||||
|
func NormalizeLogLevel(level string) string {
|
||||||
|
switch strings.ToLower(strings.TrimSpace(level)) {
|
||||||
|
case LogLevelError, "err", "fatal":
|
||||||
|
return LogLevelError
|
||||||
|
case LogLevelWarning, "warn":
|
||||||
|
return LogLevelWarning
|
||||||
|
case LogLevelDebug, "trace":
|
||||||
|
return LogLevelDebug
|
||||||
|
default:
|
||||||
|
return LogLevelInfo
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// LogPanel — панель «Логи» (спец 12.9): поток системных логов с уровнями.
|
||||||
//
|
//
|
||||||
// Содержит строки лога в буфере ограниченного размера; при превышении лимита
|
// Хранит записи (уровень + строка) в буфере ограниченного размера; при
|
||||||
// старые строки отбрасываются. Контейнер виджета отдаётся окну для встраивания
|
// превышении лимита старые записи отбрасываются. Реализация обязана
|
||||||
// в таб «Логи» — реализация (Fyne) сама владеет своим виджетом.
|
// поддерживать фильтр по уровням: выпадающий список с чекбоксами
|
||||||
|
// (error/warning/info/debug), по умолчанию отмечены все уровни; показываются
|
||||||
|
// только записи выбранных уровней, изменение набора сразу обновляет ленту.
|
||||||
|
// Контейнер виджета отдаётся окну для встраивания в таб «Логи» — реализация
|
||||||
|
// (Fyne) сама владеет своим виджетом.
|
||||||
type LogPanel interface {
|
type LogPanel interface {
|
||||||
// Append добавляет строку лога. Реализация обязана перекладывать работу
|
// Append добавляет запись лога: level — уровень из events.LogLine
|
||||||
// на поток Fyne (fyne.Do / fyne.DoAndWait), т.к. вызывается из горутин Hub.
|
// (нормализуется через NormalizeLogLevel), text — строка лога.
|
||||||
Append(text string)
|
// Реализация обязана перекладывать работу на поток Fyne (fyne.Do /
|
||||||
|
// fyne.DoAndWait), т.к. вызывается из горутин Hub.
|
||||||
|
Append(level, text string)
|
||||||
}
|
}
|
||||||
|
|
||||||
// NilLogPanel — no-op реализация LogPanel для headless-режима и тестов.
|
// NilLogPanel — no-op реализация LogPanel для headless-режима и тестов.
|
||||||
@@ -26,4 +62,165 @@ type NilLogPanel struct{}
|
|||||||
// NewNilLogPanel создаёт NilLogPanel.
|
// NewNilLogPanel создаёт NilLogPanel.
|
||||||
func NewNilLogPanel() *NilLogPanel { return &NilLogPanel{} }
|
func NewNilLogPanel() *NilLogPanel { return &NilLogPanel{} }
|
||||||
|
|
||||||
func (NilLogPanel) Append(string) {}
|
func (NilLogPanel) Append(string, string) {}
|
||||||
|
|
||||||
|
// LogEntry — запись буфера панели «Логи»: канонический уровень + строка.
|
||||||
|
type LogEntry struct {
|
||||||
|
Level string
|
||||||
|
Text string
|
||||||
|
}
|
||||||
|
|
||||||
|
// compactThreshold — порог, при котором «мёртвые» (вытесненные сдвигом)
|
||||||
|
// записи физически убираются из буфера пачкой. Вытеснение лениво: оно лишь
|
||||||
|
// инкрементирует start, поэтому не копирует хвост массива на каждую
|
||||||
|
// вытесненную запись (иначе trim был бы O(n²)). Уборка выполняется редко.
|
||||||
|
const compactThreshold = 256
|
||||||
|
|
||||||
|
// LogBuffer — ограниченный буфер записей лога с фильтрацией по уровням
|
||||||
|
// (спец 12.9). Чистая логика без Fyne: используется Fyne-реализацией
|
||||||
|
// LogPanel и тестируется в internal/ui.
|
||||||
|
//
|
||||||
|
// Не синхронизирована: потокобезопасность обеспечивает владелец (все вызовы
|
||||||
|
// на потоке Fyne).
|
||||||
|
type LogBuffer struct {
|
||||||
|
entries []LogEntry
|
||||||
|
start int // индекс первого «живого» элемента (ленивое вытеснение)
|
||||||
|
total int // суммарный объём текста живых записей [start..], байты
|
||||||
|
maxLen int // лимит суммарного объёма текста записей, байты
|
||||||
|
|
||||||
|
// Инкрементальная отрисовка (RenderNew): rendered — сколько живых записей
|
||||||
|
// от начала уже выведено в текст ленты; stale — вытеснение задело уже
|
||||||
|
// выведенные записи, так что текст ленты нельзя дополнить дельтом и нужен
|
||||||
|
// полный пересбор. Эти поля поддерживаются только рендером.
|
||||||
|
rendered int
|
||||||
|
stale bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewLogBuffer создаёт буфер ёмкостью maxLen байт текста записей.
|
||||||
|
func NewLogBuffer(maxLen int) *LogBuffer {
|
||||||
|
return &LogBuffer{maxLen: maxLen}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Append добавляет запись (уровень нормализуется через NormalizeLogLevel,
|
||||||
|
// хвостовые переводы строк срезаются). При превышении лимита старые записи
|
||||||
|
// вытесняются. Сообщает, было ли вытеснение/обрезка — владелец должен
|
||||||
|
// перерисовать ленту, даже если новая запись не видима.
|
||||||
|
func (b *LogBuffer) Append(level, text string) bool {
|
||||||
|
b.entries = append(b.entries, LogEntry{
|
||||||
|
Level: NormalizeLogLevel(level),
|
||||||
|
Text: strings.TrimRight(text, "\r\n"),
|
||||||
|
})
|
||||||
|
b.total += len(b.entries[len(b.entries)-1].Text) + 1
|
||||||
|
return b.trim()
|
||||||
|
}
|
||||||
|
|
||||||
|
// trim удерживает суммарный объём текста живых записей в пределах maxLen,
|
||||||
|
// лениво вытесняя старые (инкремент start); одиночная запись длиннее лимита
|
||||||
|
// обрезается до последних maxLen байт. Сообщает, было ли вытеснение/обрезка.
|
||||||
|
func (b *LogBuffer) trim() bool {
|
||||||
|
dropped := false
|
||||||
|
live := len(b.entries) - b.start
|
||||||
|
evicted := 0
|
||||||
|
for b.total > b.maxLen && live > 1 {
|
||||||
|
b.total -= len(b.entries[b.start].Text) + 1
|
||||||
|
b.start++
|
||||||
|
live--
|
||||||
|
evicted++
|
||||||
|
dropped = true
|
||||||
|
}
|
||||||
|
// Если вытеснена хотя бы одна уже выведенная запись (b.rendered), текст
|
||||||
|
// ленты устарел: дельта не может убрать верхние строки, нужен полный
|
||||||
|
// пересбор. rendered при этом уменьшается на число вытесненных.
|
||||||
|
if evicted > 0 && b.rendered > 0 {
|
||||||
|
b.stale = true
|
||||||
|
}
|
||||||
|
if evicted >= b.rendered {
|
||||||
|
b.rendered = 0
|
||||||
|
} else {
|
||||||
|
b.rendered -= evicted
|
||||||
|
}
|
||||||
|
// Компакция «мёртвых» записей пачкой, а не на каждом append.
|
||||||
|
if b.start >= compactThreshold {
|
||||||
|
b.entries = b.entries[b.start:]
|
||||||
|
b.start = 0
|
||||||
|
}
|
||||||
|
if n := len(b.entries); n > b.start && len(b.entries[n-1].Text) > b.maxLen {
|
||||||
|
// Обрезка последней записи меняет уже выведенный текст, если она
|
||||||
|
// была отрисована.
|
||||||
|
if b.rendered == n-b.start {
|
||||||
|
b.stale = true
|
||||||
|
}
|
||||||
|
b.entries[n-1].Text = b.entries[n-1].Text[len(b.entries[n-1].Text)-b.maxLen:]
|
||||||
|
b.total = b.maxLen + 1
|
||||||
|
dropped = true
|
||||||
|
}
|
||||||
|
return dropped
|
||||||
|
}
|
||||||
|
|
||||||
|
// Entries возвращает копию списка живых записей буфера.
|
||||||
|
func (b *LogBuffer) Entries() []LogEntry {
|
||||||
|
out := make([]LogEntry, len(b.entries)-b.start)
|
||||||
|
copy(out, b.entries[b.start:])
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
// Render строит текст ленты только из живых записей уровней, отмеченных в
|
||||||
|
// selected (уровень → показывать). Пустой или nil набор → пустая лента.
|
||||||
|
// Полный рендер: используется для первичной отрисовки и после смены фильтра.
|
||||||
|
func (b *LogBuffer) Render(selected map[string]bool) string {
|
||||||
|
if selected == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return b.renderFrom(b.start, selected)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ResetRendered помечает все живые записи как уже выведенные. Зовётся после
|
||||||
|
// полного рендера (SetText(Render)), чтобы следующий RenderNew вернул только
|
||||||
|
// новые записи.
|
||||||
|
func (b *LogBuffer) ResetRendered() {
|
||||||
|
b.rendered = len(b.entries) - b.start
|
||||||
|
b.stale = false
|
||||||
|
}
|
||||||
|
|
||||||
|
// RenderNew отдаёт текст ещё не выведенных видимых записей и флаг, требует
|
||||||
|
// ли слой полного пересбора ленты. Если вытеснение задело уже выведенные
|
||||||
|
// записи (stale), возвращаемый текст — полный рендер с нуля (и флаг true), а
|
||||||
|
// не дельта. Иначе — только текст новых записей, которым можно дополнить
|
||||||
|
// текущий текст ленты.
|
||||||
|
func (b *LogBuffer) RenderNew(selected map[string]bool) (text string, full bool) {
|
||||||
|
if selected == nil {
|
||||||
|
return "", false
|
||||||
|
}
|
||||||
|
live := len(b.entries) - b.start
|
||||||
|
if b.stale {
|
||||||
|
b.stale = false
|
||||||
|
b.rendered = live
|
||||||
|
return b.renderFrom(b.start, selected), true
|
||||||
|
}
|
||||||
|
from := b.start + b.rendered
|
||||||
|
var s strings.Builder
|
||||||
|
for i := from; i < len(b.entries); i++ {
|
||||||
|
e := b.entries[i]
|
||||||
|
if !selected[e.Level] {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
s.WriteString(e.Text)
|
||||||
|
s.WriteByte('\n')
|
||||||
|
}
|
||||||
|
b.rendered = live
|
||||||
|
return s.String(), false
|
||||||
|
}
|
||||||
|
|
||||||
|
// renderFrom строит текст записей, начиная с индекса from, с учётом selected.
|
||||||
|
func (b *LogBuffer) renderFrom(from int, selected map[string]bool) string {
|
||||||
|
var s strings.Builder
|
||||||
|
for i := from; i < len(b.entries); i++ {
|
||||||
|
e := b.entries[i]
|
||||||
|
if !selected[e.Level] {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
s.WriteString(e.Text)
|
||||||
|
s.WriteByte('\n')
|
||||||
|
}
|
||||||
|
return s.String()
|
||||||
|
}
|
||||||
|
|||||||
@@ -6,7 +6,315 @@ import "testing"
|
|||||||
// вызывается из горутин Hub и из тестов без Fyne.
|
// вызывается из горутин Hub и из тестов без Fyne.
|
||||||
func TestNilLogPanelAppendNoop(t *testing.T) {
|
func TestNilLogPanelAppendNoop(t *testing.T) {
|
||||||
p := NewNilLogPanel()
|
p := NewNilLogPanel()
|
||||||
p.Append("line 1")
|
p.Append(LogLevelError, "line 1")
|
||||||
p.Append("")
|
p.Append("", "")
|
||||||
p.Append("line 3")
|
p.Append(LogLevelDebug, "line 3")
|
||||||
|
}
|
||||||
|
|
||||||
|
// NormalizeLogLevel приводит любые уровни к каноническим четырём:
|
||||||
|
// синонимы маппятся, неизвестные/пустые значения считаются info.
|
||||||
|
func TestNormalizeLogLevel(t *testing.T) {
|
||||||
|
cases := []struct {
|
||||||
|
in string
|
||||||
|
want string
|
||||||
|
}{
|
||||||
|
{LogLevelError, LogLevelError},
|
||||||
|
{"ERROR", LogLevelError},
|
||||||
|
{" err ", LogLevelError},
|
||||||
|
{"fatal", LogLevelError},
|
||||||
|
{LogLevelWarning, LogLevelWarning},
|
||||||
|
{"WARN", LogLevelWarning},
|
||||||
|
{"Warning", LogLevelWarning},
|
||||||
|
{LogLevelInfo, LogLevelInfo},
|
||||||
|
{"INFO", LogLevelInfo},
|
||||||
|
{LogLevelDebug, LogLevelDebug},
|
||||||
|
{"DEBUG", LogLevelDebug},
|
||||||
|
{"trace", LogLevelDebug},
|
||||||
|
// события LogWriter без явного уровня и прочий мусор → info
|
||||||
|
{"log", LogLevelInfo},
|
||||||
|
{"", LogLevelInfo},
|
||||||
|
{" ", LogLevelInfo},
|
||||||
|
{"unknown", LogLevelInfo},
|
||||||
|
}
|
||||||
|
for _, c := range cases {
|
||||||
|
if got := NormalizeLogLevel(c.in); got != c.want {
|
||||||
|
t.Errorf("NormalizeLogLevel(%q) = %q, want %q", c.in, got, c.want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// LogLevels — ровно четыре уровня фильтра в фиксированном порядке.
|
||||||
|
func TestLogLevelsContract(t *testing.T) {
|
||||||
|
want := []string{LogLevelError, LogLevelWarning, LogLevelInfo, LogLevelDebug}
|
||||||
|
if len(LogLevels) != len(want) {
|
||||||
|
t.Fatalf("len(LogLevels) = %d, want %d", len(LogLevels), len(want))
|
||||||
|
}
|
||||||
|
for i := range want {
|
||||||
|
if LogLevels[i] != want[i] {
|
||||||
|
t.Fatalf("LogLevels = %v, want %v", LogLevels, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// allLevels — набор «отмечены все уровни» (состояние фильтра по умолчанию).
|
||||||
|
func allLevels() map[string]bool {
|
||||||
|
m := make(map[string]bool, len(LogLevels))
|
||||||
|
for _, lvl := range LogLevels {
|
||||||
|
m[lvl] = true
|
||||||
|
}
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
// По умолчанию (все уровни отмечены) рендерится полный лог; уровни
|
||||||
|
// нормализуются при добавлении.
|
||||||
|
func TestLogBufferRenderAllByDefault(t *testing.T) {
|
||||||
|
b := NewLogBuffer(1000)
|
||||||
|
b.Append("error", "e1")
|
||||||
|
b.Append("warn", "w1") // синоним → warning
|
||||||
|
b.Append("log", "i1") // неизвестный уровень → info
|
||||||
|
b.Append("trace", "d1") // синоним → debug
|
||||||
|
|
||||||
|
got := b.Render(allLevels())
|
||||||
|
want := "e1\nw1\ni1\nd1\n"
|
||||||
|
if got != want {
|
||||||
|
t.Fatalf("Render(all) = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
|
||||||
|
entries := b.Entries()
|
||||||
|
wantLvls := []string{LogLevelError, LogLevelWarning, LogLevelInfo, LogLevelDebug}
|
||||||
|
for i, e := range entries {
|
||||||
|
if e.Level != wantLvls[i] {
|
||||||
|
t.Fatalf("entry[%d].Level = %q, want %q", i, e.Level, wantLvls[i])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Показываются только записи выбранных уровней.
|
||||||
|
func TestLogBufferRenderSelectedOnly(t *testing.T) {
|
||||||
|
b := NewLogBuffer(1000)
|
||||||
|
b.Append(LogLevelError, "err line")
|
||||||
|
b.Append(LogLevelWarning, "warn line")
|
||||||
|
b.Append(LogLevelInfo, "info line")
|
||||||
|
b.Append(LogLevelDebug, "debug line")
|
||||||
|
|
||||||
|
onlyErrors := map[string]bool{LogLevelError: true}
|
||||||
|
if got, want := b.Render(onlyErrors), "err line\n"; got != want {
|
||||||
|
t.Fatalf("Render(error only) = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
|
||||||
|
errorsAndDebug := map[string]bool{LogLevelError: true, LogLevelDebug: true}
|
||||||
|
if got, want := b.Render(errorsAndDebug), "err line\ndebug line\n"; got != want {
|
||||||
|
t.Fatalf("Render(error+debug) = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ни один уровень не выбран — лента пуста (и это не ошибка).
|
||||||
|
func TestLogBufferRenderEmptySelection(t *testing.T) {
|
||||||
|
b := NewLogBuffer(1000)
|
||||||
|
b.Append(LogLevelError, "err line")
|
||||||
|
b.Append(LogLevelInfo, "info line")
|
||||||
|
|
||||||
|
if got := b.Render(map[string]bool{}); got != "" {
|
||||||
|
t.Fatalf("Render(none) = %q, want empty", got)
|
||||||
|
}
|
||||||
|
if got := b.Render(nil); got != "" {
|
||||||
|
t.Fatalf("Render(nil) = %q, want empty", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Фильтрация применяется и к уже загруженным записям: смена набора уровней
|
||||||
|
// меняет результат рендера того же буфера без новых Append.
|
||||||
|
func TestLogBufferRefilterLoadedEntries(t *testing.T) {
|
||||||
|
b := NewLogBuffer(1000)
|
||||||
|
b.Append(LogLevelError, "err line")
|
||||||
|
b.Append(LogLevelInfo, "info line")
|
||||||
|
|
||||||
|
all := allLevels()
|
||||||
|
if got, want := b.Render(all), "err line\ninfo line\n"; got != want {
|
||||||
|
t.Fatalf("Render(all) = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
delete(all, LogLevelInfo)
|
||||||
|
if got, want := b.Render(all), "err line\n"; got != want {
|
||||||
|
t.Fatalf("Render(error) after refilter = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
all[LogLevelInfo] = true // вернули галочку — запись снова видима
|
||||||
|
if got, want := b.Render(all), "err line\ninfo line\n"; got != want {
|
||||||
|
t.Fatalf("Render(all) after re-check = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Новые записи сразу попадают в ленту своего уровня.
|
||||||
|
func TestLogBufferNewEntriesVisibleImmediately(t *testing.T) {
|
||||||
|
b := NewLogBuffer(1000)
|
||||||
|
sel := map[string]bool{LogLevelError: true}
|
||||||
|
|
||||||
|
b.Append(LogLevelError, "first")
|
||||||
|
if got, want := b.Render(sel), "first\n"; got != want {
|
||||||
|
t.Fatalf("Render = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
|
||||||
|
b.Append(LogLevelInfo, "noise") // невидимый уровень — ленту не меняет
|
||||||
|
if got, want := b.Render(sel), "first\n"; got != want {
|
||||||
|
t.Fatalf("Render after invisible append = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
|
||||||
|
b.Append(LogLevelError, "second")
|
||||||
|
if got, want := b.Render(sel), "first\nsecond\n"; got != want {
|
||||||
|
t.Fatalf("Render after visible append = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Буфер ограничен: старые записи вытесняются, суммарный объём ≤ лимита.
|
||||||
|
func TestLogBufferBounded(t *testing.T) {
|
||||||
|
b := NewLogBuffer(10)
|
||||||
|
for i := 0; i < 5; i++ {
|
||||||
|
b.Append(LogLevelInfo, "0123456789") // каждая запись — ровно лимит
|
||||||
|
}
|
||||||
|
entries := b.Entries()
|
||||||
|
if len(entries) != 1 {
|
||||||
|
t.Fatalf("len(entries) = %d, want 1 (старые вытеснены)", len(entries))
|
||||||
|
}
|
||||||
|
if entries[0].Text != "0123456789" {
|
||||||
|
t.Fatalf("text = %q", entries[0].Text)
|
||||||
|
}
|
||||||
|
|
||||||
|
// одиночная запись длиннее лимита обрезается до последних maxLen символов
|
||||||
|
b2 := NewLogBuffer(4)
|
||||||
|
b2.Append(LogLevelInfo, "abcdefgh")
|
||||||
|
if got := b2.Entries()[0].Text; got != "efgh" {
|
||||||
|
t.Fatalf("oversized entry text = %q, want %q", got, "efgh")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Буфер держит только живые записи: ленивое вытеснение (start) не даёт
|
||||||
|
// «мёртвым» записям попасть в Entries/Render.
|
||||||
|
func TestLogBufferLiveEntriesOnly(t *testing.T) {
|
||||||
|
b := NewLogBuffer(30)
|
||||||
|
for i := 0; i < 3; i++ {
|
||||||
|
b.Append(LogLevelInfo, "0123456789") // 10 байт + \n каждая
|
||||||
|
}
|
||||||
|
// сумма 33 > 30 → вытесняется первая запись, остаются две живые
|
||||||
|
entries := b.Entries()
|
||||||
|
if len(entries) != 2 {
|
||||||
|
t.Fatalf("len(Entries) = %d, want 2 (мёртвая запись скрыта)", len(entries))
|
||||||
|
}
|
||||||
|
if entries[0].Text != "0123456789" || entries[1].Text != "0123456789" {
|
||||||
|
t.Fatalf("unexpected live entries: %+v", entries)
|
||||||
|
}
|
||||||
|
if got, want := b.Render(allLevels()), "0123456789\n0123456789\n"; got != want {
|
||||||
|
t.Fatalf("Render = %q, want %q", got, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Кап считается по байтам текста, а не по числу записей: короткие записи
|
||||||
|
// не вытесняются, пока их суммарный объём не превысит лимит.
|
||||||
|
func TestLogBufferBytesCap(t *testing.T) {
|
||||||
|
b := NewLogBuffer(100)
|
||||||
|
for i := 0; i < 3; i++ {
|
||||||
|
b.Append(LogLevelInfo, "hello") // 5 байт + \n
|
||||||
|
}
|
||||||
|
if got := len(b.Entries()); got != 3 {
|
||||||
|
t.Fatalf("len(Entries) = %d, want 3 (все помещаются)", got)
|
||||||
|
}
|
||||||
|
|
||||||
|
// одна большая запись вытесняет всё старое (live=1) и обрезается
|
||||||
|
b2 := NewLogBuffer(8)
|
||||||
|
b2.Append(LogLevelInfo, "short")
|
||||||
|
b2.Append(LogLevelInfo, "verylongtext")
|
||||||
|
entries := b2.Entries()
|
||||||
|
if len(entries) != 1 {
|
||||||
|
t.Fatalf("len(Entries) = %d, want 1", len(entries))
|
||||||
|
}
|
||||||
|
if entries[0].Text != "longtext" {
|
||||||
|
t.Fatalf("oversized entry text = %q, want %q", entries[0].Text, "longtext")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Первый RenderNew отдаёт весь видимый текст, последующие — только новые
|
||||||
|
// записи (инкрементальная отрисовка: повторный вызов без Append пуст).
|
||||||
|
func TestLogBufferRenderNewDelta(t *testing.T) {
|
||||||
|
b := NewLogBuffer(1000)
|
||||||
|
sel := allLevels()
|
||||||
|
|
||||||
|
b.Append(LogLevelError, "e1")
|
||||||
|
b.Append(LogLevelInfo, "i1")
|
||||||
|
text, full := b.RenderNew(sel)
|
||||||
|
if full {
|
||||||
|
t.Fatalf("первый RenderNew не должен требовать полного пересбора, got full")
|
||||||
|
}
|
||||||
|
if text != "e1\ni1\n" {
|
||||||
|
t.Fatalf("первый RenderNew = %q, want %q", text, "e1\ni1\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Без новых записей — пустой дельта.
|
||||||
|
text, full = b.RenderNew(sel)
|
||||||
|
if full || text != "" {
|
||||||
|
t.Fatalf("повторный RenderNew без новых записей = (%q, %v), want (\"\", false)", text, full)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Новые записи — только их текст, без повторной отдачи старых.
|
||||||
|
b.Append(LogLevelInfo, "i2")
|
||||||
|
text, _ = b.RenderNew(sel)
|
||||||
|
if text != "i2\n" {
|
||||||
|
t.Fatalf("дельта RenderNew = %q, want %q", text, "i2\n")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// RenderNew применяет фильтр: невидимые уровни не попадают в дельту.
|
||||||
|
func TestLogBufferRenderNewFiltered(t *testing.T) {
|
||||||
|
b := NewLogBuffer(1000)
|
||||||
|
sel := map[string]bool{LogLevelError: true}
|
||||||
|
b.Append(LogLevelError, "e1")
|
||||||
|
b.Append(LogLevelInfo, "i1")
|
||||||
|
b.Append(LogLevelError, "e2")
|
||||||
|
|
||||||
|
text, _ := b.RenderNew(sel)
|
||||||
|
if text != "e1\ne2\n" {
|
||||||
|
t.Fatalf("RenderNew(error only) = %q, want %q", text, "e1\ne2\n")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// RenderNew уважает ResetRendered: после полного рендера новые записи
|
||||||
|
// добавляются дельтом, а не вытесняют уже выведенный текст.
|
||||||
|
func TestLogBufferRenderNewAfterReset(t *testing.T) {
|
||||||
|
b := NewLogBuffer(1000)
|
||||||
|
sel := allLevels()
|
||||||
|
b.Append(LogLevelInfo, "old")
|
||||||
|
b.Append(LogLevelInfo, "base")
|
||||||
|
|
||||||
|
text, full := b.RenderNew(sel)
|
||||||
|
if full || text != "old\nbase\n" {
|
||||||
|
t.Fatalf("RenderNew = (%q, %v), want (\"old\\nbase\\n\", false)", text, full)
|
||||||
|
}
|
||||||
|
b.ResetRendered() // имитация полной пересборки ленты
|
||||||
|
|
||||||
|
b.Append(LogLevelInfo, "new")
|
||||||
|
text, full = b.RenderNew(sel)
|
||||||
|
if full || text != "new\n" {
|
||||||
|
t.Fatalf("RenderNew после Reset = (%q, %v), want (\"new\\n\", false)", text, full)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Вытеснение уже выведенных записей помечает рендер как требующий полного
|
||||||
|
// пересбора (stale): дельта не может убрать верхние строки.
|
||||||
|
func TestLogBufferRenderNewEvictionStale(t *testing.T) {
|
||||||
|
b := NewLogBuffer(25) // записи по 10 байт + \n: помещаются 2, 3-я вытесняет 1-ю
|
||||||
|
sel := allLevels()
|
||||||
|
|
||||||
|
b.Append(LogLevelInfo, "0123456789")
|
||||||
|
b.Append(LogLevelInfo, "0123456789")
|
||||||
|
text, full := b.RenderNew(sel)
|
||||||
|
if full || text != "0123456789\n0123456789\n" {
|
||||||
|
t.Fatalf("RenderNew = (%q, %v), want two lines", text, full)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Третья запись вытесняет первую (уже отрисованную) — stale.
|
||||||
|
b.Append(LogLevelInfo, "0123456789")
|
||||||
|
text, full = b.RenderNew(sel)
|
||||||
|
if !full {
|
||||||
|
t.Fatalf("RenderNew после вытеснения = (full=%v), want full=true", full)
|
||||||
|
}
|
||||||
|
if text != "0123456789\n0123456789\n" {
|
||||||
|
t.Fatalf("полный текст после вытеснения = %q, want последние две записи", text)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
43
internal/ui/session.go
Normal file
43
internal/ui/session.go
Normal file
@@ -0,0 +1,43 @@
|
|||||||
|
package ui
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/kamelion/ratatoskr-go/internal/chat"
|
||||||
|
"github.com/kamelion/ratatoskr-go/internal/storage"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Session — одна вкладка-сессия UI: независимый диалог с LLM.
|
||||||
|
//
|
||||||
|
// Каждая вкладка имеет собственный UserID и Address, поэтому core видит её как
|
||||||
|
// отдельный чат со своим активным task (политика «одна задача на чат» работает
|
||||||
|
// из коробки). Свободная сессия (BoundTask==0) — in-memory диалог с моделью
|
||||||
|
// вне машины задач; task-сессия (BoundTask>0) — диалог конкретной задачи, у
|
||||||
|
// которой UserID == task.ChatID (чат задачи уникален).
|
||||||
|
type Session struct {
|
||||||
|
UserID chat.UserID
|
||||||
|
Address chat.Address
|
||||||
|
Title string
|
||||||
|
BoundTask int64 // 0 = свободный чат; >0 — задача, к которой привязана вкладка
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewFreeSession создаёт свободную сессию (без задачи). userID/address должны
|
||||||
|
// быть уникальными (например "ui://sess-<n>"). Транскрипт живёт только в памяти.
|
||||||
|
func NewFreeSession(userID chat.UserID, address chat.Address) *Session {
|
||||||
|
return &Session{UserID: userID, Address: address, Title: "Новый чат", BoundTask: 0}
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewTaskSession создаёт сессию, привязанную к задаче. UserID == task.ChatID,
|
||||||
|
// чтобы Router/core находили активную задачу этой вкладки. address — уникальный
|
||||||
|
// адрес доставки вкладки (не должен совпадать с UserID других вкладок).
|
||||||
|
func NewTaskSession(address chat.Address, t storage.Task) *Session {
|
||||||
|
title := t.Title
|
||||||
|
if title == "" {
|
||||||
|
title = "Задача #" + Itoa(t.ID)
|
||||||
|
}
|
||||||
|
return &Session{UserID: chat.UserID(t.ChatID), Address: address, Title: title, BoundTask: t.ID}
|
||||||
|
}
|
||||||
|
|
||||||
|
// SessionID — префикс идентификаторов UI-сессий (и UserID, и Address).
|
||||||
|
const SessionID = "ui://sess-"
|
||||||
|
|
||||||
|
// IsFreeSession — true, если вкладка не привязана к задаче.
|
||||||
|
func IsFreeSession(s *Session) bool { return s != nil && s.BoundTask == 0 }
|
||||||
55
internal/ui/session_test.go
Normal file
55
internal/ui/session_test.go
Normal file
@@ -0,0 +1,55 @@
|
|||||||
|
package ui
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/kamelion/ratatoskr-go/internal/chat"
|
||||||
|
"github.com/kamelion/ratatoskr-go/internal/storage"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNewFreeSession(t *testing.T) {
|
||||||
|
s := NewFreeSession(chat.UserID("ui://sess-1"), chat.Address("ui://sess-1"))
|
||||||
|
if s.UserID != chat.UserID("ui://sess-1") {
|
||||||
|
t.Errorf("UserID = %q, want ui://sess-1", s.UserID)
|
||||||
|
}
|
||||||
|
if s.Address != chat.Address("ui://sess-1") {
|
||||||
|
t.Errorf("Address = %q, want ui://sess-1", s.Address)
|
||||||
|
}
|
||||||
|
if s.BoundTask != 0 {
|
||||||
|
t.Errorf("BoundTask = %d, want 0", s.BoundTask)
|
||||||
|
}
|
||||||
|
if !IsFreeSession(s) {
|
||||||
|
t.Error("free session must be free")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNewTaskSessionUsesTaskChatID(t *testing.T) {
|
||||||
|
task := &storage.Task{ID: 7, ChatID: "ui://sess-3", Title: "Фича"}
|
||||||
|
s := NewTaskSession(chat.Address("ui://sess-3"), *task)
|
||||||
|
if s.UserID != chat.UserID("ui://sess-3") {
|
||||||
|
t.Errorf("UserID = %q, want task.ChatID", s.UserID)
|
||||||
|
}
|
||||||
|
if s.BoundTask != 7 {
|
||||||
|
t.Errorf("BoundTask = %d, want 7", s.BoundTask)
|
||||||
|
}
|
||||||
|
if s.Title != "Фича" {
|
||||||
|
t.Errorf("Title = %q, want Фича", s.Title)
|
||||||
|
}
|
||||||
|
if IsFreeSession(s) {
|
||||||
|
t.Error("task session must not be free")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestNewTaskSessionTitleFallback(t *testing.T) {
|
||||||
|
task := &storage.Task{ID: 3, ChatID: "ui://sess-3"}
|
||||||
|
s := NewTaskSession(chat.Address("ui://sess-3"), *task)
|
||||||
|
if s.Title != "Задача #3" {
|
||||||
|
t.Errorf("Title = %q, want Задача #3", s.Title)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestIsFreeSessionNil(t *testing.T) {
|
||||||
|
if IsFreeSession(nil) {
|
||||||
|
t.Error("nil is not free")
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -31,7 +31,9 @@ import (
|
|||||||
"os/exec"
|
"os/exec"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"runtime"
|
"runtime"
|
||||||
|
"sort"
|
||||||
"strings"
|
"strings"
|
||||||
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -50,6 +52,11 @@ type Updater struct {
|
|||||||
// Dir — каталог рядом с бинарём (для .new/.old и companion-метаданных).
|
// Dir — каталог рядом с бинарём (для .new/.old и companion-метаданных).
|
||||||
// Ставится app из os.Executable(); если пуст — используется каталог Workdir.
|
// Ставится app из os.Executable(); если пуст — используется каталог Workdir.
|
||||||
Dir string
|
Dir string
|
||||||
|
|
||||||
|
// client — общий HTTP-клиент (keep-alive), чтобы проверки/скачивания
|
||||||
|
// переиспользовали соединения, а не создавали новое на каждый запрос.
|
||||||
|
client *http.Client
|
||||||
|
clientMu sync.Mutex
|
||||||
}
|
}
|
||||||
|
|
||||||
// Result — результат Check.
|
// Result — результат Check.
|
||||||
@@ -118,16 +125,26 @@ func (u *Updater) versionsURL() string {
|
|||||||
return base + "/api/v1/packages/" + url.PathEscape(u.Owner) + "/generic/" + url.PathEscape(u.Package)
|
return base + "/api/v1/packages/" + url.PathEscape(u.Owner) + "/generic/" + url.PathEscape(u.Package)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// httpClient возвращает общий клиент (keep-alive), инициализируя лениво.
|
||||||
|
func (u *Updater) httpClient() *http.Client {
|
||||||
|
u.clientMu.Lock()
|
||||||
|
defer u.clientMu.Unlock()
|
||||||
|
if u.client == nil {
|
||||||
|
u.client = &http.Client{Timeout: 30 * time.Second}
|
||||||
|
}
|
||||||
|
return u.client
|
||||||
|
}
|
||||||
|
|
||||||
// httpGet скачивает файл по URL бэкенда. При Token непустом — Basic/токен-заголовок.
|
// httpGet скачивает файл по URL бэкенда. При Token непустом — Basic/токен-заголовок.
|
||||||
func (u *Updater) httpGet(url string) ([]byte, error) {
|
func (u *Updater) httpGet(ctx context.Context, url string) ([]byte, error) {
|
||||||
req, err := http.NewRequest(http.MethodGet, url, nil)
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
if u.Token != "" {
|
if u.Token != "" {
|
||||||
req.Header.Set("Authorization", "token "+u.Token)
|
req.Header.Set("Authorization", "token "+u.Token)
|
||||||
}
|
}
|
||||||
resp, err := (&http.Client{Timeout: 30 * time.Second}).Do(req)
|
resp, err := u.httpClient().Do(req)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -138,6 +155,61 @@ func (u *Updater) httpGet(url string) ([]byte, error) {
|
|||||||
return io.ReadAll(resp.Body)
|
return io.ReadAll(resp.Body)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// fileExists проверяет наличие файла по URL без скачивания тела: HEAD,
|
||||||
|
// а при 405/501 (сервер не поддерживает HEAD) — fallback на GET с Range байт 0-0.
|
||||||
|
// Возвращает (false, nil) при 404/410 — файла нет.
|
||||||
|
func (u *Updater) fileExists(ctx context.Context, url string) (bool, error) {
|
||||||
|
req, err := http.NewRequestWithContext(ctx, http.MethodHead, url, nil)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
if u.Token != "" {
|
||||||
|
req.Header.Set("Authorization", "token "+u.Token)
|
||||||
|
}
|
||||||
|
resp, err := u.httpClient().Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
resp.Body.Close()
|
||||||
|
switch {
|
||||||
|
case resp.StatusCode == http.StatusNotFound || resp.StatusCode == http.StatusGone:
|
||||||
|
return false, nil
|
||||||
|
case resp.StatusCode == http.StatusOK || resp.StatusCode == http.StatusPartialContent:
|
||||||
|
return true, nil
|
||||||
|
case resp.StatusCode == http.StatusMethodNotAllowed || resp.StatusCode == http.StatusNotImplemented:
|
||||||
|
// Gitea может не отвечать на HEAD — проверяем GET с Range 0-0 без чтения тела.
|
||||||
|
return u.fileExistsByRange(ctx, url)
|
||||||
|
default:
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// fileExistsByRange проверяет наличие файла GET'ом с Range: bytes=0-0.
|
||||||
|
// Тело не читается: достаточно лишь первых байт заголовков ответа.
|
||||||
|
func (u *Updater) fileExistsByRange(ctx context.Context, url string) (bool, error) {
|
||||||
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
if u.Token != "" {
|
||||||
|
req.Header.Set("Authorization", "token "+u.Token)
|
||||||
|
}
|
||||||
|
req.Header.Set("Range", "bytes=0-0")
|
||||||
|
resp, err := u.httpClient().Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
resp.Body.Close()
|
||||||
|
switch {
|
||||||
|
case resp.StatusCode == http.StatusNotFound || resp.StatusCode == http.StatusGone:
|
||||||
|
return false, nil
|
||||||
|
case resp.StatusCode == http.StatusOK || resp.StatusCode == http.StatusPartialContent:
|
||||||
|
return true, nil
|
||||||
|
default:
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Check определяет, есть ли более свежая версия в Gitea Packages.
|
// Check определяет, есть ли более свежая версия в Gitea Packages.
|
||||||
// Не скачивает бинарь. Ошибка (U1) возвращается в Result.Err — вызывающий
|
// Не скачивает бинарь. Ошибка (U1) возвращается в Result.Err — вызывающий
|
||||||
// решает, логировать и пропустить.
|
// решает, логировать и пропустить.
|
||||||
@@ -167,8 +239,12 @@ type pkgVersion struct {
|
|||||||
// ResolveLatest определяет идентификатор новейшей применимой версии пакета.
|
// ResolveLatest определяет идентификатор новейшей применимой версии пакета.
|
||||||
// Бинарь/метаданные читаем из КОНКРЕТНОЙ версии, а не из pseudo-`latest`,
|
// Бинарь/метаданные читаем из КОНКРЕТНОЙ версии, а не из pseudo-`latest`,
|
||||||
// чтобы companion-файлы и бинарь всегда брались из одного снимка.
|
// чтобы companion-файлы и бинарь всегда брались из одного снимка.
|
||||||
|
//
|
||||||
|
// Применимость версии проверяем НАЛИЧИЕМ бинаря платформы (HEAD без тела),
|
||||||
|
// а не скачиванием полного файла: при N версиях это N запросов заголовков
|
||||||
|
// вместо N×(размер бинаря) байт.
|
||||||
func (u *Updater) ResolveLatest(ctx context.Context) (string, error) {
|
func (u *Updater) ResolveLatest(ctx context.Context) (string, error) {
|
||||||
b, err := u.httpGet(u.versionsURL())
|
b, err := u.httpGet(ctx, u.versionsURL())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
@@ -176,30 +252,28 @@ func (u *Updater) ResolveLatest(ctx context.Context) (string, error) {
|
|||||||
if err := json.Unmarshal(b, &vers); err != nil {
|
if err := json.Unmarshal(b, &vers); err != nil {
|
||||||
return "", ue(U1, "list "+u.versionsURL(), err)
|
return "", ue(U1, "list "+u.versionsURL(), err)
|
||||||
}
|
}
|
||||||
|
// новые версии — с большим ID; идём с новейшей и берём первую с бинарём.
|
||||||
|
sort.SliceStable(vers, func(i, j int) bool { return vers[i].ID > vers[j].ID })
|
||||||
want := PlatformFilename()
|
want := PlatformFilename()
|
||||||
// выбираем самую свежую версию (макс. id) класса commit-*, в которой есть бинарь.
|
|
||||||
best := ""
|
|
||||||
var bestID int64
|
|
||||||
for _, v := range vers {
|
for _, v := range vers {
|
||||||
if !strings.HasPrefix(v.Ver, "commit-") {
|
if !strings.HasPrefix(v.Ver, "commit-") {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
// версия применима, только если в ней опубликован бинарь нашей платформы
|
exists, ferr := u.fileExists(ctx, u.fileURL(v.Ver, want))
|
||||||
if _, err := u.httpGet(u.fileURL(v.Ver, want)); err != nil {
|
if ferr != nil {
|
||||||
continue
|
continue // сетевые ошибки пробы не роняют проверку
|
||||||
}
|
}
|
||||||
if v.ID > bestID {
|
if exists {
|
||||||
bestID = v.ID
|
return v.Ver, nil
|
||||||
best = v.Ver
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return best, nil
|
return "", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Download скачивает бинарь конкретной версии во временный файл и возвращает
|
// Download скачивает бинарь конкретной версии во временный файл и возвращает
|
||||||
// путь к нему. Файл: <Dir>/.ratatoskr.<ver>.new.
|
// путь к нему. Файл: <Dir>/.ratatoskr.<ver>.new.
|
||||||
func (u *Updater) Download(ctx context.Context, version string) (string, error) {
|
func (u *Updater) Download(ctx context.Context, version string) (string, error) {
|
||||||
b, err := u.httpGet(u.fileURL(version, PlatformFilename()))
|
b, err := u.httpGet(ctx, u.fileURL(version, PlatformFilename()))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err // уже U3
|
return "", err // уже U3
|
||||||
}
|
}
|
||||||
@@ -216,7 +290,7 @@ func (u *Updater) Download(ctx context.Context, version string) (string, error)
|
|||||||
|
|
||||||
// versionSum256 читает companion-файл контрольной суммы конкретной версии.
|
// versionSum256 читает companion-файл контрольной суммы конкретной версии.
|
||||||
func (u *Updater) versionSum256(ctx context.Context, version string) (string, error) {
|
func (u *Updater) versionSum256(ctx context.Context, version string) (string, error) {
|
||||||
b, err := u.httpGet(u.fileURL(version, PlatformFilename()+".sha256"))
|
b, err := u.httpGet(ctx, u.fileURL(version, PlatformFilename()+".sha256"))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", err
|
return "", err
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ import (
|
|||||||
"os"
|
"os"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
"sync/atomic"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -41,7 +42,7 @@ func mockGitea(t *testing.T, bin []byte, version, checksum string) *httptest.Ser
|
|||||||
_ = json.NewEncoder(w).Encode(versions)
|
_ = json.NewEncoder(w).Encode(versions)
|
||||||
})
|
})
|
||||||
mux.HandleFunc("/api/packages/", func(w http.ResponseWriter, r *http.Request) {
|
mux.HandleFunc("/api/packages/", func(w http.ResponseWriter, r *http.Request) {
|
||||||
if r.Method != http.MethodGet {
|
if r.Method != http.MethodGet && r.Method != http.MethodHead {
|
||||||
http.Error(w, "method", http.StatusMethodNotAllowed)
|
http.Error(w, "method", http.StatusMethodNotAllowed)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -58,7 +59,10 @@ func mockGitea(t *testing.T, bin []byte, version, checksum string) *httptest.Ser
|
|||||||
http.NotFound(w, r)
|
http.NotFound(w, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
if r.Method == http.MethodGet {
|
||||||
_, _ = w.Write(body)
|
_, _ = w.Write(body)
|
||||||
|
}
|
||||||
|
// HEAD — просто 200, тело не пишем
|
||||||
})
|
})
|
||||||
return httptest.NewServer(mux)
|
return httptest.NewServer(mux)
|
||||||
}
|
}
|
||||||
@@ -133,6 +137,104 @@ func TestCheck_ServerDown(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestResolveLatest_HeadNotFullDownload проверяет, что ResolveLatest проверяет
|
||||||
|
// наличие бинаря HEAD'ом и НЕ скачивает полное тело бинаря (прошлая версия
|
||||||
|
// читала каждый файл целиком — O(N)×размер бинаря).
|
||||||
|
func TestResolveLatest_HeadNotFullDownload(t *testing.T) {
|
||||||
|
name := PlatformFilename()
|
||||||
|
_ = name
|
||||||
|
var headReqs, bodyReqs int64
|
||||||
|
mux := http.NewServeMux()
|
||||||
|
mux.HandleFunc("/api/v1/packages/", func(w http.ResponseWriter, _ *http.Request) {
|
||||||
|
_ = json.NewEncoder(w).Encode([]map[string]any{
|
||||||
|
{"id": 1, "version": "commit-aaa1111"},
|
||||||
|
{"id": 2, "version": "commit-abc1234"},
|
||||||
|
})
|
||||||
|
})
|
||||||
|
mux.HandleFunc("/api/packages/", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
parts := strings.Split(strings.Trim(r.URL.Path, "/"), "/")
|
||||||
|
if len(parts) < 6 {
|
||||||
|
http.NotFound(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ver := parts[len(parts)-2]
|
||||||
|
fileName := parts[len(parts)-1]
|
||||||
|
if ver != "commit-abc1234" || fileName != name {
|
||||||
|
http.NotFound(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
switch r.Method {
|
||||||
|
case http.MethodHead:
|
||||||
|
atomic.AddInt64(&headReqs, 1)
|
||||||
|
case http.MethodGet:
|
||||||
|
atomic.AddInt64(&bodyReqs, 1)
|
||||||
|
default:
|
||||||
|
http.Error(w, "method", http.StatusMethodNotAllowed)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
srv := httptest.NewServer(mux)
|
||||||
|
defer srv.Close()
|
||||||
|
|
||||||
|
u := &Updater{BaseURL: srv.URL, Owner: "k", Package: "p", CurrentVersion: "v1", Dir: t.TempDir()}
|
||||||
|
ver, err := u.ResolveLatest(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ResolveLatest err = %v", err)
|
||||||
|
}
|
||||||
|
if ver != "commit-abc1234" {
|
||||||
|
t.Errorf("ResolveLatest = %q, want commit-abc1234", ver)
|
||||||
|
}
|
||||||
|
if atomic.LoadInt64(&headReqs) == 0 {
|
||||||
|
t.Error("ResolveLatest не делал HEAD-проб на файлы")
|
||||||
|
}
|
||||||
|
if atomic.LoadInt64(&bodyReqs) != 0 {
|
||||||
|
t.Errorf("ResolveLatest скачал тело бинарника: %d полных GET", atomic.LoadInt64(&bodyReqs))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestResolveLatest_SkipsBinarylessVersion проверяет фоллбэк: новейшая версия
|
||||||
|
// без бинаря (разные матрицы платформ публикуются не все сразу) пропускается,
|
||||||
|
// берётся следующая, где файл есть.
|
||||||
|
func TestResolveLatest_SkipsBinarylessVersion(t *testing.T) {
|
||||||
|
name := PlatformFilename()
|
||||||
|
bin := []byte("binary")
|
||||||
|
mux := http.NewServeMux()
|
||||||
|
mux.HandleFunc("/api/v1/packages/", func(w http.ResponseWriter, _ *http.Request) {
|
||||||
|
_ = json.NewEncoder(w).Encode([]map[string]any{
|
||||||
|
{"id": 1, "version": "commit-aaa1111"},
|
||||||
|
{"id": 2, "version": "commit-mid2222"},
|
||||||
|
{"id": 3, "version": "commit-new3333"},
|
||||||
|
})
|
||||||
|
})
|
||||||
|
mux.HandleFunc("/api/packages/", func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
parts := strings.Split(strings.Trim(r.URL.Path, "/"), "/")
|
||||||
|
if len(parts) < 6 {
|
||||||
|
http.NotFound(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
ver := parts[len(parts)-2]
|
||||||
|
fileName := parts[len(parts)-1]
|
||||||
|
// бинарь есть только у commit-mid2222 — новейшие 1 и 3 пропускаются
|
||||||
|
if ver != "commit-mid2222" || fileName != name {
|
||||||
|
http.NotFound(w, r)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if r.Method != http.MethodHead {
|
||||||
|
_, _ = w.Write(bin)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
srv := httptest.NewServer(mux)
|
||||||
|
defer srv.Close()
|
||||||
|
|
||||||
|
u := &Updater{BaseURL: srv.URL, Owner: "k", Package: "p", CurrentVersion: "v0", Dir: t.TempDir()}
|
||||||
|
ver, err := u.ResolveLatest(context.Background())
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ResolveLatest err = %v", err)
|
||||||
|
}
|
||||||
|
if ver != "commit-mid2222" {
|
||||||
|
t.Errorf("ResolveLatest = %q, want commit-mid2222 (фоллбэк от версии без бинаря)", ver)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestDownload_And_Verify_Good(t *testing.T) {
|
func TestDownload_And_Verify_Good(t *testing.T) {
|
||||||
bin := []byte("ratatoskr-binary-content-v2")
|
bin := []byte("ratatoskr-binary-content-v2")
|
||||||
u, dir := testUpdater(t, bin, "commit-new12345", "")
|
u, dir := testUpdater(t, bin, "commit-new12345", "")
|
||||||
|
|||||||
206
internal/worker/postmortem.go
Normal file
206
internal/worker/postmortem.go
Normal file
@@ -0,0 +1,206 @@
|
|||||||
|
package worker
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"strings"
|
||||||
|
"text/template"
|
||||||
|
|
||||||
|
"github.com/kamelion/ratatoskr-go/internal/events"
|
||||||
|
"github.com/kamelion/ratatoskr-go/internal/storage"
|
||||||
|
)
|
||||||
|
|
||||||
|
// postMortemAgent — имя постмортем-агента (файл agents/postmortem.md).
|
||||||
|
const postMortemAgent = "postmortem"
|
||||||
|
|
||||||
|
// traceOutputMax — обрезка вывода сессии в постмортем-промпте (чтобы промпт
|
||||||
|
// не превращался в полные транскрипты и не переполнял контекст модели).
|
||||||
|
const traceOutputMax = 6000
|
||||||
|
|
||||||
|
// postMortemPromptTemplate — промпт для постмортем-агента после failed/timeout:
|
||||||
|
// задача + сессии dev/reviewer. Ожидается резюме простым текстом на русском.
|
||||||
|
var postMortemPromptTemplate = template.Must(template.New("postmortem").Parse(`Ты — постмортем-аналитик в конвейере Ratatoskr. Задача завершилась неудачей ({{.Status}}). Проанализируй сессии агентов dev/reviewer и дай резюме: почему так случилось и что сделать, чтобы не повторялось.
|
||||||
|
|
||||||
|
**Задача:**
|
||||||
|
{{if .Title}}Название: {{.Title}}{{end}}
|
||||||
|
{{if .Goal}}Цель: {{.Goal}}{{end}}
|
||||||
|
{{if .Repos}}
|
||||||
|
Репозитории:
|
||||||
|
{{- range .Repos}}
|
||||||
|
- {{.}}
|
||||||
|
{{- end}}
|
||||||
|
{{end}}
|
||||||
|
{{if .Why}}Зачем: {{.Why}}{{end}}
|
||||||
|
{{if .AC}}Критерии готовности (AC):
|
||||||
|
{{.AC}}{{end}}
|
||||||
|
|
||||||
|
**Итоговый статус задачи:** {{.Status}}
|
||||||
|
|
||||||
|
**Сессии субагентов:**
|
||||||
|
{{.Sessions}}
|
||||||
|
|
||||||
|
Ответь ПРОСТЫМ ТЕКСТОМ на русском, без JSON и разметки. Формат:
|
||||||
|
|
||||||
|
Почему так случилось:
|
||||||
|
- <причина 1>
|
||||||
|
- <причина 2>
|
||||||
|
|
||||||
|
Что сделать, чтобы это не повторялось:
|
||||||
|
- <рекомендация 1>
|
||||||
|
- <рекомендация 2>`))
|
||||||
|
|
||||||
|
// PostMortemPromptData — данные для рендера постмортем-промпта.
|
||||||
|
type PostMortemPromptData struct {
|
||||||
|
Title string
|
||||||
|
Goal string
|
||||||
|
Repos []string
|
||||||
|
Why string
|
||||||
|
AC string
|
||||||
|
Status storage.Status
|
||||||
|
Sessions string
|
||||||
|
}
|
||||||
|
|
||||||
|
// RenderPostMortemPrompt собирает промпт для постмортем-агента.
|
||||||
|
func RenderPostMortemPrompt(data PostMortemPromptData) (string, error) {
|
||||||
|
var buf strings.Builder
|
||||||
|
if err := postMortemPromptTemplate.Execute(&buf, data); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return buf.String(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// postMortemsText форматирует сессии dev/reviewer в секцию промпта.
|
||||||
|
func postMortemsText(traces []storage.Trace) string {
|
||||||
|
var b strings.Builder
|
||||||
|
for _, tr := range traces {
|
||||||
|
b.WriteString("\n=== Агент: " + tr.Agent + " (статус " + string(tr.Status) + ") ===\n")
|
||||||
|
if tr.SessionID != "" {
|
||||||
|
b.WriteString("session_id: " + tr.SessionID + "\n")
|
||||||
|
}
|
||||||
|
if strings.TrimSpace(tr.Prompt) != "" {
|
||||||
|
b.WriteString("-- Промпт агента --\n")
|
||||||
|
b.WriteString(tr.Prompt)
|
||||||
|
b.WriteString("\n")
|
||||||
|
}
|
||||||
|
if strings.TrimSpace(tr.Output) != "" {
|
||||||
|
b.WriteString("-- Вывод агента --\n")
|
||||||
|
b.WriteString(truncateTrace(tr.Output, traceOutputMax))
|
||||||
|
b.WriteString("\n")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if b.Len() == 0 {
|
||||||
|
return "(сессии dev/reviewer не найдены — вероятна инфраструктурная ошибка до запуска агентов)"
|
||||||
|
}
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// truncateTrace обрезает длинный текст до последних n символов (релевантен
|
||||||
|
// хвост: вердикт/ошибка агента в конце вывода).
|
||||||
|
func truncateTrace(s string, n int) string {
|
||||||
|
if len(s) <= n {
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
if n <= 0 {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return "(вывод обрезан)\n" + s[len(s)-n:]
|
||||||
|
}
|
||||||
|
|
||||||
|
// hasPostMortemTrace возвращает true, если у задачи уже есть постмортем-trace
|
||||||
|
// (защита от повторного запуска при повторных прогонах/retry).
|
||||||
|
func (w *Worker) hasPostMortemTrace(ctx context.Context, taskID int64) bool {
|
||||||
|
if w.Store == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, err := w.Store.GetLatestTrace(ctx, taskID, postMortemAgent)
|
||||||
|
return err == nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// postMortem запускает постмортем-агент для задачи, завершившейся failed/timeout:
|
||||||
|
// собирает сессии dev/reviewer из трасс, даёт агенту анализ, сохраняет результат
|
||||||
|
// как trace agent=postmortem и уведомляет владельца задачи резюме.
|
||||||
|
//
|
||||||
|
// Статус задачи НЕ меняется (failed/timeout остаётся достигнутым); собственные
|
||||||
|
// сбои постмортема не влияют на исход задачи — только логируются.
|
||||||
|
func (w *Worker) postMortem(ctx context.Context, task *storage.Task) {
|
||||||
|
if w.Store == nil || w.Runner == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if w.hasPostMortemTrace(ctx, task.ID) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
traces, err := w.Store.GetTraces(ctx, task.ID)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("worker: task %d: постмортем: трассы: %v", task.ID, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
// Анализируем только сессии агентов конвейера (dev/reviewer).
|
||||||
|
var sessions []storage.Trace
|
||||||
|
for _, tr := range traces {
|
||||||
|
if tr.Agent == "dev" || tr.Agent == "reviewer" {
|
||||||
|
sessions = append(sessions, *tr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
prompt, pErr := RenderPostMortemPrompt(PostMortemPromptData{
|
||||||
|
Title: task.Title,
|
||||||
|
Goal: task.Goal,
|
||||||
|
Repos: task.EffectiveRepos(),
|
||||||
|
Why: task.Why,
|
||||||
|
AC: task.AC,
|
||||||
|
Status: task.Status,
|
||||||
|
Sessions: postMortemsText(sessions),
|
||||||
|
})
|
||||||
|
if pErr != nil {
|
||||||
|
log.Printf("worker: task %d: постмортем: рендер промпта: %v", task.ID, pErr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// лог-событие для UI-панели «Состояние».
|
||||||
|
w.publish(events.AgentActivity{TaskID: task.ID, Agent: postMortemAgent, Stage: "postmortem"})
|
||||||
|
|
||||||
|
tr := &storage.Trace{TaskID: task.ID, Agent: postMortemAgent, Prompt: prompt}
|
||||||
|
traceID, aErr := w.Store.AppendTrace(ctx, tr)
|
||||||
|
if aErr != nil {
|
||||||
|
log.Printf("worker: task %d: постмортем: create trace: %v", task.ID, aErr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
res, rErr := w.Runner.Run(w.runCtx(ctx, task.ID), prompt, w.Worktree, postMortemAgent, "")
|
||||||
|
if rErr != nil {
|
||||||
|
log.Printf("worker: task %d: постмортем: запуск: %v", task.ID, rErr)
|
||||||
|
w.finalizeTrace(ctx, traceID, storage.TraceFailed, rErr.Error())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if res.SessionID != "" {
|
||||||
|
_ = w.Store.UpdateTraceSessionID(ctx, traceID, res.SessionID)
|
||||||
|
}
|
||||||
|
|
||||||
|
output := strings.TrimSpace(res.Stdout)
|
||||||
|
status := storage.TraceSuccess
|
||||||
|
if res.RC != 0 || output == "" {
|
||||||
|
status = storage.TraceFailed
|
||||||
|
if output == "" {
|
||||||
|
output = "(постмортем-агент не вернул текст)"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
log.Printf("worker: task %d: постмортем: %s", task.ID, truncateTrace(output, 2000))
|
||||||
|
w.finalizeTrace(ctx, traceID, status, output)
|
||||||
|
|
||||||
|
if res.RC == 0 && output != "" {
|
||||||
|
text := fmt.Sprintf("Задача #%d: 🔍 анализ (после %s)\n%s", task.ID, task.Status, output)
|
||||||
|
w.notify(ctx, task, text)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// postMortemAfter — defer-хук из runTask: запускает постмортем, если задача
|
||||||
|
// завершилась failed/timeout. Собственные ошибки постмортема не мешают
|
||||||
|
// исходному результату задачи (возвращаемый *error только читается).
|
||||||
|
func (w *Worker) postMortemAfter(ctx context.Context, task *storage.Task, _ *error) {
|
||||||
|
if task.Status != storage.StatusFailed && task.Status != storage.StatusTimeout {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
w.postMortem(ctx, task)
|
||||||
|
}
|
||||||
@@ -4,6 +4,7 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"log"
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
@@ -96,6 +97,10 @@ func (w *Worker) runReviewer(ctx context.Context, taskID int64, cwd, prompt stri
|
|||||||
}
|
}
|
||||||
w.finalizeTrace(ctx, traceID, ts, output)
|
w.finalizeTrace(ctx, traceID, ts, output)
|
||||||
|
|
||||||
|
if v != nil {
|
||||||
|
log.Printf("reviewer: вердикт: %s", formatReviewVerdict(v))
|
||||||
|
}
|
||||||
|
|
||||||
if v == nil {
|
if v == nil {
|
||||||
// Невалидный/пустой вывод или RC!=0: не парсится — сигнал на retry.
|
// Невалидный/пустой вывод или RC!=0: не парсится — сигнал на retry.
|
||||||
return nil, output, traceID, nil
|
return nil, output, traceID, nil
|
||||||
@@ -119,3 +124,35 @@ func parseReviewVerdict(out string) (*reviewVerdict, error) {
|
|||||||
}
|
}
|
||||||
return &v, nil
|
return &v, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// formatReviewVerdict собирает человекочитаемое однострочное описание вердикта
|
||||||
|
// ревьюера (без JSON-разметки) для панели «Логи».
|
||||||
|
func formatReviewVerdict(v *reviewVerdict) string {
|
||||||
|
var b strings.Builder
|
||||||
|
b.WriteString("passed=")
|
||||||
|
b.WriteString(boolElse(v.Passed, "yes", "no"))
|
||||||
|
if len(v.CriticalIssues) > 0 {
|
||||||
|
b.WriteString(", critical=[")
|
||||||
|
b.WriteString(strings.Join(v.CriticalIssues, " | "))
|
||||||
|
b.WriteString("]")
|
||||||
|
}
|
||||||
|
if len(v.SolidViolations) > 0 {
|
||||||
|
b.WriteString(", solid=[")
|
||||||
|
b.WriteString(strings.Join(v.SolidViolations, " | "))
|
||||||
|
b.WriteString("]")
|
||||||
|
}
|
||||||
|
if len(v.Comments) > 0 {
|
||||||
|
b.WriteString(", comments=[")
|
||||||
|
b.WriteString(strings.Join(v.Comments, " | "))
|
||||||
|
b.WriteString("]")
|
||||||
|
}
|
||||||
|
return b.String()
|
||||||
|
}
|
||||||
|
|
||||||
|
// boolElse возвращает строку для логического значения (без вызова тернарника).
|
||||||
|
func boolElse(cond bool, yes, no string) string {
|
||||||
|
if cond {
|
||||||
|
return yes
|
||||||
|
}
|
||||||
|
return no
|
||||||
|
}
|
||||||
|
|||||||
@@ -196,6 +196,10 @@ func (w *Worker) pollAndDispatch(ctx context.Context) error {
|
|||||||
// dev-агент реализует, reviewer строго проверяет весь дифф ветки; при не-проходе
|
// dev-агент реализует, reviewer строго проверяет весь дифф ветки; при не-проходе
|
||||||
// dev дорабатывает по комментариям; прошло → push ветки + success.
|
// dev дорабатывает по комментариям; прошло → push ветки + success.
|
||||||
func (w *Worker) runTask(ctx context.Context, task *storage.Task) (err error) {
|
func (w *Worker) runTask(ctx context.Context, task *storage.Task) (err error) {
|
||||||
|
// 0. постмортем-анализ: если задача завершилась failed/timeout — после
|
||||||
|
// выхода из runTask (всех путей) запускаем анализ сессий dev/reviewer.
|
||||||
|
defer w.postMortemAfter(ctx, task, &err)
|
||||||
|
|
||||||
// 1. проверяем статус
|
// 1. проверяем статус
|
||||||
if task.Status != storage.StatusApproved {
|
if task.Status != storage.StatusApproved {
|
||||||
return fmt.Errorf("%w: task %d status=%q", ErrLaunch, task.ID, task.Status)
|
return fmt.Errorf("%w: task %d status=%q", ErrLaunch, task.ID, task.Status)
|
||||||
@@ -277,6 +281,8 @@ func (w *Worker) runTask(ctx context.Context, task *storage.Task) (err error) {
|
|||||||
|
|
||||||
output := res.Stdout
|
output := res.Stdout
|
||||||
|
|
||||||
|
log.Printf("worker: task %d: dev вердикт (хвост): %s", task.ID, truncateTrace(output, 2000))
|
||||||
|
|
||||||
// 5b. dev не завершился успешно (RC!=0) → фиксируем без ревью.
|
// 5b. dev не завершился успешно (RC!=0) → фиксируем без ревью.
|
||||||
switch res.RC {
|
switch res.RC {
|
||||||
case 0:
|
case 0:
|
||||||
|
|||||||
@@ -54,6 +54,11 @@ type mockRunnerWorker struct {
|
|||||||
// (последний вечный). Позволяет смоделировать fail-then-pass.
|
// (последний вечный). Позволяет смоделировать fail-then-pass.
|
||||||
reviewSequence []*opencode.Result
|
reviewSequence []*opencode.Result
|
||||||
reviewCount int
|
reviewCount int
|
||||||
|
|
||||||
|
// postMortemResult — результат постмортем-агента; nil → RC:0 с тестовым
|
||||||
|
// резюме. postMortemCount — сколько раз постмортем вызывался.
|
||||||
|
postMortemResult *opencode.Result
|
||||||
|
postMortemCount int
|
||||||
}
|
}
|
||||||
|
|
||||||
func (m *mockRunnerWorker) Run(_ context.Context, _, _, agent, _ string) (*opencode.Result, error) {
|
func (m *mockRunnerWorker) Run(_ context.Context, _, _, agent, _ string) (*opencode.Result, error) {
|
||||||
@@ -71,6 +76,13 @@ func (m *mockRunnerWorker) Run(_ context.Context, _, _, agent, _ string) (*openc
|
|||||||
}
|
}
|
||||||
return &opencode.Result{RC: 0, Stdout: `{"passed":true,"comments":[]}`}, nil
|
return &opencode.Result{RC: 0, Stdout: `{"passed":true,"comments":[]}`}, nil
|
||||||
}
|
}
|
||||||
|
if agent == postMortemAgent {
|
||||||
|
m.postMortemCount++
|
||||||
|
if m.postMortemResult != nil {
|
||||||
|
return m.postMortemResult, m.err
|
||||||
|
}
|
||||||
|
return &opencode.Result{RC: 0, Stdout: "Почему так случилось:\n- тестовое резюме\nЧто сделать:\n- исправить"}, nil
|
||||||
|
}
|
||||||
return m.result, m.err
|
return m.result, m.err
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -402,12 +414,17 @@ func TestWorkerIterationsLimitNotification(t *testing.T) {
|
|||||||
if len(texts) == 0 {
|
if len(texts) == 0 {
|
||||||
t.Fatal("нет уведомлений")
|
t.Fatal("нет уведомлений")
|
||||||
}
|
}
|
||||||
last := texts[len(texts)-1]
|
// где-то в списке есть уведомление о failed с упоминанием лимита итераций
|
||||||
if !strings.Contains(last, "failed") {
|
// (над ним — уведомление postmortem, поэтому последним оно не обязано быть).
|
||||||
t.Errorf("последнее уведомление = %q, want упоминание failed", last)
|
var limitNotif string
|
||||||
|
for _, txt := range texts {
|
||||||
|
if strings.Contains(txt, "failed") && strings.Contains(txt, "итераци") {
|
||||||
|
limitNotif = txt
|
||||||
|
break
|
||||||
}
|
}
|
||||||
if !strings.Contains(last, "итераци") {
|
}
|
||||||
t.Errorf("последнее уведомление = %q, want упоминание лимита итераций", last)
|
if limitNotif == "" {
|
||||||
|
t.Errorf("нет уведомления о лимите итераций (failed+итераци): %#v", texts)
|
||||||
}
|
}
|
||||||
// ровно одно уведомление о failed (running→failed не задваивается)
|
// ровно одно уведомление о failed (running→failed не задваивается)
|
||||||
var failedCount int
|
var failedCount int
|
||||||
@@ -419,9 +436,20 @@ func TestWorkerIterationsLimitNotification(t *testing.T) {
|
|||||||
if failedCount != 1 {
|
if failedCount != 1 {
|
||||||
t.Errorf("уведомлений о failed = %d, want ровно 1: %#v", failedCount, texts)
|
t.Errorf("уведомлений о failed = %d, want ровно 1: %#v", failedCount, texts)
|
||||||
}
|
}
|
||||||
|
// есть постмортем-уведомление с анализом
|
||||||
|
var pmCount int
|
||||||
|
for _, txt := range texts {
|
||||||
|
if strings.Contains(txt, "анализ (после failed)") {
|
||||||
|
pmCount++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if pmCount != 1 {
|
||||||
|
t.Errorf("постмортем-уведомлений = %d, want ровно 1: %#v", pmCount, texts)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// TestWorkerTimeoutNotification — RC=-1 (таймаут dev) → уведомление о timeout.
|
// TestWorkerTimeoutNotification — RC=-1 (таймаут dev) → уведомление о timeout
|
||||||
|
// и постмортем-уведомление с анализом.
|
||||||
func TestWorkerTimeoutNotification(t *testing.T) {
|
func TestWorkerTimeoutNotification(t *testing.T) {
|
||||||
s := setupWorkerDB(t)
|
s := setupWorkerDB(t)
|
||||||
task := createReadyTask(t, s, "notif-timeout")
|
task := createReadyTask(t, s, "notif-timeout")
|
||||||
@@ -439,13 +467,21 @@ func TestWorkerTimeoutNotification(t *testing.T) {
|
|||||||
_ = w.runTask(ctx, task)
|
_ = w.runTask(ctx, task)
|
||||||
|
|
||||||
prefix := "Задача #" + strconv.FormatInt(task.ID, 10)
|
prefix := "Задача #" + strconv.FormatInt(task.ID, 10)
|
||||||
|
texts := notifTexts(n)
|
||||||
|
if len(texts) != 3 {
|
||||||
|
t.Fatalf("уведомлений = %d, want 3: %#v", len(texts), texts)
|
||||||
|
}
|
||||||
want := []string{prefix + ": running", prefix + ": timeout"}
|
want := []string{prefix + ": running", prefix + ": timeout"}
|
||||||
if got := notifTexts(n); !reflect.DeepEqual(got, want) {
|
if !reflect.DeepEqual(texts[:2], want) {
|
||||||
t.Errorf("уведомления = %#v, want %#v", got, want)
|
t.Errorf("первые уведомления = %#v, want %#v", texts[:2], want)
|
||||||
|
}
|
||||||
|
if !strings.Contains(texts[2], "🔍 анализ (после timeout)") {
|
||||||
|
t.Errorf("постмортем-уведомление = %q, want упоминание «анализ (после timeout)»", texts[2])
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// TestWorkerSpawnErrorNotification — сбой запуска dev → уведомление о failed.
|
// TestWorkerSpawnErrorNotification — сбой запуска dev → уведомление о failed
|
||||||
|
// и постмортем-уведомление.
|
||||||
func TestWorkerSpawnErrorNotification(t *testing.T) {
|
func TestWorkerSpawnErrorNotification(t *testing.T) {
|
||||||
s := setupWorkerDB(t)
|
s := setupWorkerDB(t)
|
||||||
task := createReadyTask(t, s, "notif-spawn")
|
task := createReadyTask(t, s, "notif-spawn")
|
||||||
@@ -463,9 +499,16 @@ func TestWorkerSpawnErrorNotification(t *testing.T) {
|
|||||||
_ = w.runTask(ctx, task)
|
_ = w.runTask(ctx, task)
|
||||||
|
|
||||||
prefix := "Задача #" + strconv.FormatInt(task.ID, 10)
|
prefix := "Задача #" + strconv.FormatInt(task.ID, 10)
|
||||||
|
texts := notifTexts(n)
|
||||||
|
if len(texts) != 3 {
|
||||||
|
t.Fatalf("уведомлений = %d, want 3: %#v", len(texts), texts)
|
||||||
|
}
|
||||||
want := []string{prefix + ": running", prefix + ": failed"}
|
want := []string{prefix + ": running", prefix + ": failed"}
|
||||||
if got := notifTexts(n); !reflect.DeepEqual(got, want) {
|
if !reflect.DeepEqual(texts[:2], want) {
|
||||||
t.Errorf("уведомления = %#v, want %#v", got, want)
|
t.Errorf("первые уведомления = %#v, want %#v", texts[:2], want)
|
||||||
|
}
|
||||||
|
if !strings.Contains(texts[2], "анализ (после failed)") {
|
||||||
|
t.Errorf("постмортем-уведомление = %q, want упоминание «анализ (после failed)»", texts[2])
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -529,9 +572,10 @@ func TestWorkerTimeout(t *testing.T) {
|
|||||||
s := setupWorkerDB(t)
|
s := setupWorkerDB(t)
|
||||||
task := createReadyTask(t, s, "slow")
|
task := createReadyTask(t, s, "slow")
|
||||||
|
|
||||||
|
runner := &mockRunnerWorker{result: &opencode.Result{RC: -1, Stdout: ""}}
|
||||||
w := &Worker{
|
w := &Worker{
|
||||||
Store: s,
|
Store: s,
|
||||||
Runner: &mockRunnerWorker{result: &opencode.Result{RC: -1, Stdout: ""}},
|
Runner: runner,
|
||||||
Worktree: t.TempDir(),
|
Worktree: t.TempDir(),
|
||||||
}
|
}
|
||||||
seedFakeRepo(t, w.Worktree, "slow")
|
seedFakeRepo(t, w.Worktree, "slow")
|
||||||
@@ -551,11 +595,20 @@ func TestWorkerTimeout(t *testing.T) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("get traces: %v", err)
|
t.Fatalf("get traces: %v", err)
|
||||||
}
|
}
|
||||||
if len(traces) != 1 {
|
if len(traces) != 2 {
|
||||||
t.Fatalf("got %d traces, want 1", len(traces))
|
t.Fatalf("got %d traces, want 2 (dev + postmortem)", len(traces))
|
||||||
}
|
}
|
||||||
if traces[0].Status != storage.TraceTimeout {
|
if traces[0].Status != storage.TraceTimeout {
|
||||||
t.Errorf("trace status = %q, want timeout", traces[0].Status)
|
t.Errorf("trace[0] status = %q, want timeout", traces[0].Status)
|
||||||
|
}
|
||||||
|
if traces[1].Agent != postMortemAgent {
|
||||||
|
t.Errorf("trace[1] agent = %q, want postmortem", traces[1].Agent)
|
||||||
|
}
|
||||||
|
if traces[1].Status != storage.TraceSuccess {
|
||||||
|
t.Errorf("trace[1] status = %q, want success", traces[1].Status)
|
||||||
|
}
|
||||||
|
if runner.postMortemCount != 1 {
|
||||||
|
t.Errorf("postmortem запускался %d раз, want 1", runner.postMortemCount)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -563,9 +616,10 @@ func TestWorkerSpawnError(t *testing.T) {
|
|||||||
s := setupWorkerDB(t)
|
s := setupWorkerDB(t)
|
||||||
task := createReadyTask(t, s, "spawn-fail")
|
task := createReadyTask(t, s, "spawn-fail")
|
||||||
|
|
||||||
|
runner := &mockRunnerWorker{err: errors.New("opencode not found")}
|
||||||
w := &Worker{
|
w := &Worker{
|
||||||
Store: s,
|
Store: s,
|
||||||
Runner: &mockRunnerWorker{err: errors.New("opencode not found")},
|
Runner: runner,
|
||||||
Worktree: t.TempDir(),
|
Worktree: t.TempDir(),
|
||||||
}
|
}
|
||||||
seedFakeRepo(t, w.Worktree, "spawn-fail")
|
seedFakeRepo(t, w.Worktree, "spawn-fail")
|
||||||
@@ -585,11 +639,17 @@ func TestWorkerSpawnError(t *testing.T) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("get traces: %v", err)
|
t.Fatalf("get traces: %v", err)
|
||||||
}
|
}
|
||||||
if len(traces) != 1 {
|
if len(traces) != 2 {
|
||||||
t.Fatalf("got %d traces, want 1", len(traces))
|
t.Fatalf("got %d traces, want 2 (dev-failed + postmortem)", len(traces))
|
||||||
}
|
}
|
||||||
if traces[0].Status != storage.TraceFailed {
|
if traces[0].Status != storage.TraceFailed {
|
||||||
t.Errorf("trace status = %q, want failed", traces[0].Status)
|
t.Errorf("trace[0] status = %q, want failed", traces[0].Status)
|
||||||
|
}
|
||||||
|
if traces[1].Agent != postMortemAgent {
|
||||||
|
t.Errorf("trace[1] agent = %q, want postmortem", traces[1].Agent)
|
||||||
|
}
|
||||||
|
if runner.postMortemCount != 1 {
|
||||||
|
t.Errorf("postmortem запускался %d раз, want 1", runner.postMortemCount)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -619,8 +679,8 @@ func TestWorkerNonZeroExit(t *testing.T) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("get traces: %v", err)
|
t.Fatalf("get traces: %v", err)
|
||||||
}
|
}
|
||||||
if len(traces) != 1 {
|
if len(traces) != 2 {
|
||||||
t.Fatalf("got %d traces, want 1", len(traces))
|
t.Fatalf("got %d traces, want 2 (dev + postmortem)", len(traces))
|
||||||
}
|
}
|
||||||
if traces[0].Status != storage.TraceFailed {
|
if traces[0].Status != storage.TraceFailed {
|
||||||
t.Errorf("trace status = %q, want failed", traces[0].Status)
|
t.Errorf("trace status = %q, want failed", traces[0].Status)
|
||||||
@@ -628,6 +688,9 @@ func TestWorkerNonZeroExit(t *testing.T) {
|
|||||||
if traces[0].Output != "error" {
|
if traces[0].Output != "error" {
|
||||||
t.Errorf("output = %q, want error", traces[0].Output)
|
t.Errorf("output = %q, want error", traces[0].Output)
|
||||||
}
|
}
|
||||||
|
if traces[1].Agent != postMortemAgent {
|
||||||
|
t.Errorf("trace[1] agent = %q, want postmortem", traces[1].Agent)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestWorkerBadStatus(t *testing.T) {
|
func TestWorkerBadStatus(t *testing.T) {
|
||||||
@@ -882,3 +945,191 @@ func TestWorkerSemaphore(t *testing.T) {
|
|||||||
t.Errorf("после освобождения слота success = %d, want 2", len(success))
|
t.Errorf("после освобождения слота success = %d, want 2", len(success))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TestWorkerPostMortemSkippedOnSuccess — успешный трейд НЕ запускает постмортем:
|
||||||
|
// причина анализа — только failed/timeout.
|
||||||
|
func TestWorkerPostMortemSkippedOnSuccess(t *testing.T) {
|
||||||
|
s := setupWorkerDB(t)
|
||||||
|
task := createReadyTask(t, s, "pm-ok")
|
||||||
|
|
||||||
|
runner := &mockRunnerWorker{result: &opencode.Result{RC: 0, Stdout: "done", SessionID: "s"}}
|
||||||
|
w := &Worker{
|
||||||
|
Store: s,
|
||||||
|
Runner: runner,
|
||||||
|
Worktree: t.TempDir(),
|
||||||
|
}
|
||||||
|
seedFakeRepo(t, w.Worktree, "pm-ok")
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
if err := w.runTask(ctx, task); err != nil {
|
||||||
|
t.Fatalf("runTask: %v", err)
|
||||||
|
}
|
||||||
|
task, _ = s.GetTask(ctx, task.ID)
|
||||||
|
if task.Status != storage.StatusSuccess {
|
||||||
|
t.Fatalf("status = %q, want success", task.Status)
|
||||||
|
}
|
||||||
|
if runner.postMortemCount != 0 {
|
||||||
|
t.Errorf("postmortem запускался %d раз, want 0 при success", runner.postMortemCount)
|
||||||
|
}
|
||||||
|
traces, _ := s.GetTraces(ctx, task.ID)
|
||||||
|
for _, tr := range traces {
|
||||||
|
if tr.Agent == postMortemAgent {
|
||||||
|
t.Errorf("есть неожиданный postmortem-trace при success")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestWorkerPostMortemFailureDoesNotChangeStatus — сбой самого постмортема не
|
||||||
|
// влияет на статус задачи (остаётся failed) и фиксируется как failed-трасса.
|
||||||
|
func TestWorkerPostMortemFailureDoesNotChangeStatus(t *testing.T) {
|
||||||
|
s := setupWorkerDB(t)
|
||||||
|
task := createReadyTask(t, s, "pm-fail")
|
||||||
|
|
||||||
|
runner := &mockRunnerWorker{
|
||||||
|
// dev падает при спавне → failed; постмортем тоже падает.
|
||||||
|
err: errors.New("opencode not found"),
|
||||||
|
postMortemResult: &opencode.Result{RC: 1, Stdout: ""},
|
||||||
|
}
|
||||||
|
w := &Worker{
|
||||||
|
Store: s,
|
||||||
|
Runner: runner,
|
||||||
|
Worktree: t.TempDir(),
|
||||||
|
}
|
||||||
|
seedFakeRepo(t, w.Worktree, "pm-fail")
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
_ = w.runTask(ctx, task)
|
||||||
|
|
||||||
|
task, _ = s.GetTask(ctx, task.ID)
|
||||||
|
if task.Status != storage.StatusFailed {
|
||||||
|
t.Errorf("status = %q, want failed (постмортем не должен менять статус)", task.Status)
|
||||||
|
}
|
||||||
|
traces, err := s.GetTraces(ctx, task.ID)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("get traces: %v", err)
|
||||||
|
}
|
||||||
|
if len(traces) != 2 {
|
||||||
|
t.Fatalf("traces = %d, want 2 (dev-failed + postmortem)", len(traces))
|
||||||
|
}
|
||||||
|
if traces[1].Agent != postMortemAgent {
|
||||||
|
t.Errorf("trace[1] agent = %q, want postmortem", traces[1].Agent)
|
||||||
|
}
|
||||||
|
if traces[1].Status != storage.TraceFailed {
|
||||||
|
t.Errorf("trace[1] status = %q, want failed (сбой постмортема)", traces[1].Status)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestWorkerPostMortemNotRepeated — если у задачи уже есть postmortem-trace
|
||||||
|
// (например, от прошлого прогона), повторный постмортем не запускается.
|
||||||
|
func TestWorkerPostMortemNotRepeated(t *testing.T) {
|
||||||
|
s := setupWorkerDB(t)
|
||||||
|
task := createReadyTask(t, s, "pm-repeat")
|
||||||
|
|
||||||
|
ctx := context.Background()
|
||||||
|
if _, err := s.AppendTrace(ctx, &storage.Trace{
|
||||||
|
TaskID: task.ID,
|
||||||
|
Agent: postMortemAgent,
|
||||||
|
Prompt: "старый анализ",
|
||||||
|
Output: "старое резюме",
|
||||||
|
}); err != nil {
|
||||||
|
t.Fatalf("seed postmortem trace: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
runner := &mockRunnerWorker{err: errors.New("opencode not found")}
|
||||||
|
w := &Worker{
|
||||||
|
Store: s,
|
||||||
|
Runner: runner,
|
||||||
|
Worktree: t.TempDir(),
|
||||||
|
}
|
||||||
|
seedFakeRepo(t, w.Worktree, "pm-repeat")
|
||||||
|
|
||||||
|
_ = w.runTask(ctx, task)
|
||||||
|
|
||||||
|
task, _ = s.GetTask(ctx, task.ID)
|
||||||
|
if task.Status != storage.StatusFailed {
|
||||||
|
t.Fatalf("status = %q, want failed", task.Status)
|
||||||
|
}
|
||||||
|
if runner.postMortemCount != 0 {
|
||||||
|
t.Errorf("postmortem запускался %d раз, want 0 (уже был trace)", runner.postMortemCount)
|
||||||
|
}
|
||||||
|
// количество postmortem-трасс не выросло
|
||||||
|
traces, _ := s.GetTraces(ctx, task.ID)
|
||||||
|
var pm int
|
||||||
|
for _, tr := range traces {
|
||||||
|
if tr.Agent == postMortemAgent {
|
||||||
|
pm++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if pm != 1 {
|
||||||
|
t.Errorf("postmortem-трасс = %d, want 1 (без дубля)", pm)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestRenderPostMortemPrompt — промпт постмортема включает задачу и сессии.
|
||||||
|
func TestRenderPostMortemPrompt(t *testing.T) {
|
||||||
|
tr := storage.Trace{
|
||||||
|
Agent: "dev",
|
||||||
|
Status: storage.TraceTimeout,
|
||||||
|
Prompt: "промпт dev",
|
||||||
|
Output: "вывод dev",
|
||||||
|
}
|
||||||
|
prompt, err := RenderPostMortemPrompt(PostMortemPromptData{
|
||||||
|
Title: "Таймаут-задача",
|
||||||
|
Goal: "сделать",
|
||||||
|
Repos: []string{"calc"},
|
||||||
|
AC: "работает",
|
||||||
|
Status: storage.StatusTimeout,
|
||||||
|
Sessions: postMortemsText([]storage.Trace{tr}),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("render: %v", err)
|
||||||
|
}
|
||||||
|
for _, want := range []string{"Таймаут-задача", "timeout", "=== Агент: dev", "промпт dev", "вывод dev"} {
|
||||||
|
if !strings.Contains(prompt, want) {
|
||||||
|
t.Errorf("промпт не содержит %q", want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestFormatReviewVerdict — человекочитаемое описание вердикта ревьюера.
|
||||||
|
func TestFormatReviewVerdict(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
v *reviewVerdict
|
||||||
|
want []string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "passed with everything",
|
||||||
|
v: &reviewVerdict{
|
||||||
|
Passed: true,
|
||||||
|
CriticalIssues: []string{"bug A"},
|
||||||
|
SolidViolations: []string{"v1", "v2"},
|
||||||
|
Comments: []string{"comment"},
|
||||||
|
},
|
||||||
|
want: []string{"passed=yes", "critical=[bug A]", "solid=[v1 | v2]", "comments=[comment]"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "failed with critical only",
|
||||||
|
v: &reviewVerdict{
|
||||||
|
Passed: false,
|
||||||
|
CriticalIssues: []string{"bug A", "bug B"},
|
||||||
|
},
|
||||||
|
want: []string{"passed=no", "critical=[bug A | bug B]"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "empty verdict",
|
||||||
|
v: &reviewVerdict{},
|
||||||
|
want: []string{"passed=no"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
got := formatReviewVerdict(tt.v)
|
||||||
|
for _, w := range tt.want {
|
||||||
|
if !strings.Contains(got, w) {
|
||||||
|
t.Errorf("formatReviewVerdict = %q, want contain %q", got, w)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -9,12 +9,21 @@ CI собирает ТОЛЬКО headless-linux (см. .gitea/workflows/ci.yaml)
|
|||||||
|
|
||||||
Требования:
|
Требования:
|
||||||
- Go с доступным C-компилятором (CGO_ENABLED=1). WinLibs gcc: см. память
|
- Go с доступным C-компилятором (CGO_ENABLED=1). WinLibs gcc: см. память
|
||||||
`mem:toolchain/cgo-winlibs-gcc` — bin нужно добавить в PATH перед запуском.
|
`mem:binator/cgo-winlibs-gcc`. Каталог с gcc.exe можно задать через
|
||||||
|
GCC_PATH (ниже); если не задан — gcc должен быть уже в PATH.
|
||||||
- Файл scripts/.gitea-creds (git-ignored, см. ниже) с токенами.
|
- Файл scripts/.gitea-creds (git-ignored, см. ниже) с токенами.
|
||||||
Формат (по одной `ключ=значение` на строку):
|
Формат (по одной `ключ=значение` на строку):
|
||||||
GITEA_TOKEN=write:packages-токен
|
GITEA_TOKEN=write:packages-токен
|
||||||
UPDATE_TOKEN=read:package-токен
|
UPDATE_TOKEN=read:package-токен
|
||||||
GIT_MAIN_URL=http://gitea.hal9000.home
|
GIT_MAIN_URL=http://gitea.hal9000.home
|
||||||
|
PROJECT_PATH=..\..\foo (опционально) корень проекта (где лежат cmd/
|
||||||
|
и go.mod); относительно папки scripts/ или
|
||||||
|
абсолютный; без него — папка рядом с
|
||||||
|
скриптом
|
||||||
|
GCC_PATH=D:\MinGW\bin (опционально) каталог, где лежит gcc.exe;
|
||||||
|
дописывается в PATH перед проверкой и
|
||||||
|
сборкой; абсолютный или относительно
|
||||||
|
scripts/; без него — gcc из текущего PATH
|
||||||
- Ветка уже запушена в `main` (VERSION = commit-<sha7> от текущего HEAD).
|
- Ветка уже запушена в `main` (VERSION = commit-<sha7> от текущего HEAD).
|
||||||
|
|
||||||
Пример:
|
Пример:
|
||||||
@@ -24,22 +33,13 @@ CI собирает ТОЛЬКО headless-linux (см. .gitea/workflows/ci.yaml)
|
|||||||
$ErrorActionPreference = "Stop"
|
$ErrorActionPreference = "Stop"
|
||||||
|
|
||||||
$ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
|
$ScriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
|
||||||
$RepoRoot = Split-Path -Parent $ScriptDir
|
|
||||||
$CredsFile = Join-Path $ScriptDir ".gitea-creds"
|
$CredsFile = Join-Path $ScriptDir ".gitea-creds"
|
||||||
$Package = "ratatoskr"
|
$Package = "ratatoskr"
|
||||||
$Owner = "kamelion"
|
$Owner = "kamelion"
|
||||||
$Filename = "ratatoskr-windows-amd64.exe"
|
$Filename = "ratatoskr-windows-amd64.exe"
|
||||||
|
|
||||||
Set-Location $RepoRoot
|
# --- 0. Проверка флага CGO_ENABLED (не зависит от creds) ---
|
||||||
|
|
||||||
# --- 0. Проверка C-тулчейна ---
|
|
||||||
if ($env:CGO_ENABLED -eq "0") { Write-Error "CGO_ENABLED=0 — нужен C-компилятор (MinGW). Уберите его из env." }
|
if ($env:CGO_ENABLED -eq "0") { Write-Error "CGO_ENABLED=0 — нужен C-компилятор (MinGW). Уберите его из env." }
|
||||||
if (-not (Get-Command gcc -ErrorAction SilentlyContinue)) {
|
|
||||||
Write-Host "gcc не найден в PATH. Пример (WinLibs):"
|
|
||||||
Write-Host ' $p = "<...>\mingw64\bin"; $env:Path = "$p;$env:Path"'
|
|
||||||
Write-Error "C-компилятор gcc не найден."
|
|
||||||
}
|
|
||||||
Write-Host "[ok] gcc: $((gcc --version | Select-Object -First 1))"
|
|
||||||
|
|
||||||
# --- 1. Чтение секретов ---
|
# --- 1. Чтение секретов ---
|
||||||
if (-not (Test-Path $CredsFile)) {
|
if (-not (Test-Path $CredsFile)) {
|
||||||
@@ -59,6 +59,42 @@ $giteaUrl = $creds["GIT_MAIN_URL"]
|
|||||||
if (-not $giteaToken) { Write-Error "В $CredsFile нет GITEA_TOKEN (нужен write:packages)." }
|
if (-not $giteaToken) { Write-Error "В $CredsFile нет GITEA_TOKEN (нужен write:packages)." }
|
||||||
if (-not $giteaUrl) { Write-Error "В $CredsFile нет GIT_MAIN_URL." }
|
if (-not $giteaUrl) { Write-Error "В $CredsFile нет GIT_MAIN_URL." }
|
||||||
|
|
||||||
|
# --- 1a. C-тулчейн (gcc): GCC_PATH из creds дописывается в PATH ---
|
||||||
|
# Путь абсолютный либо относительно папки scripts/; без GCC_PATH — gcc
|
||||||
|
# берётся из уже установленного PATH.
|
||||||
|
$gccPath = $creds["GCC_PATH"]
|
||||||
|
if ($gccPath -ne "" -and $gccPath) {
|
||||||
|
$gccAbs = Join-Path $ScriptDir $gccPath
|
||||||
|
$gccBin = (Resolve-Path -LiteralPath $gccAbs -ErrorAction SilentlyContinue).Path
|
||||||
|
if (-not $gccBin) {
|
||||||
|
$gccAbs = $gccPath # не нашлось относительно scripts/ — пробуем как абсолютный
|
||||||
|
$gccBin = (Resolve-Path -LiteralPath $gccAbs -ErrorAction SilentlyContinue).Path
|
||||||
|
}
|
||||||
|
if (-not $gccBin) { Write-Error "GCC_PATH не найден: $gccPath (или относительно $ScriptDir)" }
|
||||||
|
$env:Path = "$gccBin;$env:Path"
|
||||||
|
Write-Host "[gcc] PATH += $gccBin"
|
||||||
|
}
|
||||||
|
if (-not (Get-Command gcc -ErrorAction SilentlyContinue)) {
|
||||||
|
Write-Host "gcc не найден в PATH. Пример (WinLibs):"
|
||||||
|
Write-Host ' D:\MinGW\bin, затем:
|
||||||
|
GCC_PATH=D:\MinGW\bin — в scripts/.gitea-creds'
|
||||||
|
Write-Error "C-компилятор gcc не найден."
|
||||||
|
}
|
||||||
|
Write-Host "[ok] gcc: $((gcc --version | Select-Object -First 1))"
|
||||||
|
|
||||||
|
# --- 1b. Корень целевого проекта ---
|
||||||
|
# PROJECT_PATH из .gitea-creds (относительно папки scripts/) либо, если не задан,
|
||||||
|
# папка рядом со скриптом (прежнее поведение).
|
||||||
|
$projPath = $creds["PROJECT_PATH"]
|
||||||
|
if ($projPath -ne "" -and $projPath) {
|
||||||
|
$RepoRoot = (Resolve-Path -LiteralPath (Join-Path $ScriptDir $projPath)).Path
|
||||||
|
if (-not $RepoRoot) { Write-Error "PROJECT_PATH не найден: $projPath (относительно $ScriptDir)" }
|
||||||
|
} else {
|
||||||
|
$RepoRoot = Split-Path -Parent $ScriptDir
|
||||||
|
}
|
||||||
|
Set-Location $RepoRoot
|
||||||
|
Write-Host "[root] $RepoRoot"
|
||||||
|
|
||||||
# --- 2. Версия = commit-<sha7> текущего HEAD (совпадает с именованием CI) ---
|
# --- 2. Версия = commit-<sha7> текущего HEAD (совпадает с именованием CI) ---
|
||||||
$sha = (git rev-parse --short HEAD).Trim()
|
$sha = (git rev-parse --short HEAD).Trim()
|
||||||
if (-not $sha) { Write-Error "git rev-parse --short HEAD не дал хэш." }
|
if (-not $sha) { Write-Error "git rev-parse --short HEAD не дал хэш." }
|
||||||
|
|||||||
Reference in New Issue
Block a user