sign.vim 7.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250
  1. " For older Vims without sign_place() the plugin has to manaage the sign ids.
  2. let s:first_sign_id = 3000
  3. let s:next_sign_id = s:first_sign_id
  4. " Remove-all-signs optimisation requires Vim 7.3.596+.
  5. let s:supports_star = v:version > 703 || (v:version == 703 && has("patch596"))
  6. function! gitgutter#sign#enable() abort
  7. let old_signs = g:gitgutter_signs
  8. let g:gitgutter_signs = 1
  9. call gitgutter#highlight#define_sign_text_highlights()
  10. if !old_signs && !g:gitgutter_highlight_lines && !g:gitgutter_highlight_linenrs
  11. call gitgutter#all(1)
  12. endif
  13. endfunction
  14. function! gitgutter#sign#disable() abort
  15. let g:gitgutter_signs = 0
  16. call gitgutter#highlight#define_sign_text_highlights()
  17. if !g:gitgutter_highlight_lines && !g:gitgutter_highlight_linenrs
  18. call gitgutter#sign#clear_signs(bufnr(''))
  19. endif
  20. endfunction
  21. function! gitgutter#sign#toggle() abort
  22. if g:gitgutter_signs
  23. call gitgutter#sign#disable()
  24. else
  25. call gitgutter#sign#enable()
  26. endif
  27. endfunction
  28. " Removes gitgutter's signs from the buffer being processed.
  29. function! gitgutter#sign#clear_signs(bufnr) abort
  30. if exists('*sign_unplace')
  31. call sign_unplace('gitgutter', {'buffer': a:bufnr})
  32. return
  33. endif
  34. call s:find_current_signs(a:bufnr)
  35. let sign_ids = map(values(gitgutter#utility#getbufvar(a:bufnr, 'gitgutter_signs')), 'v:val.id')
  36. call s:remove_signs(a:bufnr, sign_ids, 1)
  37. call gitgutter#utility#setbufvar(a:bufnr, 'gitgutter_signs', {})
  38. endfunction
  39. " Updates gitgutter's signs in the buffer being processed.
  40. "
  41. " modified_lines: list of [<line_number (number)>, <name (string)>]
  42. " where name = 'added|removed|modified|modified_removed'
  43. function! gitgutter#sign#update_signs(bufnr, modified_lines) abort
  44. if exists('*sign_unplace')
  45. " Vim is (hopefully) now quick enough to remove all signs then place new ones.
  46. call sign_unplace('gitgutter', {'buffer': a:bufnr})
  47. let modified_lines = s:handle_double_hunk(a:modified_lines)
  48. let signs = map(copy(modified_lines), '{'.
  49. \ '"buffer": a:bufnr,'.
  50. \ '"group": "gitgutter",'.
  51. \ '"name": s:highlight_name_for_change(v:val[1]),'.
  52. \ '"lnum": v:val[0],'.
  53. \ '"priority": g:gitgutter_sign_priority'.
  54. \ '}')
  55. if exists('*sign_placelist')
  56. call sign_placelist(signs)
  57. return
  58. endif
  59. for sign in signs
  60. call sign_place(0, sign.group, sign.name, sign.buffer, {'lnum': sign.lnum, 'priority': sign.priority})
  61. endfor
  62. return
  63. endif
  64. " Derive a delta between the current signs and the ones we want.
  65. " Remove signs from lines that no longer need a sign.
  66. " Upsert the remaining signs.
  67. call s:find_current_signs(a:bufnr)
  68. let new_gitgutter_signs_line_numbers = map(copy(a:modified_lines), 'v:val[0]')
  69. let obsolete_signs = s:obsolete_gitgutter_signs_to_remove(a:bufnr, new_gitgutter_signs_line_numbers)
  70. call s:remove_signs(a:bufnr, obsolete_signs, s:remove_all_old_signs)
  71. call s:upsert_new_gitgutter_signs(a:bufnr, a:modified_lines)
  72. endfunction
  73. "
  74. " Internal functions
  75. "
  76. function! s:find_current_signs(bufnr) abort
  77. let gitgutter_signs = {} " <line_number (string)>: {'id': <id (number)>, 'name': <name (string)>}
  78. if !g:gitgutter_sign_allow_clobber
  79. let other_signs = [] " [<line_number (number),...]
  80. endif
  81. if exists('*getbufinfo')
  82. let bufinfo = getbufinfo(a:bufnr)[0]
  83. let signs = has_key(bufinfo, 'signs') ? bufinfo.signs : []
  84. else
  85. let signs = []
  86. redir => signlines
  87. silent execute "sign place buffer=" . a:bufnr
  88. redir END
  89. for signline in filter(split(signlines, '\n')[2:], 'v:val =~# "="')
  90. " Typical sign line before v8.1.0614: line=88 id=1234 name=GitGutterLineAdded
  91. " We assume splitting is faster than a regexp.
  92. let components = split(signline)
  93. call add(signs, {
  94. \ 'lnum': str2nr(split(components[0], '=')[1]),
  95. \ 'id': str2nr(split(components[1], '=')[1]),
  96. \ 'name': split(components[2], '=')[1]
  97. \ })
  98. endfor
  99. endif
  100. for sign in signs
  101. if sign.name =~# 'GitGutter'
  102. " Remove orphaned signs (signs placed on lines which have been deleted).
  103. " (When a line is deleted its sign lingers. Subsequent lines' signs'
  104. " line numbers are decremented appropriately.)
  105. if has_key(gitgutter_signs, sign.lnum)
  106. execute "sign unplace" gitgutter_signs[sign.lnum].id
  107. endif
  108. let gitgutter_signs[sign.lnum] = {'id': sign.id, 'name': sign.name}
  109. else
  110. if !g:gitgutter_sign_allow_clobber
  111. call add(other_signs, sign.lnum)
  112. endif
  113. endif
  114. endfor
  115. call gitgutter#utility#setbufvar(a:bufnr, 'gitgutter_signs', gitgutter_signs)
  116. if !g:gitgutter_sign_allow_clobber
  117. call gitgutter#utility#setbufvar(a:bufnr, 'other_signs', other_signs)
  118. endif
  119. endfunction
  120. " Returns a list of [<id (number)>, ...]
  121. " Sets `s:remove_all_old_signs` as a side-effect.
  122. function! s:obsolete_gitgutter_signs_to_remove(bufnr, new_gitgutter_signs_line_numbers) abort
  123. let signs_to_remove = [] " list of [<id (number)>, ...]
  124. let remove_all_signs = 1
  125. let old_gitgutter_signs = gitgutter#utility#getbufvar(a:bufnr, 'gitgutter_signs')
  126. for line_number in keys(old_gitgutter_signs)
  127. if index(a:new_gitgutter_signs_line_numbers, str2nr(line_number)) == -1
  128. call add(signs_to_remove, old_gitgutter_signs[line_number].id)
  129. else
  130. let remove_all_signs = 0
  131. endif
  132. endfor
  133. let s:remove_all_old_signs = remove_all_signs
  134. return signs_to_remove
  135. endfunction
  136. function! s:remove_signs(bufnr, sign_ids, all_signs) abort
  137. if a:all_signs && s:supports_star && (g:gitgutter_sign_allow_clobber || empty(gitgutter#utility#getbufvar(a:bufnr, 'other_signs')))
  138. execute "sign unplace * buffer=" . a:bufnr
  139. else
  140. for id in a:sign_ids
  141. execute "sign unplace" id
  142. endfor
  143. endif
  144. endfunction
  145. function! s:upsert_new_gitgutter_signs(bufnr, modified_lines) abort
  146. if !g:gitgutter_sign_allow_clobber
  147. let other_signs = gitgutter#utility#getbufvar(a:bufnr, 'other_signs')
  148. endif
  149. let old_gitgutter_signs = gitgutter#utility#getbufvar(a:bufnr, 'gitgutter_signs')
  150. let modified_lines = s:handle_double_hunk(a:modified_lines)
  151. for line in modified_lines
  152. let line_number = line[0] " <number>
  153. if g:gitgutter_sign_allow_clobber || index(other_signs, line_number) == -1 " don't clobber others' signs
  154. let name = s:highlight_name_for_change(line[1])
  155. if !has_key(old_gitgutter_signs, line_number) " insert
  156. let id = s:next_sign_id()
  157. execute "sign place" id "line=" . line_number "name=" . name "buffer=" . a:bufnr
  158. else " update if sign has changed
  159. let old_sign = old_gitgutter_signs[line_number]
  160. if old_sign.name !=# name
  161. execute "sign place" old_sign.id "name=" . name "buffer=" . a:bufnr
  162. end
  163. endif
  164. endif
  165. endfor
  166. " At this point b:gitgutter_gitgutter_signs is out of date.
  167. endfunction
  168. " Handle special case where the first line is the site of two hunks:
  169. " lines deleted above at the start of the file, and lines deleted
  170. " immediately below.
  171. function! s:handle_double_hunk(modified_lines)
  172. if a:modified_lines[0:1] == [[1, 'removed_first_line'], [1, 'removed']]
  173. return [[1, 'removed_above_and_below']] + a:modified_lines[2:]
  174. endif
  175. return a:modified_lines
  176. endfunction
  177. function! s:next_sign_id() abort
  178. let next_id = s:next_sign_id
  179. let s:next_sign_id += 1
  180. return next_id
  181. endfunction
  182. " Only for testing.
  183. function! gitgutter#sign#reset()
  184. let s:next_sign_id = s:first_sign_id
  185. endfunction
  186. function! s:highlight_name_for_change(text) abort
  187. if a:text ==# 'added'
  188. return 'GitGutterLineAdded'
  189. elseif a:text ==# 'removed'
  190. return 'GitGutterLineRemoved'
  191. elseif a:text ==# 'removed_first_line'
  192. return 'GitGutterLineRemovedFirstLine'
  193. elseif a:text ==# 'modified'
  194. return 'GitGutterLineModified'
  195. elseif a:text ==# 'modified_removed'
  196. return 'GitGutterLineModifiedRemoved'
  197. elseif a:text ==# 'removed_above_and_below'
  198. return 'GitGutterLineRemovedAboveAndBelow'
  199. endif
  200. endfunction