samer@3: " commentary.vim - Comment stuff out
samer@3: " Maintainer:   Tim Pope <http://tpo.pe/>
samer@3: " Version:      1.1
samer@3: " GetLatestVimScripts: 3695 1 :AutoInstall: commentary.vim
samer@3: 
samer@3: if exists("g:loaded_commentary") || &cp || v:version < 700
samer@3:   finish
samer@3: endif
samer@3: let g:loaded_commentary = 1
samer@3: 
samer@3: function! s:go(type,...) abort
samer@3:   if a:0
samer@3:     let [lnum1, lnum2] = [a:type, a:1]
samer@3:   else
samer@3:     let [lnum1, lnum2] = [line("'["), line("']")]
samer@3:   endif
samer@3: 
samer@3:   let [l, r] = split(substitute(substitute(&commentstring,'\S\zs%s',' %s',''),'%s\ze\S','%s ',''),'%s',1)
samer@3:   let uncomment = 2
samer@3:   for lnum in range(lnum1,lnum2)
samer@3:     let line = matchstr(getline(lnum),'\S.*\s\@<!')
samer@3:     if line != '' && (stridx(line,l) || line[strlen(line)-strlen(r) : -1] != r)
samer@3:       let uncomment = 0
samer@3:     endif
samer@3:   endfor
samer@3: 
samer@3:   for lnum in range(lnum1,lnum2)
samer@3:     let line = getline(lnum)
samer@3:     if strlen(r) > 2 && l.r !~# '\\'
samer@3:       let line = substitute(line,
samer@3:             \'\M'.r[0:-2].'\zs\d\*\ze'.r[-1:-1].'\|'.l[0].'\zs\d\*\ze'.l[1:-1],
samer@3:             \'\=substitute(submatch(0)+1-uncomment,"^0$\\|^-\\d*$","","")','g')
samer@3:     endif
samer@3:     if uncomment
samer@3:       let line = substitute(line,'\S.*\s\@<!','\=submatch(0)[strlen(l):-strlen(r)-1]','')
samer@3:     else
samer@3:       let line = substitute(line,'^\%('.matchstr(getline(lnum1),'^\s*').'\|\s*\)\zs.*\S\@<=','\=l.submatch(0).r','')
samer@3:     endif
samer@3:     call setline(lnum,line)
samer@3:   endfor
samer@3: endfunction
samer@3: 
samer@3: function! s:undo()
samer@3:   let [l, r] = split(substitute(substitute(&commentstring,'\S\zs%s',' %s',''),'%s\ze\S','%s ',''),'%s',1)
samer@3:   let lnums = [line('.')+1, line('.')-2]
samer@3:   for [index, dir, bound, line] in [[0, -1, 1, ''], [1, 1, line('$'), '']]
samer@3:     while lnums[index] != bound && line ==# '' || !(stridx(line,l) || line[strlen(line)-strlen(r) : -1] != r)
samer@3:       let lnums[index] += dir
samer@3:       let line = matchstr(getline(lnums[index]+dir),'\S.*\s\@<!')
samer@3:     endwhile
samer@3:   endfor
samer@3:   call s:go(lnums[0], lnums[1])
samer@3:   silent! call repeat#set("\<Plug>CommentaryUndo")
samer@3: endfunction
samer@3: 
samer@3: xnoremap <silent> <Plug>Commentary     :<C-U>call <SID>go(line("'<"),line("'>"))<CR>
samer@3: nnoremap <silent> <Plug>Commentary     :<C-U>set opfunc=<SID>go<CR>g@
samer@3: nnoremap <silent> <Plug>CommentaryLine :<C-U>set opfunc=<SID>go<Bar>exe 'norm! 'v:count1.'g@_'<CR>
samer@3: nnoremap <silent> <Plug>CommentaryUndo :<C-U>call <SID>undo()<CR>
samer@3: 
samer@3: if !hasmapto('<Plug>Commentary') || maparg('gc','n') ==# ''
samer@3:   xmap gc  <Plug>Commentary
samer@3:   nmap gc  <Plug>Commentary
samer@3:   nmap gcc <Plug>CommentaryLine
samer@3:   nmap gcu <Plug>CommentaryUndo
samer@3: endif
samer@3: 
samer@3: if maparg('\\','n') ==# '' && maparg('\','n') ==# ''
samer@3:   xmap \\  <Plug>Commentary
samer@3:   nmap \\  <Plug>Commentary
samer@3:   nmap \\\ <Plug>CommentaryLine
samer@3:   nmap \\u <Plug>CommentaryUndo
samer@3: endif
samer@3: 
samer@3: " vim:set et sw=2: