ON-457: wip revoke role handler

OriumNetwork · Aug 30, 2023 · 588bf9a · 588bf9a
1 parent a9de8f2
commit 588bf9a
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 7 deletions.
diff --git a/src/erc7432/role/revoke-handler.ts b/src/erc7432/role/revoke-handler.ts
@@ -1,8 +1,31 @@
+import { log } from '@graphprotocol/graph-ts'
+import { RoleRevoked } from '../../../generated/Roles/ERC7432Roles'
+import { Nft, Role } from '../../../generated/schema'
+import { generateId } from '../../utils/helper'
+
 export function handleRoleRevoked(event: RoleRevoked): void {
-  let entity = Role.load(event.params.role.toHex())
+  const tokenId = event.params._tokenId.toString()
+  const address = event.address.toHexString()
+
+  const nftId = generateId(tokenId, address)
+  const nft = Nft.load(nftId)
+
+  if (!nft) {
+    log.info('NFT {} does not exist, skipping transfer...', [nftId])
+    return
+  }
+
+  const nftOwner = nft.owner
+
+  if (nftOwner != address) {
+    log.info('NFT {} is not owned by {}, skipping transfer...', [address])
+    return
+  }
+
+  if (event.address!) let entity = Role.load(event.params._role.toHex())
 
   if (entity == null) {
-    entity = new Role(event.params.role.toHex())
+    entity = new Role(event.params._role.toHex())
   }
 
   entity.account = event.params.account.toHex()

diff --git a/subgraph-goerli.yaml b/subgraph-goerli.yaml
@@ -31,7 +31,7 @@ dataSources:
     source:
       address: '0x8680fC3Df31805B1F7f8BfB702f723460d360FE9'
       abi: ERC7432Roles
-      startBlock: 8099655
+      startBlock: 9568669
     mapping:
       kind: ethereum/events
       apiVersion: 0.0.6
@@ -44,8 +44,8 @@ dataSources:
         - name: ERC7432Roles
           file: ./abis/ERC7432Roles.json
       eventHandlers:
-        - event: RoleGranted(indexed bytes32,indexed address,indexed uint256, address, uint64, bytes)
+        - event: RoleGranted(indexed bytes32,indexed address,indexed uint256,address,uint64,bytes)
           handler: handleRoleGranted
-        - event: RoleRevoked(indexed bytes32,indexed address,indexed uint256, address)
+        - event: RoleRevoked(indexed bytes32,indexed address,indexed uint256,address)
           handler: handleRoleRevoked
       file: ./src/erc7432/index.ts
diff --git a/subgraph-mumbai.yaml b/subgraph-mumbai.yaml
@@ -22,8 +22,8 @@ dataSources:
         - name: ERC7432Roles
           file: ./abis/ERC7432Roles.json
       eventHandlers:
-        - event: RoleGranted(indexed bytes32,indexed address,indexed uint256, address, uint64, bytes)
+        - event: RoleGranted(indexed bytes32,indexed address,indexed uint256,address,uint64,bytes)
           handler: handleRoleGranted
-        - event: RoleRevoked(indexed bytes32,indexed address,indexed uint256, address)
+        - event: RoleRevoked(indexed bytes32,indexed address,indexed uint256,address)
           handler: handleRoleRevoked
       file: ./src/erc7432/index.ts