Carlos Gutierrez
0d1aa591e5
Major performance improvements for graph visualization: Cache System: - Add cache.lua module that stores parsed links/tags per file - Only re-parse files that have been modified (mtime check) - Cache stored in .ideadrop-graph-cache.json in idea_dir - Fast file scanning using vim.fs.find when available Layout Optimizations: - Reduce max iterations from 300 to 100 - Faster convergence with adjusted parameters - Barnes-Hut approximation for large graphs (100+ nodes) - Cache math functions locally for speed - Skip distant node pairs in repulsion calculation - Reuse visible_nodes array across iterations New Commands: - :IdeaGraph rebuild - Force full cache rebuild - :IdeaGraphClearCache - Clear cache file This makes opening the graph nearly instant for previously scanned vaults, similar to Obsidian's behavior.
329 lines
8.8 KiB
Lua
329 lines
8.8 KiB
Lua
-- ideaDrop.nvim/core/init.lua
|
|
-- Core modules
|
|
local config = require("ideaDrop.core.config")
|
|
|
|
-- UI modules
|
|
local sidebar = require("ideaDrop.ui.sidebar")
|
|
local tree = require("ideaDrop.ui.tree")
|
|
local graph = require("ideaDrop.ui.graph")
|
|
|
|
-- Feature modules
|
|
local list = require("ideaDrop.features.list")
|
|
local tags = require("ideaDrop.features.tags")
|
|
local search = require("ideaDrop.features.search")
|
|
|
|
-- Utility modules
|
|
local keymaps = require("ideaDrop.utils.keymaps")
|
|
|
|
local M = {}
|
|
|
|
---@class IdeaDrop
|
|
---@field setup fun(user_opts: IdeaDropConfig): nil
|
|
|
|
---@class IdeaDropConfig
|
|
---@field idea_dir string Directory where idea files will be stored
|
|
|
|
---Setup function for ideaDrop.nvim
|
|
---@param user_opts IdeaDropConfig|nil User configuration options
|
|
---@return nil
|
|
function M.setup(user_opts)
|
|
config.setup(user_opts)
|
|
|
|
-- Command to open ideas in floating window (original behavior)
|
|
vim.api.nvim_create_user_command("Idea", function(opts)
|
|
local arg = opts.args
|
|
local idea_dir = config.options.idea_dir
|
|
|
|
if arg == "listAll" then
|
|
list.list_all()
|
|
elseif arg ~= "" then
|
|
-- Ensure directory exists (even for nested folders)
|
|
local filename = arg:match("%.md$") and arg or (arg .. ".md")
|
|
local full_path = idea_dir .. "/" .. filename
|
|
|
|
-- Create parent folders if needed
|
|
local folder = vim.fn.fnamemodify(full_path, ":h")
|
|
if vim.fn.isdirectory(folder) == 0 then
|
|
vim.fn.mkdir(folder, "p")
|
|
end
|
|
|
|
sidebar.open(full_path, filename, false)
|
|
else
|
|
-- Default to today's idea file
|
|
local path = string.format("%s/%s.md", idea_dir, os.date("%Y-%m-%d"))
|
|
sidebar.open(path, nil, false)
|
|
end
|
|
end, {
|
|
nargs = "?",
|
|
complete = function()
|
|
return { "listAll" }
|
|
end,
|
|
desc = "Open today's idea, a named idea, or list all (in floating window)",
|
|
})
|
|
|
|
-- Command to open ideas in current buffer
|
|
vim.api.nvim_create_user_command("IdeaBuffer", function(opts)
|
|
local arg = opts.args
|
|
local idea_dir = config.options.idea_dir
|
|
|
|
if arg == "listAll" then
|
|
list.list_all()
|
|
elseif arg ~= "" then
|
|
-- Ensure directory exists (even for nested folders)
|
|
local filename = arg:match("%.md$") and arg or (arg .. ".md")
|
|
local full_path = idea_dir .. "/" .. filename
|
|
|
|
-- Create parent folders if needed
|
|
local folder = vim.fn.fnamemodify(full_path, ":h")
|
|
if vim.fn.isdirectory(folder) == 0 then
|
|
vim.fn.mkdir(folder, "p")
|
|
end
|
|
|
|
sidebar.open(full_path, filename, true)
|
|
else
|
|
-- Default to today's idea file
|
|
local path = string.format("%s/%s.md", idea_dir, os.date("%Y-%m-%d"))
|
|
sidebar.open(path, nil, true)
|
|
end
|
|
end, {
|
|
nargs = "?",
|
|
complete = function()
|
|
return { "listAll" }
|
|
end,
|
|
desc = "Open today's idea, a named idea, or list all (in current buffer)",
|
|
})
|
|
|
|
-- Command to open ideas in persistent right-side buffer
|
|
vim.api.nvim_create_user_command("IdeaRight", function(opts)
|
|
local arg = opts.args
|
|
local idea_dir = config.options.idea_dir
|
|
|
|
if arg == "listAll" then
|
|
list.list_all()
|
|
elseif arg ~= "" then
|
|
-- Ensure directory exists (even for nested folders)
|
|
local filename = arg:match("%.md$") and arg or (arg .. ".md")
|
|
local full_path = idea_dir .. "/" .. filename
|
|
|
|
-- Create parent folders if needed
|
|
local folder = vim.fn.fnamemodify(full_path, ":h")
|
|
if vim.fn.isdirectory(folder) == 0 then
|
|
vim.fn.mkdir(folder, "p")
|
|
end
|
|
|
|
sidebar.open_right_side(full_path, filename)
|
|
else
|
|
-- Default to today's idea file
|
|
local path = string.format("%s/%s.md", idea_dir, os.date("%Y-%m-%d"))
|
|
sidebar.open_right_side(path, nil)
|
|
end
|
|
end, {
|
|
nargs = "?",
|
|
complete = function()
|
|
return { "listAll" }
|
|
end,
|
|
desc = "Open today's idea, a named idea, or list all (in persistent right-side buffer)",
|
|
})
|
|
|
|
-- Command to open the tree view for browsing ideas
|
|
vim.api.nvim_create_user_command("IdeaTree", function()
|
|
tree.open_tree_window(function(selected_file)
|
|
-- When a file is selected from the tree, open it in the right-side buffer
|
|
if selected_file then
|
|
local filename = vim.fn.fnamemodify(selected_file, ":t")
|
|
sidebar.open_right_side(selected_file, filename)
|
|
end
|
|
end)
|
|
end, {
|
|
desc = "Open tree view to browse and select idea files",
|
|
})
|
|
|
|
-- Tag-related commands
|
|
vim.api.nvim_create_user_command("IdeaTags", function()
|
|
tags.show_tag_picker(function(selected_tag)
|
|
if selected_tag then
|
|
tags.show_files_with_tag(selected_tag)
|
|
end
|
|
end)
|
|
end, {
|
|
desc = "Show all tags and browse files by tag",
|
|
})
|
|
|
|
vim.api.nvim_create_user_command("IdeaAddTag", function(opts)
|
|
local tag = opts.args
|
|
if tag == "" then
|
|
vim.notify("❌ Please provide a tag name", vim.log.levels.ERROR)
|
|
return
|
|
end
|
|
|
|
-- Get current file from right-side buffer or prompt for file
|
|
local current_file = sidebar.get_current_file()
|
|
if current_file then
|
|
tags.add_tag(current_file, tag)
|
|
else
|
|
vim.notify("❌ No active idea file. Open an idea first.", vim.log.levels.ERROR)
|
|
end
|
|
end, {
|
|
nargs = 1,
|
|
desc = "Add a tag to the current idea file",
|
|
})
|
|
|
|
vim.api.nvim_create_user_command("IdeaRemoveTag", function(opts)
|
|
local tag = opts.args
|
|
if tag == "" then
|
|
vim.notify("❌ Please provide a tag name", vim.log.levels.ERROR)
|
|
return
|
|
end
|
|
|
|
-- Get current file from right-side buffer or prompt for file
|
|
local current_file = sidebar.get_current_file()
|
|
if current_file then
|
|
tags.remove_tag(current_file, tag)
|
|
else
|
|
vim.notify("❌ No active idea file. Open an idea first.", vim.log.levels.ERROR)
|
|
end
|
|
end, {
|
|
nargs = 1,
|
|
desc = "Remove a tag from the current idea file",
|
|
})
|
|
|
|
vim.api.nvim_create_user_command("IdeaSearchTag", function(opts)
|
|
local tag = opts.args
|
|
if tag == "" then
|
|
vim.notify("❌ Please provide a tag name", vim.log.levels.ERROR)
|
|
return
|
|
end
|
|
|
|
tags.show_files_with_tag(tag)
|
|
end, {
|
|
nargs = 1,
|
|
desc = "Search for files with a specific tag",
|
|
})
|
|
|
|
-- Search-related commands
|
|
vim.api.nvim_create_user_command("IdeaSearch", function(opts)
|
|
local query = opts.args
|
|
if query == "" then
|
|
vim.notify("❌ Please provide a search query", vim.log.levels.ERROR)
|
|
return
|
|
end
|
|
|
|
search.fuzzy_search(query)
|
|
end, {
|
|
nargs = 1,
|
|
desc = "Fuzzy search through idea titles and content",
|
|
})
|
|
|
|
vim.api.nvim_create_user_command("IdeaSearchContent", function(opts)
|
|
local query = opts.args
|
|
if query == "" then
|
|
vim.notify("❌ Please provide a search query", vim.log.levels.ERROR)
|
|
return
|
|
end
|
|
|
|
search.search_in_content(query)
|
|
end, {
|
|
nargs = 1,
|
|
desc = "Search only in idea content",
|
|
})
|
|
|
|
vim.api.nvim_create_user_command("IdeaSearchTitle", function(opts)
|
|
local query = opts.args
|
|
if query == "" then
|
|
vim.notify("❌ Please provide a search query", vim.log.levels.ERROR)
|
|
return
|
|
end
|
|
|
|
search.search_by_title(query)
|
|
end, {
|
|
nargs = 1,
|
|
desc = "Search only in idea titles",
|
|
})
|
|
|
|
-- Graph visualization commands
|
|
vim.api.nvim_create_user_command("IdeaGraph", function(opts)
|
|
local arg = opts.args
|
|
|
|
if arg == "close" then
|
|
graph.close()
|
|
elseif arg == "refresh" then
|
|
graph.refresh()
|
|
elseif arg == "animate" then
|
|
graph.open({ animate = true })
|
|
elseif arg == "rebuild" then
|
|
graph.open({ force_rebuild = true })
|
|
else
|
|
graph.open()
|
|
end
|
|
end, {
|
|
nargs = "?",
|
|
complete = function()
|
|
return { "close", "refresh", "animate", "rebuild" }
|
|
end,
|
|
desc = "Open Obsidian-style graph visualization of notes and links",
|
|
})
|
|
|
|
vim.api.nvim_create_user_command("IdeaGraphClearCache", function()
|
|
local cache = require("ideaDrop.ui.graph.cache")
|
|
cache.clear()
|
|
vim.notify("🗑️ Graph cache cleared", vim.log.levels.INFO)
|
|
end, {
|
|
desc = "Clear the graph cache to force full rebuild",
|
|
})
|
|
|
|
vim.api.nvim_create_user_command("IdeaGraphFilter", function(opts)
|
|
local args = vim.split(opts.args, " ", { trimempty = true })
|
|
|
|
if #args < 2 then
|
|
vim.notify("Usage: :IdeaGraphFilter <tag|folder> <value>", vim.log.levels.ERROR)
|
|
return
|
|
end
|
|
|
|
local filter_type = args[1]
|
|
local filter_value = args[2]
|
|
|
|
if filter_type ~= "tag" and filter_type ~= "folder" then
|
|
vim.notify("Filter type must be 'tag' or 'folder'", vim.log.levels.ERROR)
|
|
return
|
|
end
|
|
|
|
-- If graph is open, apply filter
|
|
if graph.is_open() then
|
|
local graph_data = graph.get_graph()
|
|
if graph_data then
|
|
local data_module = require("ideaDrop.ui.graph.data")
|
|
data_module.apply_filter(graph_data, filter_type, filter_value)
|
|
graph.refresh()
|
|
end
|
|
else
|
|
-- Open graph with filter
|
|
graph.open()
|
|
vim.defer_fn(function()
|
|
local graph_data = graph.get_graph()
|
|
if graph_data then
|
|
local data_module = require("ideaDrop.ui.graph.data")
|
|
data_module.apply_filter(graph_data, filter_type, filter_value)
|
|
graph.refresh()
|
|
end
|
|
end, 100)
|
|
end
|
|
end, {
|
|
nargs = "+",
|
|
complete = function(_, cmd_line, _)
|
|
local args = vim.split(cmd_line, " ", { trimempty = true })
|
|
if #args <= 2 then
|
|
return { "tag", "folder" }
|
|
end
|
|
return {}
|
|
end,
|
|
desc = "Filter graph by tag or folder",
|
|
})
|
|
|
|
-- Set up keymaps
|
|
keymaps.setup()
|
|
|
|
vim.notify("ideaDrop loaded!", vim.log.levels.INFO)
|
|
end
|
|
|
|
return M
|