feat: add name, accessPrice, remainingAccess and isUserStrict properties to Contact

Author: PierreJeanjacquot

src/utils/subgraphQuery.ts

@@ -21,13 +21,14 @@ const checkProtectedDataQuery = gql`
       orderDirection: desc
     ) {
       id
+      name
     }
   }
 `;
 
 export const getValidContact = async (
   graphQLClient: GraphQLClient,
-  contacts: Contact[]
+  contacts: Omit<Contact, 'name'>[]
 ): Promise<Contact[]> => {
   if (contacts.length === 0) {
     return [];
@@ -66,13 +67,17 @@ export const getValidContact = async (
     );
 
     // Convert protectedData[] into Contact[] using the map for constant time lookups
-    return protectedDataList.map(({ id }) => {
+    return protectedDataList.map(({ id, name }) => {
       const contact = contactsMap.get(id);
       if (contact) {
         return {
           address: id,
+          name: name,
+          remainingAccess: contact.remainingAccess,
+          accessPrice: contact.accessPrice,
           owner: contact.owner,
           accessGrantTimestamp: contact.accessGrantTimestamp,
+          isUserStrict: contact.isUserStrict,
         };
       }
     });

src/web3mail/fetchUserContacts.ts

@@ -1,3 +1,6 @@
+import { PublishedDatasetorder } from 'iexec/IExecOrderbookModule';
+import { ZeroAddress } from 'ethers';
+import { IExec } from 'iexec';
 import { ANY_DATASET_ADDRESS } from '../config/config.js';
 import { handleIfProtocolError, WorkflowError } from '../utils/errors.js';
 import { autoPaginateRequest } from '../utils/paginate.js';
@@ -63,7 +66,7 @@ export const fetchUserContacts = async ({
     ]);
 
     const orders = dappOrders.concat(whitelistOrders);
-    const myContacts: Contact[] = [];
+    const myContacts: Omit<Contact, 'name'>[] = [];
     let web3DappResolvedAddress = vDappAddressOrENS;
     if (isEnsTest(vDappAddressOrENS)) {
       web3DappResolvedAddress = await iexec.ens.resolveName(vDappAddressOrENS);
@@ -78,7 +81,10 @@ export const fetchUserContacts = async ({
         const contact = {
           address: order.order.dataset.toLowerCase(),
           owner: order.signer.toLowerCase(),
+          remainingAccess: order.remaining,
+          accessPrice: order.order.datasetprice,
           accessGrantTimestamp: order.publicationTimestamp,
+          isUserStrict: order.order.requesterrestrict !== ZeroAddress,
         };
         myContacts.push(contact);
       }
@@ -102,7 +108,12 @@ async function fetchAllOrdersByApp({
   userAddress,
   appAddress,
   isUserStrict,
-}) {
+}: {
+  iexec: IExec;
+  userAddress: string;
+  appAddress: string;
+  isUserStrict: boolean;
+}): Promise<PublishedDatasetorder[]> {
   const ordersFirstPage = iexec.orderbook.fetchDatasetOrderbook(
     ANY_DATASET_ADDRESS,
     {

src/web3mail/types.ts

@@ -20,6 +20,10 @@ export type Contact = {
   address: Address;
   owner: Address;
   accessGrantTimestamp: TimeStamp;
+  isUserStrict: boolean;
+  name: string;
+  remainingAccess: number;
+  accessPrice: number;
 };
 
 export type SendEmailParams = {
@@ -59,6 +63,7 @@ export type SendEmailResponse = {
  */
 export type ProtectedDataQuery = {
   id: string;
+  name: string;
 };
 
 export type GraphQLResponse = {

tests/e2e/fetchUserContacts.test.ts

@@ -43,6 +43,33 @@ describe('web3mail.fetchMyContacts()', () => {
   }, 4 * MAX_EXPECTED_BLOCKTIME + MAX_EXPECTED_WEB2_SERVICES_TIME);
 
   describe('when no access is granted', () => {
+    it('contacts should contain name, accessPrice, remainingAccess, owner, accessGrantTimestamp, and isUserStrict', async () => {
+      const userWithAccess = Wallet.createRandom().address;
+
+      await web3mail.init();
+      // eslint-disable-next-line @typescript-eslint/dot-notation
+      const authorizedApp = web3mail['dappAddressOrENS'];
+
+      await dataProtector.grantAccess({
+        authorizedApp: authorizedApp,
+        protectedData: protectedData1.address,
+        authorizedUser: userWithAccess,
+      });
+
+      const contacts = await web3mail.fetchUserContacts({
+        userAddress: userWithAccess,
+        isUserStrict: true,
+      });
+      expect(contacts.length).toBe(1);
+      expect(contacts[0].address).toBe(protectedData1.address.toLowerCase());
+      expect(contacts[0].owner).toBeDefined();
+      expect(contacts[0].accessPrice).toBe(0);
+      expect(contacts[0].remainingAccess).toBe(1);
+      expect(contacts[0].accessGrantTimestamp).toBeDefined();
+      expect(contacts[0].isUserStrict).toBe(true);
+      expect(contacts[0].name).toBe('test do not use');
+    });
+
     it(
       'should return an empty contact array',
       async () => {

tests/unit/utils/subgraphQuery.test.ts

@@ -8,7 +8,7 @@ import type { Contact } from '../../../src/web3mail/types.js';
 
 describe('getValidContact', () => {
   // Define the variables in the outermost scope
-  let contacts: Contact[];
+  let contacts: Omit<Contact, 'name'>[];
   let graphQLClient: GraphQLClient;
 
   beforeAll(() => {
@@ -18,16 +18,25 @@ describe('getValidContact', () => {
         address: 'address1',
         owner: 'owner1',
         accessGrantTimestamp: '2023-06-08T09:32:29.761Z',
+        remainingAccess: 1,
+        accessPrice: 0,
+        isUserStrict: true,
       },
       {
         address: 'address2',
         owner: 'owner2',
         accessGrantTimestamp: '2023-06-09T14:21:17.231Z',
+        remainingAccess: 1,
+        accessPrice: 0,
+        isUserStrict: true,
       },
       {
         address: 'address3',
         owner: 'owner3',
         accessGrantTimestamp: '2023-06-10T14:21:17.231Z',
+        remainingAccess: 1,
+        accessPrice: 0,
+        isUserStrict: true,
       },
     ];
 
@@ -38,7 +47,10 @@ describe('getValidContact', () => {
   it('should fetch valid contacts', async () => {
     // Mock the request method of the GraphQLClient instance
     jest.spyOn(graphQLClient, 'request').mockResolvedValue({
-      protectedDatas: [{ id: 'address1' }, { id: 'address3' }],
+      protectedDatas: [
+        { id: 'address1', name: 'Contact 1' },
+        { id: 'address3', name: 'Contact 3' },
+      ],
     });
 
     const validContacts = await getValidContact(graphQLClient, contacts);
@@ -48,11 +60,19 @@ describe('getValidContact', () => {
         address: 'address1',
         owner: 'owner1',
         accessGrantTimestamp: '2023-06-08T09:32:29.761Z',
+        name: 'Contact 1',
+        remainingAccess: 1,
+        accessPrice: 0,
+        isUserStrict: true,
       },
       {
         address: 'address3',
         owner: 'owner3',
         accessGrantTimestamp: '2023-06-10T14:21:17.231Z',
+        name: 'Contact 3',
+        remainingAccess: 1,
+        accessPrice: 0,
+        isUserStrict: true,
       },
     ]);