chore: manage dips in allocation actions

Author: pcarranzav

packages/indexer-common/src/indexer-management/allocations.ts

@@ -99,20 +99,22 @@ export type TransactionResult =
   | ActionFailure[]
 
 export class AllocationManager {
-  private dipsManager: DipsManager
+  private dipsManager: DipsManager | null = null
   constructor(
     private logger: Logger,
     private models: IndexerManagementModels,
     private graphNode: GraphNode,
     private network: Network,
   ) {
-    this.dipsManager = new DipsManager(
-      this.logger,
-      this.models,
-      this.graphNode,
-      this.network,
-      this,
-    )
+    if (this.network.specification.indexerOptions.dipperEndpoint) {
+      this.dipsManager = new DipsManager(
+        this.logger,
+        this.models,
+        this.graphNode,
+        this.network,
+        this,
+      )
+    }
   }
 
   async executeBatch(actions: Action[]): Promise<AllocationResult[]> {
@@ -521,6 +523,14 @@ export class AllocationManager {
       await upsertIndexingRule(logger, this.models, indexingRule)
     }
 
+    if (this.dipsManager) {
+      await this.dipsManager.tryUpdateAgreementAllocation(
+        deployment,
+        null,
+        createAllocationEventLogs.allocationID,
+      )
+    }
+
     return {
       actionID,
       type: 'allocate',
@@ -677,6 +687,15 @@ export class AllocationManager {
 
     await upsertIndexingRule(logger, this.models, neverIndexingRule)
 
+    if (this.dipsManager) {
+      await this.dipsManager.tryCancelAgreement(allocationID)
+      await this.dipsManager.tryUpdateAgreementAllocation(
+        allocation.subgraphDeployment.id.toString(),
+        allocationID,
+        null,
+      )
+    }
+
     return {
       actionID,
       type: 'unallocate',
@@ -976,6 +995,14 @@ export class AllocationManager {
       await upsertIndexingRule(logger, this.models, indexingRule)
     }
 
+    if (this.dipsManager) {
+      await this.dipsManager.tryUpdateAgreementAllocation(
+        subgraphDeploymentID.toString(),
+        allocationID,
+        createAllocationEventLogs.allocationID,
+      )
+    }
+
     return {
       actionID,
       type: 'reallocate',

packages/indexer-common/src/indexing-fees/dips.ts

@@ -69,13 +69,13 @@ export class DipsManager {
   }
   // Update the current and last allocation ids for an agreement if it exists
   async tryUpdateAgreementAllocation(
-    oldAllocationId: string,
+    deploymentId: string,
+    oldAllocationId: string | null,
     newAllocationId: string | null,
   ) {
     const agreement = await this.models.IndexingAgreement.findOne({
       where: {
-        current_allocation_id: oldAllocationId,
-        cancelled_at: null,
+        subgraph_deployment_id: deploymentId,
       },
     })
     if (agreement) {