builder.vim 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246
  1. " MIT License. Copyright (c) 2013-2021 Bailey Ling et al.
  2. " vim: et ts=2 sts=2 sw=2
  3. scriptencoding utf-8
  4. let s:prototype = {}
  5. function! s:prototype.split(...) dict
  6. call add(self._sections, ['|', a:0 ? a:1 : '%='])
  7. endfunction
  8. function! s:prototype.add_section_spaced(group, contents) dict
  9. let spc = empty(a:contents) ? '' : g:airline_symbols.space
  10. call self.add_section(a:group, spc.a:contents.spc)
  11. endfunction
  12. function! s:prototype.add_section(group, contents) dict
  13. call add(self._sections, [a:group, a:contents])
  14. endfunction
  15. function! s:prototype.add_raw(text) dict
  16. call add(self._sections, ['', a:text])
  17. endfunction
  18. function! s:prototype.insert_section(group, contents, position) dict
  19. call insert(self._sections, [a:group, a:contents], a:position)
  20. endfunction
  21. function! s:prototype.insert_raw(text, position) dict
  22. call insert(self._sections, ['', a:text], a:position)
  23. endfunction
  24. function! s:prototype.get_position() dict
  25. return len(self._sections)
  26. endfunction
  27. function! airline#builder#get_prev_group(sections, i)
  28. let x = a:i - 1
  29. while x >= 0
  30. let group = a:sections[x][0]
  31. if group != '' && group != '|'
  32. return group
  33. endif
  34. let x = x - 1
  35. endwhile
  36. return ''
  37. endfunction
  38. function! airline#builder#get_next_group(sections, i)
  39. let x = a:i + 1
  40. let l = len(a:sections)
  41. while x < l
  42. let group = a:sections[x][0]
  43. if group != '' && group != '|'
  44. return group
  45. endif
  46. let x = x + 1
  47. endwhile
  48. return ''
  49. endfunction
  50. function! s:prototype.build() dict
  51. let side = 1
  52. let line = ''
  53. let i = 0
  54. let length = len(self._sections)
  55. let split = 0
  56. let is_empty = 0
  57. let prev_group = ''
  58. while i < length
  59. let section = self._sections[i]
  60. let group = section[0]
  61. let contents = section[1]
  62. let pgroup = prev_group
  63. let prev_group = airline#builder#get_prev_group(self._sections, i)
  64. if group ==# 'airline_c' && &buftype ==# 'terminal' && self._context.active
  65. let group = 'airline_term'
  66. elseif group ==# 'airline_c' && !self._context.active && has_key(self._context, 'bufnr')
  67. let group = 'airline_c'. self._context.bufnr
  68. elseif prev_group ==# 'airline_c' && !self._context.active && has_key(self._context, 'bufnr')
  69. let prev_group = 'airline_c'. self._context.bufnr
  70. endif
  71. if is_empty
  72. let prev_group = pgroup
  73. endif
  74. let is_empty = s:section_is_empty(self, contents)
  75. if is_empty
  76. " need to fix highlighting groups, since we
  77. " have skipped a section, we actually need
  78. " the previous previous group and so the
  79. " separator goes from the previous previous group
  80. " to the current group
  81. let pgroup = group
  82. endif
  83. if group == ''
  84. let line .= contents
  85. elseif group == '|'
  86. let side = 0
  87. let line .= contents
  88. let split = 1
  89. else
  90. if prev_group == ''
  91. let line .= '%#'.group.'#'
  92. elseif split
  93. if !is_empty
  94. let line .= s:get_transitioned_separator(self, prev_group, group, side)
  95. endif
  96. let split = 0
  97. else
  98. if !is_empty
  99. let line .= s:get_separator(self, prev_group, group, side)
  100. endif
  101. endif
  102. let line .= is_empty ? '' : s:get_accented_line(self, group, contents)
  103. endif
  104. let i = i + 1
  105. endwhile
  106. if !self._context.active
  107. "let line = substitute(line, '%#airline_c#', '%#airline_c'.self._context.bufnr.'#', '')
  108. let line = substitute(line, '%#.\{-}\ze#', '\0_inactive', 'g')
  109. endif
  110. return line
  111. endfunction
  112. function! airline#builder#should_change_group(group1, group2)
  113. if a:group1 == a:group2
  114. return 0
  115. endif
  116. let color1 = airline#highlighter#get_highlight(a:group1)
  117. let color2 = airline#highlighter#get_highlight(a:group2)
  118. return color1[1] != color2[1] || color1[0] != color2[0]
  119. \ || color1[2] != color2[2] || color1[3] != color2[3]
  120. endfunction
  121. function! s:get_transitioned_separator(self, prev_group, group, side)
  122. let line = ''
  123. if get(a:self._context, 'tabline', 0) && get(g:, 'airline#extensions#tabline#alt_sep', 0) && a:group ==# 'airline_tabsel' && a:side
  124. call airline#highlighter#add_separator(a:prev_group, a:group, 0)
  125. let line .= '%#'.a:prev_group.'_to_'.a:group.'#'
  126. let line .= a:self._context.right_sep.'%#'.a:group.'#'
  127. else
  128. call airline#highlighter#add_separator(a:prev_group, a:group, a:side)
  129. let line .= '%#'.a:prev_group.'_to_'.a:group.'#'
  130. let line .= a:side ? a:self._context.left_sep : a:self._context.right_sep
  131. let line .= '%#'.a:group.'#'
  132. endif
  133. return line
  134. endfunction
  135. function! s:get_separator(self, prev_group, group, side)
  136. if airline#builder#should_change_group(a:prev_group, a:group)
  137. return s:get_transitioned_separator(a:self, a:prev_group, a:group, a:side)
  138. else
  139. return a:side ? a:self._context.left_alt_sep : a:self._context.right_alt_sep
  140. endif
  141. endfunction
  142. function! s:get_accented_line(self, group, contents)
  143. if a:self._context.active
  144. " active window
  145. let contents = []
  146. let content_parts = split(a:contents, '__accent')
  147. for cpart in content_parts
  148. let accent = matchstr(cpart, '_\zs[^#]*\ze')
  149. call add(contents, cpart)
  150. endfor
  151. let line = join(contents, a:group)
  152. let line = substitute(line, '__restore__', a:group, 'g')
  153. else
  154. " inactive window
  155. let line = substitute(a:contents, '%#__accent[^#]*#', '', 'g')
  156. let line = substitute(line, '%#__restore__#', '', 'g')
  157. endif
  158. return line
  159. endfunction
  160. function! s:section_is_empty(self, content)
  161. let start=1
  162. " do not check for inactive windows or the tabline
  163. if a:self._context.active == 0
  164. return 0
  165. elseif get(a:self._context, 'tabline', 0)
  166. return 0
  167. endif
  168. " only check, if airline#skip_empty_sections == 1
  169. if get(g:, 'airline_skip_empty_sections', 0) == 0
  170. return 0
  171. endif
  172. " only check, if airline#skip_empty_sections == 1
  173. if get(w:, 'airline_skip_empty_sections', -1) == 0
  174. return 0
  175. endif
  176. " special case: When the content is %=, that is the
  177. " separation marker, which switches between left- and
  178. " right-aligned content.
  179. " Consider that to be empty, so that the previous previous
  180. " group is correctly remembered in the builder() function
  181. if empty(a:content) || a:content is# '%='
  182. return 1
  183. endif
  184. let stripped = substitute(a:content,
  185. \ '\(%{.*}\|%#__accent_[^#]*#\|%#__restore__#\|%( \| %)\)', '', 'g')
  186. if !empty(stripped)
  187. return 0 " There is content in the statusline
  188. endif
  189. let exprlist = []
  190. call substitute(a:content, '%{\([^}]*\)}', '\=add(exprlist, submatch(1))', 'g')
  191. for expr in exprlist
  192. try
  193. " catch all exceptions, just in case
  194. if !empty(eval(expr))
  195. return 0
  196. endif
  197. catch
  198. return 0
  199. endtry
  200. endfor
  201. return 1
  202. endfunction
  203. function! airline#builder#new(context)
  204. let builder = copy(s:prototype)
  205. let builder._context = a:context
  206. let builder._sections = []
  207. call extend(builder._context, {
  208. \ 'left_sep': g:airline_left_sep,
  209. \ 'left_alt_sep': g:airline_left_alt_sep,
  210. \ 'right_sep': g:airline_right_sep,
  211. \ 'right_alt_sep': g:airline_right_alt_sep,
  212. \ }, 'keep')
  213. return builder
  214. endfunction