envio 3.0.0-alpha.3 → 3.0.0-alpha.5

package/src/PgStorage.res.mjs

@@ -1,6 +1,7 @@
 // Generated by ReScript, PLEASE EDIT WITH CARE
 
 import * as Fs from "fs";
+import * as Env from "./Env.res.mjs";
 import * as Path from "path";
 import * as $$Array from "rescript/lib/es6/array.js";
 import * as Table from "./db/Table.res.mjs";
@@ -13,6 +14,7 @@ import * as Js_dict from "rescript/lib/es6/js_dict.js";
 import * as Logging from "./Logging.res.mjs";
 import * as $$Promise from "./bindings/Promise.res.mjs";
 import * as Internal from "./Internal.res.mjs";
+import Postgres from "postgres";
 import * as Belt_Array from "rescript/lib/es6/belt_Array.js";
 import * as Prometheus from "./Prometheus.res.mjs";
 import * as Caml_option from "rescript/lib/es6/caml_option.js";
@@ -26,6 +28,24 @@ import * as Caml_js_exceptions from "rescript/lib/es6/caml_js_exceptions.js";
 
 var getCacheRowCountFnName = "get_cache_row_count";
 
+function makeClient() {
+  return Postgres({
+              host: Env.Db.host,
+              port: Env.Db.port,
+              database: Env.Db.database,
+              username: Env.Db.user,
+              password: Env.Db.password,
+              ssl: Env.Db.ssl,
+              max: 2,
+              onnotice: Env.userLogLevel === "warn" || Env.userLogLevel === "error" ? undefined : (function (_str) {
+                    
+                  }),
+              transform: {
+                undefined: null
+              }
+            });
+}
+
 function makeCreateIndexQuery(tableName, indexFields, pgSchema) {
   var indexName = tableName + "_" + indexFields.join("_");
   var index = Belt_Array.map(indexFields, (function (idx) {
@@ -697,7 +717,7 @@ function make(sql, pgHost, pgSchema, pgPort, pgUser, pgDatabase, pgPassword, isH
   var psqlExecOptions = {
     env: psqlExecOptions_env
   };
-  var cacheDirPath = Path.resolve("..", ".envio", "cache");
+  var cacheDirPath = Path.resolve(".envio", "cache");
   var isInitialized = async function () {
     var envioTables = await sql.unsafe("SELECT table_schema FROM information_schema.tables WHERE table_schema = '" + pgSchema + "' AND (table_name = 'event_sync_state' OR table_name = '" + InternalTable.Chains.table.tableName + "');");
     return Utils.$$Array.notEmpty(envioTables);
@@ -1075,6 +1095,7 @@ var maxItemsPerQuery = 500;
 
 export {
   getCacheRowCountFnName ,
+  makeClient ,
   makeCreateIndexQuery ,
   makeCreateTableIndicesQuery ,
   makeCreateTableQuery ,

package/src/Types.ts

@@ -1,6 +1,6 @@
 export type Invalid = never;
 
-export type Address = string;
+export type Address = `0x${string}`;
 
 export type Logger = {
   readonly debug: (

package/src/UserContext.res

@@ -261,7 +261,6 @@ let handlerTraps: Utils.Proxy.traps<contextParams> = {
       )
 
     | "isPreload" => params.isPreload->Utils.magic
-    | "chains" => params.chains->Utils.magic
     | "chain" =>
       let chainId = params.item->Internal.getItemChainId
       params.chains->Utils.Dict.dangerouslyGetByIntNonOption(chainId)->Utils.magic

package/src/UserContext.res.mjs

@@ -148,8 +148,6 @@ var handlerTraps = {
         case "chain" :
             var chainId = Internal.getItemChainId(params.item);
             return params.chains[chainId];
-        case "chains" :
-            return params.chains;
         case "effect" :
             return initEffect(params);
         case "isPreload" :

package/src/Utils.res

@@ -32,6 +32,19 @@ module Object = {
 
   @val @scope("Object")
   external defineProperty: ('obj, string, propertyDescriptor<'a>) => 'obj = "defineProperty"
+
+  // Property descriptor with required value field (no boxing)
+  type enumerablePropertyDescriptor<'a> = {
+    enumerable: bool,
+    value: 'a,
+  }
+
+  @val @scope("Object")
+  external definePropertyWithValue: ('obj, string, enumerablePropertyDescriptor<'a>) => 'obj =
+    "defineProperty"
+
+  @val @scope("Object")
+  external createNullObject: (@as(json`null`) _, unit) => 'a = "create"
 }
 
 module Error = {
@@ -175,6 +188,15 @@ module Dict = {
     }
   `)
 
+  let unsafeDeleteUndefinedFieldsInPlace: 'a => unit = %raw(`(dict) => {
+      for (var key in dict) {
+        if (dict[key] === undefined) {
+          delete dict[key];
+        }
+      }
+    }
+  `)
+
   let updateImmutable: (
     dict<'a>,
     string,
@@ -472,6 +494,12 @@ module Schema = {
     ->(magic: S.t<Js.Json.t> => S.t<Js.Date.t>)
     ->S.preprocess(_ => {serializer: date => date->magic->Js.Date.toISOString})
 
+  // ClickHouse expects timestamps as numbers (milliseconds), not ISO strings
+  let clickHouseDate =
+    S.json(~validate=false)
+    ->(magic: S.t<Js.Json.t> => S.t<Js.Date.t>)
+    ->S.preprocess(_ => {serializer: date => date->magic->Js.Date.getTime})
+
   // When trying to serialize data to Json pg type, it will fail with
   // PostgresError: column "params" is of type json but expression is of type boolean
   // If there's bool or null on the root level. It works fine as object field values.

package/src/Utils.res.mjs

@@ -151,6 +151,14 @@ var deleteInPlace = ((dict, key) => {
       delete dict[key];
     });
 
+var unsafeDeleteUndefinedFieldsInPlace = ((dict) => {
+      for (var key in dict) {
+        if (dict[key] === undefined) {
+          delete dict[key];
+        }
+      }
+    });
+
 var updateImmutable = ((dict, key, value) => ({...dict, [key]: value}));
 
 var shallowCopy = ((dict) => ({...dict}));
@@ -172,6 +180,7 @@ var Dict = {
   size: size,
   isEmpty: isEmpty,
   deleteInPlace: deleteInPlace,
+  unsafeDeleteUndefinedFieldsInPlace: unsafeDeleteUndefinedFieldsInPlace,
   updateImmutable: updateImmutable,
   shallowCopy: shallowCopy,
   incrementByInt: incrementByInt
@@ -449,6 +458,14 @@ var dbDate = S$RescriptSchema.preprocess(S$RescriptSchema.json(false), (function
               };
       }));
 
+var clickHouseDate = S$RescriptSchema.preprocess(S$RescriptSchema.json(false), (function (param) {
+        return {
+                s: (function (date) {
+                    return date.getTime();
+                  })
+              };
+      }));
+
 function coerceToJsonPgType(schema) {
   return S$RescriptSchema.preprocess(schema, (function (s) {
                 var match = s.schema.t;
@@ -485,6 +502,7 @@ var Schema = {
   getNonOptionalFieldNames: getNonOptionalFieldNames,
   getCapitalizedFieldNames: getCapitalizedFieldNames,
   dbDate: dbDate,
+  clickHouseDate: clickHouseDate,
   coerceToJsonPgType: coerceToJsonPgType
 };
 

package/src/bindings/ClickHouse.res

@@ -98,6 +98,36 @@ let getClickHouseFieldType = (
   isNullable ? `Nullable(${baseType})` : baseType
 }
 
+// Creates an entity schema from table definition, using clickHouseDate for Date fields
+let makeClickHouseEntitySchema = (table: Table.table): S.t<Internal.entity> => {
+  S.schema(s => {
+    let dict = Js.Dict.empty()
+    table.fields->Belt.Array.forEach(field => {
+      switch field {
+      | Field(f) => {
+          let fieldName = f->Table.getDbFieldName
+          let fieldSchema = switch f.fieldType {
+          | Date => {
+              let dateSchema = Utils.Schema.clickHouseDate->S.toUnknown
+              if f.isNullable {
+                S.null(dateSchema)->S.toUnknown
+              } else if f.isArray {
+                S.array(dateSchema)->S.toUnknown
+              } else {
+                dateSchema
+              }
+            }
+          | _ => f.fieldSchema
+          }
+          dict->Js.Dict.set(fieldName, s.matches(fieldSchema))
+        }
+      | DerivedFrom(_) => () // Skip derived fields
+      }
+    })
+    dict->Utils.magic
+  })
+}
+
 let setCheckpointsOrThrow = async (client, ~batch: Batch.t, ~database: string) => {
   let checkpointsCount = batch.checkpointIds->Array.length
   if checkpointsCount === 0 {
@@ -154,7 +184,7 @@ let setUpdatesOrThrow = async (
           )}\``,
         convertOrThrow: S.compile(
           S.union([
-            EntityHistory.makeSetUpdateSchema(entityConfig.schema),
+            EntityHistory.makeSetUpdateSchema(makeClickHouseEntitySchema(entityConfig.table)),
             S.object(s => {
               s.tag(EntityHistory.changeFieldName, EntityHistory.RowAction.DELETE)
               Change.Delete({

package/src/bindings/ClickHouse.res.mjs

@@ -76,6 +76,31 @@ function getClickHouseFieldType(fieldType, isNullable, isArray) {
   }
 }
 
+function makeClickHouseEntitySchema(table) {
+  return S$RescriptSchema.schema(function (s) {
+              var dict = {};
+              Belt_Array.forEach(table.fields, (function (field) {
+                      if (field.TAG !== "Field") {
+                        return ;
+                      }
+                      var f = field._0;
+                      var fieldName = Table.getDbFieldName(f);
+                      var match = f.fieldType;
+                      var fieldSchema;
+                      if (typeof match !== "object" && match === "Date") {
+                        var dateSchema = Utils.Schema.clickHouseDate;
+                        fieldSchema = f.isNullable ? S$RescriptSchema.$$null(dateSchema) : (
+                            f.isArray ? S$RescriptSchema.array(dateSchema) : dateSchema
+                          );
+                      } else {
+                        fieldSchema = f.fieldSchema;
+                      }
+                      dict[fieldName] = s.m(fieldSchema);
+                    }));
+              return dict;
+            });
+}
+
 async function setCheckpointsOrThrow(client, batch, database) {
   var checkpointsCount = batch.checkpointIds.length;
   if (checkpointsCount === 0) {
@@ -120,7 +145,7 @@ async function setUpdatesOrThrow(client, updates, entityConfig, database) {
   } else {
     var cache_tableName = database + ".\`" + EntityHistory.historyTableName(entityConfig.name, entityConfig.index) + "\`";
     var cache_convertOrThrow = S$RescriptSchema.compile(S$RescriptSchema.union([
-              EntityHistory.makeSetUpdateSchema(entityConfig.schema),
+              EntityHistory.makeSetUpdateSchema(makeClickHouseEntitySchema(entityConfig.table)),
               S$RescriptSchema.object(function (s) {
                     s.tag(EntityHistory.changeFieldName, "DELETE");
                     return {
@@ -264,6 +289,7 @@ async function resume(client, database, checkpointId) {
 
 export {
   getClickHouseFieldType ,
+  makeClickHouseEntitySchema ,
   setCheckpointsOrThrow ,
   setUpdatesOrThrow ,
   makeCreateHistoryTableQuery ,

package/src/bindings/Ethers.res

@@ -1,16 +1,3 @@
-type abi = EvmTypes.Abi.t
-
-let makeAbi = (abi: Js.Json.t): abi => abi->Utils.magic
-
-@deprecated("Use Address.t instead. The type will be removed in v3")
-type ethAddress = Address.t
-@deprecated("Use Address.Evm.fromStringOrThrow instead. The function will be removed in v3")
-let getAddressFromStringUnsafe = Address.Evm.fromStringOrThrow
-@deprecated("Use Address.toString instead. The function will be removed in v3")
-let ethAddressToString = Address.toString
-@deprecated("Use Address.schema instead. The function will be removed in v3")
-let ethAddressSchema = Address.schema
-
 type txHash = string
 
 module Constants = {
@@ -18,35 +5,6 @@ module Constants = {
   @module("ethers") @scope("ethers") external zeroAddress: Address.t = "ZeroAddress"
 }
 
-module Addresses = {
-  @genType
-  let mockAddresses = [
-    "0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266",
-    "0x70997970C51812dc3A010C7d01b50e0d17dc79C8",
-    "0x3C44CdDdB6a900fa2b585dd299e03d12FA4293BC",
-    "0x90F79bf6EB2c4f870365E785982E1f101E93b906",
-    "0x15d34AAf54267DB7D7c367839AAf71A00a2C6A65",
-    "0x9965507D1a55bcC2695C58ba16FB37d819B0A4dc",
-    "0x976EA74026E726554dB657fA54763abd0C3a0aa9",
-    "0x14dC79964da2C08b23698B3D3cc7Ca32193d9955",
-    "0x23618e81E3f5cdF7f54C3d65f7FBc0aBf5B21E8f",
-    "0xa0Ee7A142d267C1f36714E4a8F75612F20a79720",
-    "0xBcd4042DE499D14e55001CcbB24a551F3b954096",
-    "0x71bE63f3384f5fb98995898A86B02Fb2426c5788",
-    "0xFABB0ac9d68B0B445fB7357272Ff202C5651694a",
-    "0x1CBd3b2770909D4e10f157cABC84C7264073C9Ec",
-    "0xdF3e18d64BC6A983f673Ab319CCaE4f1a57C7097",
-    "0xcd3B766CCDd6AE721141F452C550Ca635964ce71",
-    "0x2546BcD3c84621e976D8185a91A922aE77ECEc30",
-    "0xbDA5747bFD65F08deb54cb465eB87D40e51B197E",
-    "0xdD2FD4581271e230360230F9337D5c0430Bf44C0",
-    "0x8626f6940E2eb28930eFb4CeF49B2d1F2C9C1199",
-  ]->Belt.Array.map(getAddressFromStringUnsafe)
-  @genType
-  let defaultAddress =
-    mockAddresses[0]
-}
-
 module Filter = {
   type t
 }
@@ -144,41 +102,43 @@ module JsonRpcProvider = {
   @send
   external getLogs: (t, ~filter: Filter.t) => promise<array<log>> = "getLogs"
 
-  @send
-  external getTransaction: (t, ~transactionHash: string) => promise<transaction> = "getTransaction"
-
   let makeGetTransactionFields = (~getTransactionByHash, ~lowercaseAddresses: bool) => async (
     log: log,
-  ): promise<unknown> => {
-    let transaction = await getTransactionByHash(log.transactionHash)
+  ): promise<Internal.evmTransactionFields> => {
+    let transaction: Internal.evmTransactionFields = await getTransactionByHash(log.transactionHash)
     // Mutating should be fine, since the transaction isn't used anywhere else outside the function
     let fields: {..} = transaction->Obj.magic
 
-    // Make it compatible with HyperSync transaction fields
+    // RPC may return null for transactionIndex on pending transactions
     fields["transactionIndex"] = log.transactionIndex
-    fields["input"] = fields["data"]
 
     // NOTE: this is wasteful if these fields are not selected in the users config.
     //       There might be a better way to do this in the `makeThrowingGetEventTransaction` function rather based on the schema.
     //       However this is not extremely expensive and good enough for now (only on rpc sync also).
-    if lowercaseAddresses {
-      open Js.Nullable
-      switch fields["from"] {
-      | Value(from) => fields["from"] = from->Js.String2.toLowerCase
-      | Undefined => ()
-      | Null => ()
-      }
-      switch fields["to"] {
-      | Value(to) => fields["to"] = to->Js.String2.toLowerCase
-      | Undefined => ()
-      | Null => ()
-      }
-      switch fields["contractAddress"] {
-      | Value(contractAddress) =>
-        fields["contractAddress"] = contractAddress->Js.String2.toLowerCase
-      | Undefined => ()
-      | Null => ()
-      }
+    open Js.Nullable
+    switch fields["from"] {
+    | Value(from) =>
+      fields["from"] = lowercaseAddresses
+        ? from->Js.String2.toLowerCase->Address.unsafeFromString
+        : from->Address.Evm.fromStringOrThrow
+    | Undefined => ()
+    | Null => ()
+    }
+    switch fields["to"] {
+    | Value(to) =>
+      fields["to"] = lowercaseAddresses
+        ? to->Js.String2.toLowerCase->Address.unsafeFromString
+        : to->Address.Evm.fromStringOrThrow
+    | Undefined => ()
+    | Null => ()
+    }
+    switch fields["contractAddress"] {
+    | Value(contractAddress) =>
+      fields["contractAddress"] = lowercaseAddresses
+        ? contractAddress->Js.String2.toLowerCase->Address.unsafeFromString
+        : contractAddress->Address.Evm.fromStringOrThrow
+    | Undefined => ()
+    | Null => ()
     }
 
     fields->Obj.magic

package/src/bindings/Ethers.res.mjs

@@ -2,52 +2,10 @@
 
 import * as Ethers from "ethers";
 import * as Address from "../Address.res.mjs";
-import * as Belt_Array from "rescript/lib/es6/belt_Array.js";
-import * as Caml_array from "rescript/lib/es6/caml_array.js";
 import * as Caml_option from "rescript/lib/es6/caml_option.js";
 
-function makeAbi(abi) {
-  return abi;
-}
-
-var getAddressFromStringUnsafe = Address.Evm.fromStringOrThrow;
-
-function ethAddressToString(prim) {
-  return prim;
-}
-
 var Constants = {};
 
-var mockAddresses = Belt_Array.map([
-      "0xf39Fd6e51aad88F6F4ce6aB8827279cffFb92266",
-      "0x70997970C51812dc3A010C7d01b50e0d17dc79C8",
-      "0x3C44CdDdB6a900fa2b585dd299e03d12FA4293BC",
-      "0x90F79bf6EB2c4f870365E785982E1f101E93b906",
-      "0x15d34AAf54267DB7D7c367839AAf71A00a2C6A65",
-      "0x9965507D1a55bcC2695C58ba16FB37d819B0A4dc",
-      "0x976EA74026E726554dB657fA54763abd0C3a0aa9",
-      "0x14dC79964da2C08b23698B3D3cc7Ca32193d9955",
-      "0x23618e81E3f5cdF7f54C3d65f7FBc0aBf5B21E8f",
-      "0xa0Ee7A142d267C1f36714E4a8F75612F20a79720",
-      "0xBcd4042DE499D14e55001CcbB24a551F3b954096",
-      "0x71bE63f3384f5fb98995898A86B02Fb2426c5788",
-      "0xFABB0ac9d68B0B445fB7357272Ff202C5651694a",
-      "0x1CBd3b2770909D4e10f157cABC84C7264073C9Ec",
-      "0xdF3e18d64BC6A983f673Ab319CCaE4f1a57C7097",
-      "0xcd3B766CCDd6AE721141F452C550Ca635964ce71",
-      "0x2546BcD3c84621e976D8185a91A922aE77ECEc30",
-      "0xbDA5747bFD65F08deb54cb465eB87D40e51B197E",
-      "0xdD2FD4581271e230360230F9337D5c0430Bf44C0",
-      "0x8626f6940E2eb28930eFb4CeF49B2d1F2C9C1199"
-    ], getAddressFromStringUnsafe);
-
-var defaultAddress = Caml_array.get(mockAddresses, 0);
-
-var Addresses = {
-  mockAddresses: mockAddresses,
-  defaultAddress: defaultAddress
-};
-
 var Filter = {};
 
 function toFilter(combinedFilter) {
@@ -81,26 +39,23 @@ function makeGetTransactionFields(getTransactionByHash, lowercaseAddresses) {
   return async function (log) {
     var transaction = await getTransactionByHash(log.transactionHash);
     transaction.transactionIndex = log.transactionIndex;
-    transaction.input = transaction.data;
-    if (lowercaseAddresses) {
-      var from = transaction.from;
-      if (from === null || from === undefined) {
-        from === null;
-      } else {
-        transaction.from = from.toLowerCase();
-      }
-      var to = transaction.to;
-      if (to === null || to === undefined) {
-        to === null;
-      } else {
-        transaction.to = to.toLowerCase();
-      }
-      var contractAddress = transaction.contractAddress;
-      if (contractAddress === null || contractAddress === undefined) {
-        contractAddress === null;
-      } else {
-        transaction.contractAddress = contractAddress.toLowerCase();
-      }
+    var from = transaction.from;
+    if (from === null || from === undefined) {
+      from === null;
+    } else {
+      transaction.from = lowercaseAddresses ? from.toLowerCase() : Address.Evm.fromStringOrThrow(from);
+    }
+    var to = transaction.to;
+    if (to === null || to === undefined) {
+      to === null;
+    } else {
+      transaction.to = lowercaseAddresses ? to.toLowerCase() : Address.Evm.fromStringOrThrow(to);
+    }
+    var contractAddress = transaction.contractAddress;
+    if (contractAddress === null || contractAddress === undefined) {
+      contractAddress === null;
+    } else {
+      transaction.contractAddress = lowercaseAddresses ? contractAddress.toLowerCase() : Address.Evm.fromStringOrThrow(contractAddress);
     }
     return transaction;
   };
@@ -112,19 +67,12 @@ var JsonRpcProvider = {
   makeGetTransactionFields: makeGetTransactionFields
 };
 
-var ethAddressSchema = Address.schema;
-
 export {
-  makeAbi ,
-  getAddressFromStringUnsafe ,
-  ethAddressToString ,
-  ethAddressSchema ,
   Constants ,
-  Addresses ,
   Filter ,
   CombinedFilter ,
   logToMinimumParseableLogData ,
   Network ,
   JsonRpcProvider ,
 }
-/* mockAddresses Not a pure module */
+/* ethers Not a pure module */

package/src/bindings/Postgres.gen.ts

@@ -0,0 +1,8 @@
+/* TypeScript file generated from Postgres.res by genType. */
+
+/* eslint-disable */
+/* tslint:disable */
+
+import type {Sql as $$sql} from 'postgres';
+
+export type sql = $$sql;

package/src/bindings/Postgres.res

@@ -1,3 +1,6 @@
+// Only needed for some old tests
+// Remove @genType in the future
+@genType.import(("postgres", "Sql"))
 type sql
 
 type undefinedTransform = | @as(undefined) Undefined | @as(null) Null

package/src/bindings/Postgres.res.d.mts

@@ -0,0 +1,5 @@
+// This is to prevent tsc --noEmit from failing
+// when importing code from .res.mjs files in genType .ts files
+// After we upgrade GenType and it starts to include ts-ignore,
+// the line can be removed.
+export type sql = any;

package/src/bindings/RescriptMocha.res

@@ -0,0 +1,123 @@
+module Assert = {
+  type assertion<'a> = ('a, 'a, ~message: string=?) => unit
+
+  @module("assert") external equal: assertion<'a> = "equal"
+  @module("assert") external notEqual: assertion<'a> = "notEqual"
+
+  @module("assert") external deepEqual: assertion<'a> = "deepEqual"
+  @module("assert")
+  external notDeepEqual: assertion<'a> = "notDeepEqual"
+
+  @module("assert") external strictEqual: assertion<'a> = "strictEqual"
+  @module("assert")
+  external notStrictEqual: assertion<'a> = "notStrictEqual"
+
+  @module("assert")
+  external deepStrictEqual: assertion<'a> = "deepStrictEqual"
+  @module("assert")
+  external notDeepStrictEqual: assertion<'a> = "notDeepStrictEqual"
+
+  @module("assert") external ifError: 'a => unit = "ifError"
+
+  @module("assert")
+  external throws: (unit => 'a, ~error: 'error=?, ~message: string=?) => unit = "throws"
+  @module("assert")
+  external doesNotThrow: (unit => 'a, ~error: 'error=?, ~message: string=?) => unit = "doesNotThrow"
+
+  @module("assert")
+  external rejects: (unit => promise<'a>, ~error: 'error=?, ~message: string=?) => promise<unit> =
+    "rejects"
+
+  @module("assert") external ok: (bool, ~message: string=?) => unit = "ok"
+  @module("assert") external fail: string => 'a = "fail"
+}
+
+/* Mocha bindings on `this` for `describe` and `it` functions */
+module This = {
+  @val external timeout: int => unit = "this.timeout"
+  @val external retries: int => unit = "this.retries"
+  @val external slow: int => unit = "this.slow"
+  @val external skip: unit => unit = "this.skip"
+}
+
+@val
+external describe: (string, unit => unit) => unit = "describe"
+@val
+external describe_only: (string, unit => unit) => unit = "describe.only"
+@val
+external describe_skip: (string, unit => unit) => unit = "describe.skip"
+
+@val
+external it: (string, unit => unit) => unit = "it"
+@val
+external it_only: (string, unit => unit) => unit = "it.only"
+@val
+external it_skip: (string, unit => unit) => unit = "it.skip"
+@val
+external before: (unit => unit) => unit = "before"
+@val
+external after: (unit => unit) => unit = "after"
+@val
+external beforeEach: (unit => unit) => unit = "beforeEach"
+@val
+external afterEach: (unit => unit) => unit = "afterEach"
+@val
+external beforeWithTitle: (string, unit => unit) => unit = "before"
+@val
+external afterWithTitle: (string, unit => unit) => unit = "after"
+@val
+external beforeEachWithTitle: (string, unit => unit) => unit = "beforeEach"
+@val
+external afterEachWithTitle: (string, unit => unit) => unit = "afterEach"
+
+module Async = {
+  @val
+  external it: (string, unit => promise<unit>) => unit = "it"
+  @val
+  external it_only: (string, unit => promise<unit>) => unit = "it.only"
+  @val
+  external it_skip: (string, unit => promise<unit>) => unit = "it.skip"
+  @val
+  external before: (unit => promise<unit>) => unit = "before"
+  @val
+  external after: (unit => promise<unit>) => unit = "after"
+  @val
+  external beforeEach: (unit => promise<unit>) => unit = "beforeEach"
+  @val
+  external afterEach: (unit => promise<unit>) => unit = "afterEach"
+  @val
+  external beforeWithTitle: (string, unit => promise<unit>) => unit = "before"
+  @val
+  external afterWithTitle: (string, unit => promise<unit>) => unit = "after"
+  @val
+  external beforeEachWithTitle: (string, unit => promise<unit>) => unit = "beforeEach"
+  @val
+  external afterEachWithTitle: (string, unit => promise<unit>) => unit = "afterEach"
+}
+
+module DoneCallback = {
+  type doneCallback = Js.Nullable.t<Js.Exn.t> => unit
+
+  @val
+  external it: (string, doneCallback => unit) => unit = "it"
+  @val
+  external it_only: (string, doneCallback => unit) => unit = "it.only"
+  @val
+  external it_skip: (string, doneCallback => unit) => unit = "it.skip"
+  @val
+  external before: (doneCallback => unit) => unit = "before"
+  @val
+  external after: (doneCallback => unit) => unit = "after"
+  @val
+  external beforeEach: (doneCallback => unit) => unit = "beforeEach"
+  @val
+  external afterEach: (doneCallback => unit) => unit = "afterEach"
+  @val
+  external beforeWithTitle: (string, doneCallback => unit) => unit = "before"
+  @val
+  external afterWithTitle: (string, doneCallback => unit) => unit = "after"
+  @val
+  external beforeEachWithTitle: (string, doneCallback => unit) => unit = "beforeEach"
+  @val
+  external afterEachWithTitle: (string, doneCallback => unit) => unit = "afterEach"
+}