Move proposal creation / voted event listeners to own hook

src/hooks/DAO/loaders/governance/useAzoriusProposalListeners.ts

@@ -0,0 +1,249 @@
+import { LinearERC20Voting, LinearERC721Voting } from '@fractal-framework/fractal-contracts';
+import { TypedListener } from '@fractal-framework/fractal-contracts/dist/typechain-types/common';
+import {
+  Azorius,
+  ProposalCreatedEvent,
+} from '@fractal-framework/fractal-contracts/dist/typechain-types/contracts/azorius/Azorius';
+import { VotedEvent as ERC20VotedEvent } from '@fractal-framework/fractal-contracts/dist/typechain-types/contracts/azorius/LinearERC20Voting';
+import { VotedEvent as ERC721VotedEvent } from '@fractal-framework/fractal-contracts/dist/typechain-types/contracts/azorius/LinearERC721Voting';
+import { BigNumber } from 'ethers';
+import { Dispatch, useEffect, useMemo } from 'react';
+import { useFractal } from '../../../../providers/App/AppProvider';
+import { FractalGovernanceAction } from '../../../../providers/App/governance/action';
+import { useEthersProvider } from '../../../../providers/Ethers/hooks/useEthersProvider';
+import {
+  ProposalMetadata,
+  VotingStrategyType,
+  DecodedTransaction,
+  FractalActions,
+} from '../../../../types';
+import { Providers } from '../../../../types/network';
+import {
+  getProposalVotesSummary,
+  mapProposalCreatedEventToProposal,
+  decodeTransactions,
+} from '../../../../utils';
+import useSafeContracts from '../../../safe/useSafeContracts';
+import { useSafeDecoder } from '../../../utils/useSafeDecoder';
+
+const proposalCreatedEventListener = (
+  azoriusContract: Azorius,
+  erc20StrategyContract: LinearERC20Voting | undefined,
+  erc721StrategyContract: LinearERC721Voting | undefined,
+  provider: Providers,
+  strategyType: VotingStrategyType,
+  decode: (value: string, to: string, data?: string | undefined) => Promise<DecodedTransaction[]>,
+  dispatch: Dispatch<FractalActions>,
+): TypedListener<ProposalCreatedEvent> => {
+  return async (_strategyAddress, proposalId, proposer, transactions, metadata) => {
+    console.log({ metadata });
+    if (!metadata) {
+      return;
+    }
+
+    const metaDataEvent: ProposalMetadata = JSON.parse(metadata);
+    const proposalData = {
+      metaData: {
+        title: metaDataEvent.title,
+        description: metaDataEvent.description,
+        documentationUrl: metaDataEvent.documentationUrl,
+      },
+      transactions: transactions,
+      decodedTransactions: await decodeTransactions(decode, transactions),
+    };
+
+    const proposal = await mapProposalCreatedEventToProposal(
+      erc20StrategyContract,
+      erc721StrategyContract,
+      strategyType,
+      proposalId,
+      proposer,
+      azoriusContract,
+      provider,
+      Promise.resolve(undefined),
+      Promise.resolve(undefined),
+      Promise.resolve(undefined),
+      proposalData,
+    );
+
+    console.log({ proposal });
+
+    dispatch({
+      type: FractalGovernanceAction.UPDATE_PROPOSALS_NEW,
+      payload: proposal,
+    });
+  };
+};
+
+const erc20VotedEventListener = (
+  erc20StrategyContract: LinearERC20Voting,
+  strategyType: VotingStrategyType,
+  dispatch: Dispatch<FractalActions>,
+): TypedListener<ERC20VotedEvent> => {
+  return async (voter, proposalId, voteType, weight) => {
+    const votesSummary = await getProposalVotesSummary(
+      erc20StrategyContract,
+      undefined,
+      strategyType,
+      BigNumber.from(proposalId),
+    );
+
+    dispatch({
+      type: FractalGovernanceAction.UPDATE_NEW_AZORIUS_ERC20_VOTE,
+      payload: {
+        proposalId: proposalId.toString(),
+        voter,
+        support: voteType,
+        weight,
+        votesSummary,
+      },
+    });
+  };
+};
+
+const erc721VotedEventListener = (
+  erc721StrategyContract: LinearERC721Voting,
+  strategyType: VotingStrategyType,
+  dispatch: Dispatch<FractalActions>,
+): TypedListener<ERC721VotedEvent> => {
+  return async (voter, proposalId, voteType, tokenAddresses, tokenIds) => {
+    const votesSummary = await getProposalVotesSummary(
+      undefined,
+      erc721StrategyContract,
+      strategyType,
+      BigNumber.from(proposalId),
+    );
+
+    dispatch({
+      type: FractalGovernanceAction.UPDATE_NEW_AZORIUS_ERC721_VOTE,
+      payload: {
+        proposalId: proposalId.toString(),
+        voter,
+        support: voteType,
+        tokenAddresses,
+        tokenIds: tokenIds.map(tokenId => tokenId.toString()),
+        votesSummary,
+      },
+    });
+  };
+};
+
+export const useAzoriusProposalListeners = () => {
+  const {
+    action,
+    governanceContracts: {
+      azoriusContractAddress,
+      ozLinearVotingContractAddress,
+      erc721LinearVotingContractAddress,
+    },
+  } = useFractal();
+
+  const baseContracts = useSafeContracts();
+  const provider = useEthersProvider();
+  const decode = useSafeDecoder();
+
+  const azoriusContract = useMemo(() => {
+    if (!baseContracts || !azoriusContractAddress) {
+      return;
+    }
+
+    return baseContracts.fractalAzoriusMasterCopyContract.asProvider.attach(azoriusContractAddress);
+  }, [azoriusContractAddress, baseContracts]);
+
+  const strategyType = useMemo(() => {
+    if (ozLinearVotingContractAddress) {
+      return VotingStrategyType.LINEAR_ERC20;
+    } else if (erc721LinearVotingContractAddress) {
+      return VotingStrategyType.LINEAR_ERC721;
+    } else {
+      return undefined;
+    }
+  }, [ozLinearVotingContractAddress, erc721LinearVotingContractAddress]);
+
+  const erc20StrategyContract = useMemo(() => {
+    if (!baseContracts || !ozLinearVotingContractAddress) {
+      return undefined;
+    }
+
+    return baseContracts.linearVotingMasterCopyContract.asProvider.attach(
+      ozLinearVotingContractAddress,
+    );
+  }, [baseContracts, ozLinearVotingContractAddress]);
+
+  const erc721StrategyContract = useMemo(() => {
+    if (!baseContracts || !erc721LinearVotingContractAddress) {
+      return undefined;
+    }
+
+    return baseContracts.linearVotingERC721MasterCopyContract.asProvider.attach(
+      erc721LinearVotingContractAddress,
+    );
+  }, [baseContracts, erc721LinearVotingContractAddress]);
+
+  useEffect(() => {
+    if (!azoriusContract || !provider || !strategyType) {
+      return;
+    }
+
+    const proposalCreatedFilter = azoriusContract.filters.ProposalCreated();
+    const listener = proposalCreatedEventListener(
+      azoriusContract,
+      erc20StrategyContract,
+      erc721StrategyContract,
+      provider,
+      strategyType,
+      decode,
+      action.dispatch,
+    );
+
+    azoriusContract.on(proposalCreatedFilter, listener);
+
+    return () => {
+      azoriusContract.off(proposalCreatedFilter, listener);
+    };
+  }, [
+    action.dispatch,
+    azoriusContract,
+    decode,
+    erc20StrategyContract,
+    erc721StrategyContract,
+    provider,
+    strategyType,
+  ]);
+
+  useEffect(() => {
+    if (strategyType !== VotingStrategyType.LINEAR_ERC20 || !erc20StrategyContract) {
+      return;
+    }
+
+    const votedEvent = erc20StrategyContract.filters.Voted();
+    const listener = erc20VotedEventListener(erc20StrategyContract, strategyType, action.dispatch);
+
+    console.log('creating erc20 voted listener');
+    erc20StrategyContract.on(votedEvent, listener);
+
+    return () => {
+      console.log('removing erc20 voted listener');
+      erc20StrategyContract.off(votedEvent, listener);
+    };
+  }, [action.dispatch, erc20StrategyContract, strategyType]);
+
+  useEffect(() => {
+    if (strategyType !== VotingStrategyType.LINEAR_ERC721 || !erc721StrategyContract) {
+      return;
+    }
+
+    const votedEvent = erc721StrategyContract.filters.Voted();
+    const listener = erc721VotedEventListener(
+      erc721StrategyContract,
+      strategyType,
+      action.dispatch,
+    );
+
+    erc721StrategyContract.on(votedEvent, listener);
+
+    return () => {
+      erc721StrategyContract.off(votedEvent, listener);
+    };
+  }, [action.dispatch, erc721StrategyContract, strategyType]);
+};