From 4a561537d1f503adc912119647770299daec8000 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ensar=20Saraj=C4=8Di=C4=87?= <es.ensar@gmail.com>
Date: Sun, 24 Oct 2021 22:09:40 +0200
Subject: [PATCH] Add basic vimwiki extension for generating directory specific
 index

---
 .../config/nvim/lua/vimwiki_extensions.lua    | 78 +++++++++++++++++++
 symlinks/config/nvim/plugin/vimwiki.vim       |  3 +
 2 files changed, 81 insertions(+)
 create mode 100644 symlinks/config/nvim/lua/vimwiki_extensions.lua

diff --git a/symlinks/config/nvim/lua/vimwiki_extensions.lua b/symlinks/config/nvim/lua/vimwiki_extensions.lua
new file mode 100644
index 0000000..750655d
--- /dev/null
+++ b/symlinks/config/nvim/lua/vimwiki_extensions.lua
@@ -0,0 +1,78 @@
+-------------------------------------------------------------------------------
+--     - Vimwiki extensions library -
+--     Relies on vimwiki-reviews-lua for its vimwiki API
+-------------------------------------------------------------------------------
+local api = require('vimwiki_reviews.vimwiki_api')
+local utils = require('vimwiki_reviews.utils')
+local templates = require('vimwiki_reviews.templates')
+local Path = require('plenary.path')
+local scandir = require('plenary.scandir')
+
+local M = {}
+
+-- Gets path to any dir of provided vimwiki (by index)
+function M.get_vimwiki_subdir(vimwiki_index, directory)
+  vimwiki_index = api.normalize_vimwiki_index(vimwiki_index)
+  local vimwiki = vim.g.vimwiki_list[vimwiki_index]
+
+  return vimwiki.path .. directory .. '/'
+end
+
+function M.get_directory_index(vimwiki_index, directory)
+  local dir = M.get_vimwiki_subdir(vimwiki_index, directory)
+  local path = Path:new(dir):expand()
+
+  local entries = scandir.scan_dir(
+    path,
+    {
+      hidden = false,
+      add_dirs = false,
+      respect_gitignore = true,
+      depth = 1
+    })
+
+  local index = {}
+
+  for _,entry in pairs(entries) do
+    local filename = utils.get_filename_from_path(entry)
+    local ext = api.get_vimwiki_extension(vimwiki_index)
+    local noext = string.gsub(filename, ext, '')
+
+    index[noext] = filename
+  end
+
+  return index
+end
+
+-- Open subdirectory index file
+function M.open_subdirectory_index_file(vimwiki_index, subdirectory)
+  local dir = M.get_vimwiki_subdir(vimwiki_index, subdirectory)
+  local ext = api.get_vimwiki_extension(vimwiki_index)
+  local filename = dir .. 'index' .. ext
+  vim.cmd('edit ' .. filename)
+
+  local index = M.get_directory_index(vimwiki_index, subdirectory)
+
+  local builder = templates.for_vimwiki(vimwiki_index)
+
+  local lines = {
+    builder.header(1, subdirectory:sub(1,1):upper()..subdirectory:sub(2)),
+    '',
+  }
+
+  -- Add items
+  for title, fname in pairs(index) do
+    if (title ~= 'index')
+    then
+      title = string.gsub(title, '-', ' ')
+      title = title:sub(1,1):upper()..title:sub(2)
+      table.insert(lines, builder.list_item(builder.link(fname, title)))
+    end
+  end
+
+  local buf = vim.api.nvim_get_current_buf()
+  vim.api.nvim_buf_set_lines(buf, 0, -1, false, {})  -- Clear out
+  vim.api.nvim_buf_set_lines(buf, 0, 0, false, lines)  -- Put new contents
+end
+
+return M
diff --git a/symlinks/config/nvim/plugin/vimwiki.vim b/symlinks/config/nvim/plugin/vimwiki.vim
index e3697b6..b18dd90 100644
--- a/symlinks/config/nvim/plugin/vimwiki.vim
+++ b/symlinks/config/nvim/plugin/vimwiki.vim
@@ -27,3 +27,6 @@ let g:vimwiki_global_ext = 0
 
 " Prettier checkboxes
 let g:vimwiki_listsyms = '✗○◐●✓'
+
+
+command! -count=0 -nargs=1 VimwikiOpenSubdirectoryIndex :lua require('vimwiki_extensions').open_subdirectory_index_file(<count>, <f-args>)