import { mkdtempSync, rmSync } from "node:fs";

import os from "node:os";

import path from "node:path";

import type { OpenClawConfig } from "openclaw/plugin-sdk/config-contracts";

import { resetInboundDedupe } from "openclaw/plugin-sdk/reply-runtime";

import type { GetReplyOptions, MsgContext } from "openclaw/plugin-sdk/reply-runtime";

import { beforeEach, vi, type Mock } from "vitest";

import { afterEach, beforeEach, vi, type Mock } from "vitest";

import type { TelegramBotDeps } from "./bot-deps.js";

import {

resetTopicNameCacheForTest,

}

const saveMediaBufferSpy: Mock = vi.fn(defaultSaveMediaBuffer);

let mediaHarnessStoreRoot: string | undefined;

function ensureMediaHarnessStoreRoot(): string {

mediaHarnessStoreRoot ??= mkdtempSync(path.join(os.tmpdir(), "openclaw-telegram-media-e2e-"));

return mediaHarnessStoreRoot;

}

function cleanupMediaHarnessStoreRoot(): void {

if (!mediaHarnessStoreRoot) {

return;

}

rmSync(mediaHarnessStoreRoot, { recursive: true, force: true });

mediaHarnessStoreRoot = undefined;

}

export function setNextSavedMediaPath(params: {

path: string;

channels: { telegram: { dmPolicy: "open", allowFrom: ["*"] } },

}) as OpenClawConfig) as TelegramBotDeps["getRuntimeConfig"],

resolveStorePath: vi.fn(

(storePath?: string) => storePath ?? "/tmp/telegram-media-sessions.json",

(storePath?: string) => storePath ?? path.join(ensureMediaHarnessStoreRoot(), "sessions.json"),

) as TelegramBotDeps["resolveStorePath"],

readChannelAllowFromStore: vi.fn(async () => []) as TelegramBotDeps["readChannelAllowFromStore"],

upsertChannelPairingRequest: vi.fn(async () => ({

};

beforeEach(() => {

cleanupMediaHarnessStoreRoot();

ensureMediaHarnessStoreRoot();

resetInboundDedupe();

topicNameStoresForTest.clear();

resetTopicNameCacheForTest();

resetReadRemoteMediaBufferMock();

});

afterEach(() => {

cleanupMediaHarnessStoreRoot();

});

vi.doMock("./bot.runtime.js", () => ({

...telegramBotRuntimeForTest,

}));

return {

...actual,

readSessionUpdatedAt: () => undefined,

resolveStorePath: (storePath?: string) => storePath ?? "/tmp/sessions.json",

resolveStorePath: (storePath?: string) =>

storePath ?? path.join(ensureMediaHarnessStoreRoot(), "sessions.json"),

};

});