import { withTempHome as withTempHomeBase } from "../plugin-sdk/test-helpers/temp-home.js";

import { resolveBundledPluginsDir } from "../plugins/bundled-dir.js";

import { resetPluginLoaderTestStateForTest } from "../plugins/loader.test-fixtures.js";

import { resetProviderRuntimeHookCacheForTest } from "../plugins/provider-runtime.js";

import { resolveOwningPluginIdsForProvider } from "../plugins/providers.js";

import type { MockFn } from "../test-utils/vitest-mock-fn.js";

import { resetModelsJsonReadyCacheForTest } from "./models-config-state.js";

export function installModelsConfigTestHooks(opts?: {

restoreFetch?: boolean;

resetPluginLoaderState?: boolean;

resetProviderRuntimeHookCache?: boolean;

}) {

let previousHome: string | undefined;

let previousOpenClawAgentDir: string | undefined;

let previousPiCodingAgentDir: string | undefined;

const originalFetch = globalThis.fetch;

const shouldResetPluginLoaderState = opts?.resetPluginLoaderState !== false;

const shouldResetProviderRuntimeHookCache = opts?.resetProviderRuntimeHookCache !== false;

beforeEach(() => {

previousHome = process.env.HOME;

resetPluginLoaderTestStateForTest();

}

resetModelsJsonReadyCacheForTest();

if (shouldResetProviderRuntimeHookCache) {

resetProviderRuntimeHookCacheForTest();

}

});

afterEach(() => {

resetPluginLoaderTestStateForTest();

}

resetModelsJsonReadyCacheForTest();

if (shouldResetProviderRuntimeHookCache) {

resetProviderRuntimeHookCacheForTest();

}

if (opts?.restoreFetch && originalFetch) {

globalThis.fetch = originalFetch;

}