replace dao ids

aragon · Jul 18, 2023 · 06ccbca · 06ccbca
1 parent a708fb4
commit 06ccbca
Show file tree

Hide file tree

Showing 4 changed files with 40 additions and 36 deletions.
diff --git a/packages/subgraph/src/plugin/pluginSetupProcessor.ts b/packages/subgraph/src/plugin/pluginSetupProcessor.ts
@@ -1,4 +1,5 @@
 import {Bytes, log} from '@graphprotocol/graph-ts';
+import {getDaoId} from '@aragon/subgraph-commons';
 import {
   InstallationApplied,
   InstallationPrepared,
@@ -20,7 +21,7 @@ import {
 } from './utils';
 
 export function handleInstallationPrepared(event: InstallationPrepared): void {
-  let dao = event.params.dao.toHexString();
+  let daoId = getDaoId(event.params.dao);
   let plugin = event.params.plugin.toHexString();
   let setupId = event.params.preparedSetupId.toHexString();
   let pluginRepo = event.params.pluginSetupRepo.toHexString();
@@ -30,9 +31,9 @@ export function handleInstallationPrepared(event: InstallationPrepared): void {
     event.params.versionTag.build
   );
 
-  let installationId = getPluginInstallationId(dao, plugin);
+  let installationId = getPluginInstallationId(daoId, plugin);
   if (!installationId) {
-    log.error('Failed to get installationId', [dao, plugin]);
+    log.error('Failed to get installationId', [daoId, plugin]);
     return;
   }
 
@@ -46,7 +47,7 @@ export function handleInstallationPrepared(event: InstallationPrepared): void {
   let preparationEntity = new PluginPreparation(preparationId);
   preparationEntity.installation = installationId.toHexString();
   preparationEntity.creator = event.params.sender;
-  preparationEntity.dao = dao;
+  preparationEntity.dao = daoId;
   preparationEntity.preparedSetupId = event.params.preparedSetupId;
   preparationEntity.pluginRepo = event.params.pluginSetupRepo.toHexString();
   preparationEntity.pluginVersion = pluginVersionId;
@@ -77,26 +78,26 @@ export function handleInstallationPrepared(event: InstallationPrepared): void {
     pluginEntity = new PluginInstallation(installationId.toHexString());
   }
   pluginEntity.state = 'InstallationPrepared';
-  pluginEntity.dao = dao;
+  pluginEntity.dao = daoId;
   pluginEntity.save();
 
-  addPlugin(dao, event.params.plugin);
+  addPlugin(daoId, event.params.plugin);
 }
 
 export function handleInstallationApplied(event: InstallationApplied): void {
-  let dao = event.params.dao.toHexString();
+  let daoId = getDaoId(event.params.dao);
   let plugin = event.params.plugin.toHexString();
-  let installationId = getPluginInstallationId(dao, plugin);
+  let installationId = getPluginInstallationId(daoId, plugin);
   if (!installationId) {
-    log.error('Failed to get installationId', [dao, plugin]);
+    log.error('Failed to get installationId', [daoId, plugin]);
     return;
   }
   let preparationId = `${installationId.toHexString()}_${event.params.preparedSetupId.toHexString()}`;
 
   let pluginEntity = PluginInstallation.load(installationId.toHexString());
   if (!pluginEntity) {
     pluginEntity = new PluginInstallation(installationId.toHexString());
-    pluginEntity.dao = dao;
+    pluginEntity.dao = daoId;
   }
 
   let pluginPreparationEntity = PluginPreparation.load(preparationId);
@@ -112,7 +113,7 @@ export function handleInstallationApplied(event: InstallationApplied): void {
 }
 
 export function handleUpdatePrepared(event: UpdatePrepared): void {
-  let dao = event.params.dao.toHexString();
+  let daoId = getDaoId(event.params.dao);
   let plugin = event.params.setupPayload.plugin.toHexString();
   let setupId = event.params.preparedSetupId.toHexString();
   let pluginRepo = event.params.pluginSetupRepo.toHexString();
@@ -123,9 +124,9 @@ export function handleUpdatePrepared(event: UpdatePrepared): void {
     event.params.versionTag.build
   );
 
-  let installationId = getPluginInstallationId(dao, plugin);
+  let installationId = getPluginInstallationId(daoId, plugin);
   if (!installationId) {
-    log.error('Failed to get installationId', [dao, plugin]);
+    log.error('Failed to get installationId', [daoId, plugin]);
     return;
   }
 
@@ -139,7 +140,7 @@ export function handleUpdatePrepared(event: UpdatePrepared): void {
   let preparationEntity = new PluginPreparation(preparationId);
   preparationEntity.installation = installationId.toHexString();
   preparationEntity.creator = event.params.sender;
-  preparationEntity.dao = dao;
+  preparationEntity.dao = daoId;
   preparationEntity.preparedSetupId = event.params.preparedSetupId;
   preparationEntity.pluginRepo = event.params.pluginSetupRepo.toHexString();
   preparationEntity.pluginVersion = pluginVersionId;
@@ -168,27 +169,27 @@ export function handleUpdatePrepared(event: UpdatePrepared): void {
   let pluginEntity = PluginInstallation.load(installationId.toHexString());
   if (!pluginEntity) {
     pluginEntity = new PluginInstallation(installationId.toHexString());
-    pluginEntity.dao = dao;
+    pluginEntity.dao = daoId;
   }
 
   pluginEntity.state = 'UpdatePrepared';
   pluginEntity.save();
 }
 
 export function handleUpdateApplied(event: UpdateApplied): void {
-  let dao = event.params.dao.toHexString();
+  let daoId = getDaoId(event.params.dao);
   let plugin = event.params.plugin.toHexString();
-  let installationId = getPluginInstallationId(dao, plugin);
+  let installationId = getPluginInstallationId(daoId, plugin);
   if (!installationId) {
-    log.error('Failed to get installationId', [dao, plugin]);
+    log.error('Failed to get installationId', [daoId, plugin]);
     return;
   }
   let preparationId = `${installationId.toHexString()}_${event.params.preparedSetupId.toHexString()}`;
 
   let pluginEntity = PluginInstallation.load(installationId.toHexString());
   if (!pluginEntity) {
     pluginEntity = new PluginInstallation(installationId.toHexString());
-    pluginEntity.dao = dao;
+    pluginEntity.dao = daoId;
   }
 
   let pluginPreparationEntity = PluginPreparation.load(preparationId);
@@ -202,13 +203,13 @@ export function handleUpdateApplied(event: UpdateApplied): void {
   pluginEntity.state = 'Installed';
   pluginEntity.save();
 
-  addPlugin(dao, event.params.plugin);
+  addPlugin(daoId, event.params.plugin);
 }
 
 export function handleUninstallationPrepared(
   event: UninstallationPrepared
 ): void {
-  let dao = event.params.dao.toHexString();
+  let daoId = getDaoId(event.params.dao);
   let plugin = event.params.setupPayload.plugin.toHexString();
   let setupId = event.params.preparedSetupId.toHexString();
   let pluginRepo = event.params.pluginSetupRepo.toHexString();
@@ -219,9 +220,9 @@ export function handleUninstallationPrepared(
     event.params.versionTag.build
   );
 
-  let installationId = getPluginInstallationId(dao, plugin);
+  let installationId = getPluginInstallationId(daoId, plugin);
   if (!installationId) {
-    log.error('Failed to get installationId', [dao, plugin]);
+    log.error('Failed to get installationId', [daoId, plugin]);
     return;
   }
 
@@ -230,7 +231,7 @@ export function handleUninstallationPrepared(
   let preparationEntity = new PluginPreparation(preparationId);
   preparationEntity.installation = installationId.toHexString();
   preparationEntity.creator = event.params.sender;
-  preparationEntity.dao = dao;
+  preparationEntity.dao = daoId;
   preparationEntity.preparedSetupId = event.params.preparedSetupId;
   preparationEntity.pluginRepo = event.params.pluginSetupRepo.toHexString();
   preparationEntity.pluginVersion = pluginVersionId;
@@ -258,7 +259,7 @@ export function handleUninstallationPrepared(
   let pluginEntity = PluginInstallation.load(installationId.toHexString());
   if (!pluginEntity) {
     pluginEntity = new PluginInstallation(installationId.toHexString());
-    pluginEntity.dao = dao;
+    pluginEntity.dao = daoId;
   }
   pluginEntity.state = 'UninstallPrepared';
   pluginEntity.save();
@@ -267,19 +268,19 @@ export function handleUninstallationPrepared(
 export function handleUninstallationApplied(
   event: UninstallationApplied
 ): void {
-  let dao = event.params.dao.toHexString();
+  let daoId = getDaoId(event.params.dao);
   let plugin = event.params.plugin.toHexString();
-  let installationId = getPluginInstallationId(dao, plugin);
+  let installationId = getPluginInstallationId(daoId, plugin);
   if (!installationId) {
-    log.error('Failed to get installationId', [dao, plugin]);
+    log.error('Failed to get installationId', [daoId, plugin]);
     return;
   }
   let preparationId = `${installationId.toHexString()}_${event.params.preparedSetupId.toHexString()}`;
 
   let pluginEntity = PluginInstallation.load(installationId.toHexString());
   if (!pluginEntity) {
     pluginEntity = new PluginInstallation(installationId.toHexString());
-    pluginEntity.dao = dao;
+    pluginEntity.dao = daoId;
   }
   pluginEntity.appliedPreparation = preparationId;
   pluginEntity.state = 'Uninstalled';

diff --git a/packages/subgraph/src/utils/tokens/erc20.ts b/packages/subgraph/src/utils/tokens/erc20.ts
@@ -1,4 +1,5 @@
 import {Address, BigInt, Bytes, Value, ethereum} from '@graphprotocol/graph-ts';
+import {getDaoId} from '@aragon/subgraph-commons';
 import {
   ERC20Balance,
   ERC20Contract,
@@ -108,7 +109,7 @@ export function updateERC20Balance(
     return;
   }
 
-  let daoId = dao.toHexString();
+  let daoId = getDaoId(dao);
   let balanceId = daoId.concat('_').concat(token.toHexString());
 
   let erc20Balance = ERC20Balance.load(balanceId);
@@ -175,7 +176,7 @@ export function handleERC20Action(
     amount = tuple[2].toBigInt();
   }
 
-  let daoId = dao.toHexString();
+  let daoId = getDaoId(dao);
 
   let id = getTransferId(
     event.transaction.hash,
@@ -231,7 +232,7 @@ export function handleERC20Deposit(
     return;
   }
 
-  let daoId = dao.toHexString();
+  let daoId = getDaoId(dao);
 
   let id = getTransferId(event.transaction.hash, event.transactionLogIndex, 0);
 

diff --git a/packages/subgraph/src/utils/tokens/erc721.ts b/packages/subgraph/src/utils/tokens/erc721.ts
@@ -1,4 +1,5 @@
 import {Address, BigInt, Bytes, ethereum} from '@graphprotocol/graph-ts';
+import {getDaoId} from '@aragon/subgraph-commons';
 import {
   ERC721Balance,
   ERC721Contract,
@@ -103,7 +104,7 @@ export function handleERC721Received(
   let from = tuple[1].toAddress();
   let tokenId = tuple[2].toBigInt();
 
-  let daoId = dao.toHexString();
+  let daoId = getDaoId(dao);
 
   updateERC721Balance(
     daoId,
@@ -173,7 +174,7 @@ export function handleERC721Action(
   let to = tuple[1].toAddress();
   let tokenId = tuple[2].toBigInt();
 
-  let daoId = dao.toHexString();
+  let daoId = getDaoId(dao);
 
   let transferId = getTransferId(
     event.transaction.hash,

diff --git a/packages/subgraph/src/utils/tokens/eth.ts b/packages/subgraph/src/utils/tokens/eth.ts
@@ -1,4 +1,5 @@
 import {Address, BigInt, ethereum} from '@graphprotocol/graph-ts';
+import {getDaoId} from '@aragon/subgraph-commons';
 import {NativeBalance, NativeTransfer} from '../../../generated/schema';
 import {ADDRESS_ZERO} from '../constants';
 import {getTransferId, TransferType} from './common';
@@ -33,7 +34,7 @@ export function handleNativeDeposit(
   reference: string,
   event: ethereum.Event
 ): void {
-  let daoId = dao.toHexString();
+  let daoId = getDaoId(dao);
 
   let id = getTransferId(event.transaction.hash, event.transactionLogIndex, 0);
 
@@ -69,7 +70,7 @@ export function handleNativeAction(
   actionIndex: number,
   event: ethereum.Event
 ): void {
-  let daoId = dao.toHexString();
+  let daoId = getDaoId(dao);
 
   let id = getTransferId(
     event.transaction.hash,