feat(cloud): cloud task service & rendering (#932)

Author: tatoalo

apps/twig/src/api/posthogClient.ts

@@ -354,7 +354,7 @@ export class PostHogAPIClient {
           task_id: taskId,
           id: runId,
         },
-        body: updates,
+        body: updates as Record<string, unknown>,
       },
     );
     return data as unknown as TaskRun;
@@ -383,6 +383,40 @@ export class PostHogAPIClient {
     }
   }
 
+  async getTaskRunSessionLogs(
+    taskId: string,
+    runId: string,
+    options?: { limit?: number; after?: string },
+  ): Promise<StoredLogEntry[]> {
+    try {
+      const teamId = await this.getTeamId();
+      const url = new URL(
+        `${this.api.baseUrl}/api/projects/${teamId}/tasks/${taskId}/runs/${runId}/session_logs/`,
+      );
+      url.searchParams.set("limit", String(options?.limit ?? 5000));
+      if (options?.after) {
+        url.searchParams.set("after", options.after);
+      }
+      const response = await this.api.fetcher.fetch({
+        method: "get",
+        url,
+        path: `/api/projects/${teamId}/tasks/${taskId}/runs/${runId}/session_logs/`,
+      });
+
+      if (!response.ok) {
+        log.warn(
+          `Failed to fetch session logs: ${response.status} ${response.statusText}`,
+        );
+        return [];
+      }
+
+      return (await response.json()) as StoredLogEntry[];
+    } catch (err) {
+      log.warn("Failed to fetch task run session logs", err);
+      return [];
+    }
+  }
+
   async getTaskLogs(taskId: string): Promise<StoredLogEntry[]> {
     try {
       const task = (await this.getTask(taskId)) as unknown as Task;

apps/twig/src/main/di/container.ts

@@ -2,6 +2,7 @@ import "reflect-metadata";
 import { Container } from "inversify";
 import { AgentService } from "../services/agent/service.js";
 import { AppLifecycleService } from "../services/app-lifecycle/service.js";
+import { CloudTaskService } from "../services/cloud-task/service.js";
 import { ConnectivityService } from "../services/connectivity/service.js";
 import { ContextMenuService } from "../services/context-menu/service.js";
 import { DeepLinkService } from "../services/deep-link/service.js";
@@ -32,6 +33,7 @@ export const container = new Container({
 
 container.bind(MAIN_TOKENS.AgentService).to(AgentService);
 container.bind(MAIN_TOKENS.AppLifecycleService).to(AppLifecycleService);
+container.bind(MAIN_TOKENS.CloudTaskService).to(CloudTaskService);
 container.bind(MAIN_TOKENS.ConnectivityService).to(ConnectivityService);
 container.bind(MAIN_TOKENS.ContextMenuService).to(ContextMenuService);
 container.bind(MAIN_TOKENS.DeepLinkService).to(DeepLinkService);

apps/twig/src/main/di/tokens.ts

@@ -8,6 +8,7 @@ export const MAIN_TOKENS = Object.freeze({
   // Services
   AgentService: Symbol.for("Main.AgentService"),
   AppLifecycleService: Symbol.for("Main.AppLifecycleService"),
+  CloudTaskService: Symbol.for("Main.CloudTaskService"),
   ConnectivityService: Symbol.for("Main.ConnectivityService"),
   ContextMenuService: Symbol.for("Main.ContextMenuService"),
 

apps/twig/src/main/services/cloud-task/schemas.ts

@@ -0,0 +1,45 @@
+import { z } from "zod";
+import type { CloudTaskUpdatePayload, TaskRun } from "../../../shared/types.js";
+
+export type { CloudTaskUpdatePayload };
+
+// --- Terminal statuses ---
+
+export const TERMINAL_STATUSES = ["completed", "failed", "cancelled"] as const;
+
+// --- Events ---
+
+export const CloudTaskEvent = {
+  Update: "cloud-task-update",
+} as const;
+
+export interface CloudTaskEvents {
+  [CloudTaskEvent.Update]: CloudTaskUpdatePayload;
+}
+
+export type TaskRunStatus = TaskRun["status"];
+
+// --- tRPC Schemas ---
+
+export const watchInput = z.object({
+  taskId: z.string(),
+  runId: z.string(),
+  apiHost: z.string(),
+  teamId: z.number(),
+});
+
+export type WatchInput = z.infer<typeof watchInput>;
+
+export const unwatchInput = z.object({
+  taskId: z.string(),
+  runId: z.string(),
+});
+
+export const updateTokenInput = z.object({
+  token: z.string(),
+});
+
+export const onUpdateInput = z.object({
+  taskId: z.string(),
+  runId: z.string(),
+});