From 2730051a3ff2eaf34e392f2304bab217d884fc30 Mon Sep 17 00:00:00 2001
From: undyamon <148129121+undyamon@users.noreply.github.com>
Date: Sat, 30 Dec 2023 15:58:34 +0100
Subject: Fix module search to include runtime (#20)

---
 lua/tasks/utils.lua | 13 ++++---------
 1 file changed, 4 insertions(+), 9 deletions(-)

(limited to 'lua/tasks/utils.lua')

diff --git a/lua/tasks/utils.lua b/lua/tasks/utils.lua
index fab3be7..be60cf6 100644
--- a/lua/tasks/utils.lua
+++ b/lua/tasks/utils.lua
@@ -1,5 +1,3 @@
-local scandir = require('plenary.scandir')
-local Path = require('plenary.path')
 local utils = {}
 
 local args_regex = vim.regex([[\s\%(\%([^'"]*\(['"]\)[^'"]*\1\)*[^'"]*$\)\@=]])
@@ -60,14 +58,11 @@ end
 
 ---@return table
 function utils.get_module_names()
-  local module_dir = Path:new(debug.getinfo(1).source:sub(2)):parent() / 'module'
-
   local modules = {}
-  for _, entry in ipairs(scandir.scan_dir(module_dir.filename, { depth = 1 })) do
-    -- Strip full path and extension
-    local extension_len = 4
-    local parent_offset = 2
-    table.insert(modules, entry:sub(#Path:new(entry):parent().filename + parent_offset, #entry - extension_len))
+
+  local runtime_files = vim.api.nvim_get_runtime_file('lua/tasks/module/*.lua', true)
+  for _, file in ipairs(runtime_files) do
+    table.insert(modules, vim.fn.fnamemodify(file, ':t:r'))
   end
 
   return modules
-- 
cgit v1.2.3