feat: discover snippet: use lsp to get module name

Author: Marc Jakobi

lua/haskell-snippets/module.lua

@@ -27,13 +27,9 @@ local has_treesitter, parsers = pcall(require, 'nvim-treesitter.parsers')
 local hs_lang = has_treesitter and parsers.ft_to_lang('haskell')
 
 local function get_module_name_node()
-  if #vim.lsp.get_active_clients { bufnr = 0 } > 0 then
-    for _, lens in pairs(vim.lsp.codelens.get(0)) do
-      local name = lens.command.title:match('module (.*) where')
-      if name then
-        return sn(nil, { text(name) })
-      end
-    end
+  local module_name = util.lsp_get_module_name()
+  if module_name then
+    return sn(nil, { text(module_name) })
   end
   return sn(nil, { insert(1) })
 end

lua/haskell-snippets/pragmas.lua

@@ -21,6 +21,9 @@ local sn = ls.snippet_node
 local text = ls.text_node
 local insert = ls.insert_node
 local choice = ls.choice_node
+local dynamic = ls.dynamic_node
+
+local util = require('haskell-snippets.util')
 
 pragmas.prag = s({
   trig = 'prag',
@@ -90,6 +93,18 @@ pragmas.lang = s({
 })
 table.insert(pragmas.all, pragmas.lang)
 
+local function get_module_name()
+  local module_name = util.lsp_get_module_name()
+  if module_name then
+    return sn(nil, {
+      insert(1, module_name),
+    })
+  end
+  return sn(nil, {
+    insert(1, 'Spec'),
+  })
+end
+
 pragmas.discover = s({
   trig = 'discover',
   dscr = 'hspec/sydtest discover GHC option',
@@ -100,7 +115,7 @@ pragmas.discover = s({
     text('sydtest'),
   }),
   text('-discover -optF --module-name='),
-  insert(2, 'Spec'),
+  dynamic(2, get_module_name),
   text(' #-}'),
 })
 table.insert(pragmas.all, pragmas.discover)

lua/haskell-snippets/util.lua

@@ -66,4 +66,16 @@ function util.indent_newline_insert(txt, extra_indent)
   end
 end
 
+---@return string|nil
+function util.lsp_get_module_name()
+  if #vim.lsp.get_active_clients { bufnr = 0 } > 0 then
+    for _, lens in pairs(vim.lsp.codelens.get(0)) do
+      local name = lens.command.title:match('module (.*) where')
+      if name then
+        return name
+      end
+    end
+  end
+end
+
 return util