import type { Mock } from "vitest";

import { vi } from "vitest";

import type { OpenClawConfig } from "../config/types.openclaw.js";

import type { PluginInstallRecord } from "../config/types.plugins.js";

import { createCliRuntimeCapture } from "./test-runtime-capture.js";

type UnknownMock = Mock<(...args: unknown[]) => unknown>;

(typeof import("../plugins/marketplace.js"))["listMarketplacePlugins"];

type ResolveMarketplaceInstallShortcutFn =

(typeof import("../plugins/marketplace.js"))["resolveMarketplaceInstallShortcut"];

type LoadPluginInstallRecordsParams = { config?: OpenClawConfig };

type PluginInstallRecordMap = Record<string, PluginInstallRecord>;

let mockInstalledPluginIndexInstallRecords: PluginInstallRecordMap = {};

function clonePluginInstallRecords(records: PluginInstallRecordMap): PluginInstallRecordMap {

return structuredClone(records);

}

// oxlint-disable-next-line typescript/no-unnecessary-type-parameters -- Test helper preserves mock call and result types.

function invokeMock<TArgs extends unknown[], TResult>(mock: unknown, ...args: TArgs): TResult {

export const resolveMarketplaceInstallShortcut: Mock<ResolveMarketplaceInstallShortcutFn> = vi.fn();

export const enablePluginInConfig: UnknownMock = vi.fn();

export const recordPluginInstall: UnknownMock = vi.fn();

export const loadInstalledPluginIndexInstallRecords: AsyncUnknownMock = vi.fn(

async (...args: unknown[]) => {

const params = args[0] as LoadPluginInstallRecordsParams | undefined;

return structuredClone(params?.config?.plugins?.installs ?? {});

},

export const loadInstalledPluginIndexInstallRecords: AsyncUnknownMock = vi.fn(async () =>

clonePluginInstallRecords(mockInstalledPluginIndexInstallRecords),

);

export const writePersistedInstalledPluginIndexInstallRecords: AsyncUnknownMock = vi.fn(

async () => undefined,

async (records: unknown) => {

mockInstalledPluginIndexInstallRecords = clonePluginInstallRecords(

(records ?? {}) as PluginInstallRecordMap,

);

},

);

export const clearPluginManifestRegistryCache: UnknownMock = vi.fn();

export const loadPluginManifestRegistry: UnknownMock = vi.fn();

export { runtimeErrors, runtimeLogs };

export function setInstalledPluginIndexInstallRecords(records: PluginInstallRecordMap): void {

mockInstalledPluginIndexInstallRecords = clonePluginInstallRecords(records);

}

function restoreRuntimeCaptureMocks() {

defaultRuntime.log.mockReset();

defaultRuntime.log.mockImplementation((...args: unknown[]) => {

resolveMarketplaceInstallShortcut.mockReset();

enablePluginInConfig.mockReset();

recordPluginInstall.mockReset();

mockInstalledPluginIndexInstallRecords = {};

loadInstalledPluginIndexInstallRecords.mockReset();

writePersistedInstalledPluginIndexInstallRecords.mockReset();

clearPluginManifestRegistryCache.mockReset();

recordPluginInstall.mockImplementation(

((cfg: OpenClawConfig) => cfg) as (...args: unknown[]) => unknown,

);

loadInstalledPluginIndexInstallRecords.mockImplementation(async (...args: unknown[]) => {

const params = args[0] as LoadPluginInstallRecordsParams | undefined;

return structuredClone(params?.config?.plugins?.installs ?? {});

loadInstalledPluginIndexInstallRecords.mockImplementation(async () =>

clonePluginInstallRecords(mockInstalledPluginIndexInstallRecords),

);

writePersistedInstalledPluginIndexInstallRecords.mockImplementation(async (records: unknown) => {

mockInstalledPluginIndexInstallRecords = clonePluginInstallRecords(

(records ?? {}) as PluginInstallRecordMap,

);

});

writePersistedInstalledPluginIndexInstallRecords.mockResolvedValue(undefined);

loadPluginManifestRegistry.mockReturnValue({

plugins: [],

diagnostics: [],