Make records editable

lua/telescope/_extensions/telescope_knot.lua

@@ -1,6 +1,8 @@
-local pickers = require("telescope.pickers")
-local finders = require("telescope.finders")
+local action_state = require('telescope.actions.state')
+local actions = require('telescope.actions')
 local conf = require("telescope.config").values
+local finders = require("telescope.finders")
+local pickers = require("telescope.pickers")
 local previewers = require("telescope.previewers")
 local log = require('plenary.log').new {
   plugin = 'telescope_knot',
@@ -8,8 +10,8 @@ local log = require('plenary.log').new {
 }
 
 local _get_name = function(record)
-    local name = record.rtype .. " " .. record.name .. " " .. record.data
-    return name
+  local name = record.name .. " (" .. record.rtype .. "): " .. record.data
+  return name
 end
 
 local _filter_records = function(rec, records)
@@ -18,7 +20,8 @@ local _filter_records = function(rec, records)
   for _, record in ipairs(records) do
     local name = _get_name(record)
     if name == record_name then
-      r = { "ZONE", record.zone, "RTYPE", record.rtype, "NAME", record.name, "DATA", record.data, "TTL", record.ttl, "URL", record.url }
+      r = { "ZONE: " .. record.zone, "RTYPE: " .. record.rtype, "NAME: " .. record.name, "DATA: " .. record.data, "TTL: " ..
+      record.ttl, "URL: " .. record.url }
     end
   end
   return r
@@ -57,6 +60,55 @@ telescope_knot.zones = function(opts)
       end
 
     }),
+    attach_mappings = function(prompt_bufnr)
+      actions.select_default:replace(function()
+        local entry = action_state.get_selected_entry()
+        local record = _filter_records(entry, records)
+        actions.close(prompt_bufnr)
+        local buffer = vim.api.nvim_create_buf(false, true)
+        vim.api.nvim_buf_set_lines(buffer, 0, 0, false, record)
+        local width = vim.api.nvim_win_get_width(0)
+        local height = vim.api.nvim_win_get_height(0) - 10
+        local _ = vim.api.nvim_open_win(buffer, true,
+          { relative = 'win', bufpos = { 0, 0 }, width = width, height = height, border = 'rounded' })
+        vim.api.nvim_create_autocmd({ "BufWinLeave" }, {
+          buffer = buffer,
+          callback = function(_)
+            local new = vim.api.nvim_buf_get_lines(buffer, 0, -1, false)
+            if new ~= record then
+              local cleanold = {}
+              local changed = {}
+              for _, v in ipairs(record) do
+                if v ~= nil and v ~= "" then
+                  local key = v:match("%w+"):lower()
+                  cleanold[key] = v:gsub("%w+%:%s", "")
+                end
+              end
+              local cleannew = {}
+              for _, v in ipairs(new) do
+                if v ~= nil and v ~= "" then
+                  local key = v:match("%w+"):lower()
+                  cleannew[key] = v:gsub("%w+%:%s", "")
+                  if cleanold[key] ~= cleannew[key] and (key ~= "url" or key ~= "zone") then
+                    table.insert(changed, " -a " .. key .. "=" .. cleannew[key])
+                  end
+                end
+              end
+              if changed ~= {} then
+                local command = "knotctl update -z" .. cleanold.zone ..
+                    " -t" .. cleanold.ttl ..
+                    " -n" .. cleanold.name ..
+                    " -r" .. cleanold.rtype ..
+                    " -d" .. cleanold.data ..
+                    table.concat(changed, " ")
+                log.info(vim.fn.system(command))
+              end
+            end
+          end
+        })
+      end)
+      return true
+    end,
   }):find()
 end
 return require('telescope').register_extension({