import { Agent } from 'agents';
import { Hono } from 'hono';
import { SSEStreamingApi } from 'hono/streaming';
import { BlankEnv } from 'hono/types';
import { MCPClientManager } from 'agents/mcp/client';

type MCPClientConnection = MCPClientManager["mcpConnections"][string];

type ConstructorType<T> = new (...args: any[]) => T;
interface InstrumentedProperties {
    _activeStreams: Set<SSEStreamingApi>;
    _mcpConnections?: Map<string, MCPClientConnection>;
    _fiberRouter?: Hono;
}
/**
 * Mixin factory that adds instruments the Agent class with observability capabilities
 * allowing it to be viewed in the Fiberplane agents playground
 *
 * Usage:
 * ```typescript
 * export class MyCoreAgent extends Agent<Env, MyState> {
 *   // Your agent implementation
 * }
 *
 * // Creates a new class that extends MyCoreAgent with observability features
 * const MyAgent = withInstrumentation(MyCoreAgent);
 * ```
 */
declare function withInstrumentation<BaseAgent extends Agent<BlankEnv, unknown>>(BaseClass: ConstructorType<BaseAgent>): ConstructorType<BaseAgent & InstrumentedProperties>;

interface FiberplaneEntryWrapperOptions {
    customPath?: string;
}
/**
 * Creates a fetch handler that serves the Fiberplane app
 */
declare function fiberplane<E = unknown>(userFetch: (request: Request, env: E, ctx: ExecutionContext) => Promise<Response>, options?: FiberplaneEntryWrapperOptions): (request: Request, env: E, ctx: ExecutionContext) => Promise<Response>;

export { type ConstructorType, fiberplane, withInstrumentation };