Tasks: Track time lock config (#2)

Co-authored-by: Facu Spagnuolo <[email protected]>

schema.graphql

@@ -83,6 +83,9 @@ type Task @entity {
   priorityFeeLimit: BigInt!
   txCostLimitPct: BigInt!
   txCostLimit: BigInt!
+  timeLockDelay: BigInt!
+  timeLockExecutionPeriod: BigInt!
+  timeLockExpiration: BigInt!
 }
 
 type Movement @entity {

src/Deployer.ts

@@ -82,6 +82,9 @@ export function handleTaskDeployed(event: TaskDeployed): void {
   task.priorityFeeLimit = BigInt.zero()
   task.txCostLimitPct = BigInt.zero()
   task.txCostLimit = BigInt.zero()
+  task.timeLockDelay = BigInt.zero()
+  task.timeLockExecutionPeriod = BigInt.zero()
+  task.timeLockExpiration = BigInt.zero()
   task.save()
 
   TaskTemplate.create(event.params.instance)

src/Task.ts

@@ -8,6 +8,9 @@ import {
   PriorityFeeLimitSet,
   TxCostLimitPctSet,
   TxCostLimitSet, 
+  TimeLockDelaySet,
+  TimeLockExecutionPeriodSet,
+  TimeLockExpirationSet 
 } from '../types/templates/Task/Task'
 import { Task as TaskContract } from '../types/templates/Task/Task'
 
@@ -52,6 +55,30 @@ export function handleTxCostLimitSet(event: TxCostLimitSet): void {
   task.save()
 }
 
+export function handleTimeLockDelaySet(event : TimeLockDelaySet): void {
+  let task = Task.load(event.address.toHexString())
+  if (task == null) return log.warning('Missing task entity {}', [event.address.toHexString()])
+
+  task.timeLockDelay = event.params.delay
+  task.save()
+}
+
+export function handleTimeLockExecutionPeriodSet(event: TimeLockExecutionPeriodSet): void {
+  let task = Task.load(event.address.toHexString())
+  if (task == null) return log.warning('Missing task entity {}', [event.address.toHexString()])
+
+  task.timeLockExecutionPeriod = event.params.period
+  task.save()
+}
+
+export function handleTimeLockExpirationSet(event: TimeLockExpirationSet): void {
+  let task = Task.load(event.address.toHexString())
+  if (task == null) return log.warning('Missing task entity {}', [event.address.toHexString()])
+
+  task.timeLockExpiration = event.params.expiration
+  task.save()
+}
+
 export function getSmartVault(address: Address): Address {
   let taskContract = TaskContract.bind(address)
   let smartVaultCall = taskContract.try_smartVault()
@@ -87,3 +114,4 @@ export function getExecutionType(address: Address): Bytes {
   log.warning('EXECUTION_TYPE() call reverted for task {}', [address.toHexString()])
   return Bytes.fromUTF8('')
 }
+

subgraph.template.yaml

@@ -186,4 +186,10 @@ templates:
           handler: handleTxCostLimitPctSet
         - event: TxCostLimitSet(uint256)
           handler: handleTxCostLimitSet
+        - event: TimeLockDelaySet(uint256)
+          handler: handleTimeLockDelaySet
+        - event: TimeLockExecutionPeriodSet(uint256)
+          handler: handleTimeLockExecutionPeriodSet
+        - event: TimeLockExpirationSet(uint256)
+          handler: handleTimeLockExpirationSet
       file: ./src/Task.ts