added a function to cancel a service

TalentLayer · Apr 19, 2024 · b0db026 · b0db026
1 parent 2427f52
commit b0db026
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 3 deletions.
diff --git a/packages/client/src/services/__tests__/services.spec.ts b/packages/client/src/services/__tests__/services.spec.ts
@@ -68,7 +68,6 @@ describe('Service', () => {
             const serviceDetails = testServiceDetails;
             const userId = testUserId;
             const existingServiceId = testServiceId;
-
 
             // Act
             const response = await service.update(serviceDetails, userId, parseInt(existingServiceId));
@@ -84,6 +83,27 @@ describe('Service', () => {
         })
     })
 
+    describe('cancel', () => {
+        it('should cancel an existing service', async () => {
+            // Arrange
+            const userId = testUserId;
+            const serviceId = parseInt(testServiceId);
+
+            // Act
+            const response = await service.cancel(userId, serviceId);
+
+            // Assert
+            expect(response).toEqual(testAddress);
+            expect(mockViemClient.writeContract).toHaveBeenCalledWith(
+                'talentLayerService',
+                'cancelService',
+                [userId, serviceId]
+            );
+
+
+        })
+    })
+
     describe('search', () => {
         it('should reaturn services based on criteria', async () => {
             // Arranage

diff --git a/packages/client/src/services/index.ts b/packages/client/src/services/index.ts
@@ -1,3 +1,4 @@
+import { Hash } from 'viem';
 import GraphQLClient from '../graphql';
 import IPFSClient from '../ipfs';
 import { Logger } from '../logger';
@@ -27,6 +28,7 @@ export interface IService {
   updloadServiceDataToIpfs(serviceData: ServiceDetails): Promise<string>;
   getServices(params: IProps): Promise<any>;
   search(params: IProps): Promise<any>;
+  cancel(userId: string, serviceId: number): Promise<Hash>;
 }
 
 /**
@@ -168,14 +170,14 @@ export class Service {
     serviceDetails: ServiceDetails,
     userId: string,
     existingServiceId: number,
-  ): Promise<ClientTransactionResponse> {   
+  ): Promise<ClientTransactionResponse> {
 
     const cid = await this.updloadServiceDataToIpfs(serviceDetails);
 
     const tx = await this.viemClient.writeContract(
       'talentLayerService',
       'updateServiceData',
-      [userId, existingServiceId, cid], 
+      [userId, existingServiceId, cid],
     );
 
     if (cid && tx) {
@@ -184,4 +186,27 @@ export class Service {
 
     throw new Error('Unable to update service');
   }
+
+  /**
+   * Cancel an existing service.
+   * @param {string} userId - Id of the user cancelling the service
+   * @param {number} serviceId - Id of the service being updated
+   * @returns {Promise<Hash>} - A promise that resolves to the transaction hash for the cancelService function call
+  */
+  public async cancel(
+    userId: string,
+    serviceId: number
+  ): Promise<Hash> {
+    const tx = await this.viemClient.writeContract(
+      'talentLayerService',
+      'cancelService',
+      [userId, serviceId]
+    )
+
+    if (tx) {
+      return tx;
+    }
+
+    throw new Error('Unable to cancel service');
+  }
 }