import { getGlobalHookRunner } from "../../plugins/hook-runner-global.js";

import { __testing as cliBackendsTesting } from "../cli-backends.js";

import { hashCliSessionText } from "../cli-session.js";

import { buildActiveImageGenerationTaskPromptContextForSession } from "../image-generation-task-status.js";

import { buildActiveMusicGenerationTaskPromptContextForSession } from "../music-generation-task-status.js";

import { buildActiveVideoGenerationTaskPromptContextForSession } from "../video-generation-task-status.js";

import {

isActiveVideoGenerationTask: vi.fn(() => false),

}));

vi.mock("../image-generation-task-status.js", () => ({

IMAGE_GENERATION_TASK_KIND: "image_generation",

buildActiveImageGenerationTaskPromptContextForSession: vi.fn(() => undefined),

buildImageGenerationTaskStatusDetails: vi.fn(() => ({})),

buildImageGenerationTaskStatusText: vi.fn(() => ""),

findActiveImageGenerationTaskForSession: vi.fn(() => undefined),

getImageGenerationTaskProviderId: vi.fn(() => undefined),

isActiveImageGenerationTask: vi.fn(() => false),

}));

vi.mock("../music-generation-task-status.js", () => ({

MUSIC_GENERATION_TASK_KIND: "music_generation",

buildActiveMusicGenerationTaskPromptContextForSession: vi.fn(() => undefined),

const mockBuildActiveVideoGenerationTaskPromptContextForSession = vi.mocked(

buildActiveVideoGenerationTaskPromptContextForSession,

);

const mockBuildActiveImageGenerationTaskPromptContextForSession = vi.mocked(

buildActiveImageGenerationTaskPromptContextForSession,

);

const mockBuildActiveMusicGenerationTaskPromptContextForSession = vi.mocked(

buildActiveMusicGenerationTaskPromptContextForSession,

);

});

mockGetGlobalHookRunner.mockReturnValue(null);

getRuntimeConfigMock.mockReturnValue({});

mockBuildActiveImageGenerationTaskPromptContextForSession.mockReturnValue(undefined);

mockBuildActiveVideoGenerationTaskPromptContextForSession.mockReturnValue(undefined);

mockBuildActiveMusicGenerationTaskPromptContextForSession.mockReturnValue(undefined);

});

cliBackendsTesting.resetDepsForTest();

getRuntimeConfigMock.mockReset();

mockGetGlobalHookRunner.mockReset();

mockBuildActiveImageGenerationTaskPromptContextForSession.mockReset();

mockBuildActiveVideoGenerationTaskPromptContextForSession.mockReset();

mockBuildActiveMusicGenerationTaskPromptContextForSession.mockReset();

vi.unstubAllEnvs();

it("applies direct-run prepend system context helpers on the CLI path", async () => {

const { dir, sessionFile } = createSessionFile();

try {

mockBuildActiveImageGenerationTaskPromptContextForSession.mockReturnValue(

"active image task",

);

mockBuildActiveVideoGenerationTaskPromptContextForSession.mockReturnValue(

"active video task",

);

});

expect(context.systemPrompt).toBe(

"active video task\n\nhook prepend system\n\nhook system\n\nCurrent model identity: test-cli/test-model. If asked what model you are, answer with this value for the current run.",

"active image task\n\nactive video task\n\nhook prepend system\n\nhook system\n\nCurrent model identity: test-cli/test-model. If asked what model you are, answer with this value for the current run.",

);

expect(mockBuildActiveImageGenerationTaskPromptContextForSession).toHaveBeenCalledWith(

"agent:main:test",

);

expect(mockBuildActiveVideoGenerationTaskPromptContextForSession).toHaveBeenCalledWith(

"agent:main:test",