feat: support loading after directories (#4)

doc/rtp_nvim.txt

@@ -12,4 +12,11 @@ rtp_nvim.source_rtp_dir({dir})                         *rtp_nvim.source_rtp_dir*
         {dir}  (string)  The runtime directory to source
 
 
+rtp_nvim.source_after_plugin_dir({dir})       *rtp_nvim.source_after_plugin_dir*
+    Source the `after` scripts
+
+    Parameters: ~
+        {dir}  (string)  The runtime directory to source
+
+
 vim:tw=78:ts=8:noet:ft=help:norl:

lua/rtp_nvim.lua

@@ -23,6 +23,7 @@ local _sourced_ftdetect = {}
 local RtpSourceDir = {
     plugin = "plugin",
     ftdetect = "ftdetect",
+    after_plugin = vim.fs.joinpath("after", "plugin"),
 }
 
 ---Recursively iterate over a directory's children
@@ -77,11 +78,22 @@ local function source_ftdetect(dir)
     end
 end
 
+---@param dir string
+local function source_after_plugin(dir)
+    source(RtpSourceDir.after_plugin, dir)
+end
+
 ---Source the `plugin` and `ftdetect` directories.
 ---@param dir string The runtime directory to source
 function rtp_nvim.source_rtp_dir(dir)
     source_plugin(dir)
     source_ftdetect(dir)
 end
 
+---Source the `after` scripts
+---@param dir string The runtime directory to source
+function rtp_nvim.source_after_plugin_dir(dir)
+    source_after_plugin(dir)
+end
+
 return rtp_nvim

spec/fixtures/after/plugin/test_after_plugin.lua

@@ -0,0 +1 @@
+vim.g.test_after_plugin = (vim.g.test_after_plugin or 0) + 1

spec/rtp_spec.lua

@@ -4,11 +4,16 @@ describe("rtp_nvim", function()
     it("sources plugin scripts multiple times and ftdetect scripts once", function()
         assert.is_nil(vim.g.test_plugin)
         assert.is_nil(vim.g.test_ftdetect)
+        assert.is_nil(vim.g.test_after_plugin)
         rtp.source_rtp_dir(fixture_dir)
         assert.same(1, vim.g.test_plugin)
         assert.same(1, vim.g.test_ftdetect)
         rtp.source_rtp_dir(fixture_dir)
         assert.same(2, vim.g.test_plugin)
         assert.same(1, vim.g.test_ftdetect)
+        rtp.source_after_plugin_dir(fixture_dir)
+        assert.same(1, vim.g.test_after_plugin)
+        rtp.source_after_plugin_dir(fixture_dir)
+        assert.same(2, vim.g.test_after_plugin)
     end)
 end)