/* Copyright 2026 Marimo. All rights reserved. */

import { type Mock, vi } from "vitest";
import type { EditRequests, RunRequests } from "@/core/network/types";

type RequestClient = {
  [K in keyof (EditRequests & RunRequests)]: Mock<
    (EditRequests & RunRequests)[K]
  >;
};

export const MockRequestClient = {
  create(overrides?: Partial<RequestClient>): RequestClient {
    return {
      // Edit requests
      sendComponentValues: vi.fn().mockResolvedValue({}),
      sendModelValue: vi.fn().mockResolvedValue({}),
      sendRename: vi.fn().mockResolvedValue({}),
      sendRestart: vi.fn().mockResolvedValue({}),
      sendDocumentTransaction: vi.fn().mockResolvedValue({}),
      sendSave: vi.fn().mockResolvedValue({}),
      sendCopy: vi.fn().mockResolvedValue({}),
      sendStdin: vi.fn().mockResolvedValue({}),
      sendFormat: vi.fn().mockResolvedValue({ codes: {} }),
      sendInterrupt: vi.fn().mockResolvedValue({}),
      sendShutdown: vi.fn().mockResolvedValue({}),
      sendRun: vi.fn().mockResolvedValue({}),
      sendRunScratchpad: vi.fn().mockResolvedValue({}),
      sendInstantiate: vi.fn().mockResolvedValue({}),
      sendDeleteCell: vi.fn().mockResolvedValue({}),
      sendCodeCompletionRequest: vi.fn().mockResolvedValue({ items: [] }),
      saveUserConfig: vi.fn().mockResolvedValue({}),
      saveAppConfig: vi.fn().mockResolvedValue({}),
      saveCellConfig: vi.fn().mockResolvedValue({}),
      sendFunctionRequest: vi.fn().mockResolvedValue({}),
      sendInstallMissingPackages: vi.fn().mockResolvedValue({}),
      readCode: vi.fn().mockResolvedValue({ contents: "" }),
      readSnippets: vi.fn().mockResolvedValue({ snippets: [] }),
      previewDatasetColumn: vi.fn().mockResolvedValue({}),
      previewSQLTable: vi.fn().mockResolvedValue({}),
      previewSQLTableList: vi.fn().mockResolvedValue({ tables: [] }),
      previewSQLSchemaList: vi.fn().mockResolvedValue({ schemas: [] }),
      previewDataSourceConnection: vi.fn().mockResolvedValue({}),
      discoverDataSources: vi.fn().mockResolvedValue({}),
      validateSQL: vi.fn().mockResolvedValue({}),
      openFile: vi.fn().mockResolvedValue({}),
      getUsageStats: vi.fn().mockResolvedValue({}),
      getEnvironmentInfo: vi
        .fn()
        .mockRejectedValue(new Error("Environment information unavailable")),
      getFileRoots: vi.fn().mockResolvedValue({ roots: [] }),
      sendPdb: vi.fn().mockResolvedValue({}),
      sendSetBreakpoints: vi.fn().mockResolvedValue({}),
      sendListFiles: vi.fn().mockResolvedValue({ files: [] }),
      sendSearchFiles: vi
        .fn()
        .mockResolvedValue({ files: [], query: "", total_found: 0 }),
      sendCreateFileOrFolder: vi.fn().mockResolvedValue({}),
      sendDeleteFileOrFolder: vi.fn().mockResolvedValue({}),
      sendCopyFileOrFolder: vi.fn().mockResolvedValue({}),
      sendRenameFileOrFolder: vi.fn().mockResolvedValue({}),
      sendUpdateFile: vi.fn().mockResolvedValue({}),
      sendFileDetails: vi.fn().mockResolvedValue({}),
      openTutorial: vi.fn().mockResolvedValue({}),
      getRecentFiles: vi.fn().mockResolvedValue({ files: [] }),
      getWorkspaceFiles: vi.fn().mockResolvedValue({ files: [] }),
      getRunningNotebooks: vi.fn().mockResolvedValue({ files: [] }),
      shutdownSession: vi.fn().mockResolvedValue({}),
      getExportAvailability: vi.fn().mockResolvedValue({
        source: "server",
        formats: [
          {
            format: "html",
            dependenciesAvailable: true,
            missingPackages: [],
            missingSetup: [],
          },
          {
            format: "markdown",
            dependenciesAvailable: true,
            missingPackages: [],
            missingSetup: [],
          },
          {
            format: "ipynb",
            dependenciesAvailable: true,
            missingPackages: [],
            missingSetup: [],
          },
          {
            format: "pdf",
            dependenciesAvailable: true,
            missingPackages: [],
            missingSetup: [],
          },
          {
            format: "script",
            dependenciesAvailable: true,
            missingPackages: [],
            missingSetup: [],
          },
        ],
      }),
      installExportRequirements: vi.fn().mockResolvedValue({
        source: "server",
        formats: [],
      }),
      exportAsHTML: vi.fn().mockResolvedValue({
        contents: "",
        filename: "notebook.html",
        mediaType: "text/html; charset=utf-8",
      }),
      exportAsIPYNB: vi.fn().mockResolvedValue({
        contents: "",
        filename: "notebook.ipynb",
        mediaType: "text/plain; charset=utf-8",
      }),
      exportAsMarkdown: vi.fn().mockResolvedValue({
        contents: "",
        filename: "notebook.md",
        mediaType: "text/plain; charset=utf-8",
      }),
      exportAsScript: vi.fn().mockResolvedValue({
        contents: "",
        filename: "notebook.script.py",
        mediaType: "text/plain; charset=utf-8",
      }),
      exportAsPDF: vi.fn().mockResolvedValue({
        contents: new Blob(),
        filename: "notebook.pdf",
        mediaType: "application/pdf",
      }),
      autoExportAsHTML: vi.fn().mockResolvedValue({}),
      autoExportAsMarkdown: vi.fn().mockResolvedValue({}),
      autoExportAsIPYNB: vi.fn().mockResolvedValue({}),
      updateCellOutputs: vi.fn().mockResolvedValue({}),
      addPackage: vi.fn().mockResolvedValue({}),
      removePackage: vi.fn().mockResolvedValue({}),
      getPackageList: vi.fn().mockResolvedValue({
        packages: [],
      }),
      getSandbox: vi
        .fn()
        .mockResolvedValue({ backend: null, manifest: null, filename: null }),
      updateManifest: vi
        .fn()
        .mockResolvedValue({ backend: null, manifest: null, filename: null }),
      syncSandbox: vi.fn().mockResolvedValue({ success: true }),
      getDependencyTree: vi.fn().mockResolvedValue({
        tree: null,
        context: { kind: "package-manager", name: "pip" },
      }),
      listSecretKeys: vi.fn().mockResolvedValue({ keys: [] }),
      writeSecret: vi.fn().mockResolvedValue({}),
      invokeAiTool: vi.fn().mockResolvedValue({}),
      clearCache: vi.fn().mockResolvedValue(null),
      getCacheInfo: vi.fn().mockResolvedValue(null),
      listStorageEntries: vi.fn().mockResolvedValue(null),
      downloadStorage: vi.fn().mockResolvedValue(null),
      ...overrides,
    };
  },
};