local Cache = require("lazy.core.cache")

local M = {}

---@type table<string, string>
M.hashes = {}

---@param modname string
---@param modpath string
---@return any
function M.load(modname, modpath)
  local err
  ---@type (string|fun())?
  local chunk = Cache.get(modname)

  local hash = Cache.hash(modpath)
  if hash ~= M.hashes[modname] then
    M.hashes[modname] = hash
    Cache.del(modname)
    chunk = nil
  end

  if chunk then
    chunk, err = load(chunk --[[@as string]], "@" .. modpath, "b")
  else
    vim.schedule(function()
      vim.notify("loadfile(" .. modname .. ")")
    end)
    chunk, err = loadfile(modpath)
    if chunk and not err then
      Cache.set(modname, string.dump(chunk))
    end
  end

  if chunk then
    return chunk()
  else
    error(err)
  end
end

function M.setup()
  -- load cache
  local value = Cache.get("cache.modules")
  if value then
    M.hashes = vim.json.decode(value)
  end

  -- preload core modules
  local root = vim.fn.fnamemodify(debug.getinfo(1, "S").source:sub(2), ":p:h:h")
  for _, name in ipairs({ "util", "config", "loader", "plugin", "handler" }) do
    local modname = "lazy.core." .. name
    ---@diagnostic disable-next-line: no-unknown
    package.preload[modname] = function()
      return M.load(modname, root .. "/core/" .. name:gsub("%.", "/") .. ".lua")
    end
  end
  return M
end

function M.save()
  Cache.set("cache.modules", vim.json.encode(M.hashes))
end

return M