@nymphjs/client 1.0.0-beta.11 → 1.0.0-beta.111

package/src/HttpRequester.ts

@@ -1,25 +1,42 @@
+import {
+  fetchEventSource,
+  EventStreamContentType,
+} from 'fetch-event-source-hperrin';
+
 export type HttpRequesterEventType = 'request' | 'response';
 export type HttpRequesterRequestCallback = (
   requester: HttpRequester,
   url: string,
-  options: RequestInit
+  options: RequestInit,
 ) => void;
 export type HttpRequesterResponseCallback = (
   requester: HttpRequester,
   response: Response,
-  text: string
+  text: string,
+) => void;
+export type HttpRequesterIteratorCallback = (
+  requester: HttpRequester,
+  url: string,
+  headers: Record<string, string>,
 ) => void;
 export type HttpRequesterRequestOptions = {
   url: string;
+  headers?: { [k: string]: any };
   data: { [k: string]: any };
   dataType: string;
 };
 
+export interface AbortableAsyncIterator<
+  T extends any = any,
+> extends AsyncIterable<T> {
+  abortController: AbortController;
+}
+
 export default class HttpRequester {
   private fetch: WindowOrWorkerGlobalScope['fetch'];
-  private xsrfToken: string | null = null;
   private requestCallbacks: HttpRequesterRequestCallback[] = [];
   private responseCallbacks: HttpRequesterResponseCallback[] = [];
+  private iteratorCallbacks: HttpRequesterIteratorCallback[] = [];
 
   static makeUrl(url: string, data: { [k: string]: any }) {
     if (!data) {
@@ -45,14 +62,18 @@ export default class HttpRequester {
     callback: T extends 'request'
       ? HttpRequesterRequestCallback
       : T extends 'response'
-      ? HttpRequesterResponseCallback
-      : never
+        ? HttpRequesterResponseCallback
+        : T extends 'iterator'
+          ? HttpRequesterIteratorCallback
+          : never,
   ) {
     const prop = (event + 'Callbacks') as T extends 'request'
       ? 'requestCallbacks'
       : T extends 'response'
-      ? 'responseCallbacks'
-      : never;
+        ? 'responseCallbacks'
+        : T extends 'iterator'
+          ? 'iteratorCallbacks'
+          : never;
     if (!(prop in this)) {
       throw new Error('Invalid event type.');
     }
@@ -66,14 +87,18 @@ export default class HttpRequester {
     callback: T extends 'request'
       ? HttpRequesterRequestCallback
       : T extends 'response'
-      ? HttpRequesterResponseCallback
-      : never
+        ? HttpRequesterResponseCallback
+        : T extends 'iterator'
+          ? HttpRequesterIteratorCallback
+          : never,
   ) {
     const prop = (event + 'Callbacks') as T extends 'request'
       ? 'requestCallbacks'
       : T extends 'response'
-      ? 'responseCallbacks'
-      : never;
+        ? 'responseCallbacks'
+        : T extends 'iterator'
+          ? 'iteratorCallbacks'
+          : never;
     if (!(prop in this)) {
       return false;
     }
@@ -86,10 +111,6 @@ export default class HttpRequester {
     return true;
   }
 
-  setXsrfToken(xsrfToken: string | null) {
-    this.xsrfToken = xsrfToken;
-  }
-
   async GET(opt: HttpRequesterRequestOptions) {
     return await this._httpRequest('GET', opt);
   }
@@ -98,6 +119,10 @@ export default class HttpRequester {
     return await this._httpRequest('POST', opt);
   }
 
+  async POST_ITERATOR(opt: HttpRequesterRequestOptions) {
+    return await this._iteratorRequest('POST', opt);
+  }
+
   async PUT(opt: HttpRequesterRequestOptions) {
     return await this._httpRequest('PUT', opt);
   }
@@ -112,7 +137,7 @@ export default class HttpRequester {
 
   async _httpRequest(
     method: 'GET' | 'POST' | 'PUT' | 'PATCH' | 'DELETE',
-    opt: HttpRequesterRequestOptions
+    opt: HttpRequesterRequestOptions,
   ) {
     const dataString = JSON.stringify(opt.data);
     let url = opt.url;
@@ -123,7 +148,7 @@ export default class HttpRequester {
     }
     const options: RequestInit = {
       method,
-      headers: {},
+      headers: opt.headers ?? {},
       credentials: 'include',
     };
 
@@ -137,18 +162,13 @@ export default class HttpRequester {
       this.requestCallbacks[i] && this.requestCallbacks[i](this, url, options);
     }
 
-    if (this.xsrfToken !== null) {
-      (options.headers as Record<string, string>)['X-Xsrf-Token'] =
-        this.xsrfToken;
-    }
-
     const response = await this.fetch(url, options);
     let text: string;
     try {
       text = await response.text();
     } catch (e: any) {
       throw new InvalidResponseError(
-        'Server response did not contain valid text body.'
+        'Server response did not contain valid text body.',
       );
     }
     if (!response.ok) {
@@ -170,12 +190,12 @@ export default class HttpRequester {
       throw response.status < 200
         ? new InformationalError(response, errObj)
         : response.status < 300
-        ? new SuccessError(response, errObj)
-        : response.status < 400
-        ? new RedirectError(response, errObj)
-        : response.status < 500
-        ? new ClientError(response, errObj)
-        : new ServerError(response, errObj);
+          ? new SuccessError(response, errObj)
+          : response.status < 400
+            ? new RedirectError(response, errObj)
+            : response.status < 500
+              ? new ClientError(response, errObj)
+              : new ServerError(response, errObj);
     }
     for (let i = 0; i < this.responseCallbacks.length; i++) {
       this.responseCallbacks[i] &&
@@ -192,13 +212,216 @@ export default class HttpRequester {
           throw e;
         }
         throw new InvalidResponseError(
-          'Server response was invalid: ' + JSON.stringify(text)
+          'Server response was invalid: ' + JSON.stringify(text),
         );
       }
     } else {
       return text;
     }
   }
+
+  async _iteratorRequest(
+    method: 'GET' | 'POST' | 'PUT' | 'PATCH' | 'DELETE',
+    opt: HttpRequesterRequestOptions,
+  ): Promise<AbortableAsyncIterator> {
+    const dataString = JSON.stringify(opt.data);
+    let url = opt.url;
+    if (method === 'GET') {
+      // TODO: what should this size be?
+      // && dataString.length < 1) {
+      url = HttpRequester.makeUrl(opt.url, opt.data);
+    }
+    const hasBody = method !== 'GET' && opt.data;
+    const headers: Record<string, string> = opt.headers ?? {};
+
+    if (hasBody) {
+      headers['Content-Type'] = 'application/json';
+    }
+
+    for (let i = 0; i < this.iteratorCallbacks.length; i++) {
+      this.iteratorCallbacks[i] &&
+        this.iteratorCallbacks[i](this, url, headers);
+    }
+
+    const responses: any[] = [];
+    let nextResponseResolve: (value: void) => void;
+    let nextResponseReadyPromise = new Promise<void>((res) => {
+      nextResponseResolve = res;
+    });
+    let responsesDone = false;
+    let serverResponse: Response;
+
+    const ctrl = new AbortController();
+
+    fetchEventSource(url, {
+      openWhenHidden: true,
+      fetch: this.fetch,
+
+      method,
+      headers,
+      credentials: 'include',
+      body: hasBody ? dataString : undefined,
+      signal: ctrl.signal,
+
+      async onopen(response) {
+        serverResponse = response;
+        if (response.ok) {
+          if (response.headers.get('content-type') === EventStreamContentType) {
+            throw new InvalidResponseError(
+              'Server response is not an event stream.',
+            );
+          }
+
+          // Response is ok, wait for messages.
+          return;
+        }
+
+        let text: string = '';
+        try {
+          text = await response.text();
+        } catch (e: any) {
+          // Ignore error here.
+        }
+
+        let errObj;
+        try {
+          errObj = JSON.parse(text);
+        } catch (e: any) {
+          if (!(e instanceof SyntaxError)) {
+            throw e;
+          }
+        }
+
+        if (typeof errObj !== 'object') {
+          errObj = {
+            textStatus: response.statusText,
+          };
+        }
+        errObj.status = response.status;
+        throw response.status < 200
+          ? new InformationalError(response, errObj)
+          : response.status < 300
+            ? new SuccessError(response, errObj)
+            : response.status < 400
+              ? new RedirectError(response, errObj)
+              : response.status < 500
+                ? new ClientError(response, errObj)
+                : new ServerError(response, errObj);
+      },
+
+      onmessage(event) {
+        if (event.event === 'next') {
+          let text = event.data;
+
+          if (opt.dataType === 'json') {
+            if (!text.length) {
+              responses.push(
+                new InvalidResponseError('Server response was empty.'),
+              );
+            } else {
+              try {
+                responses.push(JSON.parse(text));
+              } catch (e: any) {
+                if (!(e instanceof SyntaxError)) {
+                  responses.push(e);
+                } else {
+                  responses.push(
+                    new InvalidResponseError(
+                      'Server response was invalid: ' + JSON.stringify(text),
+                    ),
+                  );
+                }
+              }
+            }
+          } else {
+            responses.push(text);
+          }
+        } else if (event.event === 'error') {
+          let text = event.data;
+
+          let errObj;
+          try {
+            errObj = JSON.parse(text);
+          } catch (e: any) {
+            if (!(e instanceof SyntaxError)) {
+              throw e;
+            }
+          }
+
+          if (typeof errObj !== 'object') {
+            errObj = {
+              status: 500,
+              textStatus: 'Iterator Error',
+            };
+          }
+          responses.push(
+            errObj.status < 200
+              ? new InformationalError(serverResponse, errObj)
+              : errObj.status < 300
+                ? new SuccessError(serverResponse, errObj)
+                : errObj.status < 400
+                  ? new RedirectError(serverResponse, errObj)
+                  : errObj.status < 500
+                    ? new ClientError(serverResponse, errObj)
+                    : new ServerError(serverResponse, errObj),
+          );
+        } else if (event.event === 'finished') {
+          responsesDone = true;
+        } else if (event.event === 'ping') {
+          // Ignore keep-alive pings.
+          return;
+        }
+
+        const resolve = nextResponseResolve;
+        if (!responsesDone) {
+          nextResponseReadyPromise = new Promise<void>((res) => {
+            nextResponseResolve = res;
+          });
+        }
+
+        // Resolve the promise to continue any waiting iterator.
+        resolve();
+      },
+
+      onclose() {
+        responses.push(
+          new ConnectionClosedUnexpectedlyError(
+            'The connection to the server was closed unexpectedly.',
+          ),
+        );
+
+        responsesDone = true;
+        nextResponseResolve();
+      },
+
+      onerror(err) {
+        // Rethrow to stop the operation.
+        throw err;
+      },
+    }).catch((err) => {
+      responses.push(
+        new ConnectionError('The connection could not be established: ' + err),
+      );
+
+      responsesDone = true;
+      nextResponseResolve();
+    });
+
+    const iterator: AbortableAsyncIterator = {
+      abortController: ctrl,
+      async *[Symbol.asyncIterator]() {
+        do {
+          await nextResponseReadyPromise;
+
+          while (responses.length) {
+            yield responses.shift();
+          }
+        } while (!responsesDone);
+      },
+    };
+
+    return iterator;
+  }
 }
 
 export class InvalidResponseError extends Error {
@@ -208,6 +431,20 @@ export class InvalidResponseError extends Error {
   }
 }
 
+export class ConnectionClosedUnexpectedlyError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = 'ConnectionClosedUnexpectedlyError';
+  }
+}
+
+export class ConnectionError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = 'ConnectionError';
+  }
+}
+
 export class HttpError extends Error {
   status: number;
   statusText: string;
@@ -215,7 +452,7 @@ export class HttpError extends Error {
   constructor(
     name: string,
     response: Response,
-    errObj: { textStatus: string }
+    errObj: { textStatus: string },
   ) {
     super(errObj.textStatus);
     this.name = name;