hephaestus/heph.nvim/lua/heph/link.lua

--- `[[wiki-link]]` parsing and following (tech-spec §8).
---
--- The cursor grammar mirrors `heph-core`'s `extract.rs`: a span `[[target]]`
--- or `[[target|display]]`, where the resolvable name is everything left of the
--- first `|`, trimmed. Resolution goes through `node.resolve` (exact, the same
--- mapping that materializes stored `wiki` links) — never fuzzy `search`, which
--- would mis-jump.

local rpc = require("heph.rpc")
local util = require("heph.util")

local M = {}

--- The wiki target under the cursor on the current line, or nil. Scans for the
--- `[[...]]` span that contains the cursor column.
function M.target_under_cursor()
  local line = vim.api.nvim_get_current_line()
  local col = vim.api.nvim_win_get_cursor(0)[2] + 1 -- 1-based byte column
  local from = 1
  while true do
    local open_s, open_e = line:find("[[", from, true)
    if not open_s then
      return nil
    end
    local close_s, close_e = line:find("]]", open_e + 1, true)
    if not close_s then
      return nil
    end
    if col >= open_s and col <= close_e then
      local inner = line:sub(open_e + 1, close_s - 1)
      local target = inner:match("^([^|]*)") or ""
      target = target:gsub("^%s+", ""):gsub("%s+$", "")
      return (#target > 0) and target or nil
    end
    from = close_e + 1
  end
end

--- Follow the `[[link]]` under the cursor to its node, **creating** the target
--- doc if it doesn't exist yet (the zettelkasten follow-or-create gesture). The
--- newly-created doc resolves the source's previously-unresolved wiki-link, so
--- re-saving the source materializes the backlink.
function M.follow()
  local target = M.target_under_cursor()
  if not target then
    util.notify("no [[link]] under cursor", vim.log.levels.WARN)
    return
  end
  local node = rpc.call("node.resolve", { title = target })
  if not node then
    node = rpc.call("node.create", { kind = "doc", title = target, body = "" })
    util.notify("created [[" .. target .. "]]")
    -- Materialize the source's wiki-link to the new doc — it was unresolved when
    -- the source was saved, so extraction skipped it (tech-spec §5). If the
    -- source has unsaved edits, saving re-extracts and materializes it (and
    -- persists the edits); otherwise add the link directly (a no-op re-save
    -- wouldn't re-extract).
    local src = vim.api.nvim_get_current_buf()
    local src_id = vim.b[src].heph_node_id
    if src_id then
      if vim.bo[src].modified then
        pcall(require("heph.node").write, src, vim.api.nvim_buf_get_name(src))
      else
        pcall(rpc.call, "links.add", { src = src_id, dst = node.id, link_type = "wiki" })
      end
    end
  end
  require("heph.node").open(node.id)
end

--- Pick a node (by full-text search) and insert a canonical `[[NODEID]]` link at
--- the cursor — the authoring path for wiki-links-by-id (§8.4); a node id is the
--- only thing that ever enters a stored link, so there's no name ambiguity. A
--- "Create" entry mints a new doc named after the query. No-op if cancelled.
function M.insert()
  vim.ui.input({ prompt = "Link to: " }, function(query)
    if not query or query == "" then
      return
    end
    local items = {}
    for _, hit in ipairs(rpc.call("search", { query = query }) or {}) do
      items[#items + 1] = hit
    end
    items[#items + 1] = { __create = true, title = query }
    require("heph.picker").select(items, {
      prompt = "Link to",
      format = function(it)
        if it.__create then
          return "+ Create new doc: " .. it.title
        end
        return it.title .. "  [" .. (it.kind or "node") .. "]"
      end,
    }, function(choice)
      if not choice then
        return
      end
      local id, title
      if choice.__create then
        local node = rpc.call("node.create", { kind = "doc", title = choice.title })
        id, title = node.id, node.title
      else
        id, title = choice.id, choice.title
      end
      -- Insert the labelled form `[[id|Name]]` (readable + conceal-ready); it
      -- collapses to the canonical bare `[[id]]` on save (§8.4).
      vim.api.nvim_put({ "[[" .. id .. "|" .. title .. "]]" }, "c", true, true)
    end)
  end)
end

--- Attach the buffer-local follow/insert keymaps and inline-`#hashtag`
--- highlighting (only on heph:// buffers).
function M.attach(buf)
  vim.keymap.set("n", "<CR>", function()
    M.follow()
  end, { buffer = buf, desc = "heph: follow [[link]]" })
  -- Typing `[[` opens the node picker (Obsidian-style), inserting `[[NODEID]]`.
  vim.keymap.set("i", "[[", function()
    M.insert()
  end, { buffer = buf, desc = "heph: insert [[link]]" })

  -- Render inline #hashtags in italics so they stand out — matching the
  -- save-time tag detection (whitespace-prefixed `#word`, never a `# heading`).
  -- `default = true` leaves a user's own `HephHashtag` definition intact.
  vim.api.nvim_set_hl(0, "HephHashtag", { italic = true, default = true })
  vim.api.nvim_buf_call(buf, function()
    vim.cmd([[syntax match HephHashtag /\v%(^|\s)@<=#[0-9A-Za-z_-]+/ containedin=ALL]])
  end)
end

return M