refactor(mobile): keep presentation logic local

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

Author: richardsolomou

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

@@ -1,14 +1,7 @@
 import { Text } from "@components/text";
-import { getAutomationStatusPresentation } from "@posthog/core/automations/automationStatus";
 import type { TaskRun } from "@posthog/shared";
 import { View } from "react-native";
-
-const STATUS_TONE_CLASSES = {
-  neutral: "bg-gray-4 text-gray-11",
-  warning: "bg-status-warning/20 text-status-warning",
-  success: "bg-status-success/20 text-status-success",
-  error: "bg-status-error/20 text-status-error",
-} as const;
+import { getAutomationStatusPresentation } from "../utils/automationStatus";
 
 interface AutomationStatusBadgeProps {
   enabled: boolean;
@@ -25,9 +18,6 @@ export function AutomationStatusBadge({
     lastRunStatus,
     lastTaskRunStatus,
   });
-  const runStatusClassName = runStatus
-    ? STATUS_TONE_CLASSES[runStatus.tone]
-    : null;
 
   return (
     <View className="flex-row flex-wrap gap-2">
@@ -42,9 +32,9 @@ export function AutomationStatusBadge({
           {enabled ? "Enabled" : "Paused"}
         </Text>
       </View>
-      {runStatus && runStatusClassName ? (
-        <View className={`rounded px-1.5 py-0.5 ${runStatusClassName}`}>
-          <Text className={`text-xs ${runStatusClassName.split(" ")[1]}`}>
+      {runStatus ? (
+        <View className={`rounded px-1.5 py-0.5 ${runStatus.className}`}>
+          <Text className={`text-xs ${runStatus.className.split(" ")[1]}`}>
             {runStatus.label}
           </Text>
         </View>

apps/mobile/src/features/tasks/components/TaskStatusIcon.test.ts

@@ -0,0 +1,87 @@
+import type { Task } from "@posthog/shared";
+import { describe, expect, it } from "vitest";
+import { getTaskStatusIconKind } from "./taskStatusIconKind";
+
+function makeTask(latestRun?: Partial<NonNullable<Task["latest_run"]>>): Task {
+  return {
+    id: "task-1",
+    task_number: 1,
+    slug: "task-1",
+    title: "Test task",
+    description: "",
+    created_at: "2026-01-01T00:00:00Z",
+    updated_at: "2026-01-01T00:00:00Z",
+    origin_product: "code",
+    latest_run: latestRun
+      ? {
+          id: "run-1",
+          task: "task-1",
+          team: 1,
+          branch: null,
+          stage: null,
+          environment: "local",
+          status: "not_started",
+          log_url: "",
+          error_message: null,
+          output: null,
+          state: {},
+          created_at: "2026-01-01T00:00:00Z",
+          updated_at: "2026-01-01T00:00:00Z",
+          completed_at: null,
+          ...latestRun,
+        }
+      : undefined,
+  };
+}
+
+describe("getTaskStatusIconKind", () => {
+  it("prioritizes PR over cloud status", () => {
+    const task = makeTask({
+      environment: "cloud",
+      status: "in_progress",
+      output: { pr_url: "https://github.com/PostHog/code/pull/123" },
+    });
+
+    expect(getTaskStatusIconKind(task)).toBe("pr");
+  });
+
+  it("shows chat for cloud tasks without a PR, regardless of run status", () => {
+    expect(
+      getTaskStatusIconKind(
+        makeTask({ environment: "cloud", status: "queued" }),
+      ),
+    ).toBe("chat");
+    expect(
+      getTaskStatusIconKind(
+        makeTask({ environment: "cloud", status: "in_progress" }),
+      ),
+    ).toBe("chat");
+    expect(
+      getTaskStatusIconKind(
+        makeTask({ environment: "cloud", status: "completed" }),
+      ),
+    ).toBe("chat");
+    expect(
+      getTaskStatusIconKind(
+        makeTask({ environment: "cloud", status: "cancelled" }),
+      ),
+    ).toBe("chat");
+  });
+
+  it("preserves local run-state icons", () => {
+    expect(
+      getTaskStatusIconKind(
+        makeTask({ environment: "local", status: "in_progress" }),
+      ),
+    ).toBe("running");
+    expect(
+      getTaskStatusIconKind(
+        makeTask({ environment: "local", status: "failed" }),
+      ),
+    ).toBe("failed");
+  });
+
+  it("falls back to chat when a task has no run yet", () => {
+    expect(getTaskStatusIconKind(makeTask())).toBe("chat");
+  });
+});

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

@@ -1,4 +1,3 @@
-import { getTaskStatusPresentationKind } from "@posthog/core/tasks/taskStatusPresentation";
 import type { Task } from "@posthog/shared";
 import {
   ChatCircle,
@@ -11,6 +10,7 @@ import {
 import { memo, useEffect, useRef } from "react";
 import { Animated, Easing } from "react-native";
 import { useThemeColors } from "@/lib/theme";
+import { getTaskStatusIconKind } from "./taskStatusIconKind";
 
 interface TaskStatusIconProps {
   task: Task;
@@ -19,7 +19,7 @@ interface TaskStatusIconProps {
 
 function TaskStatusIconComponent({ task, size = 16 }: TaskStatusIconProps) {
   const colors = useThemeColors();
-  const iconKind = getTaskStatusPresentationKind(task);
+  const iconKind = getTaskStatusIconKind(task);
 
   const rotation = useRef(new Animated.Value(0)).current;
   const isRunning = iconKind === "running";

apps/mobile/src/features/tasks/components/taskStatusIconKind.ts

@@ -0,0 +1,41 @@
+import type { Task } from "@posthog/shared";
+
+export type TaskStatusIconKind =
+  | "pr"
+  | "completed"
+  | "failed"
+  | "running"
+  | "started"
+  | "chat";
+
+export function getTaskStatusIconKind(task: Task): TaskStatusIconKind {
+  const prUrl = task.latest_run?.output?.pr_url as string | undefined;
+  const status = task.latest_run?.status;
+  const environment = task.latest_run?.environment;
+
+  if (prUrl) {
+    return "pr";
+  }
+
+  if (environment === "cloud") {
+    return "chat";
+  }
+
+  if (status === "completed") {
+    return "completed";
+  }
+
+  if (status === "failed") {
+    return "failed";
+  }
+
+  if (status === "in_progress") {
+    return "running";
+  }
+
+  if (status === "queued") {
+    return "started";
+  }
+
+  return "chat";
+}

apps/mobile/src/features/tasks/utils/automationStatus.test.ts

@@ -0,0 +1,34 @@
+import { describe, expect, it } from "vitest";
+import { getAutomationStatusPresentation } from "./automationStatus";
+
+describe("automationStatus", () => {
+  it("shows queued when the linked task run has not started work yet", () => {
+    expect(
+      getAutomationStatusPresentation({
+        lastRunStatus: "running",
+        lastTaskRunStatus: "queued",
+      }),
+    ).toMatchObject({ label: "Queued" });
+  });
+
+  it("hides the running badge while the linked task run is actively in progress", () => {
+    expect(
+      getAutomationStatusPresentation({
+        lastRunStatus: "running",
+        lastTaskRunStatus: "in_progress",
+      }),
+    ).toBeNull();
+  });
+
+  it("hides the running badge when only the automation-level status is available", () => {
+    expect(
+      getAutomationStatusPresentation({ lastRunStatus: "running" }),
+    ).toBeNull();
+  });
+
+  it("falls back to automation status when task-run detail is unavailable", () => {
+    expect(
+      getAutomationStatusPresentation({ lastRunStatus: "success" }),
+    ).toMatchObject({ label: "Success" });
+  });
+});

…core/src/automations/automationStatus.ts …features/tasks/utils/automationStatus.tspackages/core/src/automations/automationStatus.ts renamed to apps/mobile/src/features/tasks/utils/automationStatus.ts packages/core/src/automations/automationStatus.ts renamed to apps/mobile/src/features/tasks/utils/automationStatus.ts

@@ -1,29 +1,13 @@
-export type AutomationTaskRunStatus =
-  | "not_started"
-  | "queued"
-  | "started"
-  | "in_progress"
-  | "completed"
-  | "failed"
-  | "cancelled";
+import type { TaskRun } from "@posthog/shared";
 
 export interface AutomationStatusInput {
   lastRunStatus: string | null;
-  lastTaskRunStatus?: AutomationTaskRunStatus | null;
+  lastTaskRunStatus?: TaskRun["status"] | null;
 }
 
-export type AutomationStatusTone = "neutral" | "warning" | "success" | "error";
-
-export type AutomationStatusIconKind =
-  | "queued"
-  | "success"
-  | "failed"
-  | "never-run";
-
 export interface AutomationStatusPresentation {
   label: string;
-  tone: AutomationStatusTone;
-  iconKind: AutomationStatusIconKind;
+  className: string;
 }
 
 export function getAutomationStatusPresentation({
@@ -35,24 +19,20 @@ export function getAutomationStatusPresentation({
     case "queued":
       return {
         label: "Queued",
-        tone: "warning",
-        iconKind: "queued",
+        className: "bg-status-warning/20 text-status-warning",
       };
-    case "started":
     case "in_progress":
       return null;
     case "completed":
       return {
         label: "Success",
-        tone: "success",
-        iconKind: "success",
+        className: "bg-status-success/20 text-status-success",
       };
     case "failed":
     case "cancelled":
       return {
         label: "Failed",
-        tone: "error",
-        iconKind: "failed",
+        className: "bg-status-error/20 text-status-error",
       };
     default:
       break;
@@ -64,20 +44,17 @@ export function getAutomationStatusPresentation({
     case "success":
       return {
         label: "Success",
-        tone: "success",
-        iconKind: "success",
+        className: "bg-status-success/20 text-status-success",
       };
     case "failed":
       return {
         label: "Failed",
-        tone: "error",
-        iconKind: "failed",
+        className: "bg-status-error/20 text-status-error",
       };
     default:
       return {
         label: "Never run",
-        tone: "neutral",
-        iconKind: "never-run",
+        className: "bg-gray-4 text-gray-11",
       };
   }
 }