Librarify InMemoryStore (#816)

* Move resource to npm package

* fix tests

Author: DZakh

…tes/static/codegen/src/InMemoryStore.res …ator/cli/npm/envio/src/InMemoryStore.rescodegenerator/cli/templates/static/codegen/src/InMemoryStore.res renamed to codegenerator/cli/npm/envio/src/InMemoryStore.res codegenerator/cli/templates/static/codegen/src/InMemoryStore.res renamed to codegenerator/cli/npm/envio/src/InMemoryStore.res

@@ -8,7 +8,7 @@ let hashRawEventsKey = (key: rawEventsKey) =>
   EventUtils.getEventIdKeyString(~chainId=key.chainId, ~eventId=key.eventId)
 
 module EntityTables = {
-  type t = dict<InMemoryTable.Entity.t<Entities.internalEntity>>
+  type t = dict<InMemoryTable.Entity.t<Internal.entity>>
   exception UndefinedEntity({entityName: string})
   let make = (entities: array<Internal.entityConfig>): t => {
     let init = Js.Dict.empty()
@@ -22,7 +22,7 @@ module EntityTables = {
     switch self->Utils.Dict.dangerouslyGetNonOption(entityName) {
     | Some(table) =>
       table->(
-        Utils.magic: InMemoryTable.Entity.t<Entities.internalEntity> => InMemoryTable.Entity.t<
+        Utils.magic: InMemoryTable.Entity.t<Internal.entity> => InMemoryTable.Entity.t<
           entity,
         >
       )
@@ -51,13 +51,13 @@ type effectCacheInMemTable = {
 
 type t = {
   rawEvents: InMemoryTable.t<rawEventsKey, InternalTable.RawEvents.t>,
-  entities: dict<InMemoryTable.Entity.t<Entities.internalEntity>>,
+  entities: dict<InMemoryTable.Entity.t<Internal.entity>>,
   effects: dict<effectCacheInMemTable>,
   rollbackTargetCheckpointId: option<int>,
 }
 
 let make = (
-  ~entities: array<Internal.entityConfig>=Entities.allEntities,
+  ~entities: array<Internal.entityConfig>,
   ~rollbackTargetCheckpointId=?,
 ): t => {
   rawEvents: InMemoryTable.make(~hash=hashRawEventsKey),
@@ -106,7 +106,7 @@ let isRollingBack = (inMemoryStore: t) => inMemoryStore.rollbackTargetCheckpoint
 let setBatchDcs = (inMemoryStore: t, ~batch: Batch.t, ~shouldSaveHistory) => {
   let inMemTable =
     inMemoryStore->getInMemTable(
-      ~entityConfig=module(InternalTable.DynamicContractRegistry)->Entities.entityModToInternal,
+      ~entityConfig=InternalTable.DynamicContractRegistry.config,
     )
 
   let itemIdx = ref(0)
@@ -154,3 +154,4 @@ let setBatchDcs = (inMemoryStore: t, ~batch: Batch.t, ~shouldSaveHistory) => {
     itemIdx := itemIdx.contents + checkpointEventsProcessed
   }
 }
+

codegenerator/cli/templates/dynamic/codegen/src/TestHelpers_MockDb.res.hbs

@@ -272,7 +272,7 @@ let rec makeWithInMemoryStore: InMemoryStore.t => t = (inMemoryStore: InMemorySt
     ~getStore=db =>
       db
       ->InMemoryStore.getInMemTable(
-        ~entityConfig=module(InternalTable.DynamicContractRegistry)->Entities.entityModToInternal,
+        ~entityConfig=InternalTable.DynamicContractRegistry.config,
       )
       ->(
         Utils.magic: InMemoryTable.Entity.t<Internal.entity> => InMemoryTable.Entity.t<
@@ -403,7 +403,7 @@ and makeProcessEvents = (mockDb: t, ~chainId=?) => async (
     let mockDbClone = mockDb->cloneMockDb
 
     //Construct a new instance of an in memory store to run for the given event
-    let inMemoryStore = InMemoryStore.make()
+    let inMemoryStore = InMemoryStore.make(~entities=Entities.allEntities)
     let loadManager = LoadManager.make()
     let persistence = {
       ...indexer.persistence,
@@ -626,4 +626,4 @@ new state so you can compare states before and after.
 @genType
 let //Note: It's called createMockDb over "make" to make it more intuitive in JS and TS
 
-createMockDb = () => makeWithInMemoryStore(InMemoryStore.make())
+createMockDb = () => makeWithInMemoryStore(InMemoryStore.make(~entities=Entities.allEntities))

codegenerator/cli/templates/static/codegen/src/IO.res

@@ -327,7 +327,7 @@ let executeBatch = async (
 }
 
 let prepareRollbackDiff = async (~persistence: Persistence.t, ~rollbackTargetCheckpointId) => {
-  let inMemStore = InMemoryStore.make(~rollbackTargetCheckpointId)
+  let inMemStore = InMemoryStore.make(~entities=Entities.allEntities, ~rollbackTargetCheckpointId)
 
   let deletedEntities = Js.Dict.empty()
   let setEntities = Js.Dict.empty()

codegenerator/cli/templates/static/codegen/src/globalState/GlobalState.res

@@ -955,7 +955,7 @@ let injectedTaskReducer = (
         | _ => None
         }
 
-        let inMemoryStore = rollbackInMemStore->Option.getWithDefault(InMemoryStore.make())
+        let inMemoryStore = rollbackInMemStore->Option.getWithDefault(InMemoryStore.make(~entities=Entities.allEntities))
 
         inMemoryStore->InMemoryStore.setBatchDcs(~batch, ~shouldSaveHistory)
 

scenarios/erc20_multichain_factory/pnpm-lock.yaml

@@ -39,7 +39,7 @@ describe("Chains State", () => {
       async () => {
         // This test verifies that the chains field is accessible
         // The actual integration test is in EventHandlers.res with the EmptyEvent handler
-        let inMemoryStore = InMemoryStore.make()
+        let inMemoryStore = InMemoryStore.make(~entities=Entities.allEntities)
         let loadManager = LoadManager.make()
 
         let item = MockEvents.newGravatarLog1->MockEvents.newGravatarEventToBatchItem

scenarios/test_codegen/pnpm-lock.yaml

@@ -3,7 +3,7 @@ open RescriptMocha
 describe("LoadLayer", () => {
   Async.it("Trys to load non existing entity from db", async () => {
     let storageMock = Mock.Storage.make([#loadByIdsOrThrow])
-    let inMemoryStore = InMemoryStore.make()
+    let inMemoryStore = InMemoryStore.make(~entities=Entities.allEntities)
     let loadManager = LoadManager.make()
 
     let getUser = entityId =>
@@ -36,7 +36,7 @@ describe("LoadLayer", () => {
     async () => {
       let storageMock = Mock.Storage.make([#loadByIdsOrThrow])
       let loadManager = LoadManager.make()
-      let inMemoryStore = InMemoryStore.make()
+      let inMemoryStore = InMemoryStore.make(~entities=Entities.allEntities)
 
       let getUser = entityId =>
         LoadLayer.loadById(
@@ -75,7 +75,7 @@ describe("LoadLayer", () => {
     async () => {
       let storageMock = Mock.Storage.make([#loadByIdsOrThrow])
       let loadManager = LoadManager.make()
-      let inMemoryStore = InMemoryStore.make()
+      let inMemoryStore = InMemoryStore.make(~entities=Entities.allEntities)
       let getUser = entityId =>
         LoadLayer.loadById(
           ~loadManager,

scenarios/test_codegen/test/EventOrigin_test.res

@@ -20,7 +20,7 @@ module InMemoryStore = {
   }
 
   let make = (~entities=[]) => {
-    let inMemoryStore = InMemoryStore.make()
+    let inMemoryStore = InMemoryStore.make(~entities=Entities.allEntities)
     entities->Js.Array2.forEach(((entityMod, items)) => {
       items->Js.Array2.forEach(entity => {
         inMemoryStore->setEntity(~entityMod, entity)

scenarios/test_codegen/test/LoadLayer_test.res

@@ -34,7 +34,7 @@ describe("Load and save an entity with a BigDecimal from DB", () => {
       ~pgSchema=Generated.storagePgSchema,
     )
 
-    let inMemoryStore = InMemoryStore.make()
+    let inMemoryStore = InMemoryStore.make(~entities=Entities.allEntities)
     let loadManager = LoadManager.make()
 
     let item = MockEvents.newGravatarLog1->MockEvents.newGravatarEventToBatchItem