Adding the 721 and 1155 contract impl to the Upgrade

ourzora · Jan 11, 2024 · dc2e364 · dc2e364
1 parent dd90785
commit dc2e364
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/schema.graphql b/schema.graphql
@@ -101,6 +101,7 @@ type Upgrade @entity {
 
   impl: Bytes
   version: String!
+  creatorContractImpl: Bytes
   type: String
 }
 

diff --git a/src/ERC1155Mappings/factoryMappings.ts b/src/ERC1155Mappings/factoryMappings.ts
@@ -114,6 +114,9 @@ export function handle1155FactoryUpgraded(event: Upgraded): void {
   upgrade.timestamp = event.block.timestamp;
   upgrade.impl = event.params.implementation;
   upgrade.version = creator.contractVersion();
+  // zora1155Impl didn't exist on all versions of the contract, so this
+  // handles the case it doesnt exist.
+  upgrade.creatorContractImpl = creator.try_zora1155Impl().value;
   upgrade.address = event.address;
   upgrade.type = "1155Factory";
 

diff --git a/src/ERC721Mappings/ERC721FactoryMappings.ts b/src/ERC721Mappings/ERC721FactoryMappings.ts
@@ -87,6 +87,7 @@ export function handleFactoryUpgraded(event: Upgraded): void {
   upgrade.timestamp = event.block.timestamp;
   upgrade.impl = event.params.implementation;
   upgrade.version = creator.contractVersion().toString();
+  upgrade.creatorContractImpl = creator.implementation();
   upgrade.address = event.address;
   upgrade.type = "721Factory";