agent: allow expired allocations to be reallocated with force

graphprotocol · Oct 23, 2024 · 3801b49 · 3801b49
1 parent c0933a7
commit 3801b49
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/packages/indexer-common/src/actions.ts b/packages/indexer-common/src/actions.ts
@@ -137,7 +137,15 @@ export const validateActionInputs = async (
       // allocationID must belong to active allocation
       // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
       const allocation = await networkMonitor.allocation(action.allocationID!)
-      if (allocation.status !== AllocationStatus.ACTIVE) {
+
+      const forcedReallocate =
+        action.type === ActionType.REALLOCATE &&
+        action.force === true &&
+        isZeroPOI(action.poi)
+
+      if (forcedReallocate) {
+        logger.info(`forcing reallocate of id = '${action.allocationID}'`)
+      } else if (allocation.status !== AllocationStatus.ACTIVE) {
         throw new Error(
           `An active allocation does not exist with id = '${action.allocationID}'`,
         )
@@ -151,6 +159,10 @@ export const validateActionInputs = async (
       }
     }
   }
+
+  function isZeroPOI(poi: string | undefined): boolean {
+    return (typeof poi == 'number' && poi == 0) || (typeof poi == 'string' && poi == '0')
+  }
 }
 
 export interface ActionFilter {