From a56cfbb98cea196308f75b8e0c25728c76478ac6 Mon Sep 17 00:00:00 2001
From: WaDadidou <wadadidou@gmail.com>
Date: Mon, 21 Oct 2024 19:19:36 -0400
Subject: [PATCH] feat: add useDAOProposalById

---
 packages/hooks/dao/useDAOProposalById.ts | 200 +++++++++++++++++++++++
 packages/hooks/dao/useDAOProposals.ts    |   4 +-
 2 files changed, 202 insertions(+), 2 deletions(-)
 create mode 100644 packages/hooks/dao/useDAOProposalById.ts

diff --git a/packages/hooks/dao/useDAOProposalById.ts b/packages/hooks/dao/useDAOProposalById.ts
new file mode 100644
index 0000000000..4c4c3c9370
--- /dev/null
+++ b/packages/hooks/dao/useDAOProposalById.ts
@@ -0,0 +1,200 @@
+import { GnoJSONRPCProvider } from "@gnolang/gno-js-client";
+import { useQuery, useQueryClient } from "@tanstack/react-query";
+import { useCallback } from "react";
+
+import { useDAOFirstProposalModule } from "./useDAOProposalModules";
+import { AppProposalResponse, GnoDAOProposal } from "./useDAOProposals";
+
+import { useFeedbacks } from "@/context/FeedbacksProvider";
+import { DaoProposalSingleQueryClient } from "@/contracts-clients/dao-proposal-single/DaoProposalSingle.client";
+import { Status } from "@/contracts-clients/dao-proposal-single/DaoProposalSingle.types";
+import {
+  NetworkKind,
+  mustGetNonSigningCosmWasmClient,
+  parseUserId,
+} from "@/networks";
+import { extractGnoJSONString } from "@/utils/gno";
+
+const daoProposalByIdQueryKey = (
+  daoId: string | undefined,
+  proposalId: number | undefined,
+) => ["dao-proposals", daoId, proposalId];
+
+export const useDAOProposalById = (
+  daoId: string | undefined,
+  proposalId: number | undefined,
+) => {
+  const { setToast } = useFeedbacks();
+  const [network, daoAddress] = parseUserId(daoId);
+  const { daoProposal: cosmWasmDAOProposal, ...cosmWasmOther } =
+    useCosmWasmDAOProposalById(daoId, proposalId);
+
+  const { data: gnoDAOProposal, ...gnoOther } = useQuery(
+    daoProposalByIdQueryKey(daoId, proposalId),
+    async () => {
+      try {
+        if (network?.kind !== NetworkKind.Gno)
+          throw new Error("Not a Gno network");
+        if (!proposalId) throw new Error("Missing proposal id");
+
+        const provider = new GnoJSONRPCProvider(network.endpoint);
+
+        const gnoProposal: GnoDAOProposal = extractGnoJSONString(
+          await provider.evaluateExpression(
+            daoAddress,
+            `getProposalJSON(0, ${proposalId})`,
+          ),
+        );
+
+        const title = gnoProposal.title;
+        const description = gnoProposal.description;
+        const status = gnoProposal.status.toLowerCase() as Status;
+        const proposer = gnoProposal.proposer;
+        const yesVotes = gnoProposal.votes.yes;
+        const noVotes = gnoProposal.votes.no;
+        const abstainVotes = gnoProposal.votes.abstain;
+        const threshold =
+          gnoProposal.threshold.thresholdQuorum.threshold.percent / 10000;
+        const quorum =
+          gnoProposal.threshold.thresholdQuorum.quorum.percent / 10000;
+        const actions = gnoProposal.messages.map((m) => JSON.stringify(m));
+
+        const proposal: AppProposalResponse = {
+          id: gnoProposal.id,
+          proposal: {
+            title,
+            description,
+            votes: {
+              yes: yesVotes.toString(),
+              no: noVotes.toString(),
+              abstain: abstainVotes.toString(),
+            },
+            allow_revoting: false,
+            expiration: "TODO" as any,
+            msgs: gnoProposal.messages.map((m) => ({
+              ...m,
+              gno: true,
+            })),
+            actions,
+            proposer,
+            start_height: gnoProposal.startHeight,
+            status,
+            threshold: {
+              threshold_quorum: {
+                threshold: { percent: `${threshold}` },
+                quorum: { percent: `${quorum}` },
+              },
+            },
+            total_power: gnoProposal.totalPower.toString(),
+          },
+        };
+
+        return proposal;
+      } catch (err) {
+        const message = err instanceof Error ? err.message : `${err}`;
+        setToast({
+          title:
+            "Failed to fetch the Gno DAO proposal\nThis proposal might not exist in this DAO",
+          message,
+          type: "error",
+          mode: "normal",
+        });
+        console.error(
+          "Failed to fetch the Gno DAO. This proposal might not exist in this DAO: ",
+          message,
+        );
+        return null;
+      }
+    },
+    {
+      staleTime: Infinity,
+      enabled: !!(daoId && network?.kind === NetworkKind.Gno && proposalId),
+    },
+  );
+
+  if (network?.kind === NetworkKind.Gno) {
+    return {
+      daoProposal: gnoDAOProposal,
+      ...gnoOther,
+    };
+  }
+
+  return {
+    daoProposal: cosmWasmDAOProposal,
+    ...cosmWasmOther,
+  };
+};
+
+const useCosmWasmDAOProposalById = (
+  daoId: string | undefined,
+  proposalId: number | undefined,
+) => {
+  const { setToast } = useFeedbacks();
+  const [network] = parseUserId(daoId);
+  const networkId = network?.id;
+  const { daoFirstProposalModule } = useDAOFirstProposalModule(daoId);
+  const proposalModuleAddress = daoFirstProposalModule?.address;
+
+  const { data, ...other } = useQuery(
+    daoProposalByIdQueryKey(daoId, proposalId),
+    async () => {
+      try {
+        if (!networkId) throw new Error("Missing network id");
+        if (!proposalModuleAddress)
+          throw new Error("No proposal module address");
+        if (!proposalId) throw new Error("Missing proposal id");
+
+        const cosmwasmClient = await mustGetNonSigningCosmWasmClient(networkId);
+        const daoProposalClient = new DaoProposalSingleQueryClient(
+          cosmwasmClient,
+          proposalModuleAddress,
+        );
+
+        const daoProposal = await daoProposalClient.proposal({
+          proposalId,
+        });
+
+        const proposal: AppProposalResponse = {
+          ...daoProposal,
+          proposal: {
+            ...daoProposal.proposal,
+            actions: [] as string[],
+          },
+        };
+
+        return proposal;
+      } catch (err) {
+        const message = err instanceof Error ? err.message : `${err}`;
+        setToast({
+          title:
+            "Failed to fetch the Cosmos DAO proposal\nThis proposal might not exist in this DAO",
+          message,
+          type: "error",
+          mode: "normal",
+        });
+        console.error(
+          "Failed to fetch the Cosmos DAO proposal. This proposal might not exist in this DAO: ",
+          message,
+        );
+        return null;
+      }
+    },
+    {
+      staleTime: Infinity,
+      enabled: !!(networkId && proposalModuleAddress && proposalId),
+    },
+  );
+  return { daoProposal: data, ...other };
+};
+
+export const useInvalidateDAOProposalById = (
+  daoId: string | undefined,
+  proposalId: number | undefined,
+) => {
+  const queryClient = useQueryClient();
+  return useCallback(
+    () =>
+      queryClient.invalidateQueries(daoProposalByIdQueryKey(daoId, proposalId)),
+    [queryClient, daoId, proposalId],
+  );
+};
diff --git a/packages/hooks/dao/useDAOProposals.ts b/packages/hooks/dao/useDAOProposals.ts
index 1c144c2e2b..787f0865a0 100644
--- a/packages/hooks/dao/useDAOProposals.ts
+++ b/packages/hooks/dao/useDAOProposals.ts
@@ -25,13 +25,13 @@ export type AppProposalResponse = ProposalResponse & {
   proposal: { actions: string[] };
 };
 
-type GnoProposalVotes = {
+export type GnoProposalVotes = {
   yes: number;
   no: number;
   abstain: number;
 };
 
-type GnoDAOProposal = {
+export type GnoDAOProposal = {
   id: number;
   title: string;
   description: string;