@cocorograph/hub-agent 0.5.9 → 0.5.10
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/package.json +1 -1
- package/src/main.mjs +82 -2
package/package.json
CHANGED
package/src/main.mjs
CHANGED
|
@@ -10,8 +10,8 @@
|
|
|
10
10
|
*
|
|
11
11
|
* 仕様書: ナレッジ/インフラ/cockpit-hub-hosted-integration-spec (id=6080)
|
|
12
12
|
*/
|
|
13
|
-
import { readFileSync } from "node:fs"
|
|
14
|
-
import { readFile } from "node:fs/promises"
|
|
13
|
+
import { readFileSync, watch as fsWatch } from "node:fs"
|
|
14
|
+
import { mkdir, readFile, readdir } from "node:fs/promises"
|
|
15
15
|
import os from "node:os"
|
|
16
16
|
import path from "node:path"
|
|
17
17
|
|
|
@@ -119,11 +119,16 @@ export async function startDaemon({ version, ptyModule } = {}) {
|
|
|
119
119
|
// 5s 周期で全 tmux session の状態を捕捉し、変化したものだけ session.state を push。
|
|
120
120
|
// browser がフォーカスしていない session でも常時更新するためのバックグラウンド職人。
|
|
121
121
|
const stateLoop = startStateLoop({ client, plugins, logger, intervalMs: 5_000 })
|
|
122
|
+
// bundle hook (cockpit_session_event_hook.py) が /tmp/cockpit_session_events/<name>.json
|
|
123
|
+
// に書き出す UserPromptSubmit / Stop の event を fs.watch で拾って WS push する。
|
|
124
|
+
// text マーカー判定 (detectStatusFromText) より精度が高い「ターン境界」の signal。
|
|
125
|
+
const sessionEventLoop = await startSessionEventWatcher({ client, logger })
|
|
122
126
|
|
|
123
127
|
const shutdown = async (signal) => {
|
|
124
128
|
logger.info({ signal }, "shutting down")
|
|
125
129
|
await runHookBroadcast(plugins, "onAgentStop", ctx)
|
|
126
130
|
stateLoop.stop()
|
|
131
|
+
sessionEventLoop?.stop?.()
|
|
127
132
|
ptyBridge.shutdown()
|
|
128
133
|
client.stop()
|
|
129
134
|
process.exit(0)
|
|
@@ -134,6 +139,81 @@ export async function startDaemon({ version, ptyModule } = {}) {
|
|
|
134
139
|
return { client, plugins, ptyBridge }
|
|
135
140
|
}
|
|
136
141
|
|
|
142
|
+
const SESSION_EVENTS_DIR =
|
|
143
|
+
process.env.COCKPIT_SESSION_EVENTS_DIR || "/tmp/cockpit_session_events"
|
|
144
|
+
|
|
145
|
+
/**
|
|
146
|
+
* `/tmp/cockpit_session_events/<tmux_session>.json` を fs.watch して
|
|
147
|
+
* UserPromptSubmit / Stop の event を `session.event` 型で Hub に push する。
|
|
148
|
+
*
|
|
149
|
+
* 各 .json は `{ "event": "prompt_submit" | "stop", "at": <epoch_ms> }` 形式。
|
|
150
|
+
* - bundle 配信の `cockpit_session_event_hook.py` が claude code の hook 経由で
|
|
151
|
+
* ターン境界に書き換える
|
|
152
|
+
* - capture-pane の text 判定よりタイミングが正確で attach の影響を受けない
|
|
153
|
+
*
|
|
154
|
+
* ディレクトリが無ければ作成し、起動直後に既存ファイルを 1 回読んで初期 push。
|
|
155
|
+
* 以降は変更検知のたびに該当ファイルを読み直して push する。
|
|
156
|
+
*/
|
|
157
|
+
async function startSessionEventWatcher({ client, logger }) {
|
|
158
|
+
try {
|
|
159
|
+
await mkdir(SESSION_EVENTS_DIR, { recursive: true })
|
|
160
|
+
} catch (err) {
|
|
161
|
+
logger?.warn(
|
|
162
|
+
{ err: err.message, dir: SESSION_EVENTS_DIR },
|
|
163
|
+
"session event watcher: failed to mkdir, watcher disabled",
|
|
164
|
+
)
|
|
165
|
+
return null
|
|
166
|
+
}
|
|
167
|
+
|
|
168
|
+
const pushFile = async (filename) => {
|
|
169
|
+
if (!filename || !filename.endsWith(".json")) return
|
|
170
|
+
const sessionName = filename.slice(0, -5)
|
|
171
|
+
const full = path.join(SESSION_EVENTS_DIR, filename)
|
|
172
|
+
try {
|
|
173
|
+
const text = await readFile(full, "utf-8")
|
|
174
|
+
const data = JSON.parse(text)
|
|
175
|
+
if (!data || typeof data.event !== "string") return
|
|
176
|
+
client.send({
|
|
177
|
+
type: "session.event",
|
|
178
|
+
session_name: sessionName,
|
|
179
|
+
event: data.event,
|
|
180
|
+
at: typeof data.at === "number" ? data.at : Date.now(),
|
|
181
|
+
})
|
|
182
|
+
} catch {
|
|
183
|
+
// 一時欠如 / parse 失敗等は無視 (次の変更で読み直す)
|
|
184
|
+
}
|
|
185
|
+
}
|
|
186
|
+
|
|
187
|
+
// 起動時に既存ファイルを 1 回読んで初期 push (再起動後に取りこぼし防止)
|
|
188
|
+
try {
|
|
189
|
+
const files = await readdir(SESSION_EVENTS_DIR)
|
|
190
|
+
for (const f of files) await pushFile(f)
|
|
191
|
+
} catch {
|
|
192
|
+
// 無視
|
|
193
|
+
}
|
|
194
|
+
|
|
195
|
+
let watcher = null
|
|
196
|
+
try {
|
|
197
|
+
watcher = fsWatch(SESSION_EVENTS_DIR, { persistent: false }, (_event, filename) => {
|
|
198
|
+
if (!filename) return
|
|
199
|
+
// 名前文字列で来るので path.basename 不要
|
|
200
|
+
pushFile(filename).catch(() => {})
|
|
201
|
+
})
|
|
202
|
+
} catch (err) {
|
|
203
|
+
logger?.warn(
|
|
204
|
+
{ err: err.message, dir: SESSION_EVENTS_DIR },
|
|
205
|
+
"session event watcher: fs.watch failed, watcher disabled",
|
|
206
|
+
)
|
|
207
|
+
return null
|
|
208
|
+
}
|
|
209
|
+
|
|
210
|
+
return {
|
|
211
|
+
stop() {
|
|
212
|
+
try { watcher?.close() } catch {}
|
|
213
|
+
},
|
|
214
|
+
}
|
|
215
|
+
}
|
|
216
|
+
|
|
137
217
|
/**
|
|
138
218
|
* 全 tmux session の状態を定期 capture し、変化したものだけ Hub に push する。
|
|
139
219
|
*
|