-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Implement roam origin functionality * Add origin as option to node api * Update expr_under_cursor, link_under_cursor, and node_under_cursor to support explicit window * Add open config option for node view * Update select-node to take exclusion list * Update add_origin to exclude self * Update node view to support navigating to origin * Add goto_next_node and goto_prev_node api methods
- Loading branch information
1 parent
8065e6d
commit 8772079
Showing
20 changed files
with
671 additions
and
143 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,127 @@ | ||
| ------------------------------------------------------------------------------- | ||
| -- ORIGIN.LUA | ||
| -- | ||
| -- Contains functionality tied to the roam origin api. | ||
| ------------------------------------------------------------------------------- | ||
|
|
||
| local db = require("org-roam.database") | ||
| local select_node = require("org-roam.ui.select-node") | ||
| local utils = require("org-roam.utils") | ||
|
|
||
| local ORIGIN_PROP_NAME = "ROAM_ORIGIN" | ||
|
|
||
| local M = {} | ||
|
|
||
| ---Adds an origin to the node under cursor. | ||
| ---Will replace the existing origin. | ||
| --- | ||
| ---If no `origin` is specified, a prompt is provided. | ||
| --- | ||
| ---@param opts? {origin?:string} | ||
| function M.add_origin(opts) | ||
| opts = opts or {} | ||
| utils.node_under_cursor(function(node) | ||
| if not node then return end | ||
|
|
||
| db:load_file({ path = node.file }):next(function(results) | ||
| -- Get the OrgFile instance | ||
| local file = results.file | ||
|
|
||
| -- Look for a file or headline that matches our node | ||
| local entry = utils.find_id_match(file, node.id) | ||
|
|
||
| if entry and opts.origin then | ||
| entry:set_property(ORIGIN_PROP_NAME, opts.origin) | ||
| elseif entry then | ||
| -- If no origin specified, we load up a selection dialog | ||
| -- to pick a node other than the current one | ||
| select_node({ exclude = { node.id } }, function(selection) | ||
| if selection.id then | ||
| entry:set_property(ORIGIN_PROP_NAME, selection.id) | ||
| end | ||
| end) | ||
| end | ||
|
|
||
| return file | ||
| end) | ||
| end) | ||
| end | ||
|
|
||
| ---Goes to the previous node in sequence for the node under cursor. | ||
| ---Leverages a lookup of the node using the origin of the node under cursor. | ||
| ---@param opts? {win?:integer} | ||
| function M.goto_prev_node(opts) | ||
| opts = opts or {} | ||
| local winnr = opts.win or vim.api.nvim_get_current_win() | ||
|
|
||
| ---@param node org-roam.core.file.Node|nil | ||
| local function goto_node(node) | ||
| if not node then return end | ||
| utils.goto_node({ node = node, win = winnr }) | ||
| end | ||
|
|
||
| utils.node_under_cursor(function(node) | ||
| if not node or not node.origin then return end | ||
| db:get(node.origin):next(goto_node) | ||
| end, { win = winnr }) | ||
| end | ||
|
|
||
| ---Goes to the next node in sequence for the node under cursor. | ||
| ---Leverages a lookup of nodes whose origin match the node under cursor. | ||
| ---@param opts? {win?:integer} | ||
| function M.goto_next_node(opts) | ||
| opts = opts or {} | ||
| local winnr = opts.win or vim.api.nvim_get_current_win() | ||
|
|
||
| ---@param node org-roam.core.file.Node|nil | ||
| local function goto_node(node) | ||
| if not node then return end | ||
| utils.goto_node({ node = node, win = winnr }) | ||
| end | ||
|
|
||
| utils.node_under_cursor(function(node) | ||
| if not node then return end | ||
| db:find_nodes_by_origin(node.id):next(function(nodes) | ||
| if #nodes == 0 then return nodes end | ||
| if #nodes == 1 then | ||
| goto_node(nodes[1]) | ||
| return nodes | ||
| end | ||
|
|
||
| local ids = vim.tbl_map(function(n) | ||
| return n.id | ||
| end, nodes) | ||
|
|
||
| select_node({ include = ids }, function(selection) | ||
| if selection.id then | ||
| db:get(selection.id):next(goto_node) | ||
| end | ||
| end) | ||
|
|
||
| return nodes | ||
| end) | ||
| end, { win = winnr }) | ||
| end | ||
|
|
||
| ---Removes the origin from the node under cursor. | ||
| function M.remove_origin() | ||
| utils.node_under_cursor(function(node) | ||
| if not node then return end | ||
|
|
||
| db:load_file({ path = node.file }):next(function(results) | ||
| -- Get the OrgFile instance | ||
| local file = results.file | ||
|
|
||
| -- Look for a file or headline that matches our node | ||
| local entry = utils.find_id_match(file, node.id) | ||
|
|
||
| if entry then | ||
| entry:set_property(ORIGIN_PROP_NAME, nil) | ||
| end | ||
|
|
||
| return file | ||
| end) | ||
| end) | ||
| end | ||
|
|
||
| return M |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.