@controlflow-ai/daemon 0.1.1 → 0.1.3

package/src/daemon-client.ts

@@ -1,7 +1,7 @@
 import { defaultDaemonToken, defaultDaemonUrl } from './config.js';
 import { localHeaders } from './local-auth.js';
-import type { AgentRoomSubscriptionMode, AgentRun, ApiResponse, Chat, Message, RoomChannel, RoomParticipant, RunAction } from './types.js';
-import type { CreateArtifactRequest, SendRequest } from './client.js';
+import type { AgentRoomSubscriptionMode, AgentRun, ApiResponse, Chat, HeldMessageDraft, Message, RoomChannel, RoomParticipant, RoomReminder, RoomSavedMessage, RoomTask, RoomTaskStatus, RunAction } from './types.js';
+import type { ClaimRoomTaskRequest, CreateArtifactRequest, CreateHandoffRoomRequest, CreateRoomReminderRequest, CreateRoomSavedMessageRequest, CreateRoomTasksRequest, HeldDraftResolutionResult, ListHeldDraftsRequest, ListRoomRemindersRequest, ListRoomSavedMessagesRequest, SendMessageResult, SendRequest } from './client.js';
 
 export class DaemonClient {
   readonly baseUrl: string;
@@ -17,10 +17,41 @@ export class DaemonClient {
   }
 
   async sendMessage(input: SendRequest): Promise<Message> {
-    const data = await this.post<{ message: Message }>('/local/messages', input);
+    const data = await this.sendMessageResult(input);
+    if (!data.message) throw new Error('message was held as a draft');
     return data.message;
   }
 
+  async sendMessageResult(input: SendRequest): Promise<SendMessageResult> {
+    return this.post<SendMessageResult>('/local/messages', input);
+  }
+
+  async listHeldDrafts(input: ListHeldDraftsRequest = {}): Promise<HeldMessageDraft[]> {
+    const params = new URLSearchParams();
+    if (input.agent) params.set('agent', input.agent);
+    if (input.room_id) params.set('room_id', input.room_id);
+    params.set('status', input.status ?? 'held');
+    params.set('limit', String(input.limit ?? 50));
+    const data = await this.get<{ drafts: HeldMessageDraft[] }>(`/local/held-drafts?${params}`);
+    return data.drafts;
+  }
+
+  async abandonHeldDraft(id: string): Promise<HeldDraftResolutionResult> {
+    return this.post<HeldDraftResolutionResult>(`/local/held-drafts/${encodeURIComponent(id)}/abandon`, {});
+  }
+
+  async staySilentHeldDraft(id: string): Promise<HeldDraftResolutionResult> {
+    return this.abandonHeldDraft(id);
+  }
+
+  async sendHeldDraftAnyway(id: string): Promise<HeldDraftResolutionResult> {
+    return this.post<HeldDraftResolutionResult>(`/local/held-drafts/${encodeURIComponent(id)}/send-anyway`, {});
+  }
+
+  async reviseHeldDraft(id: string, content: string): Promise<HeldDraftResolutionResult> {
+    return this.post<HeldDraftResolutionResult>(`/local/held-drafts/${encodeURIComponent(id)}/revise`, { content });
+  }
+
   async getMessages(params: URLSearchParams): Promise<Message[]> {
     const data = await this.get<{ messages: Message[] }>(`/local/messages?${params}`);
     return data.messages;
@@ -44,11 +75,85 @@ export class DaemonClient {
     return this.post<unknown>(`/local/rooms/${encodeURIComponent(room)}/agents`, input);
   }
 
+  async createHandoffRoom(input: CreateHandoffRoomRequest): Promise<{ room: Chat; message: Message | null; deliveries: unknown[]; notify: unknown }> {
+    return this.post<{ room: Chat; message: Message | null; deliveries: unknown[]; notify: unknown }>('/local/rooms/handoff', input);
+  }
+
   async createTopic(room: string, input: { name: string; created_by?: string | null }): Promise<RoomChannel> {
     const data = await this.post<{ channel: RoomChannel }>(`/local/rooms/${encodeURIComponent(room)}/topics`, input);
     return data.channel;
   }
 
+  async listRoomTasks(room: string, status: RoomTaskStatus | 'all' = 'all', limit = 50): Promise<RoomTask[]> {
+    const params = new URLSearchParams({ status, limit: String(limit) });
+    const data = await this.get<{ tasks: RoomTask[] }>(`/local/rooms/${encodeURIComponent(room)}/tasks?${params}`);
+    return data.tasks;
+  }
+
+  async createRoomTasks(room: string, input: CreateRoomTasksRequest): Promise<RoomTask[]> {
+    const data = await this.post<{ tasks: RoomTask[] }>(`/local/rooms/${encodeURIComponent(room)}/tasks`, input);
+    return data.tasks;
+  }
+
+  async claimRoomTask(room: string, taskNumber: number, input: ClaimRoomTaskRequest): Promise<RoomTask> {
+    const data = await this.post<{ task: RoomTask }>(`/local/rooms/${encodeURIComponent(room)}/tasks/${taskNumber}/claim`, input);
+    return data.task;
+  }
+
+  async unclaimRoomTask(room: string, taskNumber: number): Promise<RoomTask> {
+    const data = await this.post<{ task: RoomTask }>(`/local/rooms/${encodeURIComponent(room)}/tasks/${taskNumber}/unclaim`, {});
+    return data.task;
+  }
+
+  async updateRoomTaskStatus(room: string, taskNumber: number, status: RoomTaskStatus): Promise<RoomTask> {
+    const data = await this.patch<{ task: RoomTask }>(`/local/rooms/${encodeURIComponent(room)}/tasks/${taskNumber}`, { status });
+    return data.task;
+  }
+
+  async restartRoomAgent(room: string, agent: string): Promise<AgentRun> {
+    const data = await this.post<{ run: AgentRun }>(`/local/rooms/${encodeURIComponent(room)}/agents/${encodeURIComponent(agent)}/restart`, {});
+    return data.run;
+  }
+
+  async scheduleReminder(input: CreateRoomReminderRequest): Promise<RoomReminder> {
+    const data = await this.post<{ reminder: RoomReminder }>('/local/reminders', input);
+    return data.reminder;
+  }
+
+  async listReminders(input: ListRoomRemindersRequest = {}): Promise<RoomReminder[]> {
+    const params = new URLSearchParams();
+    params.set('status', input.status ?? 'scheduled');
+    if (input.created_by) params.set('created_by', input.created_by);
+    if (input.room_id) params.set('room_id', input.room_id);
+    params.set('limit', String(input.limit ?? 50));
+    const data = await this.get<{ reminders: RoomReminder[] }>(`/local/reminders?${params}`);
+    return data.reminders;
+  }
+
+  async cancelReminder(id: string): Promise<RoomReminder> {
+    const data = await this.post<{ reminder: RoomReminder }>(`/local/reminders/${encodeURIComponent(id)}/cancel`, {});
+    return data.reminder;
+  }
+
+  async saveMessage(input: CreateRoomSavedMessageRequest): Promise<RoomSavedMessage> {
+    const data = await this.post<{ saved_message: RoomSavedMessage }>('/local/saved-messages', input);
+    return data.saved_message;
+  }
+
+  async listSavedMessages(input: ListRoomSavedMessagesRequest = {}): Promise<RoomSavedMessage[]> {
+    const params = new URLSearchParams();
+    if (input.saved_by) params.set('saved_by', input.saved_by);
+    if (input.room_id) params.set('room_id', input.room_id);
+    params.set('limit', String(input.limit ?? 50));
+    const data = await this.get<{ saved_messages: RoomSavedMessage[] }>(`/local/saved-messages?${params}`);
+    return data.saved_messages;
+  }
+
+  async removeSavedMessage(id: string): Promise<RoomSavedMessage> {
+    const data = await this.post<{ saved_message: RoomSavedMessage }>(`/local/saved-messages/${encodeURIComponent(id)}/delete`, {});
+    return data.saved_message;
+  }
+
   async getMessage(id: number): Promise<Message> {
     const data = await this.get<{ message: Message }>(`/local/messages/${id}`);
     return data.message;
@@ -80,6 +185,14 @@ export class DaemonClient {
     });
   }
 
+  private async patch<T>(path: string, body: unknown): Promise<T> {
+    return this.request<T>(path, {
+      method: 'PATCH',
+      headers: { 'content-type': 'application/json', ...localHeaders(this.token) },
+      body: JSON.stringify(body),
+    });
+  }
+
   private async request<T>(path: string, init: RequestInit): Promise<T> {
     const response = await fetch(`${this.baseUrl}${path}`, init);
     const payload = await response.json() as ApiResponse<T>;