feat: inprogress update of balance sheet managers (#289)

Fixes #277

Author: filo87

ponder.schema.ts

@@ -941,12 +941,19 @@ export const HoldingEscrowRelations = relations(HoldingEscrow, ({ one }) => ({
   }),
 }));
 
+export const PoolManagerCrosschainInProgressTypes = [`CanManage`, `CanNotManage`] as const;
+export const PoolManagerCrosschainInProgress = onchainEnum(
+  "pool_manager_crosschain_in_progress",
+  PoolManagerCrosschainInProgressTypes
+);
+
 const PoolManagerColumns = (t: PgColumnsBuilders) => ({
   address: t.hex().notNull(),
   centrifugeId: t.text().notNull(),
   poolId: t.bigint().notNull(),
   isHubManager: t.boolean().notNull().default(false),
   isBalancesheetManager: t.boolean().notNull().default(false),
+  crosschainInProgress: PoolManagerCrosschainInProgress("pool_manager_crosschain_in_progress"),
   ...defaultColumns(t),
 });
 

src/handlers/balanceSheetHandlers.ts

@@ -110,27 +110,29 @@ multiMapper("balanceSheet:UpdateManager", async ({ event, context }) => {
 
   const centrifugeId = await BlockchainService.getCentrifugeId(context);
 
-  const { who: manager, poolId, canManage } = event.args;
+  const { who: _manager, poolId, canManage } = event.args;
 
-  const account = (await AccountService.getOrInit(
+  const managerAddress = _manager.toLowerCase().substring(0, 42) as `0x${string}`;
+
+  const _account = (await AccountService.getOrInit(
     context,
     {
-      address: manager,
+      address: managerAddress,
     },
     event
   )) as AccountService;
 
-  const { address: managerAddress } = account.read();
-
   const poolManager = (await PoolManagerService.getOrInit(
     context,
     {
       address: managerAddress,
       centrifugeId,
       poolId,
     },
-    event
+    event,
+    undefined,
+    true
   )) as PoolManagerService;
-  poolManager.setIsBalancesheetManager(canManage);
-  await poolManager.save(event);
+
+  await poolManager.setCrosschainInProgress().setIsBalancesheetManager(canManage).save(event);
 });

src/handlers/hubHandlers.ts

@@ -1,6 +1,12 @@
 import { multiMapper } from "../helpers/multiMapper";
 import { logEvent, serviceError } from "../helpers/logger";
-import { WhitelistedInvestorService, TokenService, PoolSpokeBlockchainService } from "../services";
+import {
+  WhitelistedInvestorService,
+  TokenService,
+  PoolSpokeBlockchainService,
+  PoolManagerService,
+  AccountService,
+} from "../services";
 
 multiMapper("hub:NotifyPool", async ({ event, context }) => {
   logEvent(event, context, "hub:NotifyPool");
@@ -66,6 +72,30 @@ multiMapper("hub:UpdateRestriction", async ({ event, context }) => {
   }
 });
 
+multiMapper("hub:UpdateBalanceSheetManager", async ({ event, context }) => {
+  logEvent(event, context, "hub:UpdateBalanceSheetManager");
+  const { poolId, manager: _manager, canManage, centrifugeId: spokeCentrifugeId } = event.args;
+  const manager = _manager.toLowerCase().substring(0, 42) as `0x${string}`;
+
+  const _account = (await AccountService.getOrInit(
+    context,
+    {
+      address: manager,
+    },
+    event
+  )) as AccountService;
+
+  const poolManager = (await PoolManagerService.getOrInit(
+    context,
+    { poolId, centrifugeId: spokeCentrifugeId.toString(), address: manager },
+    event,
+    undefined,
+    true
+  )) as PoolManagerService;
+  poolManager.setCrosschainInProgress(canManage ? `CanManage` : `CanNotManage`);
+  await poolManager.save(event);
+});
+
 enum RestrictionType {
   "Invalid",
   "Member",

src/services/PoolManagerService.ts

@@ -1,5 +1,6 @@
-import { PoolManager } from "ponder:schema";
+import { PoolManager, PoolManagerCrosschainInProgressTypes } from "ponder:schema";
 import { Service, mixinCommonStatics } from "./Service";
+import { serviceLog } from "../helpers/logger";
 
 /**
  * Service class for managing PoolManager entities.
@@ -27,6 +28,7 @@ export class PoolManagerService extends mixinCommonStatics(
    * @returns The service instance for method chaining
    */
   public setIsHubManager(isHubManager: boolean) {
+    serviceLog(`Setting isHubManager to ${isHubManager}`);
     this.data.isHubManager = isHubManager;
     return this;
   }
@@ -38,7 +40,22 @@ export class PoolManagerService extends mixinCommonStatics(
    * @returns The service instance for method chaining
    */
   public setIsBalancesheetManager(isBalancesheetManager: boolean) {
+    serviceLog(`Setting isBalancesheetManager to ${isBalancesheetManager}`);
     this.data.isBalancesheetManager = isBalancesheetManager;
     return this;
   }
+
+  /**
+   * Sets the crosschain progress for the manager.
+   *
+   * @param crosschainProgress - The value to set for crosschainProgress
+   * @returns The service instance for method chaining
+   */
+  public setCrosschainInProgress(
+    crosschainInProgress?: (typeof PoolManagerCrosschainInProgressTypes)[number]
+  ) {
+    this.data.crosschainInProgress = crosschainInProgress ?? null;
+    serviceLog(`Setting crosschainInProgress to ${crosschainInProgress}`);
+    return this;
+  }
 }