schema: implement task handlers

schema.graphql

@@ -75,6 +75,8 @@ type Task @entity {
   implementation: Implementation!
   environment: Environment!
   smartVault: SmartVault!
+  nextBalanceConnector: String!
+  previousBalanceConnector: String!
 }
 
 type Movement @entity {

src/Deployer.ts

@@ -66,6 +66,8 @@ export function handleTaskDeployed(event: TaskDeployed): void {
   task.implementation = implementation.id
   task.environment = environment.id
   task.smartVault = getSmartVault(event.params.instance).toHexString()
+  task.previousBalanceConnector = '0x0000000000000000000000000000000000000000000000000000000000000000'
+  task.nextBalanceConnector = '0x0000000000000000000000000000000000000000000000000000000000000000'
   task.save()
 }
 

src/Task.ts

@@ -1,7 +1,19 @@
 import { Address, log } from '@graphprotocol/graph-ts'
 
+import { Task } from '../types/schema'
+
+import { BalanceConnectorsSet } from '../types/templates/Task/Task'
 import { Task as TaskContract } from '../types/templates/Task/Task'
 
+export function handleBalanceConnectorsSet(event: BalanceConnectorsSet): void {
+  let task = Task.load(event.address.toHexString())
+  if (task == null) return log.warning('Missing task entity {}', [event.address.toHexString()])
+
+  task.previousBalanceConnector = event.params.previous.toHexString()
+  task.nextBalanceConnector = event.params.next.toHexString()
+  task.save()
+}
+
 export function getSmartVault(address: Address): Address {
   let taskContract = TaskContract.bind(address)
   let smartVaultCall = taskContract.try_smartVault()

subgraph.template.yaml

@@ -155,4 +155,7 @@ templates:
       abis:
         - name: Task
           file: ./node_modules/@mimic-fi/v3-tasks/artifacts/contracts/interfaces/ITask.sol/ITask.json
+      eventHandlers:
+        - event: BalanceConnectorsSet(indexed bytes32,indexed bytes32)
+          handler: handleBalanceConnectorsSet
       file: ./src/Task.ts