chore: add dips actions in allocations resolvers

graphprotocol · Feb 14, 2025 · 3427a3a · 3427a3a
1 parent 7b9061f
commit 3427a3a
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 4 deletions.
diff --git a/packages/indexer-common/src/indexer-management/allocations.ts b/packages/indexer-common/src/indexer-management/allocations.ts
@@ -99,7 +99,7 @@ export type TransactionResult =
   | ActionFailure[]
 
 export class AllocationManager {
-  private dipsManager: DipsManager | null = null
+  declare dipsManager: DipsManager | null
   constructor(
     private logger: Logger,
     private models: IndexerManagementModels,

diff --git a/packages/indexer-common/src/indexer-management/resolvers/allocations.ts b/packages/indexer-common/src/indexer-management/resolvers/allocations.ts
@@ -374,7 +374,7 @@ export default {
       amount: string
       protocolNetwork: string
     },
-    { network, graphNode, logger, models }: IndexerManagementResolverContext,
+    { network, graphNode, logger, models, actionManager }: IndexerManagementResolverContext,
   ): Promise<CreateAllocationResult> => {
     logger.debug('Execute createAllocation() mutation', {
       deployment,
@@ -571,6 +571,14 @@ export default {
 
       await models.IndexingRule.upsert(indexingRule)
 
+      if (actionManager?.allocationManager?.dipsManager) {
+        await actionManager.allocationManager.dipsManager.tryUpdateAgreementAllocation(
+          deployment,
+          null,
+          createAllocationEventLogs.allocationID,
+        )
+      }
+
       // Since upsert succeeded, we _must_ have a rule
       const updatedRule = await models.IndexingRule.findOne({
         where: { identifier: indexingRule.identifier },
@@ -608,7 +616,7 @@ export default {
       poi: string | undefined
       force: boolean
     },
-    { logger, models, network }: IndexerManagementResolverContext,
+    { logger, models, network, actionManager }: IndexerManagementResolverContext,
   ): Promise<CloseAllocationResult> => {
     logger.debug('Execute closeAllocation() mutation', {
       allocationID: allocation,
@@ -726,6 +734,17 @@ export default {
 
       await models.IndexingRule.upsert(offchainIndexingRule)
 
+      if (actionManager?.allocationManager?.dipsManager) {
+        await actionManager.allocationManager.dipsManager.tryCancelAgreement(
+          allocation,
+        )
+        await actionManager.allocationManager.dipsManager.tryUpdateAgreementAllocation(
+          allocationData.subgraphDeployment.id.toString(),
+          allocation,
+          null,
+        )
+      }
+
       // Since upsert succeeded, we _must_ have a rule
       const updatedRule = await models.IndexingRule.findOne({
         where: { identifier: offchainIndexingRule.identifier },
@@ -765,7 +784,7 @@ export default {
       force: boolean
       protocolNetwork: string
     },
-    { logger, models, network }: IndexerManagementResolverContext,
+    { logger, models, network, actionManager }: IndexerManagementResolverContext,
   ): Promise<ReallocateAllocationResult> => {
     logger = logger.child({
       component: 'reallocateAllocationResolver',
@@ -1032,6 +1051,14 @@ export default {
 
       await models.IndexingRule.upsert(indexingRule)
 
+      if (actionManager?.allocationManager?.dipsManager) {
+        await actionManager.allocationManager.dipsManager.tryUpdateAgreementAllocation(
+          allocationData.subgraphDeployment.id.toString(),
+          allocation,
+          createAllocationEventLogs.allocationID,
+        )
+      }
+
       // Since upsert succeeded, we _must_ have a rule
       const updatedRule = await models.IndexingRule.findOne({
         where: { identifier: indexingRule.identifier },