--
-- Spell check plugin
--

local function toggle_spell()
	vim.o.spell = not vim.o.spell
end

local function set_spell_en()
	vim.opt.spelllang = { "en_us" }
end

local function set_spell_fr()
	vim.opt.spelllang = { "fr" }
end

local function set_spell_en_fr()
	vim.opt.spelllang = { "en_us", "fr" }
end

vim.opt.spelllang = "en_us" -- Initial spell language
vim.opt.spellfile = { -- Spellfiles for zg, zw, z=, etc
	vim.fn.stdpath("config") .. "/spell/en.utf-8.add",
	vim.fn.stdpath("config") .. "/spell/fr.utf-8.add",
}

vim.api.nvim_create_user_command("SpellToggle", toggle_spell, { desc = "Toggle spell checking" })
vim.api.nvim_create_user_command("SpellSetEn", set_spell_en, { desc = "Set spell language to English" })
vim.api.nvim_create_user_command("SpellSetFr", set_spell_fr, { desc = "Set spell language to French" })
vim.api.nvim_create_user_command(
	"SpellSetEnFr",
	set_spell_en_fr,
	{ desc = "Set spell languages to English and French" }
)