Update

LuaSnip/typst/algo.lua

@@ -0,0 +1,139 @@
+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 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, t(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 = "bal", snippetType = "autosnippet" },
+    fmta(
+      [[
+    \begin{algorithm}
+      \caption{<>}
+      \label{algo:<>}
+      \begin{algorithmic}[1]
+        \Require <>
+      \end{algorithmic}
+
+    \end{algorithm}
+      ]],
+      {
+        i(1),
+        rep(1),
+        i(0),
+      }
+    ),
+    { condition = tex.in_text * line_begin }
+  ),
+  s({ trig = "ss", snippetType = "autosnippet" }, {
+    t("\\State "),
+  }, { condition = tex.in_algo * line_begin }),
+  s({ trig = "sx", snippetType = "autosnippet" }, {
+    t("\\Statex "),
+  }, { condition = tex.in_algo * line_begin }),
+  s({ trig = "return", snippetType = "autosnippet" }, {
+    t("\\Return"),
+  }, { condition = tex.in_algo }),
+  s(
+    { trig = "while", snippetType = "autosnippet" },
+    fmta(
+      [[
+      \While{<>}
+        <>
+      \EndWhile
+      ]],
+      {
+        i(1),
+        i(0),
+      }
+    ),
+    { condition = tex.in_algo * line_begin }
+  ),
+  s(
+    { trig = "for", snippetType = "autosnippet" },
+    fmta(
+      [[
+      \For{<>}
+        <>
+      \EndFor
+      ]],
+      {
+        i(1),
+        i(0),
+      }
+    ),
+    { condition = tex.in_algo * line_begin }
+  ),
+  s(
+    { trig = "if", snippetType = "autosnippet" },
+    fmta(
+      [[
+      \If{<>}
+        <>
+      \EndIf
+      ]],
+      {
+        i(1),
+        i(0),
+      }
+    ),
+    { condition = tex.in_algo * line_begin }
+  ),
+  s(
+    { trig = "elif", snippetType = "autosnippet" },
+    fmta(
+      [[
+      \ElsIf{<>}
+        <>
+      ]],
+      {
+        i(1),
+        i(0),
+      }
+    ),
+    { condition = tex.in_algo * line_begin }
+  ),
+  s(
+    { trig = "else", snippetType = "autosnippet" },
+    fmta(
+      [[
+      \Else
+        <>
+      ]],
+      {
+        i(0),
+      }
+    ),
+    { condition = tex.in_algo * line_begin }
+  ),
+  s(
+    { trig = "fn", snippetType = "autosnippet" },
+    fmta(
+      [[
+      \Function{<>}{<>}
+        <>
+      \EndFunction
+      ]],
+      {
+        i(1),
+        i(2),
+        i(0),
+      }
+    ),
+    { condition = tex.in_algo * line_begin }
+  ),
+}