refactor(mobile): reuse task API client

Move task runs and automations onto the existing API client and derive automation contracts from its generated schema.

Generated-By: PostHog Code
Task-Id: c1bbe3cf-742b-4b24-bf96-d11a18b4cf22

Author: richardsolomou

apps/mobile/src/features/tasks/api.test.ts

@@ -1,7 +1,8 @@
 import { beforeEach, describe, expect, it, vi } from "vitest";
 
-const { mockFetch } = vi.hoisted(() => ({
+const { mockFetch, mockRunTaskInCloud } = vi.hoisted(() => ({
   mockFetch: vi.fn(),
+  mockRunTaskInCloud: vi.fn(),
 }));
 
 vi.mock("expo/fetch", () => ({
@@ -34,6 +35,10 @@ vi.mock("@/lib/api", () => ({
     }),
 }));
 
+vi.mock("@/lib/posthogApiClient", () => ({
+  getPostHogApiClient: () => ({ runTaskInCloud: mockRunTaskInCloud }),
+}));
+
 import { cancelRun, HttpError, runTaskInCloud } from "./api";
 
 function bodyOf(call: unknown): Record<string, unknown> {
@@ -43,34 +48,37 @@ function bodyOf(call: unknown): Record<string, unknown> {
 
 describe("runTaskInCloud", () => {
   beforeEach(() => {
-    mockFetch.mockReset();
-    mockFetch.mockResolvedValue(
-      new Response(JSON.stringify({ id: "task-1" }), { status: 200 }),
-    );
+    mockRunTaskInCloud.mockReset();
+    mockRunTaskInCloud.mockResolvedValue({ id: "task-1" });
   });
 
   it.each([true, false])(
     "forwards auto_publish=%s to the payload",
     async (flag) => {
       await runTaskInCloud("task-1", { autoPublish: flag });
 
-      expect(bodyOf(mockFetch.mock.calls[0])).toMatchObject({
-        auto_publish: flag,
-      });
+      expect(mockRunTaskInCloud).toHaveBeenCalledWith(
+        "task-1",
+        undefined,
+        expect.objectContaining({ autoPublish: flag }),
+      );
     },
   );
 
   it("omits auto_publish when not provided", async () => {
     await runTaskInCloud("task-1", { model: "claude-opus-4-8" });
 
-    expect(bodyOf(mockFetch.mock.calls[0])).not.toHaveProperty("auto_publish");
+    expect(mockRunTaskInCloud).toHaveBeenCalledWith(
+      "task-1",
+      undefined,
+      expect.objectContaining({ autoPublish: undefined }),
+    );
   });
 
   it("sends no body for the plain initial run", async () => {
     await runTaskInCloud("task-1");
 
-    const [, init] = mockFetch.mock.calls[0] as [string, RequestInit];
-    expect(init.body).toBeUndefined();
+    expect(mockRunTaskInCloud).toHaveBeenCalledWith("task-1");
   });
 
   it("forwards the selected sandbox environment and custom image", async () => {
@@ -79,10 +87,14 @@ describe("runTaskInCloud", () => {
       customImageId: "image-123",
     });
 
-    expect(bodyOf(mockFetch.mock.calls[0])).toMatchObject({
-      sandbox_environment_id: "environment-123",
-      custom_image_id: "image-123",
-    });
+    expect(mockRunTaskInCloud).toHaveBeenCalledWith(
+      "task-1",
+      undefined,
+      expect.objectContaining({
+        sandboxEnvironmentId: "environment-123",
+        customImageId: "image-123",
+      }),
+    );
   });
 
   it("omits the sandbox environment and custom image when unset", async () => {
@@ -92,24 +104,34 @@ describe("runTaskInCloud", () => {
       customImageId: null,
     });
 
-    const body = bodyOf(mockFetch.mock.calls[0]);
-    expect(body).not.toHaveProperty("sandbox_environment_id");
-    expect(body).not.toHaveProperty("custom_image_id");
+    expect(mockRunTaskInCloud).toHaveBeenCalledWith(
+      "task-1",
+      undefined,
+      expect.objectContaining({
+        sandboxEnvironmentId: undefined,
+        customImageId: undefined,
+      }),
+    );
   });
 
   it("sends rtk_enabled=false when the run opts out", async () => {
     await runTaskInCloud("task-1", { rtkEnabled: false });
 
-    expect(bodyOf(mockFetch.mock.calls[0])).toMatchObject({
-      rtk_enabled: false,
-    });
+    expect(mockRunTaskInCloud).toHaveBeenCalledWith(
+      "task-1",
+      undefined,
+      expect.objectContaining({ rtkEnabled: false }),
+    );
   });
 
   it("omits rtk_enabled when the run keeps compression on", async () => {
     await runTaskInCloud("task-1", { rtkEnabled: true });
 
-    const [, init] = mockFetch.mock.calls[0] as [string, RequestInit];
-    expect(init.body).toBeUndefined();
+    expect(mockRunTaskInCloud).toHaveBeenCalledWith(
+      "task-1",
+      undefined,
+      expect.objectContaining({ rtkEnabled: true }),
+    );
   });
 });
 

apps/mobile/src/features/tasks/api.ts

@@ -1,4 +1,10 @@
-import type { Adapter, StoredLogEntry, Task, TaskRun } from "@posthog/shared";
+import type {
+  Adapter,
+  ExecutionMode,
+  StoredLogEntry,
+  Task,
+  TaskRun,
+} from "@posthog/shared";
 import { fetch } from "expo/fetch";
 import {
   authedFetch,
@@ -8,6 +14,7 @@ import {
   getProjectId,
   HttpError,
 } from "@/lib/api";
+import { getPostHogApiClient } from "@/lib/posthogApiClient";
 
 export { HttpError } from "@/lib/api";
 
@@ -78,84 +85,34 @@ export interface RunTaskInCloudOptions {
   autoPublish?: boolean;
   /** Only false is sent: opts the run out of rtk command-output compression. */
   rtkEnabled?: boolean;
+  sandboxEnvironmentId?: string | null;
+  customImageId?: string | null;
 }
 
 export async function runTaskInCloud(
   taskId: string,
   options?: RunTaskInCloudOptions,
 ): Promise<Task> {
-  const baseUrl = getBaseUrl();
-  const projectId = getProjectId();
-
-  // Only serialize a body when we have options to send. Sending an empty
-  // or minimal body on the initial run historically changed backend
-  // behavior, so we preserve the "no body" path for the common case.
-  const hasOptions =
-    !!options &&
-    (options.branch !== undefined ||
-      options.resumeFromRunId !== undefined ||
-      options.pendingUserMessage !== undefined ||
-      options.mode !== undefined ||
-      options.runtimeAdapter !== undefined ||
-      options.model !== undefined ||
-      options.reasoningEffort !== undefined ||
-      options.initialPermissionMode !== undefined ||
-      options.runSource !== undefined ||
-      options.signalReportId !== undefined ||
-      options.autoPublish !== undefined ||
-      options.rtkEnabled === false);
-
-  let body: string | undefined;
-  if (hasOptions) {
-    const payload: Record<string, unknown> = {
-      mode: options?.mode ?? "interactive",
-    };
-    if (options?.branch) payload.branch = options.branch;
-    if (options?.resumeFromRunId) {
-      payload.resume_from_run_id = options.resumeFromRunId;
-    }
-    if (options?.pendingUserMessage) {
-      payload.pending_user_message = options.pendingUserMessage;
-    }
-    if (options?.runtimeAdapter) {
-      payload.runtime_adapter = options.runtimeAdapter;
-      if (options?.model) payload.model = options.model;
-      if (options?.reasoningEffort) {
-        payload.reasoning_effort = options.reasoningEffort;
-      }
-    }
-    if (options?.initialPermissionMode) {
-      payload.initial_permission_mode = options.initialPermissionMode;
-    }
-    if (options?.runSource) payload.run_source = options.runSource;
-    if (options?.signalReportId)
-      payload.signal_report_id = options.signalReportId;
-    if (options?.autoPublish !== undefined) {
-      payload.auto_publish = options.autoPublish;
-    }
-    if (options?.rtkEnabled === false) {
-      payload.rtk_enabled = false;
-    }
-    body = JSON.stringify(payload);
+  if (!options) {
+    return getPostHogApiClient().runTaskInCloud(taskId);
   }
 
-  const response = await authedFetch(
-    `${baseUrl}/api/projects/${projectId}/tasks/${taskId}/run/`,
-    {
-      method: "POST",
-      body,
-    },
-  );
-
-  if (!response.ok) {
-    throw new HttpError(
-      response.status,
-      response.statusText,
-      "Failed to run task",
-    );
-  }
-
-  return await response.json();
+  return getPostHogApiClient().runTaskInCloud(taskId, options.branch, {
+    adapter: options.runtimeAdapter,
+    model: options.model,
+    reasoningLevel: options.reasoningEffort,
+    initialPermissionMode: options.initialPermissionMode as
+      | ExecutionMode
+      | undefined,
+    runSource: options.runSource,
+    signalReportId: options.signalReportId,
+    autoPublish: options.autoPublish,
+    rtkEnabled: options.rtkEnabled,
+    sandboxEnvironmentId: options.sandboxEnvironmentId ?? undefined,
+    customImageId: options.customImageId ?? undefined,
+    resumeFromRunId: options.resumeFromRunId,
+    pendingUserMessage: options.pendingUserMessage,
+  });
 }
 
 export async function getTaskRun(

apps/mobile/src/features/tasks/components/AutomationDetail.tsx

@@ -1,7 +1,7 @@
 import { Text } from "@components/text";
+import type { TaskAutomation } from "@posthog/api-client/posthog-client";
 import type { TaskRun } from "@posthog/shared";
 import { ActivityIndicator, Pressable, View } from "react-native";
-import type { TaskAutomation } from "../types";
 import { formatAutomationScheduleSummary } from "../utils/automationSchedule";
 import { getAutomationTemplatePresentation } from "../utils/automationTemplatePresentation";
 import { AutomationStatusBadge } from "./AutomationStatusBadge";

apps/mobile/src/features/tasks/components/AutomationForm.tsx

@@ -1,4 +1,5 @@
 import { Text } from "@components/text";
+import type { CreateTaskAutomationOptions } from "@posthog/api-client/posthog-client";
 import { CaretDown, GithubLogo } from "phosphor-react-native";
 import { type MutableRefObject, useEffect, useMemo, useState } from "react";
 import {
@@ -12,10 +13,7 @@ import { MarkdownText } from "@/features/chat/components/MarkdownText";
 import { useThemeColors } from "@/lib/theme";
 import { RepositoryPickerInline } from "../composer/RepositoryPickerInline";
 import { useIntegrations } from "../hooks/useIntegrations";
-import type {
-  CreateTaskAutomationOptions,
-  RepositorySelection,
-} from "../types";
+import type { RepositorySelection } from "../types";
 import {
   type AutomationScheduleDraft,
   buildCronExpression,

apps/mobile/src/features/tasks/components/AutomationItem.tsx

@@ -1,9 +1,9 @@
 import { Text } from "@components/text";
+import type { TaskAutomation } from "@posthog/api-client/posthog-client";
 import type { TaskRun } from "@posthog/shared";
 import { format, formatDistanceToNow } from "date-fns";
 import { memo } from "react";
 import { Pressable, View } from "react-native";
-import type { TaskAutomation } from "../types";
 import { formatAutomationScheduleSummary } from "../utils/automationSchedule";
 import { getAutomationTemplatePresentation } from "../utils/automationTemplatePresentation";
 import { AutomationStatusBadge } from "./AutomationStatusBadge";

apps/mobile/src/features/tasks/components/AutomationList.tsx

@@ -1,4 +1,5 @@
 import { Text } from "@components/text";
+import type { TaskAutomation } from "@posthog/api-client/posthog-client";
 import { Plus } from "phosphor-react-native";
 import {
   ActivityIndicator,
@@ -10,7 +11,6 @@ import {
 import { useThemeColors } from "@/lib/theme";
 import { useAutomations } from "../hooks/useAutomations";
 import { useTasks } from "../hooks/useTasks";
-import type { TaskAutomation } from "../types";
 import { AutomationItem } from "./AutomationItem";
 
 interface AutomationListProps {

apps/mobile/src/features/tasks/hooks/useAutomations.ts

@@ -1,12 +1,12 @@
-import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
-import { useAuthStore } from "@/features/auth";
-import { logger } from "@/lib/logger";
-import { getPostHogApiClient } from "@/lib/posthogApiClient";
 import type {
   CreateTaskAutomationOptions,
   TaskAutomation,
   UpdateTaskAutomationOptions,
-} from "../types";
+} from "@posthog/api-client/posthog-client";
+import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
+import { useAuthStore } from "@/features/auth";
+import { logger } from "@/lib/logger";
+import { getPostHogApiClient } from "@/lib/posthogApiClient";
 import { taskKeys } from "./useTasks";
 
 const log = logger.scope("automations-mutations");

apps/mobile/src/features/tasks/types.ts

@@ -5,47 +5,6 @@ import type {
   TaskRunStatus,
 } from "@posthog/shared";
 
-export interface TaskAutomation {
-  id: string;
-  name: string;
-  prompt: string;
-  repository: string;
-  github_integration?: number | null;
-  cron_expression: string;
-  timezone?: string | null;
-  template_id?: string | null;
-  enabled: boolean;
-  last_run_at: string | null;
-  last_run_status: string | null;
-  last_task_id: string | null;
-  last_task_run_id: string | null;
-  last_error: string | null;
-  created_at: string;
-  updated_at: string;
-}
-
-export interface CreateTaskAutomationOptions {
-  name: string;
-  prompt: string;
-  repository: string;
-  github_integration?: number | null;
-  cron_expression: string;
-  timezone: string;
-  enabled?: boolean;
-  template_id?: string | null;
-}
-
-export interface UpdateTaskAutomationOptions {
-  name?: string;
-  prompt?: string;
-  repository?: string;
-  github_integration?: number | null;
-  cron_expression?: string;
-  timezone?: string;
-  enabled?: boolean;
-  template_id?: string | null;
-}
-
 export interface MobileStoredLogEntry extends SharedStoredLogEntry {
   direction?: "client" | "agent";
 }

apps/mobile/src/features/tasks/utils/automationSchedule.ts

@@ -1,4 +1,4 @@
-import type { TaskAutomation } from "../types";
+import type { TaskAutomation } from "@posthog/api-client/posthog-client";
 
 export type AutomationScheduleMode =
   | "hourly"

apps/mobile/src/features/tasks/utils/automationTemplatePresentation.ts

@@ -1,5 +1,5 @@
+import type { TaskAutomation } from "@posthog/api-client/posthog-client";
 import { parseSkillTemplateId } from "../skills/skillTemplateIds";
-import type { TaskAutomation } from "../types";
 
 export interface AutomationTemplatePresentation {
   templateName: string | null;