feat(link)!: wrap link actions (normal/visual)

refactor: remove `expr` in kmp

Author: phanen

README.md

@@ -7,25 +7,18 @@
 ```
 
 ## Config
+see `config.lua`
 ```lua
-{
+local default_opts = {
   ft = { "markdown", "typst" },
   line = "<c- >",
   codeblock = "<c-e>",
   listdn = "o",
   listup = "O",
-},
+  img_link = "<leader>zi",
+  raw_link = "<leader>zj",
+}
 ```
 
 ## API
-```lua
-vim.api.nvim_create_autocmd("Filetype", {
-  pattern = { "markdown" },
-  callback = function()
-    vim.keymap.set({ "x", "n" }, "<c- >", require("mder").line, { buffer = 0 })
-    vim.keymap.set({ "x" }, "<c-e>", require("mder").codeblock, { buffer = 0 })
-    vim.keymap.set({ "n" }, "o", require("mder").listdn, { expr = true, buffer = 0 })
-    vim.keymap.set({ "n" }, "O", require("mder").listup, { expr = true, buffer = 0 })
-  end,
-})
-```
+see `commands.lua`

lua/mder/autolist.lua

@@ -1,22 +1,31 @@
+local M = {}
+
 -- auto list item
 local autolist = function(c)
   return function()
     local row = vim.api.nvim_win_get_cursor(0)[1]
     local line = vim.api.nvim_buf_get_lines(0, row - 1, row, false)[1]
-    for it, pat in pairs {
+    for it, p in pairs {
       ["-"] = "%-",
       ["+"] = "%+",
       ["*"] = "%*",
       ["="] = "%=",
     } do
-      if line:find("^%s*" .. pat .. " %[ %]") then return c .. it .. " [ ] " end
-      if line:find("^%s*" .. pat .. " ") then return c .. it .. " " end
+      if line:find("^%s*" .. p .. " %[x%]") then
+        vim.api.nvim_feedkeys(c .. it .. " [x] ", "n", false)
+        return
+      end
+      if line:find("^%s*" .. p .. " %[ %]") then
+        vim.api.nvim_feedkeys(c .. it .. " [ ] ", "n", false)
+        return
+      end
+      if line:find("^%s*" .. p .. " ") then vim.api.nvim_feedkeys(c .. it .. " ", "n", false) end
     end
-    return c
+    vim.api.nvim_feedkeys(c, "n", false)
   end
 end
 
-return {
-  listup = autolist "O",
-  listdn = autolist "o",
-}
+M.listup = autolist "O"
+M.listdn = autolist "o"
+
+return M

lua/mder/codeblock.lua

@@ -1,5 +1,3 @@
-local k = require("mder.utils").k
-
 -- surround codeblock with correct indent
 local wrap_codeblock = function()
   local vs, ve = vim.fn.getpos(".")[2], vim.fn.getpos("v")[2]
@@ -12,11 +10,11 @@ local wrap_codeblock = function()
   lines[#lines + 1] = "```"
   vim.api.nvim_buf_set_lines(0, vs, ve, false, lines)
 
-  vim.api.nvim_feedkeys(k "<esc>", "x", false)
+  vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes("<esc>", true, true, true), "x", false)
   vim.api.nvim_win_set_cursor(0, { vs + 1, 3 })
-  vim.api.nvim_feedkeys(k "A", "n", false)
+  vim.api.nvim_feedkeys("A", "n", false)
 end
 
 return {
-  codeblock = wrap_codeblock,
+  wrap_codeblock = wrap_codeblock,
 }

lua/mder/commands.lua

@@ -1,6 +1,8 @@
 return {
-  codeblock = function() return require("mder.codeblock").codeblock() end,
-  line = function() return require("mder.line").line() end,
+  codeblock = function() return require("mder.codeblock").wrap_codeblock() end,
+  line = function() return require("mder.line").toggle_lines() end,
   listdn = function() return require("mder.autolist").listdn() end,
   listup = function() return require("mder.autolist").listup() end,
+  img_link = function() return require("mder.link").wrap_or_paste_img() end,
+  raw_link = function() return require("mder.link").wrap_or_paste_raw() end,
 }

lua/mder/config.lua

@@ -1,22 +1,31 @@
+local M = {}
+
 local default_opts = {
   ft = { "markdown", "typst" },
-  line = "<c- >",
-  codeblock = "<c-e>",
-  listdn = "o",
-  listup = "O",
+  keymap = {
+    line = "<c- >",
+    codeblock = "<c-e>",
+    listdn = "o",
+    listup = "O",
+    img_link = "<leader>zi",
+    raw_link = "<leader>zj",
+  },
 }
 
-return {
-  setup = function(opts)
-    opts = vim.tbl_deep_extend("force", default_opts, opts or {})
-    vim.api.nvim_create_autocmd("Filetype", {
-      pattern = opts.ft,
-      callback = function()
-        vim.keymap.set({ "x", "n" }, opts.line, require("mder").line, { buffer = 0 })
-        vim.keymap.set({ "x" }, opts.codeblock, require("mder").codeblock, { buffer = 0 })
-        vim.keymap.set({ "n" }, opts.listdn, require("mder").listdn, { expr = true, buffer = 0 })
-        vim.keymap.set({ "n" }, opts.listup, require("mder").listup, { expr = true, buffer = 0 })
-      end,
-    })
-  end,
-}
+M.setup = function(opts)
+  opts = vim.tbl_deep_extend("force", default_opts, opts or {})
+  vim.api.nvim_create_autocmd("Filetype", {
+    pattern = opts.ft,
+    callback = function()
+      local map = function(mode, act) vim.keymap.set(mode, opts.keymap[act], require("mder")[act], { buffer = 0 }) end
+      map({ "n", "x" }, "line")
+      map({ "x" }, "codeblock")
+      map({ "n" }, "listdn")
+      map({ "n" }, "listup")
+      map({ "n", "x" }, "img_link")
+      map({ "n", "x" }, "raw_link")
+    end,
+  })
+end
+
+return M

lua/mder/line.lua

@@ -1,5 +1,3 @@
-local getvpos = require("mder.utils").getvpos
-
 local check = "%[x%]"
 local empty = "%[ %]"
 
@@ -39,13 +37,13 @@ local toggle_line = function(line)
   return make_box(line)
 end
 
-local line = function()
-  local vs, ve = getvpos()
+local toggle_lines = function()
+  local vs, ve = require("mder.utils").visual_region()
   local lines = vim.api.nvim_buf_get_lines(0, vs, ve, false)
   vim.api.nvim_buf_set_lines(0, vs, ve, false, vim.iter(lines):map(toggle_line):totable())
 end
 
 return {
-  line = line,
+  toggle_lines = toggle_lines,
   toggle_line = toggle_line,
 }

lua/mder/link.lua

@@ -0,0 +1,74 @@
+local M = {}
+
+-- wrap url, then paste it
+local paste = function(wrap_cb, url)
+  if not url then
+    local text = vim.fn.getreg "+"
+    if text == nil then return end
+    -- TODO: whatever, treat it as url now
+    url = text:gsub("\n", "")
+  end
+  url = wrap_cb(url)
+  vim.api.nvim_paste(url, true, 1)
+end
+
+-- return line with wrapped url, or nil if no url in line
+-- TODO: also skip if wrapped url exist
+-- TODO: two or more url
+local wrap_line = function(cb, line)
+  -- https://github.com/sportshead/gx.nvim/blob/77241c1508883882c027aa971b98cd0631ff2a10/lua/gx/helper.lua?plain=1#L49-L60
+  local url_patterns = {
+    "(https?://[a-zA-Z%d_/%%%-%.~@\\+#=?&:]+)",
+    "([a-zA-Z%d_/%-%.~@\\+#]+%.[a-zA-Z%d_/%%%-%.~@\\+#=?&:]+)",
+  }
+  local s, e
+  for _, p in ipairs(url_patterns) do
+    s, e = line:find(p)
+    if s then break end
+  end
+  if not s then return nil end
+  local url = line:sub(s, e) -- NOTE: may not contain 'https?://'
+  url = cb(url)
+  line = line:sub(1, s - 1) .. url .. line:sub(e + 1)
+  return line
+end
+
+-- normal mode only
+local wrap_or_paste = function(cb)
+  local line = vim.api.nvim_get_current_line()
+  local wrapped = wrap_line(cb, line)
+  if not wrapped then return paste(cb) end
+  vim.api.nvim_set_current_line(wrapped)
+end
+
+local wrap_lines = function(cb)
+  -- normal: wrap or paste
+  local mode = vim.api.nvim_get_mode().mode
+  if mode == "n" then return wrap_or_paste(cb) end
+  -- visual: linewise wrap
+  local vs, ve = require("mder.utils").visual_region()
+  local lines = vim.api.nvim_buf_get_lines(0, vs, ve, false)
+  lines = vim.tbl_map(function(line) return wrap_line(cb, line) or line end, lines)
+  vim.api.nvim_buf_set_lines(0, vs, ve, false, lines)
+end
+
+-- wrap with pos inicator...
+local img_cb = function(url, prefix, name)
+  prefix = prefix or "img:"
+  name = name or ""
+  return ("![%s%s](%s)"):format(prefix, name, url), #prefix + 3, #prefix + #name + 5
+end
+
+local raw_cb = function(url) return "<" .. url .. ">", 2 end
+
+local link_cb = function(url, prefix, name)
+  prefix = prefix or "img:"
+  name = name or ""
+  return ("[%s%s](%s)"):format(prefix, name, url), #prefix + 2, #prefix + #name + 4
+end
+
+M.wrap_or_paste_img = function() wrap_lines(img_cb) end
+M.wrap_or_paste_raw = function() wrap_lines(raw_cb) end
+M.wrap_or_paste_link = function() wrap_lines(link_cb) end
+
+return M

lua/mder/utils.lua

@@ -1,14 +1,13 @@
-local getvpos = function()
-  local vs, ve = vim.fn.getpos(".")[2], vim.fn.getpos("v")[2]
+-- 0-index, exclusive, linewise
+-- also "correct" in normal mode
+local visual_region = function()
+  local vs, ve = vim.fn.line ".", vim.fn.line "v"
   if vs > ve then
     vs, ve = ve, vs
   end
   return vs - 1, ve
 end
 
-local k = function(str) return vim.api.nvim_replace_termcodes(str, true, true, true) end
-
 return {
-  getvpos = getvpos,
-  k = k,
+  visual_region = visual_region,
 }