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

import { createStore } from "jotai";
import { expect, test } from "vitest";
import {
  configOverridesAtom,
  resolvedMarimoConfigAtom,
  userConfigAtom,
} from "../config";
import {
  AppConfigSchema,
  defaultUserConfig,
  type UserConfig,
  UserConfigSchema,
} from "../config-schema";

test("default AppConfig", () => {
  const defaultConfig = AppConfigSchema.parse({});
  expect(defaultConfig).toMatchInlineSnapshot(`
    {
      "auto_download": [],
      "sql_output": "auto",
      "width": "medium",
    }
  `);
});

test("another AppConfig", () => {
  const config = AppConfigSchema.parse({
    width: "medium",
    app_title: null,
  });
  expect(config).toMatchInlineSnapshot(`
    {
      "app_title": null,
      "auto_download": [],
      "sql_output": "auto",
      "width": "medium",
    }
  `);
});

test("default UserConfig - empty", () => {
  const defaultConfig = defaultUserConfig();
  expect(defaultConfig).toMatchInlineSnapshot(`
    {
      "ai": {
        "custom_providers": {},
        "inline_tooltip": false,
        "mode": "manual",
        "models": {
          "custom_models": [],
          "displayed_models": [],
        },
        "rules": "",
      },
      "completion": {
        "activate_on_typing": true,
        "copilot": false,
        "signature_hint_on_typing": false,
      },
      "diagnostics": {},
      "display": {
        "cell_output": "below",
        "code_editor_font_size": 14,
        "dataframes": "rich",
        "default_table_max_columns": 50,
        "default_table_page_size": 10,
        "default_width": "medium",
        "reference_highlighting": true,
        "theme": "light",
      },
      "experimental": {},
      "formatting": {
        "line_length": 79,
      },
      "keymap": {
        "destructive_delete": true,
        "overrides": {},
        "preset": "default",
      },
      "mcp": {},
      "package_management": {
        "manager": "pip",
      },
      "runtime": {
        "auto_instantiate": true,
        "auto_reload": "off",
        "default_auto_download": [],
        "default_sql_output": "auto",
        "on_cell_change": "autorun",
        "reactive_tests": true,
        "show_tracebacks": false,
        "watcher_on_save": "lazy",
      },
      "save": {
        "autosave": "after_delay",
        "autosave_delay": 1000,
        "format_on_save": false,
      },
      "server": {},
    }
  `);
});

test("default UserConfig - one level", () => {
  const defaultConfig = UserConfigSchema.parse({
    completion: {},
    save: {},
    formatting: {},
    keymap: {},
    runtime: {},
    display: {},
    experimental: {},
  });
  expect(defaultConfig).toMatchInlineSnapshot(`
    {
      "ai": {
        "custom_providers": {},
        "inline_tooltip": false,
        "mode": "manual",
        "models": {
          "custom_models": [],
          "displayed_models": [],
        },
        "rules": "",
      },
      "completion": {
        "activate_on_typing": true,
        "copilot": false,
        "signature_hint_on_typing": false,
      },
      "diagnostics": {},
      "display": {
        "cell_output": "below",
        "code_editor_font_size": 14,
        "dataframes": "rich",
        "default_table_max_columns": 50,
        "default_table_page_size": 10,
        "default_width": "medium",
        "reference_highlighting": true,
        "theme": "light",
      },
      "experimental": {},
      "formatting": {
        "line_length": 79,
      },
      "keymap": {
        "destructive_delete": true,
        "overrides": {},
        "preset": "default",
      },
      "mcp": {},
      "package_management": {
        "manager": "pip",
      },
      "runtime": {
        "auto_instantiate": true,
        "auto_reload": "off",
        "default_auto_download": [],
        "default_sql_output": "auto",
        "on_cell_change": "autorun",
        "reactive_tests": true,
        "show_tracebacks": false,
        "watcher_on_save": "lazy",
      },
      "save": {
        "autosave": "after_delay",
        "autosave_delay": 1000,
        "format_on_save": false,
      },
      "server": {},
    }
  `);

  expect(
    UserConfigSchema.parse({
      completion: {},
      save: {},
      formatting: {},
      keymap: {},
      runtime: {},
      display: {},
      experimental: {},
      diagnostics: {},
    }),
  ).toEqual(UserConfigSchema.parse({}));
});

test("default UserConfig with additional information", () => {
  const config = UserConfigSchema.parse({
    some_new_config: {
      is_new_config: true,
    },
  });
  expect(config).toEqual(
    expect.objectContaining({
      some_new_config: {
        is_new_config: true,
      },
    }),
  );
});

test("UserConfig with custom_providers", () => {
  const config = UserConfigSchema.parse({
    ai: {
      custom_providers: {
        my_provider: {
          api_key: "test-key",
          base_url: "https://api.example.com/v1",
        },
        another_provider: {
          base_url: "https://api.another.com/v1",
        },
      },
    },
  });

  expect(config.ai?.custom_providers).toEqual({
    my_provider: {
      api_key: "test-key",
      base_url: "https://api.example.com/v1",
    },
    another_provider: {
      base_url: "https://api.another.com/v1",
    },
  });
});

test("UserConfig custom_providers defaults to empty object", () => {
  const config = UserConfigSchema.parse({
    ai: {},
  });

  expect(config.ai?.custom_providers).toEqual({});
});

test("resolvedMarimoConfigAtom overrides correctly and does not mutate the original array", () => {
  const initialUserConfig = {
    completion: {
      activate_on_typing: true,
      copilot: false,
      signature_hint_on_typing: false,
    },
    save: {
      autosave: "after_delay",
      autosave_delay: 1000,
      format_on_save: false,
    },
    formatting: { line_length: 79 },
  };

  const overrides = {
    completion: { copilot: "github" },
    display: { theme: "dark" },
  };

  const store = createStore();

  store.set(userConfigAtom, initialUserConfig as UserConfig);
  store.set(configOverridesAtom, overrides);

  const result = store.get(resolvedMarimoConfigAtom);

  expect(result).toEqual({
    completion: {
      activate_on_typing: true,
      copilot: "github",
      signature_hint_on_typing: false,
    },
    save: {
      autosave: "after_delay",
      autosave_delay: 1000,
      format_on_save: false,
    },
    formatting: { line_length: 79 },
    display: { theme: "dark" },
  });

  expect(initialUserConfig).toEqual({
    completion: {
      activate_on_typing: true,
      copilot: false,
      signature_hint_on_typing: false,
    },
    save: {
      autosave: "after_delay",
      autosave_delay: 1000,
      format_on_save: false,
    },
    formatting: { line_length: 79 },
  });
});