From 1635c282c66175455a4856d6471414f908259aee Mon Sep 17 00:00:00 2001
From: Fredrik Enestad <fredrik@enestad.com>
Date: Thu, 26 Sep 2024 15:03:02 +0200
Subject: [PATCH] client type definitions

---
 runtimes/js/encore.dev/api/mod.ts | 40 ++++++++++++++++++++++++++-----
 1 file changed, 34 insertions(+), 6 deletions(-)

diff --git a/runtimes/js/encore.dev/api/mod.ts b/runtimes/js/encore.dev/api/mod.ts
index 3bd0ae28f0..035cb2ebad 100644
--- a/runtimes/js/encore.dev/api/mod.ts
+++ b/runtimes/js/encore.dev/api/mod.ts
@@ -129,12 +129,36 @@ api.raw = function raw(options: APIOptions, fn: RawHandler) {
 export interface StreamIn<Request> extends AsyncIterable<Request> {
   recv: () => Promise<Request>;
 }
+export interface StreamOutWithResponse<Request, Response>
+  extends StreamOut<Request> {
+  response: () => Promise<Response>;
+}
 
 export interface StreamOut<Response> {
   send: (msg: Response) => Promise<void>;
   close: () => Promise<void>;
 }
 
+type StreamInOutHandlerFn<HandshakeData, Request, Response> =
+  HandshakeData extends void
+    ? () => Promise<StreamInOut<Request, Response>>
+    : (data: HandshakeData) => Promise<StreamInOut<Request, Response>>;
+
+type StreamOutHandlerFn<HandshakeData, Response> = HandshakeData extends void
+  ? () => Promise<StreamIn<Response>>
+  : (data: HandshakeData) => Promise<StreamIn<Response>>;
+
+type StreamInHandlerFn<HandshakeData, Request, Response> =
+  HandshakeData extends void
+    ? Response extends void
+      ? () => Promise<StreamOutWithResponse<Request, void>>
+      : () => Promise<StreamOutWithResponse<Request, Response>>
+    : Response extends void
+      ? (data: HandshakeData) => Promise<StreamIn<Request>>
+      : (
+          data: HandshakeData
+        ) => Promise<StreamOutWithResponse<Request, Response>>;
+
 export type StreamInOut<Request, Response> = StreamIn<Request> &
   StreamOut<Response>;
 
@@ -144,23 +168,27 @@ function streamInOut<HandshakeData, Request, Response>(
     data: HandshakeData,
     stream: StreamInOut<Request, Response>
   ) => Promise<void>
-): void;
+): StreamInOutHandlerFn<HandshakeData, Response, Request>;
 function streamInOut<Request, Response>(
   options: StreamOptions,
   fn: (stream: StreamInOut<Request, Response>) => Promise<void>
-): void;
+): StreamInOutHandlerFn<void, Response, Request>;
 function streamInOut(options: StreamOptions, fn: any): typeof fn {
   return fn;
 }
 
+function streamIn<Request>(
+  options: StreamOptions,
+  fn: (stream: StreamIn<Request>) => Promise<void>
+): StreamInHandlerFn<void, Request, void>;
 function streamIn<Request, Response>(
   options: StreamOptions,
   fn: (stream: StreamIn<Request>) => Promise<Response>
-): void;
+): StreamInHandlerFn<void, Request, Response>;
 function streamIn<HandshakeData, Request, Response>(
   options: StreamOptions,
   fn: (data: HandshakeData, stream: StreamIn<Request>) => Promise<Response>
-): void;
+): StreamInHandlerFn<HandshakeData, Request, Response>;
 function streamIn(options: StreamOptions, fn: any): typeof fn {
   return fn;
 }
@@ -168,11 +196,11 @@ function streamIn(options: StreamOptions, fn: any): typeof fn {
 function streamOut<HandshakeData, Response>(
   options: StreamOptions,
   fn: (data: HandshakeData, stream: StreamOut<Response>) => Promise<void>
-): void;
+): StreamOutHandlerFn<HandshakeData, Response>;
 function streamOut<Response>(
   options: StreamOptions,
   fn: (stream: StreamOut<Response>) => Promise<void>
-): void;
+): StreamOutHandlerFn<void, Response>;
 function streamOut(options: StreamOptions, fn: any): typeof fn {
   return fn;
 }