implement permissions (#60)

* implement permissions * fix permissions type issues * fix permissions type issues * resolve confilicts
TogetherCrew · Aug 28, 2024 · a6de7cc · a6de7cc
1 parent 381f919
commit a6de7cc
Show file tree

Hide file tree

Showing 4 changed files with 161 additions and 196 deletions.
diff --git a/src/components/shared/CustomTable.tsx b/src/components/shared/CustomTable.tsx
@@ -35,9 +35,6 @@ const CustomTable: React.FC<CustomTableProps<AccessData>> = ({
   ycolumns,
   handleGrantOrRevokeAccess,
 }) => {
-  console.log('xcolumns:', xcolumns);
-  console.log('ycolumns:', ycolumns);
-
   const groupedApplications = ycolumns.reduce(
     (acc, application) => {
       if (!acc[application.applicationName]) {
@@ -89,12 +86,16 @@ const CustomTable: React.FC<CustomTableProps<AccessData>> = ({
                   );
                   return (
                     <TableCell key={colIndex} align="center">
-                      <AccessControlButton
-                        hasAccess={application?.hasPermission || false}
-                        onToggleAccess={() =>
-                          handleToggleAccess(application!, platform)
-                        }
-                      />
+                      {application ? (
+                        <AccessControlButton
+                          hasAccess={application.hasPermission}
+                          onToggleAccess={() =>
+                            handleToggleAccess(application, platform)
+                          }
+                        />
+                      ) : (
+                        <Typography variant="body2">No Data</Typography>
+                      )}
                     </TableCell>
                   );
                 })}

diff --git a/src/pages/Identifiers/Attestation/Attestation.tsx b/src/pages/Identifiers/Attestation/Attestation.tsx
@@ -138,7 +138,7 @@ export function Attestation() {
           data: linkingIdentifierRequest.message.data,
         },
         signature: linkingIdentifierRequest.signature,
-        attester: linkingIdentifierRequest.message.attester,
+        attester: linkingIdentifierRequest.message.attester as string,
         deadline: 0n,
       };
       console.log({ transformedPayload });

diff --git a/src/pages/Permissions/Permissions.tsx b/src/pages/Permissions/Permissions.tsx
@@ -1,175 +1,188 @@
-/* eslint-disable react-hooks/exhaustive-deps */
-/* eslint-disable @typescript-eslint/no-unused-vars */
-import {
-  useAccount,
-  useReadContract,
-  useReadContracts,
-  useWriteContract,
-} from 'wagmi';
-import { Abi, Address } from 'viem';
-import { useEffect, useState } from 'react';
+/* eslint-disable @typescript-eslint/no-explicit-any */
+import { useEffect, useMemo, useState } from 'react';
+import { useReadContract, useReadContracts, useWriteContract } from 'wagmi';
+import { Address, Abi } from 'viem';
+import { Backdrop, CircularProgress, Stack, Typography } from '@mui/material';
+import { useGetAttestations } from '../../services/eas/query';
 import sepoliaChainAppConctract from '../../utils/contracts/app/sepoliaChain.json';
 import sepoliaChainOidonctract from '../../utils/contracts/oid/sepoliaChain.json';
-import useSessionSigs from '../../hooks/useSessionSigs';
-import useLit from '../../hooks/LitProvider';
-import { useSigner } from '../../utils/eas-wagmi-utils';
-import { decryptAttestation, getAttestations } from '../../libs/oci';
-import CustomTable from '../../components/shared/CustomTable';
-
-interface IProvider {
-  uid: string;
-  provider: string;
-  id: string;
-}
+import { decodeAttestationData, IAttestation } from '../../libs/oci';
+import CustomTable, {
+  Platform,
+  AccessData,
+} from '../../components/shared/CustomTable';
 
 export function Permissions() {
-  const signer = useSigner();
-  const { litNodeClient } = useLit();
-  const { sessionSigs, createSessionSigs } = useSessionSigs();
-
-  const { isConnected, address, chainId } = useAccount();
-  const [applicationsArgs, setApplicationsArgs] = useState<[number, number]>([
-    0, 10,
-  ]);
-  const { writeContract } = useWriteContract();
-
-  const [providers, setProviders] = useState<IProvider[]>([]);
-  const [uids, setUids] = useState<string[]>([]);
-
-  const { data: applications } = useReadContract({
+  const { writeContract, isPending: isWriting } = useWriteContract();
+  const { data: attestationsResponse, isLoading: isLoadingAttestations } =
+    useGetAttestations();
+  const [applicationsArgs] = useState<[number, number]>([0, 10]);
+  const [attestations, setAttestations] = useState<
+    (IAttestation & { provider?: string; id?: string })[]
+  >([]);
+  const [permissionsWithUidsAndApps, setPermissionsWithUidsAndApps] = useState<
+    AccessData[]
+  >([]);
+
+  const {
+    data,
+    isLoading: isLoadingApplications,
+    error,
+  } = useReadContract({
     abi: sepoliaChainAppConctract.appContractABI,
     address: sepoliaChainAppConctract.appContractAddress as Address,
     functionName: 'getApplications',
     args: applicationsArgs,
   });
 
-  useEffect(() => {
-    if (!isConnected) throw new Error('Wallet not connected');
-  }, [isConnected, applications]);
+  if (error) {
+    throw new Error('Error fetching applications');
+  }
 
-  const contractCalls = uids.flatMap((uid) =>
-    applications.map((application) => ({
-      abi: sepoliaChainOidonctract.oidContractAbi as Abi,
-      address: sepoliaChainOidonctract.oidContractAddress as Address,
-      functionName: 'hasPermission',
-      args: [uid, application.account],
-    }))
+  const applications = useMemo(
+    () => (data as { name: string; account: string }[]) || [],
+    [data]
   );
 
-  const { data: hasPermissionsOnApp } = useReadContracts({
-    contracts: contractCalls,
-  });
-
-  console.log({ hasPermissionsOnApp });
+  useEffect(() => {
+    const processAttestations = () => {
+      if (!attestationsResponse) {
+        return;
+      }
 
-  const permissionsWithUidsAndApps =
-    hasPermissionsOnApp
-      ?.map((permissionResult, index) => {
-        const uidIndex = Math.floor(index / applications.length);
-        const appIndex = index % applications.length;
+      const attestationsData = attestationsResponse.map((attestation) => {
+        const decodedData = decodeAttestationData(attestation.data);
 
-        if (!permissionResult || typeof permissionResult.result !== 'boolean') {
-          console.error(
-            `Unexpected result format for UID: ${uids[uidIndex]} and account: ${applications[appIndex].account}`,
-            permissionResult
-          );
-          return null;
-        }
+        const providerData = decodedData.find(
+          (provider) => provider.name === 'provider'
+        );
 
         return {
-          uid: uids[uidIndex],
-          account: applications[appIndex].account,
-          applicationName: applications[appIndex].name,
-          hasPermission: permissionResult.result as boolean,
+          ...attestation,
+          provider:
+            typeof providerData?.value.value === 'string'
+              ? providerData.value.value
+              : undefined,
+          decodedData,
         };
-      })
-      .filter(Boolean) || [];
+      });
 
-  console.log({ permissionsWithUidsAndApps });
+      setAttestations(attestationsData);
+    };
 
-  // const { data: hasPermission } = useReadContract({
-  //   abi: sepoliaChainOidonctract.oidContractAbi,
-  //   address: sepoliaChainOidonctract.oidContractAddress as Address,
-  //   functionName: 'hasPermission',
-  //   args: [uid, account],
-  // });
+    processAttestations();
+  }, [attestationsResponse]);
 
-  // useEffect(() => {
-  //   if (!isConnected) throw new Error('Wallet not connected');
+  const contractCalls = useMemo(
+    () =>
+      attestations.flatMap(
+        (attestation) =>
+          applications?.map((application) => ({
+            abi: sepoliaChainOidonctract.oidContractAbi as Abi,
+            address: sepoliaChainOidonctract.oidContractAddress as Address,
+            functionName: 'hasPermission',
+            args: [attestation.id, application.account],
+          })) || []
+      ),
+    [attestations, applications]
+  );
 
-  //   console.log({ hasPermission });
-  // }, [hasPermission]);
+  const { data: hasPermissionsOnApp, isLoading: isLoadingPermissions } =
+    useReadContracts({
+      contracts: contractCalls,
+    });
 
   useEffect(() => {
-    if (isConnected && signer && chainId && litNodeClient) {
-      createSessionSigs({ signer, chainId, litNodeClient });
+    if (hasPermissionsOnApp) {
+      const uids = attestations.map((attestation) => attestation.id);
+
+      const permissions =
+        hasPermissionsOnApp
+          ?.map((permissionResult, index) => {
+            const uidIndex = Math.floor(index / applications.length);
+            const appIndex = index % applications.length;
+
+            if (
+              !permissionResult ||
+              typeof permissionResult.result !== 'boolean'
+            ) {
+              console.error(
+                `Unexpected result format for UID: ${uids[uidIndex]} and account: ${applications[appIndex].account}`,
+                permissionResult
+              );
+              return null;
+            }
+
+            return {
+              uid: uids[uidIndex],
+              account: applications[appIndex].account,
+              applicationName: applications[appIndex].name,
+              hasPermission: permissionResult.result as boolean,
+            };
+          })
+          .filter(Boolean) || [];
+
+      setPermissionsWithUidsAndApps(
+        permissions.filter(
+          (permission): permission is AccessData => permission !== null
+        )
+      );
     }
-  }, [signer, isConnected, litNodeClient, chainId, createSessionSigs]);
+  }, [hasPermissionsOnApp, attestations, applications]);
 
-  const fetchAttestations = async () => {
-    if (!address) throw new Error('No address found');
+  const providers: Platform[] = attestations.map((attestation) => ({
+    id: attestation.id || 'Unknown',
+    provider: attestation.provider || 'Unknown',
+    uid: attestation.id || 'Unknown',
+  }));
 
-    const attestations = await getAttestations(address as `0x${string}`);
-
-    const filteredUids = attestations
-      .filter((attestation) => attestation.revocationTime === 0n)
-      .map((attestation) => attestation.uid);
-
-    setUids(filteredUids);
-
-    return attestations.filter(
-      (attestation) => attestation.revocationTime === 0n
-    );
+  const handleGrantOrRevokeAccess = (application: any, platform: any) => {
+    writeContract({
+      abi: sepoliaChainOidonctract.oidContractAbi as Abi,
+      address: sepoliaChainOidonctract.oidContractAddress as Address,
+      functionName: application.hasPermission
+        ? 'revokePermission'
+        : 'grantPermission',
+      args: [platform.uid, application.account],
+    });
   };
 
-  useEffect(() => {
-    const decrypt = async () => {
-      try {
-        if (!sessionSigs || !litNodeClient)
-          throw new Error('No sessionSigs found');
-
-        const attestations = await fetchAttestations();
-
-        const decryptedProviders = (await Promise.all(
-          attestations.map((attestation) =>
-            decryptAttestation(litNodeClient, attestation, sessionSigs)
-              .then((result) => ({
-                ...result,
-                uid: attestation.uid,
-              }))
-              .catch(() => null)
-          )
-        ).then((results) =>
-          results.filter((value) => value !== null)
-        )) as IProvider[];
-
-        setProviders(decryptedProviders);
-      } catch (e) {
-        console.log(e);
-      }
-    };
-
-    decrypt();
-  }, [sessionSigs, litNodeClient]);
-
-  console.log({ providers });
+  const isLoading =
+    isWriting ||
+    isLoadingAttestations ||
+    isLoadingApplications ||
+    isLoadingPermissions;
 
   return (
     <div>
-      <div>Permissions</div>
+      <Backdrop
+        open={isLoading}
+        sx={{
+          zIndex: (theme) => theme.zIndex.drawer + 1,
+          background: '#fff',
+          color: 'black',
+        }}
+      >
+        <Stack
+          sx={{
+            display: 'flex',
+            flexDirection: 'row',
+            alignItems: 'center',
+          }}
+        >
+          <CircularProgress color="inherit" />
+          <Typography variant="h6" gutterBottom style={{ marginLeft: '15px' }}>
+            Loading...
+          </Typography>
+        </Stack>
+      </Backdrop>
+      <Typography variant="h6" gutterBottom>
+        Permissions
+      </Typography>
       <CustomTable
         xcolumns={providers}
         ycolumns={permissionsWithUidsAndApps}
-        handleGrantOrRevokeAccess={(application, platform) => {
-          writeContract({
-            abi: sepoliaChainOidonctract.oidContractAbi as Abi,
-            address: sepoliaChainOidonctract.oidContractAddress as Address,
-            functionName: application.hasPermission
-              ? 'revokePermission'
-              : 'grantPermission',
-            args: [platform.uid, application.account],
-          });
-        }}
+        handleGrantOrRevokeAccess={handleGrantOrRevokeAccess}
       />
     </div>
   );