updating lsp, workspace

Author: clpi

lua/down/mod/data/init.lua

@@ -1,11 +1,11 @@
-local mod = require 'down.mod'
-local log = require 'down.util.log'
-local config = require 'down.config'
+local config = require("down.config")
+local log = require("down.util.log")
+local mod = require("down.mod")
 
 ---@class down.mod.Data: down.Mod
-local M = mod.new 'data'
+local M = mod.new("data")
 
----@class down.mod..Data
+---@class down.mod.data.Data
 M.data = {}
 ---@type down.Store<down.File>
 M.file = {
@@ -14,7 +14,7 @@ M.file = {
 
 --- @return down.mod.Setup
 M.setup = function()
-  vim.api.nvim_create_autocmd('VimLeavePre', {
+  vim.api.nvim_create_autocmd("VimLeavePre", {
     callback = function()
       M.flush()
     end,
@@ -31,14 +31,14 @@ M.load = function() end
 
 ---@class down.mod..Config
 M.config = {
-  path = vim.fn.stdpath 'data' .. '/down.mpack',
+  path = vim.fn.stdpath("data") .. "/down.mpack",
   dir = {
-    vim = vim.fs.joinpath(vim.fn.stdpath 'data', 'down/'),
-    home = vim.fs.joinpath(os.getenv 'HOME' or '~/', '.down/'),
+    vim = vim.fs.joinpath(vim.fn.stdpath("data"), "down/"),
+    home = vim.fs.joinpath(os.getenv("HOME") or "~/", ".down/"),
   },
   file = {
-    vim = vim.fs.joinpath(vim.fn.stdpath 'data', 'down/', 'down.json'),
-    home = vim.fs.joinpath(os.getenv 'HOME' or '~/', '.down/', 'down.json'),
+    vim = vim.fs.joinpath(vim.fn.stdpath("data"), "down/", "down.json"),
+    home = vim.fs.joinpath(os.getenv("HOME") or "~/", ".down/", "down.json"),
   },
 }
 
@@ -51,11 +51,11 @@ end
 --- @return table<string>
 M.files = function(path, cond)
   local f = {}
-  local dir = path or vim.fs.root(vim.fn.cwd(), '.down/')
+  local dir = path or vim.fs.root(vim.fn.cwd(), ".down/")
   for name, type in vim.fs.dir(dir) do
-    if type == 'file' and cond or name:endswith '.md' then
+    if type == "file" and cond or name:endswith(".md") then
       table.insert(f, name)
-    elseif type == 'directory' and not name:startswith '.' then
+    elseif type == "directory" and not name:startswith(".") then
       local fs = M.get_files(M.concat(path, name))
       for _, v in ipairs(fs) do
         table.insert(f, v)
@@ -66,7 +66,7 @@ end
 
 M.directory_map = function(path, callback)
   for name, type in vim.fs.dir(path) do
-    if type == 'directory' then
+    if type == "directory" then
       M.directory_map(M.concat(path, name), callback)
     else
       callback(name, type, path)
@@ -80,22 +80,24 @@ end
 --- succeed if the directory already exists.
 ---@return boolean #If true, the directory copying succeeded
 M.copy_directory = function(old_path, new_path)
-  local file_permissions = tonumber('744', 8)
+  local file_permissions = tonumber("744", 8)
   local ok, err = vim.loop.fs_mkdir(new_path, file_permissions)
 
   if not ok then
     return ok, err ---@diagnostic disable-line -- TODO: type error workaround <pysan3>
   end
 
   for name, type in vim.fs.dir(old_path) do
-    if type == 'file' then
-      ok, err = vim.loop.fs_copyfile(M.concat(old_path, name), M.concat(new_path, name))
+    if type == "file" then
+      ok, err =
+        vim.loop.fs_copyfile(M.concat(old_path, name), M.concat(new_path, name))
 
       if not ok then
         return ok, err ---@diagnostic disable-line -- TODO: type error workaround <pysan3>
       end
-    elseif type == 'directory' and not vim.endswith(new_path, name) then
-      ok, err = M.copy_directory(M.concat(old_path, name), M.concat(new_path, name))
+    elseif type == "directory" and not vim.endswith(new_path, name) then
+      ok, err =
+        M.copy_directory(M.concat(old_path, name), M.concat(new_path, name))
 
       if not ok then
         return ok, err ---@diagnostic disable-line -- TODO: type error workaround <pysan3>
@@ -107,11 +109,11 @@ M.copy_directory = function(old_path, new_path)
 end
 --- Grabs the data present on disk and overwrites it with the data present in memory
 M.sync = function()
-  local file = io.open(M.config.path, 'r')
+  local file = io.open(M.config.path, "r")
   if not file then
     return
   end
-  local content = file:read '*a'
+  local content = file:read("*a")
   file:close()
   M.data = vim.mpack.decode and vim.mpack.decode(content)
 end
@@ -121,6 +123,7 @@ end
 ---@param data any #The data to store at the specific key
 M.put = function(key, data)
   M.data[key] = data
+  M.flush()
 end
 
 M.set = M.put
@@ -141,18 +144,20 @@ end
 M.json = function(path)
   local dir = M.config.dir.home
   local vim = M.config.dir.vim
-  vim.fn.mkdir(dir, 'p')
-  vim.fn.mkdir(vim, 'p')
-  local f = io.open(path or M.config.file.vim, 'w')
+  vim.fn.mkdir(dir, "p")
+  vim.fn.mkdir(vim, "p")
+  local f = io.open(path or M.config.file.vim, "w")
   f:write(vim.json.encode(M.data))
 end
 --- Flushes the contents in memory to the location specified
 M.flush = function(path)
-  local file = io.open(path or M.config.path, 'w')
+  local file = io.open(path or M.config.path, "w")
   if not file then
     return
   end
-  file:write(vim.mpack.encode and vim.mpack.encode(M.data) or vim.mpack.pack(M.data))
+  file:write(
+    vim.mpack.encode and vim.mpack.encode(M.data) or vim.mpack.pack(M.data)
+  )
   file:close()
 end
 

lua/down/mod/find/telescope/init.lua

@@ -1,3 +1,4 @@
+local picker = require("down.mod.find.telescope.picker")
 ---@class down.mod.find.telescope.Telescope: down.Mod
 local T = require("down.mod").new("find.telescope")
 
@@ -22,7 +23,7 @@ T.setup = function()
 end
 
 T.load = function()
-  T.picker = { down = require("down.mod.find.telescope.picker") }
+  T.picker = picker
   tel.register_extension({
     exports = T.picker.down,
   })

lua/down/mod/find/telescope/picker/header/init.lua

@@ -1 +1,48 @@
-return function() end
+local lsp = require("down.mod.lsp")
+local ws = require("down.mod.workspace")
+
+local actions = require("telescope.actions")
+local astate = require("telescope.actions.state")
+local autil = require("telescope.actions.utils")
+local entry = require("telescope.from_entry")
+local finders = require("telescope.finders")
+local gen = require("telescope.actions.generate")
+local hist = require("telescope.actions.history")
+local mt = require("telescope.actions.mt")
+local pick = require("telescope.pickers")
+local preview = require("telescope.previewers")
+local set = require("telescope.actions.set")
+local sorters = require("telescope.sorters")
+local ws = require("down.mod.workspace")
+
+---@param o table
+return function(o)
+  pick
+    .new(o or {}, {
+      prompt_title = "Down workspaces",
+      prompt_prefix = " ",
+      results_title = "Workspaces",
+      sorter = sorters.get_generic_fuzzy_sorter({}),
+      finder = finders.new_table({
+        results = ws.as_lsp_workspaces(),
+      }),
+      previewer = preview.vim_buffer_vimgrep.new({}),
+      result_display = function(entry)
+        return entry.value
+      end,
+      attach_mappings = function(prompt_bufnr, map)
+        actions.select_default:replace(function()
+          actions.close(prompt_bufnr)
+          local selection = astate.get_selected_entry()
+          ws.open(selection.value)
+        end)
+        return true
+      end,
+      theme = "dropdown",
+      layout_config = {
+        width = 0.5,
+        height = 0.4,
+      },
+    })
+    :find()
+end

lua/down/mod/find/telescope/picker/link/backlink.lua

@@ -1 +1,65 @@
-return function() end
+local action_state = require("telescope.actions.state")
+local actions = require("telescope.actions")
+local finders = require("telescope.finders")
+local pickers = require("telescope.pickers")
+local previewers = require("telescope.previewers")
+local sorters = require("telescope.sorters")
+local telescope = require("telescope")
+
+local function get_backlinks_to_file(file_path)
+  function find_backlinks(notes, target_note)
+    local backlinks = {}
+    for note, content in pairs(notes) do
+      if content:find(target_note) then
+        table.insert(backlinks, note)
+      end
+    end
+    return backlinks
+  end
+
+  local backlinks = {}
+  -- Assuming a function `find_backlinks` that returns a list of files linking to `file_path`
+  backlinks = find_backlinks(file_path)
+  return backlinks
+end
+
+local function find_backlinks(file_path)
+  -- This function should be implemented to search through the workspace
+  -- and return a list of files that contain links to `file_path`.
+  -- For now, it returns a dummy list for demonstration purposes.
+  return {
+    "file1.md",
+    "file2.md",
+    "file3.md",
+  }
+end
+
+local function backlinks_picker(opts)
+  opts = opts or {}
+  local file_path = vim.fn.expand("%:p") -- Get the current file path
+  local backlinks = get_backlinks_to_file(file_path)
+
+  pickers
+    .new(opts, {
+      prompt_title = "Backlinks to " .. file_path,
+      finder = finders.new_table({
+        results = backlinks,
+      }),
+      sorter = sorters.get_generic_fuzzy_sorter(),
+      previewer = previewers.vim_buffer_cat.new(opts),
+      attach_mappings = function(prompt_bufnr, map)
+        actions.select_default:replace(function()
+          actions.close(prompt_bufnr)
+          local selection = action_state.get_selected_entry()
+          if selection then
+            vim.cmd("edit " .. selection[1])
+          end
+        end)
+        return true
+      end,
+    })
+    :find()
+end
+
+-- To use the picker, you can call `backlinks_picker()` in your Neovim command line
+-- or map it to a keybinding in your Neovim configuration.

lua/down/mod/find/telescope/picker/link/init.lua

@@ -1 +1,66 @@
-return function() end
+local action_state = require("telescope.actions.state")
+local actions = require("telescope.actions")
+local finders = require("telescope.finders")
+local pickers = require("telescope.pickers")
+local previewers = require("telescope.previewers")
+local telescope = require("telescope")
+local conf = require("telescope.config").values
+
+local function parse_links(content, link_type)
+  local links = {}
+  local pattern
+
+  if link_type == "hyperlinks" then
+    pattern = "%b<>"
+  elseif link_type == "file_links" then
+    pattern = "%[.-%]%((.-)%)"
+  elseif link_type == "emails" then
+    pattern = "%S+@%S+"
+  else
+    return links
+  end
+
+  for link in content:gmatch(pattern) do
+    table.insert(links, link)
+  end
+
+  return links
+end
+
+local function markdown_links_picker(opts)
+  opts = opts or {}
+  local link_type = opts.link_type or "hyperlinks"
+
+  local content = vim.fn.join(vim.fn.getline(1, "$"), "\n")
+  local links = parse_links(content, link_type)
+
+  pickers
+    .new(opts, {
+      prompt_title = "Markdown Links",
+      finder = finders.new_table({
+        results = links,
+      }),
+      sorter = conf.generic_sorter(opts),
+      previewer = previewers.new_buffer_previewer({
+        define_preview = function(self, entry, status)
+          local bufnr = self.state.bufnr
+          vim.api.nvim_buf_set_lines(bufnr, 0, -1, false, { entry.value })
+        end,
+      }),
+      attach_mappings = function(prompt_bufnr, map)
+        actions.select_default:replace(function()
+          actions.close(prompt_bufnr)
+          local selection = action_state.get_selected_entry()
+          print("Selected link: " .. selection.value)
+        end)
+        return true
+      end,
+    })
+    :find()
+end
+
+telescope.register_extension({
+  exports = {
+    markdown_links = markdown_links_picker,
+  },
+})

lua/down/mod/find/telescope/picker/link/linkable.lua

@@ -1 +1,48 @@
-return function() end
+local lsp = require("down.mod.lsp")
+local ws = require("down.mod.workspace")
+
+local actions = require("telescope.actions")
+local astate = require("telescope.actions.state")
+local autil = require("telescope.actions.utils")
+local entry = require("telescope.from_entry")
+local finders = require("telescope.finders")
+local gen = require("telescope.actions.generate")
+local hist = require("telescope.actions.history")
+local mt = require("telescope.actions.mt")
+local pick = require("telescope.pickers")
+local preview = require("telescope.previewers")
+local set = require("telescope.actions.set")
+local sorters = require("telescope.sorters")
+local ws = require("down.mod.workspace")
+
+---@param o table
+return function(o)
+  pick
+    .new(o or {}, {
+      prompt_title = "Down workspaces",
+      prompt_prefix = " ",
+      results_title = "Workspaces",
+      sorter = sorters.get_generic_fuzzy_sorter({}),
+      finder = finders.new_table({
+        results = ws.as_lsp_workspaces(),
+      }),
+      previewer = preview.vim_buffer_vimgrep.new({}),
+      result_display = function(entry)
+        return entry.value
+      end,
+      attach_mappings = function(prompt_bufnr, map)
+        actions.select_default:replace(function()
+          actions.close(prompt_bufnr)
+          local selection = astate.get_selected_entry()
+          ws.open(selection.value)
+        end)
+        return true
+      end,
+      theme = "dropdown",
+      layout_config = {
+        width = 0.5,
+        height = 0.4,
+      },
+    })
+    :find()
+end