wehub-resource-sync
121 lines
3.3 KiB
TypeScript
121 lines
3.3 KiB
TypeScript
import { expect, type ElectronApplication } from '@stablyai/playwright-test'
|
|
import { existsSync, readdirSync, readFileSync } from 'node:fs'
|
|
import path from 'node:path'
|
|
import {
|
|
isAgentHookEndpointFileName,
|
|
parseAgentHookEndpointFile,
|
|
type AgentHookEndpoint
|
|
} from '../../../src/shared/agent-hook-endpoint-file'
|
|
|
|
function findEndpointEnvFile(root: string): string | null {
|
|
if (!existsSync(root)) {
|
|
return null
|
|
}
|
|
const entries = readdirSync(root, { withFileTypes: true })
|
|
for (const entry of entries) {
|
|
const fullPath = path.join(root, entry.name)
|
|
if (entry.isFile() && isAgentHookEndpointFileName(entry.name)) {
|
|
return fullPath
|
|
}
|
|
if (entry.isDirectory()) {
|
|
const nested = findEndpointEnvFile(fullPath)
|
|
if (nested) {
|
|
return nested
|
|
}
|
|
}
|
|
}
|
|
return null
|
|
}
|
|
|
|
export async function readHookEndpoint(app: ElectronApplication): Promise<AgentHookEndpoint> {
|
|
const userDataPath = await app.evaluate(({ app: electronApp }) => electronApp.getPath('userData'))
|
|
const hookRoot = path.join(userDataPath, 'agent-hooks')
|
|
let endpointPath: string | null = null
|
|
await expect
|
|
.poll(
|
|
() => {
|
|
endpointPath = findEndpointEnvFile(hookRoot)
|
|
return endpointPath
|
|
},
|
|
{
|
|
timeout: 15_000,
|
|
message: `Agent hook endpoint file not found under ${hookRoot}`
|
|
}
|
|
)
|
|
.not.toBeNull()
|
|
if (!endpointPath) {
|
|
throw new Error(`Agent hook endpoint file not found under ${hookRoot}`)
|
|
}
|
|
return parseAgentHookEndpointFile(readFileSync(endpointPath, 'utf8'))
|
|
}
|
|
|
|
export async function emitCodexHookStatus(
|
|
endpoint: AgentHookEndpoint,
|
|
status: {
|
|
paneKey: string
|
|
worktreeId: string
|
|
state: 'working' | 'done'
|
|
prompt?: string
|
|
lastAssistantMessage?: string
|
|
}
|
|
): Promise<void> {
|
|
const [tabId] = status.paneKey.split(':')
|
|
const payload =
|
|
status.state === 'working'
|
|
? {
|
|
hook_event_name: 'UserPromptSubmit',
|
|
prompt: status.prompt
|
|
}
|
|
: {
|
|
hook_event_name: 'Stop',
|
|
last_assistant_message: status.lastAssistantMessage
|
|
}
|
|
const response = await fetch(`http://127.0.0.1:${endpoint.port}/hook/codex`, {
|
|
method: 'POST',
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
'X-Orca-Agent-Hook-Token': endpoint.token
|
|
},
|
|
body: JSON.stringify({
|
|
paneKey: status.paneKey,
|
|
tabId,
|
|
worktreeId: status.worktreeId,
|
|
env: endpoint.env,
|
|
version: endpoint.version,
|
|
payload
|
|
})
|
|
})
|
|
if (response.status !== 204) {
|
|
throw new Error(`Codex hook POST returned ${response.status}`)
|
|
}
|
|
}
|
|
|
|
export async function emitGrokHookPayload(
|
|
endpoint: AgentHookEndpoint,
|
|
event: {
|
|
paneKey: string
|
|
worktreeId: string
|
|
payload: Record<string, unknown>
|
|
}
|
|
): Promise<void> {
|
|
const [tabId] = event.paneKey.split(':')
|
|
const response = await fetch(`http://127.0.0.1:${endpoint.port}/hook/grok`, {
|
|
method: 'POST',
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
'X-Orca-Agent-Hook-Token': endpoint.token
|
|
},
|
|
body: JSON.stringify({
|
|
paneKey: event.paneKey,
|
|
tabId,
|
|
worktreeId: event.worktreeId,
|
|
env: endpoint.env,
|
|
version: endpoint.version,
|
|
payload: event.payload
|
|
})
|
|
})
|
|
if (response.status !== 204) {
|
|
throw new Error(`Grok hook POST returned ${response.status}`)
|
|
}
|
|
}
|