Adding the 721 and 1155 contract impl to the Upgrade

ourzora · Jan 11, 2024 · e8c87ea · e8c87ea
1 parent 7a4531e
commit e8c87ea
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/schema.graphql b/schema.graphql
@@ -100,8 +100,9 @@ type Upgrade @entity {
   # End – default data block
 
   impl: Bytes
-  version: String
-  type: String
+  version: String!
+  creatorContractImpl: Bytes
+  type: String!
 }
 
 type EditionMetadata @entity {

diff --git a/src/ERC1155Mappings/factoryMappings.ts b/src/ERC1155Mappings/factoryMappings.ts
@@ -113,9 +113,9 @@ export function handle1155FactoryUpgraded(event: Upgraded): void {
   upgrade.block = event.block.number;
   upgrade.timestamp = event.block.timestamp;
   upgrade.impl = event.params.implementation;
-  upgrade.version = creator.contractVersion()
+  upgrade.version = creator.contractVersion();
+  upgrade.creatorContractImpl = creator.try_zora1155Impl().value;
   upgrade.address = event.address;
-  upgrade.type = "1155Factory";
 
   // Save required factories.
   factory.txn = txn;

diff --git a/src/ERC721Mappings/ERC721FactoryMappings.ts b/src/ERC721Mappings/ERC721FactoryMappings.ts
@@ -86,6 +86,8 @@ export function handleFactoryUpgraded(event: Upgraded): void {
   upgrade.block = event.block.number;
   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";