YeonGyu-Kim
0708038712
- Use shared OPENCODE_STORAGE, MESSAGE_STORAGE, PART_STORAGE constants - Make isCallerOrchestrator async with SDK fallback for beta - Fix cache implementation using Symbol sentinel - Update atlas hooks and sisyphus-junior-notepad to use async isCallerOrchestrator
26 lines
982 B
TypeScript
26 lines
982 B
TypeScript
import type { PluginInput } from "@opencode-ai/plugin"
|
|
import { createAtlasEventHandler } from "./event-handler"
|
|
import { createToolExecuteAfterHandler } from "./tool-execute-after"
|
|
import { createToolExecuteBeforeHandler } from "./tool-execute-before"
|
|
import type { AtlasHookOptions, SessionState } from "./types"
|
|
|
|
export function createAtlasHook(ctx: PluginInput, options?: AtlasHookOptions) {
|
|
const sessions = new Map<string, SessionState>()
|
|
const pendingFilePaths = new Map<string, string>()
|
|
|
|
function getState(sessionID: string): SessionState {
|
|
let state = sessions.get(sessionID)
|
|
if (!state) {
|
|
state = { promptFailureCount: 0 }
|
|
sessions.set(sessionID, state)
|
|
}
|
|
return state
|
|
}
|
|
|
|
return {
|
|
handler: createAtlasEventHandler({ ctx, options, sessions, getState }),
|
|
"tool.execute.before": createToolExecuteBeforeHandler({ ctx, pendingFilePaths }),
|
|
"tool.execute.after": createToolExecuteAfterHandler({ ctx, pendingFilePaths }),
|
|
}
|
|
}
|