Merge branch 'renerocksai:main' into main

lua/taglinks/tagutils.lua

@@ -0,0 +1,126 @@
+local Job = require("plenary.job")
+
+local M = {}
+local hashtag_re = "(^|\\s|'|\")#[a-zA-Z]+[a-zA-Z0-9/\\-_]*"
+local colon_re = "(^|\\s):[a-zA-Z]+[a-zA-Z0-9/\\-_]*:"
+local yaml_re = "(^|\\s)tags:\\s*\\[([a-zA-Z]+[a-zA-Z0-9/\\-_]*(,\\s)*)*]"
+
+local function command_find_all_tags(opts)
+    opts = opts or {}
+    opts.cwd = opts.cwd or "."
+    local re = hashtag_re
+
+    if opts.tag_notation == ":tag:" then
+        re = colon_re
+    end
+
+    if opts.tag_notation == "yaml-bare" then
+        re = yaml_re
+    end
+
+    return "rg", { "--vimgrep", "-o", re, "--", opts.cwd }
+end
+
+local function trim(s)
+    if s:sub(1, 1) == '"' or s:sub(1, 1) == "'" then
+        s = s:sub(2)
+    end
+    return (string.gsub(s, "^%s*(.-)%s*$", "%1"))
+end
+
+local function insert_tag(tbl, tag, entry)
+    entry.t = tag
+    if tbl[tag] == nil then
+        tbl[tag] = { entry }
+    else
+        tbl[tag][#tbl[tag] + 1] = entry
+    end
+end
+
+local function split(line, sep, n)
+    local startpos = 0
+    local endpos
+    local ret = {}
+    for _ = 1, n - 1 do
+        endpos = line:find(sep, startpos + 1)
+        ret[#ret + 1] = line:sub(startpos + 1, endpos - 1)
+        startpos = endpos
+    end
+    -- now the remainder
+    ret[n] = line:sub(startpos + 1)
+    return ret
+end
+
+local function yaml_to_tags(line, entry, ret)
+    local _, startpos = line:find("tags%s*:%s*%[")
+    local global_end = line:find("%]")
+
+    line = line:sub(startpos + 1, global_end)
+
+    local i = 1
+    local j
+    local prev_i
+    local tag
+    while true do
+        i, j = line:find("%s*.*%s*,", i)
+        if i == nil then
+            tag = line:sub(prev_i)
+            tag = tag:gsub("%s*(.*)%s*", "%1")
+        else
+            tag = line:sub(i, j)
+            tag = tag:gsub("%s*(.*)%s*,", "%1")
+        end
+
+        local new_entry = {}
+        new_entry.t = tag
+        new_entry.l = entry.l
+        new_entry.fn = entry.fn
+        new_entry.c = startpos + (i or prev_i)
+        insert_tag(ret, tag, new_entry)
+        if i == nil then
+            break
+        end
+        i = j + 1
+        prev_i = i
+    end
+end
+
+local function parse_entry(opts, line, ret)
+    local s = split(line, ":", 4)
+    local fn, l, c, t = s[1], s[2], s[3], s[4]
+
+    t = trim(t)
+    local entry = { fn = fn, l = l, c = c }
+
+    if opts.tag_notation == "yaml-bare" then
+        yaml_to_tags(t, entry, ret)
+    elseif opts.tag_notation == ":tag:" then
+        insert_tag(ret, t, entry)
+    else
+        insert_tag(ret, t, entry)
+    end
+end
+
+M.do_find_all_tags = function(opts)
+    local cmd, args = command_find_all_tags(opts)
+    -- print(cmd .. " " .. vim.inspect(args))
+    local ret = {}
+    local _ = Job
+        :new({
+            command = cmd,
+            args = args,
+            enable_recording = true,
+            on_exit = function(j, return_val)
+                for _, line in pairs(j:result()) do
+                    parse_entry(opts, line, ret)
+                end
+            end,
+            on_stderr = function(err, data, _)
+                print("error: " .. tostring(err) .. "data: " .. data)
+            end,
+        })
+        :sync()
+    -- print("final results: " .. vim.inspect(ret))
+    return ret
+end
+return M

lua/telekasten.lua

@@ -13,6 +13,7 @@ local themes = require("telescope.themes")
 local debug_utils = require("plenary.debug_utils")
 local filetype = require("plenary.filetype")
 local taglinks = require("taglinks.taglinks")
+local tagutils = require("taglinks.tagutils")
 
 -- declare locals for the nvim api stuff to avoid more lsp warnings
 local vim = vim
@@ -74,6 +75,10 @@ M.Cfg = {
 
     -- command palette theme: dropdown (window) or ivy (bottom panel)
     command_palette_theme = "ivy",
+
+    -- tag list theme:
+    -- get_cursor: small tag list at cursor; ivy and dropdown like above
+    show_tags_theme = "ivy",
 }
 
 local function file_exists(fname)
@@ -480,6 +485,29 @@ function picker_actions.close(opts)
     end
 end
 
+function picker_actions.paste_tag(opts)
+    return function(prompt_bufnr)
+        actions.close(prompt_bufnr)
+        local selection = action_state.get_selected_entry()
+        vim.api.nvim_put({ selection.value.tag }, "", true, true)
+        if opts.insert_after_inserting or opts.i then
+            vim.api.nvim_feedkeys("A", "m", false)
+        end
+    end
+end
+
+function picker_actions.yank_tag(opts)
+    return function(prompt_bufnr)
+        opts = opts or {}
+        if opts.close_after_yanking then
+            actions.close(prompt_bufnr)
+        end
+        local selection = action_state.get_selected_entry()
+        vim.fn.setreg('"', selection.value.tag)
+        print("yanked " .. selection.value.tag)
+    end
+end
+
 function picker_actions.paste_link(opts)
     return function(prompt_bufnr)
         actions.close(prompt_bufnr)
@@ -487,7 +515,7 @@ function picker_actions.paste_link(opts)
         local fn = path_to_linkname(selection.value)
         local title = "[[" .. fn .. "]]"
         vim.api.nvim_put({ title }, "", true, true)
-        if opts.insert_after_inserting then
+        if opts.insert_after_inserting or opts.i then
             vim.api.nvim_feedkeys("A", "m", false)
         end
     end
@@ -515,7 +543,7 @@ function picker_actions.paste_img_link(opts)
         fn = fn:gsub(M.Cfg.home .. "/", "")
         local imglink = "![](" .. fn .. ")"
         vim.api.nvim_put({ imglink }, "", true, true)
-        if opts.insert_after_inserting then
+        if opts.insert_after_inserting or opts.i then
             vim.api.nvim_feedkeys("A", "m", false)
         end
     end
@@ -725,13 +753,11 @@ local function follow_url(url)
     if vim.fn.has("mac") == 1 then
         command = format_command("open")
         vim.cmd(command)
+    elseif vim.fn.has("unix") then
+        command = format_command("xdg-open")
+        vim.cmd(command)
     else
-        if vim.fn.has("unix") then
-            command = format_command("xdg-open")
-            vim.cmd(command)
-        else
-            print("Cannot open URLs on your operating system")
-        end
+        print("Cannot open URLs on your operating system")
     end
 end
 
@@ -754,14 +780,23 @@ local function FollowLink(opts)
     local fexists
 
     -- first: check if we're in a tag or a link
-    local kind, atcol = check_for_link_or_tag()
+    local kind, atcol, tag
+
+    if opts.follow_tag ~= nil then
+        kind = "tag"
+        tag = opts.follow_tag
+    else
+        kind, atcol = check_for_link_or_tag()
+    end
 
     if kind == "tag" then
-        local tag = taglinks.get_tag_at(
-            vim.api.nvim_get_current_line(),
-            atcol,
-            M.Cfg
-        )
+        if atcol ~= nil then
+            tag = taglinks.get_tag_at(
+                vim.api.nvim_get_current_line(),
+                atcol,
+                M.Cfg
+            )
+        end
         search_mode = "tag"
         title = tag
     else
@@ -774,7 +809,6 @@ local function FollowLink(opts)
             -- we are in an external [link]
             vim.cmd("normal yi)")
             local url = vim.fn.getreg('"0')
-            print(url)
             return follow_url(url)
         end
 
@@ -1540,6 +1574,86 @@ local function ToggleTodo(opts)
     end
 end
 
+local function FindAllTags(opts)
+    opts = opts or {}
+    local i = opts.i
+    opts.cwd = M.Cfg.home
+    opts.tag_notation = M.Cfg.tag_notation
+
+    local tag_map = tagutils.do_find_all_tags(opts)
+    local taglist = {}
+
+    local max_tag_len = 0
+    for k, v in pairs(tag_map) do
+        taglist[#taglist + 1] = { tag = k, details = v }
+        if #k > max_tag_len then
+            max_tag_len = #k
+        end
+    end
+
+    if M.Cfg.show_tags_theme == "get_cursor" then
+        opts = themes.get_cursor({
+            layout_config = {
+                height = math.min(math.floor(vim.o.lines * 0.8), #taglist),
+            },
+        })
+    elseif M.Cfg.show_tags_theme == "ivy" then
+        opts = themes.get_ivy({
+            layout_config = {
+                prompt_position = "top",
+                height = math.min(math.floor(vim.o.lines * 0.8), #taglist),
+            },
+        })
+    else
+        opts = themes.get_dropdown({
+            layout_config = {
+                prompt_position = "top",
+                height = math.min(math.floor(vim.o.lines * 0.8), #taglist),
+            },
+        })
+    end
+    -- re-apply
+    opts.cwd = M.Cfg.home
+    opts.tag_notation = M.Cfg.tag_notation
+    opts.i = i
+    pickers.new(opts, {
+        prompt_title = "Tags",
+        finder = finders.new_table({
+            results = taglist,
+            entry_maker = function(entry)
+                return {
+                    value = entry,
+                    -- display = entry.tag .. ' \t (' .. #entry.details .. ' matches)',
+                    display = string.format(
+                        "%" .. max_tag_len .. "s ... (%3d matches)",
+                        entry.tag,
+                        #entry.details
+                    ),
+                    ordinal = entry.tag,
+                }
+            end,
+        }),
+        sorter = conf.generic_sorter(opts),
+        attach_mappings = function(prompt_bufnr, map)
+            actions.select_default:replace(function()
+                actions._close(prompt_bufnr, true)
+
+                -- TODO actions for insert tag, default action: search for tag
+                local selection = action_state.get_selected_entry().value.tag
+                local follow_opts = { follow_tag = selection }
+                FollowLink(follow_opts)
+            end)
+            map("i", "<c-y>", picker_actions.yank_tag(opts))
+            map("i", "<c-i>", picker_actions.paste_tag(opts))
+            map("n", "<c-y>", picker_actions.yank_tag(opts))
+            map("n", "<c-i>", picker_actions.paste_tag(opts))
+            map("n", "<c-c>", picker_actions.close(opts))
+            map("n", "<esc>", picker_actions.close(opts))
+            return true
+        end,
+    }):find()
+end
+
 -- Setup(cfg)
 --
 -- Overrides config with elements from cfg. See top of file for defaults.
@@ -1647,6 +1761,7 @@ M.insert_img_link = InsertImgLink
 M.preview_img = PreviewImg
 M.browse_media = BrowseImg
 M.taglinks = taglinks
+M.show_tags = FindAllTags
 
 -- Telekasten command, completion
 local TelekastenCmd = {
@@ -1684,6 +1799,7 @@ local TelekastenCmd = {
             { "preview image under cursor", "preview_img", M.preview_img },
             { "browse media", "browse_media", M.browse_media },
             { "panel", "panel", M.panel },
+            { "show tags", "show_tags", M.show_tags },
         }
     end,
 }