doc: add input

Makefile

@@ -20,7 +20,7 @@ test-ci: deps test
 
 # generates the documentation.
 documentation:
-	nvim --headless --noplugin -u ./scripts/minimal_init.lua -c "lua require('mini.doc').generate()" -c "qa!"
+	nvim --headless --noplugin -u ./scripts/minimal_init.lua -c "lua require('mini.doc').generate(require(\"wf.mini\").setup)" -c "qa!"
 
 # installs deps before running the documentation generation, useful for the CI.
 documentation-ci: deps documentation

doc/tags

@@ -1 +1,2 @@
 select()	wf.txt	/*select()*
+which_key()	wf.txt	/*which_key()*

doc/wf.txt

@@ -1,13 +1,19 @@
+==============================================================================
+------------------------------------------------------------------------------
+                                                                   *which_key()*
+                              `which_key`({opts})
+Parameters~
+{opts} `(optional)` `(table)`
+
+
 ==============================================================================
 ------------------------------------------------------------------------------
                                                                       *select()*
                      `select`({items}, {opts}, {on_choice})
-Define your wf setup.
-
 Parameters~
 {items} items
-{the} options for wf
-{on_choice} A callback that will be carried
+{opts} the options for wf
+{on_choice} a callback that will be carried
 Usage~
 `require("wf").select(items, opts, on_choice)`
 

lua/wf/mini.lua

@@ -0,0 +1,35 @@
+local H = {}
+
+-- Default documentation targets ----------------------------------------------
+H.input = function()
+  -- Search in current and recursively in other directories for files with
+  -- 'lua' extension
+  local res = {}
+  for _, dir_glob in ipairs({ ".", "lua/**", "after/**", "colors/**", "lua/builtin/**" }) do
+    local files = vim.fn.globpath(dir_glob, "*.lua", false, true)
+
+    -- Use full paths
+    files = vim.tbl_map(function(x)
+      return vim.fn.fnamemodify(x, ":p")
+    end, files)
+
+    -- Put 'init.lua' first among files from same directory
+    table.sort(files, function(a, b)
+      if vim.fn.fnamemodify(a, ":h") == vim.fn.fnamemodify(b, ":h") then
+        if vim.fn.fnamemodify(a, ":t") == "init.lua" then
+          return true
+        end
+        if vim.fn.fnamemodify(b, ":t") == "init.lua" then
+          return false
+        end
+      end
+
+      return a < b
+    end)
+    table.insert(res, files)
+  end
+
+  return vim.tbl_flatten(res)
+end
+
+return H