Merge pull request #16 from OriumNetwork/ON-553-refactor

ON-553: refactor

schema.graphql

@@ -4,7 +4,7 @@ enum NftType {
 }
 
 type Nft @entity {
-  id: ID! # tokenId + tokenAddress
+  id: ID! # tokenAddress + tokenId
   tokenAddress: String!
   tokenId: BigInt!
   owner: Account!
@@ -22,7 +22,7 @@ type Account @entity {
 }
 
 type RoleAssignment @entity {
-  id: ID! # rolesRegistryAddress + grantorAddress + tokenId + tokenAddress + granteeAddress + roleHash
+  id: ID! # rolesRegistryAddress + grantorAddress + tokenAddress + tokenId + granteeAddress + roleHash
   role: Role!
   nft: Nft!
   grantor: Account!
@@ -35,7 +35,7 @@ type RoleAssignment @entity {
 }
 
 type Role @entity {
-  id: ID! # rolesRegistryAddress + tokenId + tokenAddress + roleHash
+  id: ID! # rolesRegistryAddress + tokenAddress + tokenId + roleHash
   rolesRegistry: RolesRegistry!
   roleHash: Bytes!
   nft: Nft!
@@ -48,10 +48,11 @@ type RoleApproval @entity {
   grantor: Account!
   operator: Account!
   tokenAddress: String!
+  isApproved: Boolean!
 }
 
 type RolesRegistry @entity {
   id: ID! # contractAddress
   roles: [Role!] @derivedFrom(field: "rolesRegistry")
   roleApprovals: [RoleApproval!] @derivedFrom(field: "rolesRegistry")
-}
+}

src/erc1155/transfer-batch-handler.ts → src/erc-1155/transfer-batch-handler.ts

@@ -17,7 +17,7 @@ export function handleTransferBatch(event: TransferBatch): void {
   const amounts = event.params.values
 
   if (tokenIds.length != amounts.length) {
-    log.error('[erc1155][handleTransferBatch] tokenIds length {} does not match amounts length {}, tx {}', [
+    log.error('[erc-1155][handleTransferBatch] tokenIds length {} does not match amounts length {}, tx {}', [
       tokenIds.length.toString(),
       amounts.length.toString(),
       event.transaction.hash.toHex(),

src/erc1155/transfer-single-handler.ts → src/erc-1155/transfer-single-handler.ts

@@ -18,7 +18,7 @@ export function handleTransferSingle(event: TransferSingle): void {
 
   const nft = upsertERC1155Nft(tokenAddress, tokenId, amount, fromAddress, toAddress)
 
-  log.warning('[erc1155][handleTransferSingle] NFT {} amount {} transferred from {} to {} tx {}', [
+  log.warning('[erc-1155][handleTransferSingle] NFT {} amount {} transferred from {} to {} tx {}', [
     nft.id,
     amount.toString(),
     fromAddress,

src/erc-721/index.ts

@@ -0,0 +1 @@
+export { handleTransfer } from './transfer-handler'

src/erc-721/transfer-handler.ts

@@ -0,0 +1,26 @@
+import { Transfer } from '../../generated/ERC721/ERC721'
+import { upsertERC721Nft } from '../../utils'
+import { log } from '@graphprotocol/graph-ts'
+
+/** 
+@dev This handler is called when a token is transferred.
+@param event Transfer The event emitted by the contract.
+
+Example:
+    event Transfer(address indexed from, address indexed to, uint256 indexed tokenId);
+*/
+export function handleTransfer(event: Transfer): void {
+  const tokenAddress = event.address.toHex()
+  const tokenId = event.params.tokenId
+  const from = event.params.from.toHex()
+  const to = event.params.to.toHex()
+
+  upsertERC721Nft(tokenAddress, tokenId, to)
+
+  log.warning('[erc-721][handleTransfer] NFT {} transferred from {} to {} tx {}', [
+    tokenId.toString(),
+    from,
+    to,
+    event.transaction.hash.toHex(),
+  ])
+}

src/erc-7432/index.ts

@@ -0,0 +1,3 @@
+export { handleRoleGranted } from './role-granted-handler'
+export { handleRoleRevoked } from './role-revoked-handler'
+export { handleRoleApprovalForAll } from './role-approval-for-all-handler'

src/erc-7432/role-approval-for-all-handler.ts

@@ -0,0 +1,30 @@
+import { RoleApprovalForAll } from '../../generated/ERC7432/ERC7432'
+import { findOrCreateRoleApproval, findOrCreateAccount, findOrCreateRolesRegistry } from '../../utils'
+import { log } from '@graphprotocol/graph-ts'
+
+/** 
+@dev This handler is called when a role approval for all is set.
+@param event RoleApprovalForAll The event emitted by the contract.
+
+Example:
+    event RoleApprovalForAll(address indexed _tokenAddress, address indexed _operator, bool _isApproved);
+*/
+export function handleRoleApprovalForAll(event: RoleApprovalForAll): void {
+  const rolesRegistryAddress = event.address.toHex()
+  const grantorAddress = event.transaction.from.toHex()
+  const operatorAddress = event.params._operator.toHex()
+  const tokenAddress = event.params._tokenAddress.toHex()
+  const isApproved = event.params._isApproved
+
+  const grantor = findOrCreateAccount(grantorAddress)
+  const operator = findOrCreateAccount(operatorAddress)
+  const rolesRegistry = findOrCreateRolesRegistry(rolesRegistryAddress)
+  const roleApproval = findOrCreateRoleApproval(rolesRegistry, grantor, operator, tokenAddress)
+  roleApproval.isApproved = isApproved
+  roleApproval.save()
+
+  log.warning('[erc-7432][handleRoleApprovalForAll] Updated RoleAssignment Approval: {} Tx: {}', [
+    roleApproval.id,
+    event.transaction.hash.toHex(),
+  ])
+}

src/erc-7432/role-granted-handler.ts

@@ -0,0 +1,61 @@
+import { log } from '@graphprotocol/graph-ts'
+import { RoleGranted } from '../../generated/ERC7432/ERC7432'
+import { Account, Nft } from '../../generated/schema'
+import { generateERC721NftId, findOrCreateAccount, upsertRoleAssignment } from '../../utils'
+
+/** 
+@dev This handler is called when a role is granted.
+@param event RoleGranted The event emitted by the contract.
+
+Example:
+    event RoleGranted(
+        bytes32 indexed _role,
+        address indexed _tokenAddress,
+        uint256 indexed _tokenId,
+        address _grantor,
+        address _grantee,
+        uint64 _expirationDate,
+        bool _revocable,
+        bytes _data
+    );
+*/
+export function handleRoleGranted(event: RoleGranted): void {
+  const tokenId = event.params._tokenId
+  const tokenAddress = event.params._tokenAddress.toHex()
+
+  const nftId = generateERC721NftId(tokenAddress, tokenId)
+  const nft = Nft.load(nftId)
+  if (!nft) {
+    log.error('[erc-7432][handleRoleGranted] NFT {} does not exist, tx {} skipping...', [
+      nftId,
+      event.transaction.hash.toHex(),
+    ])
+    return
+  }
+
+  const grantorAddress = event.params._grantor.toHex()
+  const grantorAccount = Account.load(grantorAddress)
+  if (!grantorAccount) {
+    log.error('[erc-7432][handleRoleGranted] grantor {} does not exist, tx {} skipping...', [
+      grantorAddress,
+      event.transaction.hash.toHex(),
+    ])
+    return
+  }
+  if (grantorAccount.id != nft.owner) {
+    log.error('[erc-7432][handleRoleGranted] NFT {} is not owned by {}, tx {} skipping...', [
+      nftId,
+      grantorAccount.id,
+      event.transaction.hash.toHex(),
+    ])
+    return
+  }
+
+  const granteeAccount = findOrCreateAccount(event.params._grantee.toHex())
+  const roleAssignment = upsertRoleAssignment(event, grantorAccount, granteeAccount, nft)
+  log.warning('[erc-7432][handleRoleGranted] roleAssignment: {} NFT: {} Tx: {}', [
+    roleAssignment.id,
+    nftId,
+    event.transaction.hash.toHex(),
+  ])
+}

src/erc-7432/role-revoked-handler.ts

@@ -0,0 +1,79 @@
+import { BigInt, log } from '@graphprotocol/graph-ts'
+import { RoleRevoked } from '../../generated/ERC7432/ERC7432'
+import { Account, Nft, RoleAssignment } from '../../generated/schema'
+import { findOrCreateRole, findOrCreateRolesRegistry, generateERC721NftId, generateRoleAssignmentId } from '../../utils'
+
+/** 
+@dev This handler is called when a role is revoked.
+@param event RoleRevoked The event emitted by the contract.
+
+Example:
+    event RoleRevoked(
+        bytes32 indexed _role,
+        address indexed _tokenAddress,
+        uint256 indexed _tokenId,
+        address _revoker,
+        address _grantee
+    );
+*/
+export function handleRoleRevoked(event: RoleRevoked): void {
+  const tokenId = event.params._tokenId
+  const tokenAddress = event.params._tokenAddress.toHexString()
+
+  const nftId = generateERC721NftId(tokenAddress, tokenId)
+  const nft = Nft.load(nftId)
+  if (!nft) {
+    log.error('[erc-7432][handleRoleRevoked] NFT {} does not exist, tx {} skipping...', [
+      nftId,
+      event.transaction.hash.toHex(),
+    ])
+    return
+  }
+
+  const revokerAddress = event.params._revoker.toHex()
+  const revoker = Account.load(revokerAddress)
+  if (!revoker) {
+    log.error('[erc-7432][handleRoleGranted] revoker {} does not exist, tx {} skipping...', [
+      revokerAddress,
+      event.transaction.hash.toHex(),
+    ])
+    return
+  }
+
+  const granteeAddress = event.params._grantee.toHex()
+  const grantee = Account.load(granteeAddress)
+  if (!grantee) {
+    log.error('[erc-7432][handleRoleGranted] grantee {} does not exist, tx {} skipping...', [
+      granteeAddress,
+      event.transaction.hash.toHex(),
+    ])
+    return
+  }
+
+  const rolesRegistry = findOrCreateRolesRegistry(event.address.toHex())
+  const roleAssignmentId = generateRoleAssignmentId(rolesRegistry, revoker, grantee, nft, event.params._role)
+  const roleAssignment = RoleAssignment.load(roleAssignmentId)
+  if (!roleAssignment) {
+    log.error('[erc-7432][handleRoleRevoked] RoleAssignment {} does not exist, tx {} skipping...', [
+      roleAssignmentId,
+      event.transaction.hash.toHex(),
+    ])
+    return
+  }
+  if (event.block.timestamp > roleAssignment.expirationDate) {
+    log.error('[erc-7432][handleRoleRevoked] RoleAssignment {} already expired, tx {} skipping...', [
+      roleAssignmentId,
+      event.transaction.hash.toHex(),
+    ])
+    return
+  }
+
+  roleAssignment.expirationDate = event.block.timestamp
+  roleAssignment.save()
+
+  log.warning('[[erc-7432]handleRoleRevoked] Revoked RoleAssignment: {} NFT: {} tx: {}', [
+    roleAssignmentId,
+    nftId,
+    event.transaction.hash.toHex(),
+  ])
+}