ideadrop.nvim/lua/ideaDrop/ui/graph/data.lua

-- ideaDrop/ui/graph/data.lua
-- Graph data model: parses markdown files and builds node/edge structures

local config = require("ideaDrop.core.config")
local tags_module = require("ideaDrop.features.tags")
local types = require("ideaDrop.ui.graph.types")

---@class GraphDataModule
---@field build_graph fun(): GraphData
---@field extract_links fun(content: string): string[]
---@field resolve_link fun(link_text: string, base_dir: string): string|nil
---@field get_node_by_position fun(graph: GraphData, x: number, y: number, threshold: number): GraphNode|nil
local M = {}

---Extracts [[wiki-style links]] from markdown content
---@param content string The markdown content
---@return string[] Array of link targets (without brackets)
function M.extract_links(content)
	local links = {}
	local seen = {}

	-- Match [[link]] pattern
	for link in content:gmatch("%[%[([^%]]+)%]%]") do
		-- Handle [[link|alias]] format - take the link part
		local actual_link = link:match("^([^|]+)") or link
		-- Trim whitespace
		actual_link = actual_link:gsub("^%s*(.-)%s*$", "%1")

		if not seen[actual_link] and actual_link ~= "" then
			table.insert(links, actual_link)
			seen[actual_link] = true
		end
	end

	return links
end

---Resolves a link text to a file path
---@param link_text string The link text (without brackets)
---@param idea_dir string The idea directory path
---@param existing_files table<string, string> Map of normalized names to file paths
---@return string|nil Resolved file path or nil if not found
function M.resolve_link(link_text, idea_dir, existing_files)
	-- Normalize the link text
	local normalized = link_text:lower():gsub("%s+", "-")

	-- Try direct match first
	if existing_files[normalized] then
		return existing_files[normalized]
	end

	-- Try with .md extension
	if existing_files[normalized .. ".md"] then
		return existing_files[normalized .. ".md"]
	end

	-- Try fuzzy matching - match just the filename part
	local link_basename = vim.fn.fnamemodify(link_text, ":t"):lower():gsub("%s+", "-")
	for name, path in pairs(existing_files) do
		local file_basename = vim.fn.fnamemodify(name, ":t"):gsub("%.md$", "")
		if file_basename == link_basename or file_basename == normalized then
			return path
		end
	end

	return nil
end

---Builds a normalized name for a file (used as node ID)
---@param file_path string Full file path
---@param idea_dir string The idea directory
---@return string Normalized name (relative path without extension)
function M.normalize_file_name(file_path, idea_dir)
	local relative = file_path
	if file_path:sub(1, #idea_dir) == idea_dir then
		relative = file_path:sub(#idea_dir + 2) -- Remove idea_dir + "/"
	end
	-- Remove .md extension
	return relative:gsub("%.md$", "")
end

---Gets display name from a node ID
---@param node_id string The node ID
---@return string Display name
function M.get_display_name(node_id)
	-- Get just the filename part without path
	local name = vim.fn.fnamemodify(node_id, ":t")
	-- Capitalize and clean up
	return name:gsub("-", " "):gsub("^%l", string.upper)
end

---Builds the complete graph from markdown files
---@return GraphData
function M.build_graph()
	local idea_dir = config.options.idea_dir
	local graph = types.create_graph_data()

	-- Find all markdown files
	local files = vim.fn.glob(idea_dir .. "/**/*.md", false, true)

	if #files == 0 then
		return graph
	end

	-- Build a map of normalized names to file paths for link resolution
	local file_map = {}
	for _, file_path in ipairs(files) do
		local normalized = M.normalize_file_name(file_path, idea_dir):lower()
		file_map[normalized] = file_path

		-- Also map just the filename
		local basename = vim.fn.fnamemodify(file_path, ":t:r"):lower()
		if not file_map[basename] then
			file_map[basename] = file_path
		end
	end

	-- First pass: create all nodes
	for _, file_path in ipairs(files) do
		local node_id = M.normalize_file_name(file_path, idea_dir)
		local display_name = M.get_display_name(node_id)

		local node = types.create_node(node_id, display_name, file_path)

		-- Extract tags from file
		if vim.fn.filereadable(file_path) == 1 then
			local content = vim.fn.readfile(file_path)
			local content_str = table.concat(content, "\n")
			node.tags = tags_module.extract_tags(content_str)
		end

		graph.nodes[node_id] = node
		table.insert(graph.node_list, node)
	end

	-- Second pass: create edges from links
	local edge_set = {} -- Track unique edges (undirected)

	for _, file_path in ipairs(files) do
		if vim.fn.filereadable(file_path) == 1 then
			local content = vim.fn.readfile(file_path)
			local content_str = table.concat(content, "\n")
			local links = M.extract_links(content_str)

			local source_id = M.normalize_file_name(file_path, idea_dir)

			for _, link_text in ipairs(links) do
				local target_path = M.resolve_link(link_text, idea_dir, file_map)

				if target_path then
					local target_id = M.normalize_file_name(target_path, idea_dir)

					-- Skip self-links
					if source_id ~= target_id then
						-- Create undirected edge key (sorted)
						local edge_key
						if source_id < target_id then
							edge_key = source_id .. "|||" .. target_id
						else
							edge_key = target_id .. "|||" .. source_id
						end

						-- Only add if not already exists
						if not edge_set[edge_key] then
							edge_set[edge_key] = true

							local edge = types.create_edge(source_id, target_id)
							table.insert(graph.edges, edge)

							-- Update degrees
							if graph.nodes[source_id] then
								graph.nodes[source_id].degree = graph.nodes[source_id].degree + 1
							end
							if graph.nodes[target_id] then
								graph.nodes[target_id].degree = graph.nodes[target_id].degree + 1
							end
						end
					end
				end
			end
		end
	end

	return graph
end

---Finds a node at a given position (for mouse/cursor interaction)
---@param graph GraphData The graph data
---@param x number X coordinate
---@param y number Y coordinate
---@param threshold number Distance threshold for hit detection
---@return GraphNode|nil Node at position or nil
function M.get_node_by_position(graph, x, y, threshold)
	local closest_node = nil
	local closest_dist = threshold + 1

	for _, node in ipairs(graph.node_list) do
		if node.visible then
			local dx = node.x - x
			local dy = node.y - y
			local dist = math.sqrt(dx * dx + dy * dy)

			if dist < closest_dist then
				closest_dist = dist
				closest_node = node
			end
		end
	end

	return closest_node
end

---Gets all unique folders from the graph
---@param graph GraphData The graph data
---@return string[] Array of folder names
function M.get_folders(graph)
	local folders = {}
	local seen = {}

	for _, node in ipairs(graph.node_list) do
		if not seen[node.folder] then
			table.insert(folders, node.folder)
			seen[node.folder] = true
		end
	end

	table.sort(folders)
	return folders
end

---Gets all unique tags from the graph
---@param graph GraphData The graph data
---@return string[] Array of tag names
function M.get_tags(graph)
	local tags = {}
	local seen = {}

	for _, node in ipairs(graph.node_list) do
		for _, tag in ipairs(node.tags) do
			if not seen[tag] then
				table.insert(tags, tag)
				seen[tag] = true
			end
		end
	end

	table.sort(tags)
	return tags
end

---Applies a filter to the graph
---@param graph GraphData The graph data
---@param filter_type string|nil "tag", "folder", or nil to clear
---@param filter_value string|nil The filter value
function M.apply_filter(graph, filter_type, filter_value)
	-- Reset all visibility first
	for _, node in ipairs(graph.node_list) do
		node.visible = true
	end
	for _, edge in ipairs(graph.edges) do
		edge.visible = true
	end

	-- Apply filter if specified
	if filter_type and filter_value then
		-- First pass: hide nodes that don't match
		for _, node in ipairs(graph.node_list) do
			local matches = false

			if filter_type == "tag" then
				for _, tag in ipairs(node.tags) do
					if tag == filter_value then
						matches = true
						break
					end
				end
			elseif filter_type == "folder" then
				matches = node.folder == filter_value
			elseif filter_type == "search" then
				local search_lower = filter_value:lower()
				matches = node.name:lower():find(search_lower, 1, true) ~= nil
					or node.id:lower():find(search_lower, 1, true) ~= nil
			end

			node.visible = matches
		end

		-- Second pass: hide edges where either endpoint is hidden
		for _, edge in ipairs(graph.edges) do
			local source_visible = graph.nodes[edge.source] and graph.nodes[edge.source].visible
			local target_visible = graph.nodes[edge.target] and graph.nodes[edge.target].visible
			edge.visible = source_visible and target_visible
		end
	end
end

---Gets graph statistics
---@param graph GraphData The graph data
---@return table Statistics
function M.get_statistics(graph)
	local total_nodes = #graph.node_list
	local visible_nodes = 0
	local orphan_nodes = 0
	local total_edges = #graph.edges
	local visible_edges = 0
	local max_degree = 0
	local total_degree = 0

	for _, node in ipairs(graph.node_list) do
		if node.visible then
			visible_nodes = visible_nodes + 1
		end
		if node.degree == 0 then
			orphan_nodes = orphan_nodes + 1
		end
		if node.degree > max_degree then
			max_degree = node.degree
		end
		total_degree = total_degree + node.degree
	end

	for _, edge in ipairs(graph.edges) do
		if edge.visible then
			visible_edges = visible_edges + 1
		end
	end

	return {
		total_nodes = total_nodes,
		visible_nodes = visible_nodes,
		orphan_nodes = orphan_nodes,
		total_edges = total_edges,
		visible_edges = visible_edges,
		max_degree = max_degree,
		avg_degree = total_nodes > 0 and (total_degree / total_nodes) or 0,
	}
end

return M