feat: cleanup logging + agent logs (#166)

Author: joshsny

apps/array/src/api/generated.ts

@@ -19467,30 +19467,3 @@ export class ApiClient {
 export function createApiClient(fetcher: Fetcher, baseUrl?: string) {
   return new ApiClient(fetcher).setBaseUrl(baseUrl ?? "");
 }
-
-/**
- Example usage:
- const api = createApiClient((method, url, params) =>
-   fetch(url, { method, body: JSON.stringify(params) }).then((res) => res.json()),
- );
- api.get("/users").then((users) => console.log(users));
- api.post("/users", { body: { name: "John" } }).then((user) => console.log(user));
- api.put("/users/:id", { path: { id: 1 }, body: { name: "John" } }).then((user) => console.log(user));
-
- // With error handling
- const result = await api.get("/users/{id}", { path: { id: "123" }, withResponse: true });
- if (result.ok) {
-   // Access data directly
-   const user = result.data;
-   console.log(user);
-
-   // Or use the json() method for compatibility
-   const userFromJson = await result.json();
-   console.log(userFromJson);
- } else {
-   const error = result.data;
-   console.error(`Error ${result.status}:`, error);
- }
-*/
-
-// </ApiClient>

apps/array/src/api/posthogClient.ts

@@ -1,7 +1,10 @@
+import { logger } from "@renderer/lib/logger";
 import type { LogEntry, RepositoryConfig, Task, TaskRun } from "@shared/types";
 import { buildApiFetcher } from "./fetcher";
 import { createApiClient, type Schemas } from "./generated";
 
+const log = logger.scope("posthog-client");
+
 export class PostHogAPIClient {
   private api: ReturnType<typeof createApiClient>;
   private _teamId: number | null = null;
@@ -195,7 +198,7 @@ export class PostHogAPIClient {
       const response = await fetch(logUrl);
 
       if (!response.ok) {
-        console.warn(
+        log.warn(
           `Failed to fetch logs: ${response.status} ${response.statusText}`,
         );
         return [];
@@ -211,7 +214,7 @@ export class PostHogAPIClient {
         .split("\n")
         .map((line) => JSON.parse(line) as LogEntry);
     } catch (err) {
-      console.warn("Failed to fetch task logs from latest run", err);
+      log.warn("Failed to fetch task logs from latest run", err);
       return [];
     }
   }

apps/array/src/main/index.ts

@@ -10,6 +10,7 @@ import {
   type MenuItemConstructorOptions,
   shell,
 } from "electron";
+import "./lib/logger";
 import { ANALYTICS_EVENTS } from "../types/analytics.js";
 import { registerAgentIpc, type TaskController } from "./services/agent.js";
 import { setupAgentHotReload } from "./services/dev-reload.js";

apps/array/src/main/lib/logger.ts

@@ -0,0 +1,31 @@
+import log from "electron-log/main";
+
+// Initialize IPC transport to forward main process logs to renderer dev tools
+log.initialize();
+
+log.transports.file.level = "info";
+log.transports.console.level = "info";
+
+export const logger = {
+  info: (message: string, ...args: unknown[]) => log.info(message, ...args),
+  warn: (message: string, ...args: unknown[]) => log.warn(message, ...args),
+  error: (message: string, ...args: unknown[]) => log.error(message, ...args),
+  debug: (message: string, ...args: unknown[]) => log.debug(message, ...args),
+
+  scope: (name: string) => {
+    const scoped = log.scope(name);
+    return {
+      info: (message: string, ...args: unknown[]) =>
+        scoped.info(message, ...args),
+      warn: (message: string, ...args: unknown[]) =>
+        scoped.warn(message, ...args),
+      error: (message: string, ...args: unknown[]) =>
+        scoped.error(message, ...args),
+      debug: (message: string, ...args: unknown[]) =>
+        scoped.debug(message, ...args),
+    };
+  },
+};
+
+export type Logger = typeof logger;
+export type ScopedLogger = ReturnType<typeof logger.scope>;

apps/array/src/main/preload.ts

@@ -6,6 +6,7 @@ import type {
   TabContextMenuResult,
   TaskContextMenuResult,
 } from "./services/contextMenu.types.js";
+import "electron-log/preload";
 
 interface MessageBoxOptions {
   type?: "none" | "info" | "error" | "question" | "warning";

apps/array/src/main/services/agent.ts

@@ -2,13 +2,25 @@ import { randomUUID } from "node:crypto";
 import { mkdirSync, rmSync, symlinkSync } from "node:fs";
 import { tmpdir } from "node:os";
 import { join } from "node:path";
-import { Agent, PermissionMode } from "@posthog/agent";
+import { Agent, type OnLogCallback, PermissionMode } from "@posthog/agent";
 import {
   app,
   type BrowserWindow,
   type IpcMainInvokeEvent,
   ipcMain,
 } from "electron";
+import { logger } from "../lib/logger";
+
+const log = logger.scope("agent");
+
+const onAgentLog: OnLogCallback = (level, scope, message, data) => {
+  const scopedLog = logger.scope(scope);
+  if (data !== undefined) {
+    scopedLog[level](message, data);
+  } else {
+    scopedLog[level](message);
+  }
+};
 
 interface AgentStartParams {
   taskId: string;
@@ -96,12 +108,9 @@ export function registerAgentIpc(
           process.env.CLAUDE_CONFIG_DIR || join(app.getPath("home"), ".claude");
         const statsigPath = join(claudeConfigDir, "statsig");
         rmSync(statsigPath, { recursive: true, force: true });
-        console.log(
-          "[agent] Cleared statsig cache to work around input_examples bug",
-        );
+        log.info("Cleared statsig cache to work around input_examples bug");
       } catch (error) {
-        // Ignore errors if the folder doesn't exist
-        console.warn("[agent] Could not clear statsig cache:", error);
+        log.warn("Could not clear statsig cache:", error);
       }
 
       const taskId = randomUUID();
@@ -123,7 +132,7 @@ export function registerAgentIpc(
         if (stderrBuffer.length > 50) {
           stderrBuffer.shift();
         }
-        console.error(`[agent][claude-stderr] ${text}`);
+        log.error(`[claude-stderr] ${text}`);
         emitToRenderer({
           type: "status",
           ts: Date.now(),
@@ -146,6 +155,7 @@ export function registerAgentIpc(
         posthogApiUrl: apiHost,
         posthogProjectId: projectId,
         debug: true,
+        onLog: onAgentLog,
       });
 
       const controllerEntry: TaskController = {
@@ -181,7 +191,7 @@ export function registerAgentIpc(
               });
             }
           } catch (err) {
-            console.warn("[agent] failed to fetch task progress", err);
+            log.warn("Failed to fetch task progress", err);
           }
         };
 
@@ -214,7 +224,7 @@ export function registerAgentIpc(
             } catch {}
             symlinkSync(process.execPath, nodeSymlinkPath);
           } catch (err) {
-            console.warn("[agent] Failed to setup mock node environment", err);
+            log.warn("Failed to setup mock node environment", err);
           }
 
           const newPath = `${mockNodeDir}:${process.env.PATH || ""}`;
@@ -259,7 +269,7 @@ export function registerAgentIpc(
 
           emitToRenderer({ type: "done", success: true, ts: Date.now() });
         } catch (err) {
-          console.error("[agent] task execution failed", err);
+          log.error("Task execution failed", err);
           let errorMessage = err instanceof Error ? err.message : String(err);
           const cause =
             err instanceof Error && "cause" in err && err.cause

apps/array/src/main/services/fileWatcher.ts

@@ -7,6 +7,9 @@ import {
   type IpcMainInvokeEvent,
   ipcMain,
 } from "electron";
+import { logger } from "../lib/logger";
+
+const log = logger.scope("file-watcher");
 
 const WATCHER_IGNORE_PATTERNS = ["**/node_modules/**", "**/.git/**"];
 const DEBOUNCE_MS = 100;
@@ -72,7 +75,7 @@ class FileService {
           return a.name.localeCompare(b.name);
         });
     } catch (error) {
-      console.error("Failed to list directory:", error);
+      log.error("Failed to list directory:", error);
       return [];
     }
   }
@@ -115,7 +118,7 @@ class FileService {
       repoPath,
       (err, events) => {
         if (err) {
-          console.error("Watcher error:", err);
+          log.error("Watcher error:", err);
           return;
         }
 
@@ -162,7 +165,7 @@ class FileService {
         gitDirToWatch,
         (err, events) => {
           if (err) {
-            console.error("Git watcher error:", err);
+            log.error("Git watcher error:", err);
             return;
           }
           if (
@@ -175,7 +178,7 @@ class FileService {
         },
       );
     } catch (error) {
-      console.warn("Failed to set up git watcher:", error);
+      log.warn("Failed to set up git watcher:", error);
     }
 
     state.subscription = subscription;
@@ -196,7 +199,7 @@ class FileService {
           ignore: WATCHER_IGNORE_PATTERNS,
         });
       } catch (error) {
-        console.error("Failed to write snapshot:", error);
+        log.error("Failed to write snapshot:", error);
       }
 
       await state.subscription?.unsubscribe();

apps/array/src/main/services/folders.ts

@@ -5,8 +5,11 @@ import type {
   TaskFolderAssociation,
   WorktreeInfo,
 } from "../../shared/types";
+import { logger } from "../lib/logger";
 import { clearAllStoreData, foldersStore } from "./store";
 
+const log = logger.scope("folders");
+
 function generateFolderId(): string {
   return `folder_${Date.now()}_${Math.random().toString(36).substring(2, 9)}`;
 }
@@ -54,7 +57,7 @@ async function removeFolder(folderId: string): Promise<void> {
 
   foldersStore.set("folders", filtered);
   foldersStore.set("taskAssociations", filteredAssociations);
-  console.log(`Removed folder with ID: ${folderId}`);
+  log.debug(`Removed folder with ID: ${folderId}`);
 }
 
 async function updateFolderAccessed(folderId: string): Promise<void> {
@@ -148,7 +151,7 @@ export function registerFoldersIpc(): void {
       try {
         return await getFolders();
       } catch (error) {
-        console.error("Failed to get folders:", error);
+        log.error("Failed to get folders:", error);
         return [];
       }
     },
@@ -163,7 +166,7 @@ export function registerFoldersIpc(): void {
       try {
         return await addFolder(folderPath);
       } catch (error) {
-        console.error(`Failed to add folder ${folderPath}:`, error);
+        log.error(`Failed to add folder ${folderPath}:`, error);
         throw error;
       }
     },
@@ -175,7 +178,7 @@ export function registerFoldersIpc(): void {
       try {
         await removeFolder(folderId);
       } catch (error) {
-        console.error(`Failed to remove folder ${folderId}:`, error);
+        log.error(`Failed to remove folder ${folderId}:`, error);
         throw error;
       }
     },
@@ -187,7 +190,7 @@ export function registerFoldersIpc(): void {
       try {
         await updateFolderAccessed(folderId);
       } catch (error) {
-        console.error(`Failed to update folder with ID: ${folderId}:`, error);
+        log.error(`Failed to update folder with ID: ${folderId}:`, error);
       }
     },
   );
@@ -197,9 +200,9 @@ export function registerFoldersIpc(): void {
     async (_event: IpcMainInvokeEvent): Promise<void> => {
       try {
         clearAllStoreData();
-        console.log("Cleared all application data");
+        log.info("Cleared all application data");
       } catch (error) {
-        console.error("Failed to clear all data:", error);
+        log.error("Failed to clear all data:", error);
         throw error;
       }
     },
@@ -211,7 +214,7 @@ export function registerFoldersIpc(): void {
       try {
         return await getTaskAssociations();
       } catch (error) {
-        console.error("Failed to get task associations:", error);
+        log.error("Failed to get task associations:", error);
         return [];
       }
     },
@@ -226,7 +229,7 @@ export function registerFoldersIpc(): void {
       try {
         return await getTaskAssociation(taskId);
       } catch (error) {
-        console.error(`Failed to get task association for ${taskId}:`, error);
+        log.error(`Failed to get task association for ${taskId}:`, error);
         return null;
       }
     },
@@ -244,7 +247,7 @@ export function registerFoldersIpc(): void {
       try {
         return await setTaskAssociation(taskId, folderId, folderPath, worktree);
       } catch (error) {
-        console.error(`Failed to set task association for ${taskId}:`, error);
+        log.error(`Failed to set task association for ${taskId}:`, error);
         throw error;
       }
     },
@@ -260,7 +263,7 @@ export function registerFoldersIpc(): void {
       try {
         return await updateTaskWorktree(taskId, worktree);
       } catch (error) {
-        console.error(`Failed to update worktree for ${taskId}:`, error);
+        log.error(`Failed to update worktree for ${taskId}:`, error);
         return null;
       }
     },
@@ -272,10 +275,7 @@ export function registerFoldersIpc(): void {
       try {
         await removeTaskAssociation(taskId);
       } catch (error) {
-        console.error(
-          `Failed to remove task association for ${taskId}:`,
-          error,
-        );
+        log.error(`Failed to remove task association for ${taskId}:`, error);
         throw error;
       }
     },
@@ -287,7 +287,7 @@ export function registerFoldersIpc(): void {
       try {
         await clearTaskWorktree(taskId);
       } catch (error) {
-        console.error(`Failed to clear worktree for ${taskId}:`, error);
+        log.error(`Failed to clear worktree for ${taskId}:`, error);
         throw error;
       }
     },