-- Fennel loader, default one do not work well with NeoVim so there is custom
-- one
_G.fennel = require('fennel')

-- Load Fennel modules
local function fennel_loader(name)
  local basename = name:gsub('%.', '/')
  local paths = {"fnl/"..basename..".fnl", "fnl/"..basename.."/init.fnl"}

  for _, path in ipairs(paths) do
    local found = vim.api.nvim_get_runtime_file(path, false)
    if #found > 0 then
      return function() return fennel.dofile(found[1], {compilerEnv = _G}) end, found[1]
    end
  end

  return nil
end
table.insert(package.loaders, 1, fennel_loader)

-- Load Fennel macros
local function fennel_paths(suffixes)
  local paths = ""
  for _, dir in pairs(vim.api.nvim_get_runtime_file("fnl/", true)) do
    for _, suffix in pairs(suffixes) do
      paths = paths .. ";" .. dir .. "?" .. suffix .. ".fnl"
    end
  end

  return paths
end
fennel["path"] = fennel["path"] .. fennel_paths({"", "/init"})
fennel["macro-path"] = fennel["macro-path"] .. fennel_paths({"", "/macro-init", "/init"})

debug.traceback = fennel.traceback

-- Command-mode Fennel execution
vim.api.nvim_create_user_command('Fnl', function(arg) fennel.eval(arg.args) end, {nargs = '*'})
vim.api.nvim_create_user_command('FnlFile', function(arg) print(fennel.view(arg.args)) end, {nargs = 1})

_G.reloadable = {}

local function reload_init()
  for _, name in pairs(_G.reloadable) do
    package.loaded[name] = nil
  end

  for _, init in pairs(vim.api.nvim_get_runtime_file("init.fnl", false)) do
    fennel.dofile(init, {compilerEnv = _G})
  end
end

vim.api.nvim_create_user_command('ReloadInit', reload_init, {})

reload_init()