import { FlinkContext } from "@flink-app/flink";
import { StreamHandler, StreamHandlerModule, StreamingRouteProps } from "./types";

/**
 * Context interface for StreamingPlugin.
 * Use this when defining your application context to get type-safe access
 * to the streaming plugin via ctx.plugins.streaming
 */
export interface StreamingPluginContext {
    streaming: {
        registerStreamHandler: {
            <Ctx extends FlinkContext, T = any>(handlerModule: StreamHandlerModule<Ctx, T>): void;
            <Ctx extends FlinkContext, T = any>(handler: StreamHandler<Ctx, T>, routeProps: StreamingRouteProps): void;
        };
    };
}