Fix getting parent by ignoring injected langs and add new func to module

Author: Danielwsx64

lua/elixir_dev/module.lua

@@ -63,6 +63,31 @@ function M.get_module_public_function_nodes(bufnr, node)
 	end, do_block_node)
 end
 
+function M.module_name_by_path(path)
+	local file_path = path or vim.fn.expand("%")
+
+	if file_path and file_path ~= "" then
+		local mod_name = file_path
+			:gsub(".ex$", "")
+			:gsub(".exs$", "")
+			:gsub("/", ".")
+			:gsub("(%l)(%w*)", function(a, b)
+				return string.upper(a) .. b
+			end)
+			:gsub("_", "")
+			:gsub("Api", "API")
+			:gsub("Json", "JSON")
+			:gsub(".Controllers", "")
+			:gsub(".Views", "")
+			:gsub("Lib.", "")
+			:gsub("Test.", "")
+
+		return mod_name
+	end
+
+	return "Module"
+end
+
 function M.yank_module_name()
 	local bufnr = treesitter_utils.get_current_elixir_buf()
 

lua/elixir_dev/utils/treesitter.lua

@@ -29,16 +29,19 @@ local function hard_stop_for_inner_pipes(node)
 	return false
 end
 
+function M.get_node_at_cursor()
+	-- Ignore injected langs
+	return require("nvim-treesitter.ts_utils").get_node_at_cursor(0, true)
+end
+
 function M.is_ts_elixir_parser_enabled(bufnr)
 	local buff_active = highlighter.active[bufnr]
 
 	return buff_active and buff_active.tree._lang == "elixir" or false
 end
 
 function M.get_master_node(initial_node)
-	local ts_utils = require("nvim-treesitter.ts_utils")
-
-	local node = initial_node or ts_utils.get_node_at_cursor()
+	local node = initial_node or M.get_node_at_cursor()
 
 	local parent = node:parent()
 	local start_row = node:start()
@@ -65,15 +68,13 @@ function M.get_master_node(initial_node)
 end
 
 function M.get_parent_node(types, validation, initial_node)
-	local ts_utils = require("nvim-treesitter.ts_utils")
-
 	if not validation or type(validation) ~= "function" then
 		validation = function(_)
 			return true
 		end
 	end
 
-	local node = initial_node or ts_utils.get_node_at_cursor()
+	local node = initial_node or M.get_node_at_cursor()
 
 	while node do
 		if vim.tbl_contains(types, node:type()) and validation(node) then
@@ -87,7 +88,7 @@ function M.get_parent_node(types, validation, initial_node)
 end
 
 function M.get_all_child(validation, initial_node)
-	local node = initial_node or ts_utils.get_node_at_cursor()
+	local node = initial_node or M.get_node_at_cursor()
 
 	if not validation or type(validation) ~= "function" then
 		validation = function(_)

test/spec/elixir_dev/module_spec.lua

@@ -122,3 +122,24 @@ describe("format_node_function", function()
 		vim.api.nvim_buf_delete(bufnr, { force = true })
 	end)
 end)
+
+describe("module_name_by_path", function()
+	it("return the module name by file path", function()
+		assert.are.equal("LiveRetro.Schema", module.module_name_by_path("lib/live_retro/schema.ex"))
+
+		assert.are.equal(
+			"LiveRetroWeb.UserAuthTest",
+			module.module_name_by_path("test/live_retro_web/user_auth_test.exs")
+		)
+
+		assert.are.equal(
+			"LiveRetroWeb.PageController",
+			module.module_name_by_path("lib/live_retro_web/controllers/page_controller.ex")
+		)
+
+		assert.are.equal(
+			"LiveRetroWeb.PageControllerTest",
+			module.module_name_by_path("test/live_retro_web/controllers/page_controller_test.exs")
+		)
+	end)
+end)