protocol

type AttestationData struct {
	Proof        []byte `json:"proof"`
	PublicInputs []byte `json:"publicInputs"`
	CircuitID    string `json:"circuitId"`
	Scheme       string `json:"scheme"`
}

type CapabilityQueryPayload struct {
	Filter string `json:"filter,omitempty"` // optional tool name prefix filter
}

type CardProvider func() map[string]interface{}

type ContextSharePayload struct {
	TeamID  string                 `json:"teamId"`
	Context map[string]interface{} `json:"context"`
}

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

type HandlerConfig struct {
	Sessions *handshake.SessionStore
	Firewall *firewall.Firewall
	Executor ToolExecutor
	CardFn   CardProvider
	LocalDID string
	Logger   *zap.SugaredLogger
}

type NegotiateHandler func(ctx context.Context, peerDID string, payload NegotiatePayload) (map[string]interface{}, error)

type NegotiatePayload struct {
	SessionID string `json:"sessionId,omitempty"` // empty for initial proposal
	Action    string `json:"action"`              // "propose", "counter", "accept", "reject"
	ToolName  string `json:"toolName,omitempty"`
	Price     string `json:"price,omitempty"` // USDC amount as string
	Reason    string `json:"reason,omitempty"`
}

type OntologyHandler interface {
	HandleSchemaQuery(ctx context.Context, peerDID string, req SchemaQueryRequest) (*SchemaQueryResponse, error)
	HandleSchemaPropose(ctx context.Context, peerDID string, req SchemaProposeRequest) (*SchemaProposeResponse, error)
}

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

type PaidInvokePayload struct {
	ToolName    string                 `json:"toolName"`
	Params      map[string]interface{} `json:"params"`
	PaymentAuth map[string]interface{} `json:"paymentAuth,omitempty"`
}

type PayGateChecker interface {
	Check(peerDID, toolName string, payload map[string]interface{}) (PayGateResult, error)
}

type PayGateResult struct {
	Status       string                 // payGateStatusFree, payGateStatusVerified, payGateStatusPaymentRequired, payGateStatusInvalid, payGateStatusPostPayApproved
	Auth         interface{}            // the verified authorization (opaque to handler)
	PriceQuote   map[string]interface{} // price quote when payment required
	SettlementID string                 // deferred settlement ID for post-pay
}

type PriceQuoteResult struct {
	ToolName     string `json:"toolName"`
	Price        string `json:"price"`
	Currency     string `json:"currency"`
	USDCContract string `json:"usdcContract"`
	ChainID      int64  `json:"chainId"`
	SellerAddr   string `json:"sellerAddr"`
	QuoteExpiry  int64  `json:"quoteExpiry"`
	IsFree       bool   `json:"isFree"`
}

type RemoteAgentConfig struct {
	Name           string
	DID            string
	PeerID         peer.ID
	SessionToken   string
	Host           host.Host
	Capabilities   []string
	AttestVerifier ZKAttestVerifyFunc
	Logger         *zap.SugaredLogger
}

type Request struct {
	Type         RequestType            `json:"type"`
	SessionToken string                 `json:"sessionToken"`
	RequestID    string                 `json:"requestId"`
	Payload      map[string]interface{} `json:"payload,omitempty"`
}

type RequestType string

type Response struct {
	RequestID        string                 `json:"requestId"`
	Status           ResponseStatus         `json:"status"` // ResponseStatusOK, ResponseStatusError, ResponseStatusDenied
	Result           map[string]interface{} `json:"result,omitempty"`
	Error            string                 `json:"error,omitempty"`
	AttestationProof []byte                 `json:"attestationProof,omitempty"` // Deprecated: use Attestation
	Attestation      *AttestationData       `json:"attestation,omitempty"`
	Timestamp        time.Time              `json:"timestamp"`
}

type ResponseStatus string

type SafetyLevelChecker func(toolName string) (int, bool)

type SchemaProposeRequest struct {
	// Bundle is the JSON-encoded SchemaBundle to propose for import.
	// Uses json.RawMessage to avoid importing internal/ontology.
	Bundle json.RawMessage `json:"bundle"`

	// Reason describes why the schema elements are being proposed.
	Reason string `json:"reason,omitempty"`
}

type SchemaProposeResponse struct {
	// Action is the outcome: OntologyActionAccepted, OntologyActionPartial, or OntologyActionRejected.
	Action string `json:"action"`

	// Accepted lists the names of schema elements that were accepted.
	Accepted []string `json:"accepted,omitempty"`

	// Rejected lists the names of schema elements that were rejected.
	Rejected []string `json:"rejected,omitempty"`

	// Result is the JSON-encoded ImportResult from the ontology package.
	// Uses json.RawMessage to avoid importing internal/ontology.
	Result json.RawMessage `json:"result,omitempty"`
}

type SchemaQueryRequest struct {
	// RequestedTypes filters the response to specific entity type names.
	// Empty means return all types.
	RequestedTypes []string `json:"requestedTypes,omitempty"`

	// IncludePredicates controls whether predicate definitions are included.
	IncludePredicates bool `json:"includePredicates"`
}

type SchemaQueryResponse struct {
	// Bundle is the JSON-encoded SchemaBundle from the ontology package.
	// Uses json.RawMessage to avoid importing internal/ontology.
	Bundle json.RawMessage `json:"bundle"`
}

type SecurityEventTracker interface {
	RecordToolFailure(peerDID string)
	RecordToolSuccess(peerDID string)
}

type TeamAcceptHandler func(ctx context.Context, peerDID string, payload TeamAcceptPayload) (map[string]interface{}, error)

type TeamAcceptPayload struct {
	TeamID    string `json:"teamId"`
	MemberDID string `json:"memberDid"`
	Accepted  bool   `json:"accepted"`
	Reason    string `json:"reason,omitempty"`
}

type TeamDisbandHandler func(ctx context.Context, peerDID string, payload TeamDisbandPayload) (map[string]interface{}, error)

type TeamDisbandPayload struct {
	TeamID string `json:"teamId"`
	Reason string `json:"reason"`
}

type TeamHandler func(ctx context.Context, peerDID string, reqType RequestType, payload map[string]interface{}) (map[string]interface{}, error)

type TeamInviteHandler func(ctx context.Context, peerDID string, payload TeamInvitePayload) (map[string]interface{}, error)

type TeamInvitePayload struct {
	TeamID       string   `json:"teamId"`
	TeamName     string   `json:"teamName"`
	Goal         string   `json:"goal"`
	LeaderDID    string   `json:"leaderDid"`
	Role         string   `json:"role"`
	Capabilities []string `json:"capabilities"`
}

type TeamResultHandler func(ctx context.Context, peerDID string, payload TeamResultPayload) (map[string]interface{}, error)

type TeamResultPayload struct {
	TeamID    string                 `json:"teamId"`
	TaskID    string                 `json:"taskId"`
	MemberDID string                 `json:"memberDid"`
	Result    map[string]interface{} `json:"result,omitempty"`
	Error     string                 `json:"error,omitempty"`
	Duration  int64                  `json:"durationMs"`
}

type TeamRouter struct {
	OnInvite  TeamInviteHandler
	OnAccept  TeamAcceptHandler
	OnTask    TeamTaskHandler
	OnResult  TeamResultHandler
	OnDisband TeamDisbandHandler
}

type TeamTaskHandler func(ctx context.Context, peerDID string, payload TeamTaskPayload) (map[string]interface{}, error)

type TeamTaskPayload struct {
	TeamID   string                 `json:"teamId"`
	TaskID   string                 `json:"taskId"`
	ToolName string                 `json:"toolName"`
	Params   map[string]interface{} `json:"params"`
	Deadline time.Time              `json:"deadline,omitempty"`
}

type ToolApprovalFunc func(ctx context.Context, peerDID, toolName string, params map[string]interface{}) (bool, error)

type ToolExecutor func(ctx context.Context, toolName string, params map[string]interface{}) (map[string]interface{}, error)

type ToolInvokePayload struct {
	ToolName string                 `json:"toolName"`
	Params   map[string]interface{} `json:"params"`
}

type ZKAttestVerifyFunc func(ctx context.Context, attestation *AttestationData) (bool, error)