diagnostics (#6)

* wip on diagnostics
* add diagnostics
* enable diagnostics by default

Author: jmbuhr

README.md

@@ -44,6 +44,10 @@ It will be merged with the default options, which are shown below in the example
 ```lua
 require'quarto'.setup{
   closePreviewOnExit = true, -- close preview terminal on closing of qmd file buffer
+  diagnostics = {
+    enabled = true, -- enable diagnostics for embedded languages
+    languages = {'r', 'python', 'julia'}
+  }
 }
 ```
 

lua/quarto/init.lua

@@ -1,14 +1,23 @@
 local M = {}
-local api = vim.api
+local a = vim.api
+local q = vim.treesitter.query
 local util = require "lspconfig.util"
 
+local p = function (x)
+  print(vim.inspect(x))
+end
+
 local defaultConfig = {
   closePreviewOnExit = true,
+  diagnostics = {
+    enabled = true,
+    languages = {'r', 'python', 'julia'}
+  }
 }
 
-M.setup = function(opt)
-  M.config = vim.tbl_deep_extend('force', defaultConfig, opt or {})
-end
+M.config = defaultConfig
+
+
 
 
 local function contains(list, x)
@@ -20,7 +29,7 @@ end
 
 function M.quartoPreview()
   -- find root directory / check if it is a project
-  local buffer_path = api.nvim_buf_get_name(0)
+  local buffer_path = a.nvim_buf_get_name(0)
   local root_dir = util.root_pattern("_quarto.yml")(buffer_path)
   local cmd
   local mode
@@ -52,17 +61,145 @@ function M.quartoPreview()
 
   -- close preview terminal on exit of the quarto buffer
   if M.config.closePreviewOnExit then
-    api.nvim_create_autocmd({ "QuitPre", "WinClosed" }, {
-      buffer = api.nvim_get_current_buf(),
-      group = api.nvim_create_augroup("quartoPreview", {}),
+    a.nvim_create_autocmd({ "QuitPre", "WinClosed" }, {
+      buffer = a.nvim_get_current_buf(),
+      group = a.nvim_create_augroup("quartoPreview", {}),
       callback = function(_, _)
-        if api.nvim_buf_is_loaded(quartoOutputBuf) then
-          api.nvim_buf_delete(quartoOutputBuf, { force = true })
+        if a.nvim_buf_is_loaded(quartoOutputBuf) then
+          a.nvim_buf_delete(quartoOutputBuf, { force = true })
         end
       end
     })
   end
 end
 
 
+
+
+-- lps support
+local function lines(str)
+  local result = {}
+  for line in str:gmatch '([^\n]*)\n?' do
+    table.insert(result, line)
+  end
+  result[#result] = nil
+  return result
+end
+
+local function spaces(n)
+  local s = {}
+  for i=1,n do
+    s[i] = ' '
+  end
+  return s
+end
+
+local function get_language_content(bufnr, language)
+  -- get and parse AST
+  local language_tree = vim.treesitter.get_parser(bufnr, 'markdown')
+  local syntax_tree = language_tree:parse()
+  local root = syntax_tree[1]:root()
+
+  -- create capture
+  local query = vim.treesitter.parse_query('markdown',
+  string.gsub([[
+  (fenced_code_block
+    (info_string
+      (language) @lang
+      (#eq? @lang $language)
+    )
+    (code_fence_content) @code (#offset! @code)
+  )
+  ]], "%$(%w+)", {language=language})
+  )
+
+  -- get text ranges
+  local results = {}
+  for _, captures, metadata in query:iter_matches(root, bufnr) do
+    local text = q.get_node_text(captures[2], bufnr)
+    -- line numbers start at 0
+    -- {start line, col, end line, col}
+    local result = {
+      range = metadata.content[1],
+      -- text = lines(text)
+      text = lines(text)
+    }
+    table.insert(results, result)
+  end
+
+  return results
+end
+
+
+local function update_language_buffer(qmd_bufnr, language)
+  local language_lines = get_language_content(qmd_bufnr, language)
+  if next(language_lines) == nil then
+    return
+  end
+
+  local nmax = language_lines[#language_lines].range[3] -- last code line
+  local qmd_path = a.nvim_buf_get_name(qmd_bufnr)
+  local postfix
+  if language == 'python' then
+    postfix = '.py'
+  elseif language == 'r' then
+    postfix = '.R'
+  end
+
+  -- create buffer filled with spaces
+  local bufname_lang = qmd_path..postfix
+  local bufuri_lang = 'file://'..bufname_lang
+  local bufnr_lang = vim.uri_to_bufnr(bufuri_lang)
+  a.nvim_buf_set_name(bufnr_lang, bufname_lang)
+  a.nvim_buf_set_option(bufnr_lang,'filetype', language)
+  a.nvim_buf_set_lines(bufnr_lang, 0, -1, false, {})
+  a.nvim_buf_set_lines(bufnr_lang, 0, nmax, false, spaces(nmax))
+
+  -- write language lines
+  for _,t in ipairs(language_lines) do
+    a.nvim_buf_set_lines(bufnr_lang, t.range[1], t.range[3], false, t.text)
+  end
+  return bufnr_lang
+end
+
+
+local function enable_language_diagnostics(lang)
+  local ns  = a.nvim_create_namespace('quarto')
+  local augroup = a.nvim_create_augroup("quartoUpdate"..lang, {})
+
+  a.nvim_create_autocmd({ "TextChanged", "TextChangedI" }, {
+    -- buffer = qmd_buf,
+    pattern = '*.qmd',
+    group = augroup,
+    callback = function(args)
+      local buf = update_language_buffer(0, lang)
+      local diag = vim.diagnostic.get(buf)
+      vim.diagnostic.reset(ns, 0)
+      vim.diagnostic.set(ns, 0, diag, {})
+    end
+  })
+  a.nvim_exec_autocmds('TextChanged', {})
+end
+
+
+M.enableDiagnostics = function()
+  if M.config.diagnostics.enabled then
+    for _,lang in ipairs(M.config.diagnostics.languages) do
+      enable_language_diagnostics(lang)
+    end
+  end
+end
+
+
+M.setup = function(opt)
+  M.config = vim.tbl_deep_extend('force', defaultConfig, opt or {})
+end
+
+
+M.debug = function()
+  M.enableDiagnostics()
+  a.nvim_exec_autocmds({'TextChangedI', 'TextChanged'}, {})
+end
+
 return M
+

plugin/quarto-nvim.lua

@@ -0,0 +1,15 @@
+local a = vim.api
+
+a.nvim_create_user_command('QuartoPreview', require'quarto'.quartoPreview, {})
+
+
+a.nvim_create_autocmd({ "BufEnter" }, {
+  pattern = '*.qmd',
+  group = a.nvim_create_augroup('quarto', {}),
+  callback = function(args)
+    -- use markdown ft until quarto ft is more widespread
+    a.nvim_buf_set_option(0,'filetype', 'markdown')
+    require'quarto'.enableDiagnostics()
+  end
+})
+