envio 3.0.0-alpha.1 → 3.0.0-alpha.2

package/src/sources/HyperSyncHeightStream.res.mjs

@@ -0,0 +1,127 @@
+// Generated by ReScript, PLEASE EDIT WITH CARE
+
+import * as Utils from "../Utils.res.mjs";
+import * as Js_exn from "rescript/lib/es6/js_exn.js";
+import * as Js_dict from "rescript/lib/es6/js_dict.js";
+import * as Logging from "../Logging.res.mjs";
+import * as Belt_Int from "rescript/lib/es6/belt_Int.js";
+import * as Belt_Option from "rescript/lib/es6/belt_Option.js";
+import * as Caml_option from "rescript/lib/es6/caml_option.js";
+import * as Eventsource from "eventsource";
+
+function make(hyperSyncUrl, apiToken) {
+  var updateTimeoutId = function (eventsourceRef, timeoutIdRef, hyperSyncUrl, apiToken, heightRef, errorRef) {
+    clearTimeout(timeoutIdRef.contents);
+    var newTimeoutId = setTimeout((function () {
+            Logging.trace({
+                  msg: "Timeout fired for height stream",
+                  url: hyperSyncUrl,
+                  staleTimeMillis: 15000
+                });
+            refreshEventSource(eventsourceRef, hyperSyncUrl, apiToken, heightRef, errorRef, timeoutIdRef);
+          }), 15000);
+    timeoutIdRef.contents = newTimeoutId;
+  };
+  var refreshEventSource = function (eventsourceRef, hyperSyncUrl, apiToken, heightRef, errorRef, timeoutIdRef) {
+    var es = eventsourceRef.contents;
+    if (es !== undefined) {
+      Caml_option.valFromOption(es).close();
+    }
+    var userAgent = "hyperindex/" + Utils.EnvioPackage.value.version;
+    var es$1 = new Eventsource.EventSource(hyperSyncUrl + "/height/sse", {
+          headers: Js_dict.fromArray([
+                [
+                  "Authorization",
+                  "Bearer " + apiToken
+                ],
+                [
+                  "User-Agent",
+                  userAgent
+                ]
+              ])
+        });
+    eventsourceRef.contents = Caml_option.some(es$1);
+    updateTimeoutId(eventsourceRef, timeoutIdRef, hyperSyncUrl, apiToken, heightRef, errorRef);
+    es$1.onopen = (function () {
+        Logging.trace({
+              msg: "SSE connection opened for height stream",
+              url: hyperSyncUrl
+            });
+      });
+    es$1.onerror = (function (error) {
+        Logging.trace({
+              msg: "EventSource error",
+              error: error.message
+            });
+        errorRef.contents = Belt_Option.getWithDefault(error.message, "Unexpected no error.message");
+      });
+    es$1.addEventListener("ping", (function (_event) {
+            updateTimeoutId(eventsourceRef, timeoutIdRef, hyperSyncUrl, apiToken, heightRef, errorRef);
+            errorRef.contents = undefined;
+          }));
+    es$1.addEventListener("height", (function ($$event) {
+            var height = Belt_Int.fromString($$event.data);
+            if (height !== undefined) {
+              updateTimeoutId(eventsourceRef, timeoutIdRef, hyperSyncUrl, apiToken, heightRef, errorRef);
+              errorRef.contents = undefined;
+              heightRef.contents = height;
+            } else {
+              Logging.trace({
+                    msg: "Height was not a number in event.data",
+                    data: $$event.data
+                  });
+              errorRef.contents = "Height was not a number in event.data";
+            }
+          }));
+  };
+  var heightRef = {
+    contents: 0
+  };
+  var errorRef = {
+    contents: undefined
+  };
+  var eventsourceRef = {
+    contents: undefined
+  };
+  var timeoutIdRef = {
+    contents: setTimeout((function () {
+            
+          }), 0)
+  };
+  refreshEventSource(eventsourceRef, hyperSyncUrl, apiToken, heightRef, errorRef, timeoutIdRef);
+  return {
+          heightRef: heightRef,
+          errorRef: errorRef,
+          timeoutIdRef: timeoutIdRef,
+          eventsourceRef: eventsourceRef
+        };
+}
+
+async function getHeight(t) {
+  while(t.heightRef.contents === 0 && t.errorRef.contents === undefined) {
+    await Utils.delay(200);
+  };
+  var error = t.errorRef.contents;
+  if (error !== undefined) {
+    return Js_exn.raiseError(error);
+  } else {
+    return t.heightRef.contents;
+  }
+}
+
+function close(t) {
+  clearTimeout(t.timeoutIdRef.contents);
+  var es = t.eventsourceRef.contents;
+  if (es !== undefined) {
+    Caml_option.valFromOption(es).close();
+    return ;
+  }
+  
+}
+
+export {
+  make ,
+  getHeight ,
+  close ,
+}
+/* Utils Not a pure module */

package/src/sources/HyperSyncSource.res

@@ -159,63 +159,6 @@ type options = {
   enableQueryCaching: bool,
 }
 
-module HeightState: {
-  type t
-  let make: (HyperSyncClient.t, ~chainId: int) => Promise.t<t>
-  let getHeight: t => Promise.t<int>
-} = {
-  open HyperSyncClient
-  type t = {
-    heightStream: HeightStream.t,
-    mutable currentHeight: int,
-    mutable errMessage: option<string>,
-  }
-
-  let makeInternal = async (client: HyperSyncClient.t) => {
-    currentHeight: 0,
-    heightStream: await client.streamHeight(),
-    errMessage: None,
-  }
-
-  let start = async (state: t, ~chainId) => {
-    while true {
-      let height = await state.heightStream.recv()
-      switch height {
-      | Height({height}) => state.currentHeight = height
-      | Connected =>
-        Logging.trace({"msg": "HyperSync height stream is connected", "chainId": chainId})
-        state.errMessage = None
-      | Reconnecting({delayMillis, errorMsg}) =>
-        Logging.trace({
-          "msg": "HyperSync height stream is reconnecting",
-          "err": errorMsg,
-          "delayMillis": delayMillis,
-          "chainId": chainId,
-        })
-        state.errMessage = Some(errorMsg)
-      }
-    }
-  }
-
-  let make = async (client: HyperSyncClient.t, ~chainId) => {
-    let state = await makeInternal(client)
-    let _async = state->start(~chainId)
-    state
-  }
-
-  let getHeight = async (state: t) => {
-    while state.currentHeight == 0 && state.errMessage->Belt.Option.isNone {
-      // Just poll internally until its no longer 0
-      await Utils.delay(200)
-    }
-
-    switch state.errMessage {
-    | Some(errMessage) => Js.Exn.raiseError(errMessage)
-    | None => state.currentHeight
-    }
-  }
-}
-
 let make = (
   {
     contracts,
@@ -610,7 +553,7 @@ let make = (
 
   let malformedTokenMessage = `Your token is malformed. For more info: https://docs.envio.dev/docs/HyperSync/api-tokens.`
 
-  let heightStatePromise = HeightState.make(client, ~chainId=chain->ChainMap.Chain.toChainId)
+  let heightStream = HyperSyncHeightStream.make(~hyperSyncUrl=endpointUrl, ~apiToken)
 
   {
     name,
@@ -620,8 +563,7 @@ let make = (
     poweredByHyperSync: true,
     getBlockHashes,
     getHeightOrThrow: async () => {
-      let heightState = await heightStatePromise
-      try await heightState->HeightState.getHeight catch {
+      try await heightStream->HyperSyncHeightStream.getHeight catch {
       | Js.Exn.Error(exn)
         if exn
         ->Js.Exn.message

package/src/sources/HyperSyncSource.res.mjs

@@ -16,6 +16,7 @@ import * as Caml_exceptions from "rescript/lib/es6/caml_exceptions.js";
 import * as HyperSyncClient from "./HyperSyncClient.res.mjs";
 import * as Caml_splice_call from "rescript/lib/es6/caml_splice_call.js";
 import * as Caml_js_exceptions from "rescript/lib/es6/caml_js_exceptions.js";
+import * as HyperSyncHeightStream from "./HyperSyncHeightStream.res.mjs";
 
 function getSelectionConfig(selection, chain) {
   var nonOptionalBlockFieldNames = new Set();
@@ -120,62 +121,7 @@ function memoGetSelectionConfig(chain) {
   };
 }
 
-async function makeInternal(client) {
-  return {
-          heightStream: await client.streamHeight(),
-          currentHeight: 0,
-          errMessage: undefined
-        };
-}
-
-async function start(state, chainId) {
-  while(true) {
-    var height = await state.heightStream.recv();
-    if (typeof height !== "object") {
-      Logging.trace({
-            msg: "HyperSync height stream is connected",
-            chainId: chainId
-          });
-      state.errMessage = undefined;
-    } else if (height.type === "Height") {
-      state.currentHeight = height.height;
-    } else {
-      var errorMsg = height.errorMsg;
-      Logging.trace({
-            msg: "HyperSync height stream is reconnecting",
-            err: errorMsg,
-            delayMillis: height.delayMillis,
-            chainId: chainId
-          });
-      state.errMessage = errorMsg;
-    }
-  };
-}
-
-async function make(client, chainId) {
-  var state = await makeInternal(client);
-  start(state, chainId);
-  return state;
-}
-
-async function getHeight(state) {
-  while(state.currentHeight === 0 && Belt_Option.isNone(state.errMessage)) {
-    await Utils.delay(200);
-  };
-  var errMessage = state.errMessage;
-  if (errMessage !== undefined) {
-    return Js_exn.raiseError(errMessage);
-  } else {
-    return state.currentHeight;
-  }
-}
-
-var HeightState = {
-  make: make,
-  getHeight: getHeight
-};
-
-function make$1(param) {
+function make(param) {
   var lowercaseAddresses = param.lowercaseAddresses;
   var eventRouter = param.eventRouter;
   var shouldUseHypersyncClientDecoder = param.shouldUseHypersyncClientDecoder;
@@ -437,7 +383,7 @@ function make$1(param) {
   var getBlockHashes = function (blockNumbers, logger) {
     return HyperSync.queryBlockDataMulti(endpointUrl, apiToken, blockNumbers, logger).then(HyperSync.mapExn);
   };
-  var heightStatePromise = make(client, chain);
+  var heightStream = HyperSyncHeightStream.make(endpointUrl, apiToken);
   return {
           name: "HyperSync",
           sourceFor: "Sync",
@@ -446,9 +392,8 @@ function make$1(param) {
           pollingInterval: 32,
           getBlockHashes: getBlockHashes,
           getHeightOrThrow: (async function () {
-              var heightState = await heightStatePromise;
               try {
-                return await getHeight(heightState);
+                return await HyperSyncHeightStream.getHeight(heightStream);
               }
               catch (raw_exn){
                 var exn = Caml_js_exceptions.internalToOCamlException(raw_exn);
@@ -472,7 +417,6 @@ function make$1(param) {
 export {
   getSelectionConfig ,
   memoGetSelectionConfig ,
-  HeightState ,
-  make$1 as make,
+  make ,
 }
 /* Viem Not a pure module */