33 lines
1.4 KiB
Go
33 lines
1.4 KiB
Go
// Package activitydispatch is the single source of truth mapping the agent
|
|
// token in `ao hooks <agent> <event>` onto the function that interprets that
|
|
// agent's hook callbacks as an AO activity state.
|
|
package activitydispatch
|
|
|
|
import (
|
|
"github.com/aoagents/agent-orchestrator/backend/internal/adapters/agent/claudecode"
|
|
"github.com/aoagents/agent-orchestrator/backend/internal/adapters/agent/codex"
|
|
"github.com/aoagents/agent-orchestrator/backend/internal/adapters/agent/opencode"
|
|
"github.com/aoagents/agent-orchestrator/backend/internal/domain"
|
|
)
|
|
|
|
// DeriveFunc maps a native agent hook event and its raw stdin payload onto an AO
|
|
// activity state. ok=false means the event carries no activity signal.
|
|
type DeriveFunc func(event string, payload []byte) (domain.ActivityState, bool)
|
|
|
|
// Derivers maps the agent token in `ao hooks <agent> <event>` to its deriver.
|
|
var Derivers = map[string]DeriveFunc{
|
|
"claude-code": claudecode.DeriveActivityState,
|
|
"codex": codex.DeriveActivityState,
|
|
"opencode": opencode.DeriveActivityState,
|
|
}
|
|
|
|
// Derive looks up the deriver for an agent token and applies it. ok=false when
|
|
// the token has no registered deriver or the event carries no activity signal.
|
|
func Derive(agent, event string, payload []byte) (domain.ActivityState, bool) {
|
|
derive, found := Derivers[agent]
|
|
if !found {
|
|
return "", false
|
|
}
|
|
return derive(event, payload)
|
|
}
|