local ls = require("luasnip")
local s = ls.snippet
local sn = ls.snippet_node
local t = ls.text_node
local i = ls.insert_node
local f = ls.function_node
local d = ls.dynamic_node
local fmt = require("luasnip.extras.fmt").fmt
local fmta = require("luasnip.extras.fmt").fmta
local rep = require("luasnip.extras").rep
local line_begin = require("luasnip.extras.expand_conditions").line_begin

local tex = require("util.latex")

local get_visual = function(args, parent)
  if #parent.snippet.env.SELECT_RAW > 0 then
    return sn(nil, i(1, parent.snippet.env.SELECT_RAW))
  else -- If SELECT_RAW is empty, return a blank insert node
    return sn(nil, i(1))
  end
end

return {
  s({ trig = "...", wordTrig = false, snippetType = "autosnippet" }, {
    t("dots"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "c.", wordTrig = false, snippetType = "autosnippet" }, {
    t("dot"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "cdot.", wordTrig = false, snippetType = "autosnippet" }, {
    t("cdots"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "v.", wordTrig = false, snippetType = "autosnippet" }, {
    t("dots.v"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "iff", wordTrig = false, snippetType = "autosnippet" }, {
    t("iff"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "notin", wordTrig = false, snippetType = "autosnippet" }, {
    t("in.not"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "aa", wordTrig = false, snippetType = "autosnippet" }, {
    t("forall"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "ee", wordTrig = false, snippetType = "autosnippet" }, {
    t("exists"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "==", wordTrig = false, snippetType = "autosnippet" }, {
    t("&="),
  }, { condition = tex.in_mathzone }),
  s({ trig = "<==", wordTrig = false, snippetType = "autosnippet" }, {
    t("&<="),
  }, { condition = tex.in_mathzone }),
  s({ trig = ">==", wordTrig = false, snippetType = "autosnippet" }, {
    t("&>="),
  }, { condition = tex.in_mathzone }),
  s({ trig = "~=", wordTrig = false, snippetType = "autosnippet" }, {
    t("approx"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "~~", wordTrig = false, snippetType = "autosnippet" }, {
    t("sim"),
  }, { condition = tex.in_mathzone }),
  s({ trig = ">>", wordTrig = false, snippetType = "autosnippet" }, {
    t("gg"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "<<", wordTrig = false, snippetType = "autosnippet" }, {
    t("ll"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "cp", wordTrig = false, snippetType = "autosnippet" }, {
    t("cp"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "get", wordTrig = false, snippetType = "autosnippet" }, {
    t("gets"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "to", wordTrig = false, snippetType = "autosnippet" }, {
    t("to"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "mto", wordTrig = false, snippetType = "autosnippet", priority = 1001 }, {
    t("mapsto"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "||", wordTrig = false, snippetType = "autosnippet" }, {
    t("mid"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "mid", wordTrig = false, snippetType = "autosnippet" }, {
    t("mid"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "nmid", wordTrig = false, snippetType = "autosnippet", priority = 2000 }, {
    t("nmid"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "sr", wordTrig = false, snippetType = "autosnippet" }, {
    t("^2"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "cb", wordTrig = false, snippetType = "autosnippet" }, {
    t("^3"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "inv", wordTrig = false, snippetType = "autosnippet" }, {
    t("^{-1}"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "**", wordTrig = false, snippetType = "autosnippet" }, {
    t("^*"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "  ", wordTrig = false, snippetType = "autosnippet" }, {
    t(","),
  }, { condition = tex.in_mathzone }),
  s({ trig = "<>", wordTrig = false, snippetType = "autosnippet" }, {
    t("diamond"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "+-", wordTrig = false, snippetType = "autosnippet" }, {
    t("pm"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "-+", wordTrig = false, snippetType = "autosnippet" }, {
    t("mp"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "rhs", wordTrig = false, snippetType = "autosnippet" }, {
    t("mathrm{R.H.S}"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "lhs", wordTrig = false, snippetType = "autosnippet" }, {
    t("mathrm{L.H.S}"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "cap", wordTrig = false, snippetType = "autosnippet" }, {
    t("inter"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "cup", wordTrig = false, snippetType = "autosnippet" }, {
    t("union"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "sub", wordTrig = false, snippetType = "autosnippet" }, {
    t("subset.eq"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "suq", wordTrig = false, snippetType = "autosnippet" }, {
    t("supset.eq"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "tp", wordTrig = false, snippetType = "autosnippet" }, {
    t("^top"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "dr", wordTrig = false, snippetType = "autosnippet" }, {
    t("^dagger"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "perp", wordTrig = false, snippetType = "autosnippet" }, {
    t("perp"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "ss", wordTrig = false, snippetType = "autosnippet" }, {
    t("star"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "xx", wordTrig = false, snippetType = "autosnippet" }, {
    t("times"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "=>", wordTrig = false, snippetType = "autosnippet" }, {
    t("implies"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "llr", wordTrig = false, snippetType = "autosnippet" }, {
    t("longleftrightarrow"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "up", wordTrig = false, snippetType = "autosnippet" }, {
    t("uparrow"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "down", wordTrig = false, snippetType = "autosnippet" }, {
    t("downarrow"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "cir", wordTrig = false, snippetType = "autosnippet" }, {
    t("circ"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "iso", wordTrig = false, snippetType = "autosnippet" }, {
    t("cong"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "ihbar", wordTrig = false, snippetType = "autosnippet", priority = 2000 }, {
    t("ihbar"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "hbar", wordTrig = false, snippetType = "autosnippet", priority = 1500 }, {
    t("hbar"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "ns", wordTrig = false, snippetType = "autosnippet" }, {
    t("unlhd"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "eqv", wordTrig = false, snippetType = "autosnippet" }, {
    t("equiv"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "##", wordTrig = false, snippetType = "autosnippet" }, {
    t("#"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "ell", wordTrig = false, snippetType = "autosnippet", priority = 2000 }, {
    t("ell"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "ot", wordTrig = false, snippetType = "autosnippet" }, {
    t("times.circle"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "op", wordTrig = false, snippetType = "autosnippet" }, {
    t("xor"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "par", wordTrig = false, snippetType = "autosnippet", priority = 2000 }, {
    t("diff"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "land", snippetType = "autosnippet" }, {
    t("land"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "lor", snippetType = "autosnippet" }, {
    t("lor"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "tri", snippetType = "autosnippet" }, {
    t("triangle"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "emp", snippetType = "autosnippet", priority = 2000 }, {
    t("emptyset"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "po", snippetType = "autosnippet" }, {
    t("prop"),
  }, { condition = tex.in_mathzone }),
  s({ trig = "cong", snippetType = "autosnippet" }, {
    t("cong"),
  }, { condition = tex.in_mathzone }),
  s(
    { trig = "b|", wordTrig = false, snippetType = "autosnippet" },
    fmta(
      [[
    \Big|_(<>)
    ]],
      { i(1) }
    ),
    { condition = tex.in_mathzone }
  ),
  s(
    { trig = "jk", wordTrig = false, snippetType = "autosnippet" },
    fmta(
      [[
    _(<>)
    ]],
      { i(1) }
    ),
    { condition = tex.in_mathzone }
  ),
  s(
    { trig = "kj", wordTrig = false, snippetType = "autosnippet" },
    fmta(
      [[
    ^(<>)
    ]],
      { i(1) }
    ),
    { condition = tex.in_mathzone }
  ),
}