WIP LINK REGISTRY

Author: SlayerOfTheBad

lua/orgmode/init.lua

@@ -20,6 +20,7 @@ local auto_instance_keys = {
 ---@field capture OrgCapture
 ---@field clock OrgClock
 ---@field completion OrgCompletion
+---@field links OrgLinkHandlerRegistry
 ---@field org_mappings OrgMappings
 ---@field notifications OrgNotifications
 local Org = {}
@@ -66,6 +67,9 @@ function Org:init()
     files = self.files,
   })
   self.completion = require('orgmode.org.autocompletion'):new({ files = self.files })
+  self.links = require('orgmode.org.links'):new({
+    files = self.files,
+  })
   self.statusline_debounced = require('orgmode.utils').debounce('statusline', function()
     return self.clock:get_statusline()
   end, 300)

lua/orgmode/org/links/handlers/_internal/custom_id.lua

@@ -0,0 +1,53 @@
+local utils = require('orgmode.utils')
+local Org = require('orgmode')
+local Internal = require('orgmode.org.hyperlinks.builtin.internal')
+
+---@class OrgLinkHandlerCustomId:OrgLinkHandlerInternal
+local CustomId = Internal:new()
+
+function CustomId:new(custom_id)
+  ---@class OrgLinkHandlerCustomId
+  local this = Internal:new()
+  this.custom_id = custom_id
+  setmetatable(this, self)
+  self.__index = self
+  return this
+end
+
+---@param input string
+function CustomId.parse(input)
+  return CustomId:new(input)
+end
+
+function CustomId:__tostring()
+  return string.format('#%s', self.custom_id)
+end
+
+function CustomId:follow()
+  local headlines = Org.files:get_current_file():find_headlines_with_property_matching('custom_id', self.custom_id)
+
+  if #headlines == 0 then
+    return utils.echo_warning(('Could not find custom ID "%s".'):format(self.custom_id))
+  end
+
+  self.goto_oneof(headlines)
+end
+
+function CustomId:insert_description()
+  return self.custom_id
+end
+
+function CustomId:complete(lead, context)
+  local file = self.get_file_from_context(context)
+  local headlines = file:find_headlines_with_property_matching('CUSTOM_ID', lead)
+
+  local completions = {}
+  for _, headline in pairs(headlines) do
+    local id = headline:get_property('CUSTOM_ID')
+    table.insert(completions, self:new(id):__tostring())
+  end
+
+  return completions
+end
+
+return CustomId

lua/orgmode/org/links/handlers/_internal/headline.lua

@@ -0,0 +1,68 @@
+local utils = require('orgmode.utils')
+local Org = require('orgmode')
+local Internal = require('orgmode.org.hyperlinks.builtin.internal')
+local Id = require('orgmode.org.hyperlinks.builtin.id')
+
+---@class OrgLinkHandlerHeadline:OrgLinkHandlerInternal
+local Headline = Internal:new()
+
+function Headline:new(headline)
+  ---@class OrgLinkHandlerHeadline
+  local this = Internal:new()
+  this.headline = headline
+  setmetatable(this, self)
+  self.__index = self
+  return this
+end
+
+---@param input string
+function Headline.parse(input)
+  return Headline:new(input)
+end
+
+function Headline:__tostring()
+  return string.format('*%s', self.headline)
+end
+
+function Headline:follow()
+  local headlines = Org.files:get_current_file():find_headlines_by_title(self.headline)
+
+  if #headlines == 0 then
+    return utils.echo_warning(('Could not find headline "%s".'):format(self.headline))
+  end
+
+  self.goto_oneof(headlines)
+end
+
+function Headline:resolve()
+  local headlines = Org.files:get_current_file():find_headlines_by_title(self.headline)
+
+  if #headlines == 0 then
+    return self
+  end
+
+  local id = headlines[1]:get_property('id')
+  if not id then
+    return self
+  end
+
+  return Id:new(id):resolve()
+end
+
+function Headline:insert_description()
+  return self.headline
+end
+
+function Headline:complete(lead, context)
+  local file = self.get_file_from_context(context)
+  local headlines = file:find_headlines_by_title(lead)
+
+  local completions = {}
+  for _, headline in pairs(headlines) do
+    table.insert(completions, Headline:new(headline:get_title()):__tostring())
+  end
+
+  return completions
+end
+
+return Headline

lua/orgmode/org/links/handlers/_internal/line_number.lua

@@ -0,0 +1,28 @@
+local Internal = require('orgmode.org.hyperlinks.builtin.internal')
+
+---@class OrgLinkHandlerLineNumber:OrgLinkHandlerInternal
+local LineNumber = Internal:new()
+
+function LineNumber:new(line_number)
+  ---@class OrgLinkHandlerLineNumber
+  local this = Internal:new()
+  this.line_number = line_number
+  setmetatable(this, self)
+  self.__index = self
+  return this
+end
+
+---@param input string
+function LineNumber.parse(input)
+  return LineNumber:new(tonumber(input))
+end
+
+function LineNumber:__tostring()
+  return string.format('%d', self.line_number)
+end
+
+function LineNumber:follow()
+  vim.cmd(('normal! %dGzv'):format(self.line_number))
+end
+
+return LineNumber

lua/orgmode/org/links/handlers/_internal/plain.lua

@@ -0,0 +1,69 @@
+local utils = require('orgmode.utils')
+local Org = require('orgmode')
+local Internal = require('orgmode.org.hyperlinks.builtin.internal')
+
+---@class OrgLinkHandlerPlain:OrgLinkHandlerInternal
+local Plain = Internal:new()
+
+function Plain:new(text)
+  ---@class OrgLinkHandlerPlain
+  local this = Internal:new()
+  this.text = text
+  setmetatable(this, self)
+  self.__index = self
+  return this
+end
+
+---@param input string
+function Plain.parse(input)
+  return Plain:new(input)
+end
+
+function Plain:__tostring()
+  return self.text
+end
+
+function Plain:follow()
+  local anchors = vim.fn.matchbufline(0, ('<<<?%s[^>]*>>>?'):format(self.text), 0, '$')
+
+  if #anchors >= 1 then
+    vim.fn.cursor(anchors[1].lnum, anchors[1].byteidx)
+    return
+  end
+
+  -- TODO from here, behaviour should depend on org-link-search-must-match-exact-headline
+  -- TODO #+NAME tag support should be added, but it didn't exists yet
+
+  local plain_text_matches = vim.fn.matchbufline(0, self.text, 0, '$')
+
+  if #plain_text_matches >= 1 then
+    vim.fn.cursor(plain_text_matches[1].lnum, plain_text_matches[1].byteidx)
+    return
+  end
+
+  return utils.echo_warning(('No matches found for "%s".'):format(self.text))
+end
+
+function Plain:insert_description()
+  return self.text
+end
+
+-- TODO #+NAME tag support should be added, but it didn't exists yet
+function Plain:complete(lead, context)
+  local file = self.get_file_from_context(context)
+  local completions = {}
+
+  local anchors = file.content:gmatch(('<<<?%s[^>]*>>>?'):format(lead))
+  for anchor in anchors do
+    table.insert(completions, Plain:new(anchor):__tostring())
+  end
+
+  local headlines = file:find_headlines_by_title(lead)
+  for _, headline in pairs(headlines) do
+    table.insert(completions, Plain:new(headline:get_title()):__tostring())
+  end
+
+  return completions
+end
+
+return Plain

lua/orgmode/org/links/handlers/file.lua

@@ -0,0 +1,152 @@
+local fs = require('orgmode.utils.fs')
+local Link = require('orgmode.org.links.link_handler')
+local Id = require('orgmode.org.links.handlers.id')
+local Internal = require('orgmode.org.links.handlers.internal')
+
+---@class OrgLinkHandlerFile:OrgLinkHandler
+---@field new fun(self: OrgLinkHandlerFile, path: string, target: OrgLinkHandlerInternal | nil, prefix: boolean | nil): OrgLinkHandlerFile
+---@field parse fun(link: string, prefix: boolean | nil): OrgLinkHandlerFile | nil
+---@field path string
+---@field skip_prefix boolean
+---@field target OrgLinkHandlerInternal | nil
+local File = Link:new('file')
+
+function File:new(path, target, skip_prefix)
+  ---@class OrgLinkHandlerFile
+  local this = Link:new()
+  this.skip_prefix = skip_prefix or false
+  this.path = path
+  this.target = target
+  setmetatable(this, self)
+  self.__index = self
+  return this
+end
+
+function File.parse(input, skip_prefix)
+  if input == nil or #input == 0 then
+    return nil
+  end
+  local deliniator_start, deliniator_stop = input:find('::')
+
+  ---@type OrgLinkHandlerInternal | nil
+  local target = nil
+  local path = input
+
+  if deliniator_start then
+    ---@class OrgLinkHandlerInternal | nil
+    target = Internal.parse(input:sub(deliniator_stop + 1), true)
+    path = input:sub(0, deliniator_start - 1)
+  end
+
+  return File:new(path, target, skip_prefix)
+end
+
+-- TODO make protocol prefix optional. Based on what?
+function File:__tostring()
+  local v = ''
+  if self.skip_prefix then
+    v = ('%s'):format(self.path)
+  else
+    v = ('%s:%s'):format(self.protocol, self.path)
+  end
+
+  if self.target then
+    v = string.format('%s::%s', v, self.target)
+  end
+
+  return v
+end
+
+function File:follow()
+  vim.cmd('edit ' .. fs.get_real_path(self.path))
+
+  if self.target then
+    self.target:follow()
+  end
+end
+
+local function autocompletions_filenames(lead)
+  local Org = require('orgmode')
+  local filenames = Org.files:filenames()
+
+  local matches = {}
+  for _, f in ipairs(filenames) do
+    local realpath = fs.substitute_path(lead) or lead
+    if f:find('^' .. realpath) then
+      local path = f:gsub('^' .. realpath, lead)
+      table.insert(matches, { real = f, path = path })
+    end
+  end
+
+  print(vim.inspect(matches))
+  return matches
+end
+
+function File:resolve()
+  local Org = require('orgmode')
+  local path = fs.get_real_path(self.path)
+  if not path then
+    return self
+  end
+  local file = Org.files:get(path)
+  if not file then
+    return self
+  end
+  local id = file:get_property('id')
+  if not id then
+    return self
+  end
+
+  return Id:new(id, self.target):resolve()
+end
+
+function File:insert_description()
+  local Org = require('orgmode')
+  if self.target then
+    return self.target:insert_description()
+  end
+
+  local path = fs.get_real_path(self.path)
+  if not path then
+    return nil
+  end
+  local file = Org.files:get(path)
+  if not file then
+    return nil
+  end
+
+  return file:get_title()
+end
+
+function File:complete(lead, context)
+  context = context or {}
+  local deliniator_start, deliniator_stop = lead:find('::')
+
+  if not deliniator_start then
+    return self:_complete(lead, context)
+  else
+    local path = lead:sub(0, deliniator_start - 1)
+    local target_lead = lead:sub(deliniator_stop + 1)
+    return self:_complete_targets(path, target_lead, context)
+  end
+end
+
+function File:_complete(lead, context)
+  return vim.tbl_map(function(f)
+    return self:new(f, nil, context.skip_prefix):__tostring()
+  end, autocompletions_filenames(lead))
+end
+
+function File:_complete_targets(path, target_lead, context)
+  return vim.tbl_map(
+    function(t)
+      return self:new(path, t, context.skip_prefix):__tostring()
+    end,
+    Internal:complete(target_lead, {
+      filename = fs.get_real_path(path),
+      only_internal = true,
+    })
+  )
+end
+
+return File