Implement :CrazywallFollowRef

Author: gitpushjoe

lua/crazywall/commands/crazywall_follow_ref.lua

@@ -0,0 +1,31 @@
+local Config = require("core.config")
+local plugin_utils = require("lua.crazywall.utils")
+
+---@param plugin_state PluginState
+return function(plugin_state)
+	vim.api.nvim_create_user_command("CrazywallFollowRef", function()
+		local line = vim.api.nvim_get_current_line()
+		local column = vim.api.nvim_win_get_cursor(0)[2]
+
+		if not plugin_state:get_current_config() then
+			return vim.api.nvim_err_writeln(
+				"crazywall: Could not find config "
+					.. plugin_state.current_config_name
+					.. "."
+			)
+		end
+
+		local config, err =
+			Config:new(assert(plugin_state:get_current_config()))
+		if not config then
+			return plugin_utils.display_err(err)
+		end
+
+		plugin_state.follow_ref(
+			line,
+			column,
+			config,
+			plugin_state.current_config_name
+		)
+	end, { nargs = 0 })
+end

lua/crazywall/init.lua

@@ -1,30 +1,48 @@
 local plugin_validate = require("crazywall.validate")
 local default_config = require("core.defaults.config")
 local PluginState = require("lua.crazywall.state")
+local Config = require("core.config")
 local cmd_crazywall = require("lua.crazywall.commands.crazywall")
 local cmd_crazywall_dry = require("lua.crazywall.commands.crazywall_dry")
 local cmd_crazywall_quick = require("lua.crazywall.commands.crazywall_quick")
+local cmd_crazywall_follow_ref =
+	require("lua.crazywall.commands.crazywall_follow_ref")
 local M = {}
 
 local plugin_state = PluginState:new()
 cmd_crazywall(plugin_state)
 cmd_crazywall_dry(plugin_state)
 cmd_crazywall_quick(plugin_state)
+cmd_crazywall_follow_ref(plugin_state)
 
 M.setup = function(opts)
 	opts = opts or {}
-	local keys = { "configs", "default_config_name" }
+	local keys = { "configs", "default_config_name", "follow_ref" }
 	for key in pairs(opts) do
 		local err = plugin_validate.string_in_list(key, keys)
 		if err then
 			error(err)
 		end
 	end
+	if opts.configs then
+		for name, config_table in pairs(opts.configs) do
+			local config, err = Config:new(config_table)
+			if not config then
+				error(
+					"Error occurred when trying to build config "
+						.. name
+						.. ": \n"
+						.. err
+				)
+			end
+		end
+	end
 	plugin_state.configs = opts.configs or plugin_state.configs
 	plugin_state.configs["DEFAULT"] = plugin_state.configs.DEFAULT
 		or default_config
 	plugin_state.current_config_name = opts.default_config_name
 		or plugin_state.current_config_name
+	plugin_state.follow_ref = opts.follow_ref or plugin_state.follow_ref
 end
 
 return M

lua/crazywall/state.lua

@@ -1,6 +1,7 @@
 ---@class PluginState
 ---@field configs PartialConfigTable[]
 ---@field current_config_name string
+---@field follow_ref fun(line: string, column: integer, config: Config, config_name: string): any
 local PluginState = {}
 PluginState.__index = PluginState
 
@@ -11,6 +12,41 @@ function PluginState:new()
 	---@cast self PluginState
 	self.configs = {}
 	self.current_config_name = "DEFAULT"
+	self.follow_ref = function(line)
+		---@param path string
+		local file_exists_and_is_not_directory = function(path)
+			local stat = vim.loop.fs_stat(path)
+			return stat ~= nil and stat.type == "file"
+		end
+		--- Find text within [[]]s.
+		local match = string.match(line, "%[%[(.-)%]%]")
+		if not match then
+			return
+		end
+		local current_file_dir = vim.fn.expand("%:p:h")
+		local extension = '.' .. vim.fn.expand("%:e")
+		---@param path string
+		local expand_path = function(path)
+			return vim.fn.fnamemodify(current_file_dir .. "/" .. path, ":p")
+		end
+		---@param path string
+		local open_path = function(path)
+			vim.cmd("edit " .. vim.fn.fnameescape(path))
+		end
+		for _, path in ipairs({
+			expand_path(match .. extension),
+			expand_path(match),
+			expand_path(match .. "/_index" .. extension),
+		}) do
+			if file_exists_and_is_not_directory(path) then
+				open_path(path)
+				return
+			end
+		end
+		print(
+			"Default :CrazywallFollowRef failed. Read the docs to see how you can customize its behavior: https://github.com/gitpulljoe/crazywall.nvim"
+		)
+	end
 	return self
 end