feat(rwa): partial freeze (#2708)

kadena-community · Dec 4, 2024 · 293eb82 · 293eb82
1 parent 7dc5c60
commit 293eb82
Show file tree

Hide file tree

Showing 21 changed files with 498 additions and 68 deletions.
diff --git a/.changeset/perfect-rings-laugh.md b/.changeset/perfect-rings-laugh.md
@@ -0,0 +1,2 @@
+---
+---
diff --git a/packages/apps/rwa-demo/src/app/(app)/(isAgent)/investors/[investorAccount]/page.tsx b/packages/apps/rwa-demo/src/app/(app)/(isAgent)/investors/[investorAccount]/page.tsx
@@ -3,6 +3,7 @@
 import { DistributionForm } from '@/components/DistributionForm/DistributionForm';
 import { FreezeInvestor } from '@/components/FreezeInvestor/FreezeInvestor';
 import { InvestorInfo } from '@/components/InvestorInfo/InvestorInfo';
+import { PartiallyFreezeTokensForm } from '@/components/PartiallyFreezeTokensForm/PartiallyFreezeTokensForm';
 import { SideBarBreadcrumbs } from '@/components/SideBarBreadcrumbs/SideBarBreadcrumbs';
 import { useAsset } from '@/hooks/asset';
 import { useFreeze } from '@/hooks/freeze';
@@ -17,13 +18,19 @@ const InvestorPage = () => {
   const { paused } = useAsset();
   const params = useParams();
   const [hasOpenDistributeForm, setHasOpenDistributeForm] = useState(false);
+  const [hasOpenPartiallyFreezeForm, setHasOpenPartiallyFreezeForm] =
+    useState(false);
   const investorAccount = decodeURIComponent(params.investorAccount as string);
   const { frozen } = useFreeze({ investorAccount });
 
   const handleDistributeTokens = () => {
     setIsRightAsideExpanded(true);
     setHasOpenDistributeForm(true);
   };
+  const handlePartiallyFreezeTokens = () => {
+    setIsRightAsideExpanded(true);
+    setHasOpenPartiallyFreezeForm(true);
+  };
 
   return (
     <>
@@ -42,6 +49,15 @@ const InvestorPage = () => {
           }}
         />
       )}
+      {isRightAsideExpanded && hasOpenPartiallyFreezeForm && (
+        <PartiallyFreezeTokensForm
+          investorAccount={investorAccount}
+          onClose={() => {
+            setIsRightAsideExpanded(false);
+            setHasOpenPartiallyFreezeForm(false);
+          }}
+        />
+      )}
 
       <Stack width="100%" flexDirection="column">
         <InvestorInfo investorAccount={investorAccount} />
@@ -53,6 +69,15 @@ const InvestorPage = () => {
           >
             Distribute Tokens
           </Button>
+
+          <Button
+            startVisual={<MonoAdd />}
+            onPress={handlePartiallyFreezeTokens}
+            isDisabled={frozen || paused}
+          >
+            Partially freeze tokens
+          </Button>
+
           <FreezeInvestor investorAccount={investorAccount} />
         </Stack>
       </Stack>

diff --git a/packages/apps/rwa-demo/src/app/(app)/page.tsx b/packages/apps/rwa-demo/src/app/(app)/page.tsx
@@ -1,15 +1,17 @@
 'use client';
 
 import { AgentRootPage } from '@/components/HomePage/AgentRootPage';
+import { ComplianceOwnerRootPage } from '@/components/HomePage/ComplianceOwnerRootPage';
 import { InvestorRootPage } from '@/components/HomePage/InvestorRootPage';
 import { OwnerRootPage } from '@/components/HomePage/OwnerRootPage';
 import { useAccount } from '@/hooks/account';
 
 const Home = () => {
-  const { isAgent, isInvestor, isOwner } = useAccount();
+  const { isAgent, isInvestor, isOwner, isComplianceOwner } = useAccount();
 
   return (
     <>
+      {isComplianceOwner && <ComplianceOwnerRootPage />}
       {isOwner && <OwnerRootPage />}
       {isAgent && <AgentRootPage />}
       {isInvestor && <InvestorRootPage />}

diff --git a/packages/apps/rwa-demo/src/components/AccountProvider/AccountProvider.tsx b/packages/apps/rwa-demo/src/components/AccountProvider/AccountProvider.tsx
@@ -1,6 +1,7 @@
 'use client';
 import { getBalance as getBalanceFnc } from '@/services/getBalance';
 import { isAgent } from '@/services/isAgent';
+import { isComplianceOwner } from '@/services/isComplianceOwner';
 import { isFrozen } from '@/services/isFrozen';
 import { isInvestor } from '@/services/isInvestor';
 import { isOwner } from '@/services/isOwner';
@@ -26,6 +27,7 @@ export interface IAccountContext {
   sign: (tx: IUnsignedCommand) => Promise<ICommand | undefined>;
   isAgent: boolean;
   isOwner: boolean;
+  isComplianceOwner: boolean;
   isInvestor: boolean;
   isFrozen: boolean;
   selectAccount: (account: IWalletAccount) => void;
@@ -41,6 +43,7 @@ export const AccountContext = createContext<IAccountContext>({
   sign: async () => undefined,
   isAgent: false,
   isOwner: false,
+  isComplianceOwner: false,
   isInvestor: false,
   isFrozen: false,
   selectAccount: () => {},
@@ -52,6 +55,7 @@ export const AccountProvider: FC<PropsWithChildren> = ({ children }) => {
   const [accounts, setAccounts] = useState<IWalletAccount[]>();
   const [isMounted, setIsMounted] = useState(false);
   const [isOwnerState, setIsOwnerState] = useState(false);
+  const [isComplianceOwnerState, setIsComplianceOwnerState] = useState(false);
   const [isAgentState, setIsAgentState] = useState(false);
   const [isInvestorState, setIsInvestorState] = useState(false);
   const [isFrozenState, setIsFrozenState] = useState(false);
@@ -66,6 +70,12 @@ export const AccountProvider: FC<PropsWithChildren> = ({ children }) => {
     const resIsOwner = await isOwner({ owner: account.address });
     setIsOwnerState(!!resIsOwner);
   };
+  const checkIsComplianceOwner = async (account: IWalletAccount) => {
+    const resIsComplianceOwner = await isComplianceOwner({
+      owner: account.address,
+    });
+    setIsComplianceOwnerState(!!resIsComplianceOwner);
+  };
   const checkIsInvestor = async (account: IWalletAccount) => {
     const resIsInvestor = await isInvestor({ account });
     setIsInvestorState(!!resIsInvestor);
@@ -141,13 +151,16 @@ export const AccountProvider: FC<PropsWithChildren> = ({ children }) => {
     if (!account) {
       setIsAgentState(false);
       setIsOwnerState(false);
+      setIsComplianceOwnerState(false);
       setIsInvestorState(false);
       return;
     }
 
     // eslint-disable-next-line @typescript-eslint/no-floating-promises
     checkIsOwner(account);
     // eslint-disable-next-line @typescript-eslint/no-floating-promises
+    checkIsComplianceOwner(account);
+    // eslint-disable-next-line @typescript-eslint/no-floating-promises
     checkIsAgent(account);
     // eslint-disable-next-line @typescript-eslint/no-floating-promises
     checkIsInvestor(account);
@@ -188,6 +201,7 @@ export const AccountProvider: FC<PropsWithChildren> = ({ children }) => {
         sign,
         isMounted,
         isOwner: isOwnerState,
+        isComplianceOwner: isComplianceOwnerState,
         isAgent: isAgentState,
         isInvestor: isInvestorState,
         isFrozen: isFrozenState,

diff --git a/packages/apps/rwa-demo/src/components/DistributionForm/DistributionForm.tsx b/packages/apps/rwa-demo/src/components/DistributionForm/DistributionForm.tsx
@@ -45,6 +45,7 @@ export const DistributionForm: FC<IProps> = ({ onClose, investorAccount }) => {
   useEffect(() => {
     if (tx && resolveRef.current) {
       resolveRef.current(tx);
+      onClose();
     }
   }, [tx]);
 

diff --git a/packages/apps/rwa-demo/src/components/HomePage/ComplianceOwnerRootPage.tsx b/packages/apps/rwa-demo/src/components/HomePage/ComplianceOwnerRootPage.tsx
@@ -0,0 +1,71 @@
+'use client';
+import { SetMaxBalanceForm } from '@/components/SetMaxBalanceForm/SetMaxBalanceForm';
+import { SideBarBreadcrumbs } from '@/components/SideBarBreadcrumbs/SideBarBreadcrumbs';
+import { useAccount } from '@/hooks/account';
+import { useAsset } from '@/hooks/asset';
+import { MonoAdd } from '@kadena/kode-icons';
+import { Button, Stack } from '@kadena/kode-ui';
+import { useLayout } from '@kadena/kode-ui/patterns';
+import { useState } from 'react';
+import { SetMaxSupplyForm } from '../SetMaxSupplyForm/SetMaxSupplyForm';
+
+export const ComplianceOwnerRootPage = () => {
+  const { isComplianceOwner } = useAccount();
+  const { setIsRightAsideExpanded, isRightAsideExpanded } = useLayout();
+  const { paused } = useAsset();
+  const [hasOpenMaxBalanceForm, setHasOpenMaxBalanceForm] = useState(false);
+  const [hasOpenMaxSupplyForm, setHasOpenMaxSupplyForm] = useState(false);
+
+  const handleMaxBalanceForm = () => {
+    setIsRightAsideExpanded(true);
+    setHasOpenMaxBalanceForm(true);
+  };
+  const handleMaxSupplyForm = () => {
+    setIsRightAsideExpanded(true);
+    setHasOpenMaxSupplyForm(true);
+  };
+
+  return (
+    <Stack width="100%" flexDirection="column" gap="md">
+      <SideBarBreadcrumbs />
+
+      {isRightAsideExpanded && hasOpenMaxBalanceForm && (
+        <SetMaxBalanceForm
+          onClose={() => {
+            setIsRightAsideExpanded(false);
+            setHasOpenMaxBalanceForm(false);
+          }}
+        />
+      )}
+      {isRightAsideExpanded && hasOpenMaxSupplyForm && (
+        <SetMaxSupplyForm
+          onClose={() => {
+            setIsRightAsideExpanded(false);
+            setHasOpenMaxSupplyForm(false);
+          }}
+        />
+      )}
+
+      <Stack gap="sm">
+        {isComplianceOwner && (
+          <>
+            <Button
+              isDisabled={paused}
+              startVisual={<MonoAdd />}
+              onClick={handleMaxBalanceForm}
+            >
+              Set Max Balance
+            </Button>
+            <Button
+              isDisabled={paused}
+              startVisual={<MonoAdd />}
+              onClick={handleMaxSupplyForm}
+            >
+              Set Max Supply
+            </Button>
+          </>
+        )}
+      </Stack>
+    </Stack>
+  );
+};
diff --git a/packages/apps/rwa-demo/src/components/HomePage/InvestorRootPage.tsx b/packages/apps/rwa-demo/src/components/HomePage/InvestorRootPage.tsx
@@ -1,4 +1,5 @@
 import { useAccount } from '@/hooks/account';
+import { useFreeze } from '@/hooks/freeze';
 import { MonoCompareArrows } from '@kadena/kode-icons';
 import { Button, Stack } from '@kadena/kode-ui';
 import { useLayout } from '@kadena/kode-ui/patterns';
@@ -8,7 +9,8 @@ import { InvestorInfo } from '../InvestorInfo/InvestorInfo';
 import { TransferForm } from '../TransferForm/TransferForm';
 
 export const InvestorRootPage: FC = () => {
-  const { isInvestor, account, isFrozen } = useAccount();
+  const { isInvestor, account } = useAccount();
+  const { frozen } = useFreeze({ investorAccount: account?.address! });
   const { setIsRightAsideExpanded, isRightAsideExpanded } = useLayout();
   const [hasOpenTransferForm, setHasOpenTransferForm] = useState(false);
 
@@ -37,7 +39,7 @@ export const InvestorRootPage: FC = () => {
           <Button
             startVisual={<MonoCompareArrows />}
             onPress={handleTransferTokens}
-            isDisabled={isFrozen}
+            isDisabled={frozen}
           >
             Transfer tokens
           </Button>

diff --git a/packages/apps/rwa-demo/src/components/HomePage/OwnerRootPage.tsx b/packages/apps/rwa-demo/src/components/HomePage/OwnerRootPage.tsx
@@ -1,46 +1,13 @@
 'use client';
 import { AgentsList } from '@/components/AgentsList/AgentsList';
-import { SetComplianceForm } from '@/components/SetComplianceForm/SetComplianceForm';
 import { SideBarBreadcrumbs } from '@/components/SideBarBreadcrumbs/SideBarBreadcrumbs';
-import { useAsset } from '@/hooks/asset';
-import { MonoAdd } from '@kadena/kode-icons';
-import { Button, Stack } from '@kadena/kode-ui';
-import { useLayout } from '@kadena/kode-ui/patterns';
-import { useState } from 'react';
+import { Stack } from '@kadena/kode-ui';
 
 export const OwnerRootPage = () => {
-  const { setIsRightAsideExpanded, isRightAsideExpanded } = useLayout();
-  const { paused } = useAsset();
-  const [hasOpenComplianceForm, setHasOpenComplianceForm] = useState(false);
-
-  const handleComplianceForm = () => {
-    setIsRightAsideExpanded(true);
-    setHasOpenComplianceForm(true);
-  };
-
   return (
     <Stack width="100%" flexDirection="column" gap="md">
       <SideBarBreadcrumbs />
 
-      {isRightAsideExpanded && hasOpenComplianceForm && (
-        <SetComplianceForm
-          onClose={() => {
-            setIsRightAsideExpanded(false);
-            setHasOpenComplianceForm(false);
-          }}
-        />
-      )}
-
-      <Stack gap="sm">
-        <Button
-          isDisabled={paused}
-          startVisual={<MonoAdd />}
-          onClick={handleComplianceForm}
-        >
-          Set Compliance
-        </Button>
-      </Stack>
-
       <AgentsList />
     </Stack>
   );