aboutsummaryrefslogtreecommitdiff
path: root/vim/plugin/strip_trailing_whitespace.vim
blob: 17fff33f5451f368eeb31644cf8572e9efa2c1ba (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
"
" User-defined key mapping to strip trailing whitespace in the whole document
"
" Author: Tom Ryder <tom@sanctum.geek.nz>
" License: Same as Vim itself
"
if has('eval')

  " Define function for stripping whitespace
  function! s:StripTrailingWhitespace()

    " Iterating line number
    let l:li = 1

    " Line number of last line that had non-whitespace characters on it
    let l:lw = 0

    " Line number of the file's last line
    let l:ll = line('$')

    " Iterate over the lines
    while l:li <= l:ll

      " Get the line text
      let l:line = getline(l:li)

      " Replace the line with a subsitution of its text stripping extraneous
      " whitespace
      call setline(l:li, substitute(l:line, '\m\C\s\+$', '', 'g'))

      " If this line has any non-whitespace characters on it, update l:lw with
      " its index
      if l:line =~# '\m\C\S'
        let l:lw = l:li
      endif

      " Increment the line counter for the next iteration
      let l:li = l:li + 1
    endwhile

    " If the last non-whitespace line was before the last line proper, we can
    " delete all lines after it
    if l:lw < l:ll

      " Get the current line and column so we can return to it
      " (Yes I know about winsaveview() and winrestview(); I want this to work
      " even on very old versions of Vim if possible)
      let l:lc = line('.')
      let l:cc = col('.')

      " Delete the lines, which will move the cursor
      execute l:lw + 1.',$ delete'

      " Return the cursor to the saved position
      call cursor(l:lc, l:cc)
    endif
  endfunction

  " Create mapping proxy to the function just defined
  noremap <Plug>StripTrailingWhitespace
        \ :<C-U>call <SID>StripTrailingWhitespace()<CR>
endif