local helpers = require("personal.luasnip-helper-funcs")
local get_visual = helpers.get_visual

local line_begin = require("luasnip.extras.expand_conditions").line_begin

-- Math context detection
local tex = {}
tex.in_mathzone = function()
	return vim.fn["vimtex#syntax#in_mathzone"]() == 1
end
tex.in_text = function()
	return not tex.in_mathzone()
end

return {

	-- Equation, choice for labels
	s(
		{
			trig = "beq",
			dscr = "Expands 'beq' into an equation environment, with a choice for labels",
			snippetType = "autosnippet",
		},
		fmta(
			[[
        \begin{equation}<>
          <>
        \end{equation}
      ]],
			{
				c(1, {
					sn(
						2, -- Choose to specify an equation label
						{
							t("\\label{eq:"),
							i(1),
							t("}"),
						}
					),
					t([[]]), -- Choose no label
				}, {}),
				i(2),
			}
		)
	),

	-- Figure environment
	s(
		{ trig = "foofig", dscr = "Use 'fig' for figure environmennt, with options" },
		fmta(
			[[
        \begin{figure}<>
          \centering
          \includegraphics<>{<>}
          \caption{<>}
          \label{fig:<>}
        \end{figure}
      ]],
			{
				-- Optional [htbp] field
				c(1, {
					t([[]]), -- Choice 1, empty
					t("[htbp]"), -- Choice 2, this may be turned into a snippet
				}, {}),
				-- Options for includegraphics
				c(2, {
					t([[]]), -- Choice 1, empty
					sn(
						3, -- Choice 2, this may be turned into a snippet
						{
							t("[width="),
							i(1),
							t("\\textwidth]"),
						}
					),
				}, {}),
				i(3, "filename"),
				i(4, "text"),
				i(5, "label"),
			}
		),
		{ condition = line_begin }
	),
}