local M = {}

local theme_state_file = vim.fn.stdpath "state" .. "/theme/current"
local active_schemes = {
  cozybox = true,
  ["cozybox-light"] = true,
}

local function ensure_server_socket()
  if vim.v.servername ~= nil and vim.v.servername ~= "" then
    return
  end

  local socket_path = ("/tmp/nvim-%d.sock"):format(vim.fn.getpid())
  vim.fn.serverstart(socket_path)
end

local function apply_cozybox_overrides()
  local links = {
    { "DiffsAdd", "DiffAdd" },
    { "DiffsDelete", "DiffDelete" },
    { "DiffsChange", "DiffChange" },
    { "DiffsText", "DiffText" },
    { "DiffsClear", "Normal" },
  }

  for _, pair in ipairs(links) do
    vim.api.nvim_set_hl(0, pair[1], { link = pair[2], default = false })
  end
end

local function read_mode()
  local ok, lines = pcall(vim.fn.readfile, theme_state_file)
  if not ok or not lines or not lines[1] then return "dark" end

  local mode = vim.trim(lines[1])
  if mode == "light" then return "light" end

  return "dark"
end

local function colorscheme_for_mode(mode)
  if mode == "light" then return "cozybox-light" end

  return "cozybox"
end

function M.apply(mode)
  local next_mode = mode or read_mode()
  local next_scheme = colorscheme_for_mode(next_mode)

  if vim.o.background ~= next_mode then vim.o.background = next_mode end

  if vim.g.cozybox_theme_mode ~= next_mode or vim.g.colors_name ~= next_scheme then vim.cmd.colorscheme(next_scheme) end

  vim.g.cozybox_theme_mode = next_mode
  apply_cozybox_overrides()
end

function M.setup()
  local group = vim.api.nvim_create_augroup("cozybox_theme_sync", { clear = true })

  ensure_server_socket()

  vim.api.nvim_create_autocmd("ColorScheme", {
    group = group,
    callback = function()
      if active_schemes[vim.g.colors_name] then apply_cozybox_overrides() end
    end,
  })

  vim.api.nvim_create_autocmd({ "VimEnter", "FocusGained" }, {
    group = group,
    callback = function() M.apply() end,
  })

  vim.api.nvim_create_user_command("ThemeSync", function(opts)
    local mode = opts.args ~= "" and opts.args or nil
    M.apply(mode)
  end, {
    nargs = "?",
    complete = function() return { "dark", "light" } end,
  })

  M.apply()
end

return M