From 4b7e7cbe4f6a40afefe78eea8556fe074217695b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Karl=20Yngve=20Lerv=C3=A5g?= <karl.yngve+git@gmail.com>
Date: Thu, 18 May 2023 12:35:19 +0200
Subject: [PATCH] feat: implement visual version of WikiLinkExtractHeader

refer: #280
---
 autoload/wiki/buffer.vim |  5 ++++-
 autoload/wiki/link.vim   | 30 +++++++++++++++++++-----------
 doc/wiki.txt             |  4 +++-
 3 files changed, 26 insertions(+), 13 deletions(-)

diff --git a/autoload/wiki/buffer.vim b/autoload/wiki/buffer.vim
index 7ef0fae..0bb9bbe 100644
--- a/autoload/wiki/buffer.vim
+++ b/autoload/wiki/buffer.vim
@@ -50,7 +50,8 @@ function! s:init_buffer_commands() abort " {{{1
   command! -buffer WikiJournalIndex       call wiki#journal#make_index()
   command! -buffer WikiLinkNext           call wiki#nav#next_link()
   command! -buffer WikiLinkShow           call wiki#link#show()
-  command! -buffer WikiLinkExtractHeader  call wiki#link#set_text_from_header()
+  command! -buffer -range WikiLinkExtractHeader
+        \ call wiki#link#set_text_from_header(<range>, <line1>, <line2>)
   command! -buffer WikiLinkFollow         call wiki#link#follow()
   command! -buffer WikiLinkFollowSplit    call wiki#link#follow(expand(<q-mods>) . ' split')
   command! -buffer WikiLinkFollowTab      call wiki#link#follow('tabedit')
@@ -99,6 +100,7 @@ function! s:init_buffer_mappings() abort " {{{1
   nnoremap <silent><buffer> <plug>(wiki-link-next)            :WikiLinkNext<cr>
   nnoremap <silent><buffer> <plug>(wiki-link-show)            :WikiLinkShow<cr>
   nnoremap <silent><buffer> <plug>(wiki-link-extract-header)  :WikiLinkExtractHeader<cr>
+  xnoremap <silent><buffer> <plug>(wiki-link-extract-header)  :WikiLinkExtractHeader<cr>
   nnoremap <silent><buffer> <plug>(wiki-link-follow)          :WikiLinkFollow<cr>
   nnoremap <silent><buffer> <plug>(wiki-link-follow-split)    :WikiLinkFollowSplit<cr>
   nnoremap         <buffer> <plug>(wiki-link-follow-vsplit)   :vert WikiLinkFollowSplit<cr>
@@ -155,6 +157,7 @@ function! s:init_buffer_mappings() abort " {{{1
           \ '<plug>(wiki-link-prev)': '<s-tab>',
           \ '<plug>(wiki-link-show)': '<leader>wll',
           \ '<plug>(wiki-link-extract-header)': '<leader>wlh',
+          \ 'x_<plug>(wiki-link-extract-header)': '<leader>wlh',
           \ '<plug>(wiki-link-follow)': '<cr>',
           \ '<plug>(wiki-link-follow-split)': '<c-w><cr>',
           \ '<plug>(wiki-link-follow-vsplit)': '<c-w><tab>',
diff --git a/autoload/wiki/link.vim b/autoload/wiki/link.vim
index cb059a9..5155549 100644
--- a/autoload/wiki/link.vim
+++ b/autoload/wiki/link.vim
@@ -183,20 +183,28 @@ function! wiki#link#follow(...) abort "{{{1
 endfunction
 
 " }}}1
-function! wiki#link#set_text_from_header() abort "{{{1
-  let l:link = wiki#link#get()
-  if index(['wiki', 'journal'], l:link.scheme) < 0 | return | endif
+function! wiki#link#set_text_from_header(range, line1, line2) abort "{{{1
+  if a:range == 0
+    let l:links = [wiki#link#get()]
+  else
+    let l:links = wiki#link#get_all_from_range(a:line1, a:line2)
+  endif
 
-  let l:title = wiki#toc#get_page_title(l:link)
-  if empty(l:title) | return | endif
+  for l:link in filter(
+        \ l:links,
+        \ { _, x -> index(['wiki', 'journal'], x.scheme) >= 0 }
+        \)
+    let l:title = wiki#toc#get_page_title(l:link)
+    if empty(l:title) | return | endif
 
-  try
-    let l:new = wiki#link#templates#{l:link.type}(l:link.url, l:title, l:link)
-  catch /E117:/
-    let l:new = wiki#link#templates#wiki(l:link.url, l:title)
-  endtry
+    try
+      let l:new = wiki#link#templates#{l:link.type}(l:link.url, l:title, l:link)
+    catch /E117:/
+      let l:new = wiki#link#templates#wiki(l:link.url, l:title)
+    endtry
 
-  call l:link.replace(l:new)
+    call l:link.replace(l:new)
+  endfor
 endfunction
 
 " }}}1
diff --git a/doc/wiki.txt b/doc/wiki.txt
index 94da7f0..0110e17 100644
--- a/doc/wiki.txt
+++ b/doc/wiki.txt
@@ -1714,7 +1714,9 @@ MAPPINGS AND COMMANDS REFERENCE                       *wiki-mappings-reference*
 
 *<plug>(wiki-link-extract-header)*
 *WikiLinkExtractHeader*
-  Set link title from the first header of the target file.
+  Set link titles from the first header of the target file. In normal mode,
+  this works on the link under the cursor. In visual mode, it works on all
+  links in the current visual selection.
 
 *<plug>(wiki-page-delete)*
 *WikiPageDelete*