broker

type AgentMessage struct {
	// ID is a stable message UUID for deduplication, cancellation, and
	// ask/reply correlation. Generated by the sender; should be unique
	// per sender.
	ID string `json:"id,omitempty"`

	// From is the display name of the sender run (often same as FromRunID).
	From string `json:"from"`

	// FromRunID is the run ID of the sending agent.
	FromRunID string `json:"from_run_id"`

	// ToRunID is the run ID of the intended recipient.
	ToRunID string `json:"to_run_id,omitempty"`

	// Message is the body text.
	Message string `json:"message"`

	// Role is a display label such as "agent", "reviewer", or "supervisor".
	Role string `json:"role,omitempty"`

	// ReplyTo, when non-empty, marks this message as a reply to the
	// original message with the given ID. The broker uses this to match
	// a waiting ask.
	ReplyTo string `json:"reply_to,omitempty"`

	// ExpectsReply, when true, signals that the sender is blocking and
	// expects a reply with ReplyTo pointing back to this message's ID.
	ExpectsReply bool `json:"expects_reply,omitempty"`
}

type AskEdge struct {
	FromRunID string    `json:"from_run_id"`
	ToRunID   string    `json:"to_run_id"`
	MessageID string    `json:"message_id"`
	CreatedAt time.Time `json:"created_at"`
}

type AskReply struct {
	MessageID string          `json:"message_id"`
	FromRunID string          `json:"from_run_id"`
	Payload   json.RawMessage `json:"payload,omitempty"`
}

type Broker struct {
	// contains filtered or unexported fields
}

type ControlMessage struct {
	ID        string          `json:"id"`
	Type      string          `json:"type"`
	RunID     string          `json:"run_id"`
	FromRunID string          `json:"from_run_id,omitempty"`
	Payload   json.RawMessage `json:"payload,omitempty"`
	CreatedAt time.Time       `json:"created_at"`
}

type Envelope struct {
	FromRunID     string          `json:"from_run_id"`
	ToRunID       string          `json:"to_run_id"`
	To            string          `json:"to,omitempty"`
	Kind          string          `json:"kind"`
	CorrelationID string          `json:"correlation_id,omitempty"`
	Payload       json.RawMessage `json:"payload,omitempty"`
	CreatedAt     time.Time       `json:"created_at"`
}

type Finish struct {
	RunID        string          `json:"run_id"`
	Status       string          `json:"status"`
	Summary      string          `json:"summary"`
	FilesChanged []string        `json:"files_changed,omitempty"`
	Payload      json.RawMessage `json:"payload,omitempty"`
}

type PermissionState struct {
	RequestID string
	ToolName  string
	Desc      string
	Preview   string
	CreatedAt time.Time
}

type Recorder struct {
	// contains filtered or unexported fields
}

type Reply struct {
	RunID   string          `json:"run_id"`
	To      string          `json:"to"`
	Payload json.RawMessage `json:"payload,omitempty"`
}

type Report struct {
	RunID   string          `json:"run_id"`
	State   string          `json:"state"`
	Payload json.RawMessage `json:"payload,omitempty"`
}

type RunState struct {
	RunID               string
	Token               string
	ControlQueue        []*ControlMessage
	RegisteredAt        time.Time
	LastSeen            time.Time
	Reports             []Report
	Finishes            []Finish
	Replies             []Reply
	PermissionRequests  map[string]*PermissionState
	PermissionDecisions map[string]string // requestID -> "allow" or "deny"
	Mu                  sync.Mutex
	Notify              chan struct{}

	// Ask/reply tracking
	PendingAsks  map[string]*AskEdge      // messageID -> edge (sender is waiting)
	WaitingReply map[string]chan AskReply // messageID -> buffered(1) channel for /wait_reply

	// Session metadata for peer discovery
	Info *SessionInfo
}

type SessionInfo struct {
	RunID    string `json:"run_id"`
	Label    string `json:"label,omitempty"`
	Backend  string `json:"backend,omitempty"`
	Model    string `json:"model,omitempty"`
	Dir      string `json:"dir,omitempty"`
	Status   string `json:"status,omitempty"`
	LastSeen int64  `json:"last_seen"`
}