Feat/acp nego feature

src/acpClient.ts

@@ -1,21 +1,33 @@
 import { Address, parseEther } from "viem";
 import { io } from "socket.io-client";
-import AcpContractClient, { AcpJobPhases, MemoType } from "./acpContractClient";
+import AcpContractClient, {
+  AcpJobPhases,
+  AcpNegoStatus,
+  MemoType,
+} from "./acpContractClient";
 import { AcpAgent } from "../interfaces";
 import AcpJob from "./acpJob";
 import AcpMemo from "./acpMemo";
 import AcpJobOffering from "./acpJobOffering";
+import AcpMessage from "./acpMessage";
 import {
   IAcpClientOptions,
   IAcpJob,
   IAcpJobResponse,
   IAcpMemo,
 } from "./interfaces";
 
-enum SocketEvents {
+export interface IDeliverable {
+  type: string;
+  value: string;
+}
+
+export enum SocketEvents {
   ROOM_JOINED = "roomJoined",
   ON_EVALUATE = "onEvaluate",
   ON_NEW_TASK = "onNewTask",
+  ON_NEW_MSG = "onNewMsg",
+  ON_CREATE_MSG = "onCreateMsg",
 }
 export class EvaluateResult {
   isApproved: boolean;
@@ -29,15 +41,16 @@ export class EvaluateResult {
 
 class AcpClient {
   private acpUrl;
+  private acpJob: AcpJob | null = null;
   public acpContractClient: AcpContractClient;
   private onNewTask?: (job: AcpJob) => void;
   private onEvaluate?: (job: AcpJob) => void;
-
+  private onNewMsg?: (msg: AcpMessage, job: AcpJob) => void;
   constructor(options: IAcpClientOptions) {
     this.acpContractClient = options.acpContractClient;
     this.onNewTask = options.onNewTask;
     this.onEvaluate = options.onEvaluate || this.defaultOnEvaluate;
-
+    this.onNewMsg = options.onNewMsg;
     this.acpUrl = this.acpContractClient.config.acpUrl;
     this.init();
   }
@@ -84,9 +97,12 @@ class AcpClient {
                 memo.nextPhase
               );
             }),
-            data.phase
+            data.phase,
+            data.negoStatus
           );
 
+          this.acpJob = job;
+
           this.onEvaluate(job);
         }
       }
@@ -113,14 +129,35 @@ class AcpClient {
                 memo.nextPhase
               );
             }),
-            data.phase
+            data.phase,
+            data.negoStatus
           );
 
+          this.acpJob = job;
+
           this.onNewTask(job);
         }
       }
     );
 
+    socket.on(SocketEvents.ON_NEW_MSG, (data, callback) => {
+      callback(true);
+
+      if (this.onNewMsg && this.acpJob) {
+        this.acpJob.negoStatus = AcpNegoStatus.PENDING;
+
+        const msg = new AcpMessage(
+          data.id,
+          data.messages ?? [],
+          socket,
+          this.acpJob,
+          this.acpContractClient.walletAddress
+        );
+
+        this.onNewMsg(msg, this.acpJob);
+      }
+    });
+
     const cleanup = async () => {
       if (socket) {
         socket.disconnect();
@@ -267,7 +304,8 @@ class AcpClient {
               memo.nextPhase
             );
           }),
-          job.phase
+          job.phase,
+          job.negoStatus
         );
       });
     } catch (error) {
@@ -307,7 +345,8 @@ class AcpClient {
               memo.nextPhase
             );
           }),
-          job.phase
+          job.phase,
+          job.negoStatus
         );
       });
     } catch (error) {
@@ -346,7 +385,8 @@ class AcpClient {
               memo.nextPhase
             );
           }),
-          job.phase
+          job.phase,
+          job.negoStatus
         );
       });
     } catch (error) {
@@ -390,7 +430,8 @@ class AcpClient {
             memo.nextPhase
           );
         }),
-        job.phase
+        job.phase,
+        job.negoStatus
       );
     } catch (error) {
       throw error;

src/acpContractClient.ts

@@ -26,6 +26,13 @@ export enum AcpJobPhases {
   REJECTED = 5,
 }
 
+export enum AcpNegoStatus {
+  PENDING = "PENDING",
+  AGREED = "AGREED",
+  DISAGREED = "DISAGREED",
+  NOT_STARTED = "NOT_STARTED",
+}
+
 class AcpContractClient {
   private _sessionKeyClient: ModularAccountV2Client | undefined;
 

src/acpJob.ts

@@ -1,6 +1,6 @@
 import { Address } from "viem";
 import AcpClient from "./acpClient";
-import { AcpJobPhases } from "./acpContractClient";
+import { AcpJobPhases, AcpNegoStatus } from "./acpContractClient";
 import AcpMemo from "./acpMemo";
 
 class AcpJob {
@@ -11,7 +11,8 @@ class AcpJob {
     public providerAddress: Address,
     public evaluatorAddress: Address,
     public memos: AcpMemo[],
-    public phase: AcpJobPhases
+    public phase: AcpJobPhases,
+    public negoStatus?: AcpNegoStatus
   ) {}
 
   public get serviceRequirement() {

src/acpMessage.ts

@@ -0,0 +1,64 @@
+import { Socket } from "socket.io-client";
+import { SocketEvents } from "./acpClient";
+import AcpJob from "./acpJob";
+import { AcpJobPhases, AcpNegoStatus } from "./acpContractClient";
+import { Address } from "viem";
+
+interface Message {
+  id: number;
+  sender: Address;
+  recipient: Address;
+  content: string;
+  timestamp: number;
+}
+
+class AcpMessage {
+  constructor(
+    public id: number,
+    public messages: Message[],
+    public socket: Socket,
+    public acpJob: AcpJob | null,
+    public walletAddress: Address
+  ) {}
+
+  initOrReply(message: string) {
+    if (!this.acpJob) {
+      throw new Error("Cannot initiate or reply conversation without job");
+    }
+
+    if (this.acpJob.negoStatus !== AcpNegoStatus.PENDING) {
+      throw new Error(
+        "Cannot initiate or reply conversation in non-negotiation phase"
+      );
+    }
+
+    if (
+      this.messages.length > 0 &&
+      this.messages[this.messages.length - 1].sender === this.walletAddress
+    ) {
+      throw new Error("Cannot reply to own message");
+    }
+
+    this.socket.timeout(5000).emit(
+      SocketEvents.ON_CREATE_MSG,
+      {
+        jobId: this.acpJob.id,
+        content: message,
+        sender: this.walletAddress,
+        recipient:
+          this.messages.length > 0
+            ? this.messages[this.messages.length - 1].sender
+            : this.acpJob.providerAddress,
+      },
+      (err: any, response: any) => {
+        if (err || !response) {
+          console.log(`Message not received`, err);
+        } else {
+          console.log(`Message received`);
+        }
+      }
+    );
+  }
+}
+
+export default AcpMessage;

src/interfaces.ts

@@ -1,6 +1,11 @@
 import { Address } from "viem";
-import AcpContractClient, { AcpJobPhases, MemoType } from "./acpContractClient";
+import AcpContractClient, {
+  AcpJobPhases,
+  AcpNegoStatus,
+  MemoType,
+} from "./acpContractClient";
 import AcpJob from "./acpJob";
+import AcpMessage from "./acpMessage";
 
 export interface IDeliverable {
   type: string;
@@ -32,6 +37,7 @@ export interface IAcpJob {
     deliverable: IDeliverable | null;
     memos: IAcpMemoData[];
     createdAt: string;
+    negoStatus: AcpNegoStatus;
   };
   error?: Error;
 }
@@ -52,4 +58,5 @@ export interface IAcpClientOptions {
   acpContractClient: AcpContractClient;
   onNewTask?: (job: AcpJob) => void;
   onEvaluate?: (job: AcpJob) => void;
+  onNewMsg?: (msg: AcpMessage, job: AcpJob) => void;
 }