refactor: use subsystem logger in workspace-sync handler

This commit is contained in:
Ash Brener 2026-01-28 10:48:28 +02:00
parent 0d8c321d45
commit 5bdcbc7e32
2 changed files with 35 additions and 26 deletions

View File

@ -1,8 +1,23 @@
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import workspaceSyncHandler from "./handler.js";
import * as rclone from "../../../infra/rclone.js";
const mockLog = vi.hoisted(() => ({
info: vi.fn(),
warn: vi.fn(),
error: vi.fn(),
debug: vi.fn(),
trace: vi.fn(),
fatal: vi.fn(),
raw: vi.fn(),
child: vi.fn(),
subsystem: "workspace-sync",
}));
vi.mock("../../../logging/subsystem.js", () => ({
createSubsystemLogger: vi.fn(() => mockLog),
}));
vi.mock("../../../infra/rclone.js", () => ({
isRcloneInstalled: vi.fn(),
isRcloneConfigured: vi.fn(),
@ -18,6 +33,9 @@ vi.mock("../../../routing/session-key.js", () => ({
resolveAgentIdFromSessionKey: vi.fn(() => "default"),
}));
// Import after mocks are set up
const { default: workspaceSyncHandler } = await import("./handler.js");
describe("workspace-sync hook handler", () => {
beforeEach(() => {
vi.clearAllMocks();
@ -126,7 +144,6 @@ describe("workspace-sync hook handler", () => {
it("warns when rclone is not installed", async () => {
vi.mocked(rclone.isRcloneInstalled).mockResolvedValue(false);
const warnSpy = vi.spyOn(console, "warn").mockImplementation(() => {});
const event = {
type: "session",
@ -148,7 +165,7 @@ describe("workspace-sync hook handler", () => {
await workspaceSyncHandler(event as never);
expect(rclone.isRcloneInstalled).toHaveBeenCalled();
expect(warnSpy).toHaveBeenCalledWith("[workspace-sync] rclone not installed, skipping sync");
expect(mockLog.warn).toHaveBeenCalledWith("rclone not installed, skipping sync");
});
it("warns when rclone is not configured", async () => {
@ -166,7 +183,6 @@ describe("workspace-sync hook handler", () => {
onSessionStart: true,
onSessionEnd: false,
});
const warnSpy = vi.spyOn(console, "warn").mockImplementation(() => {});
const event = {
type: "session",
@ -187,7 +203,7 @@ describe("workspace-sync hook handler", () => {
await workspaceSyncHandler(event as never);
expect(warnSpy).toHaveBeenCalledWith(
expect(mockLog.warn).toHaveBeenCalledWith(
expect.stringContaining('rclone not configured for remote "cloud"'),
);
});
@ -208,7 +224,6 @@ describe("workspace-sync hook handler", () => {
onSessionEnd: false,
});
vi.mocked(rclone.runBisync).mockResolvedValue({ ok: true });
vi.spyOn(console, "log").mockImplementation(() => {});
const event = {
type: "session",
@ -255,7 +270,6 @@ describe("workspace-sync hook handler", () => {
onSessionEnd: true,
});
vi.mocked(rclone.runBisync).mockResolvedValue({ ok: true });
vi.spyOn(console, "log").mockImplementation(() => {});
const event = {
type: "session",
@ -298,8 +312,6 @@ describe("workspace-sync hook handler", () => {
ok: false,
error: "bisync requires --resync on first run",
});
vi.spyOn(console, "log").mockImplementation(() => {});
const warnSpy = vi.spyOn(console, "warn").mockImplementation(() => {});
const event = {
type: "session",
@ -320,8 +332,8 @@ describe("workspace-sync hook handler", () => {
await workspaceSyncHandler(event as never);
expect(warnSpy).toHaveBeenCalledWith(
expect.stringContaining("First sync requires manual --resync"),
expect(mockLog.warn).toHaveBeenCalledWith(
expect.stringContaining("first sync requires manual --resync"),
);
});
});

View File

@ -8,6 +8,7 @@
import type { MoltbotConfig } from "../../../config/config.js";
import { resolveAgentWorkspaceDir } from "../../../agents/agent-scope.js";
import { resolveAgentIdFromSessionKey } from "../../../routing/session-key.js";
import { createSubsystemLogger } from "../../../logging/subsystem.js";
import type { HookHandler } from "../../hooks.js";
import {
isRcloneInstalled,
@ -16,6 +17,8 @@ import {
runBisync,
} from "../../../infra/rclone.js";
const log = createSubsystemLogger("workspace-sync");
/**
* Sync workspace on session start or end
*/
@ -55,13 +58,13 @@ const workspaceSyncHandler: HookHandler = async (event) => {
return;
}
console.log(`[workspace-sync] Triggered on session ${event.action}`);
log.info(`triggered on session ${event.action}`);
try {
// Check if rclone is installed
const installed = await isRcloneInstalled();
if (!installed) {
console.warn("[workspace-sync] rclone not installed, skipping sync");
log.warn("rclone not installed, skipping sync");
return;
}
@ -74,15 +77,11 @@ const workspaceSyncHandler: HookHandler = async (event) => {
// Check if rclone is configured
if (!isRcloneConfigured(resolved.configPath, resolved.remoteName)) {
console.warn(
`[workspace-sync] rclone not configured for remote "${resolved.remoteName}", skipping sync`,
);
log.warn(`rclone not configured for remote "${resolved.remoteName}", skipping sync`);
return;
}
console.log(
`[workspace-sync] Syncing ${resolved.remoteName}:${resolved.remotePath} <-> ${resolved.localPath}`,
);
log.info(`syncing ${resolved.remoteName}:${resolved.remotePath} <-> ${resolved.localPath}`);
// Run sync
const result = await runBisync({
@ -96,22 +95,20 @@ const workspaceSyncHandler: HookHandler = async (event) => {
});
if (result.ok) {
console.log("[workspace-sync] Sync completed successfully");
log.info("sync completed successfully");
if (result.filesTransferred) {
console.log(`[workspace-sync] Files transferred: ${result.filesTransferred}`);
log.info(`files transferred: ${result.filesTransferred}`);
}
} else {
// Check if this is a first-run issue
if (result.error?.includes("--resync")) {
console.warn(
"[workspace-sync] First sync requires manual --resync. Run: moltbot workspace sync --resync",
);
log.warn("first sync requires manual --resync. Run: moltbot workspace sync --resync");
} else {
console.error(`[workspace-sync] Sync failed: ${result.error}`);
log.error(`sync failed: ${result.error}`);
}
}
} catch (err) {
console.error("[workspace-sync] Error:", err instanceof Error ? err.message : String(err));
log.error(`error: ${err instanceof Error ? err.message : String(err)}`);
}
};