Merge pull request #71 from aavegotchi/SUB-93-handle-whitelist-owners…

…hip-transferred SUB-93: handle whitelist ownership transferred
aavegotchi · Oct 4, 2022 · e9bb263 · e9bb263
2 parents 4f1fa6b + bdb2d62
commit e9bb263
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 4 deletions.
diff --git a/abis/AavegotchiDiamond.json b/abis/AavegotchiDiamond.json
@@ -9008,6 +9008,25 @@
         "name": "WhitelistUpdated",
         "type": "event"
     },
+    {
+        "anonymous": false,
+        "inputs": [
+            {
+                "indexed": true,
+                "internalType": "uint32",
+                "name": "whitelistId",
+                "type": "uint32"
+            },
+            {
+                "indexed": true,
+                "internalType": "address",
+                "name": "newOwner",
+                "type": "address"
+            }
+        ],
+        "name": "WhitelistOwnershipTransferred",
+        "type": "event"
+    },
     {
         "inputs": [
             {

diff --git a/src/mappings/diamond.ts b/src/mappings/diamond.ts
@@ -50,6 +50,7 @@ import {
     GotchiLendingClaimed,
     GotchiLendingAdded,
     WhitelistAccessRightSet,
+    WhitelistOwnershipTransferred,
 } from "../../generated/AavegotchiDiamond/AavegotchiDiamond";
 import {
     getOrCreateUser,
@@ -1006,6 +1007,12 @@ export function handleWhitelistUpdated(event: WhitelistUpdated): void {
     createOrUpdateWhitelist(event.params.whitelistId, event);
 }
 
+export function handleWhitelistOwnershipTransferred(
+    event: WhitelistOwnershipTransferred
+): void {
+    createOrUpdateWhitelist(event.params.whitelistId, event);
+}
+
 export function handleGotchiLendingAdd(event: GotchiLendingAdd): void {
     if (event.block.number.gt(BLOCK_DISABLE_OLD_LENDING_EVENTS)) {
         return;

diff --git a/src/utils/helpers/diamond.ts b/src/utils/helpers/diamond.ts
@@ -747,13 +747,13 @@ export function createOrUpdateWhitelist(
     if (!whitelist) {
         whitelist = new Whitelist(id.toString());
         whitelist.maxBorrowLimit = 1;
-        whitelist.ownerAddress = result.owner;
-        let user = getOrCreateUser(result.owner.toHexString());
-        user.save();
-        whitelist.owner = user.id;
         whitelist.name = name;
     }
 
+    let user = getOrCreateUser(result.owner.toHexString());
+    user.save();
+    whitelist.owner = user.id;
+    whitelist.ownerAddress = result.owner;
     whitelist.members = members.map<Bytes>((e) => e);
 
     whitelist.save();

diff --git a/subgraph.yaml b/subgraph.yaml
@@ -103,6 +103,8 @@ dataSources:
                 handler: handleWhitelistCreated
               - event: WhitelistUpdated(indexed uint32)
                 handler: handleWhitelistUpdated
+              - event: WhitelistOwnershipTransferred(indexed uint32,indexed address)
+                handler: handleWhitelistOwnershipTransferred
               - event: GotchiLendingAdd(indexed uint32)
                 handler: handleGotchiLendingAdd
               - event: GotchiLendingClaim(indexed uint32,address[],uint256[])