From ca120088c7206dfe8fc0a9e88a22470123cc574d Mon Sep 17 00:00:00 2001
From: w0rp <devw0rp@gmail.com>
Date: Sun, 7 May 2017 19:30:19 +0100
Subject: Fix #539 - Just set our highlights again when buffers are shown after
 being hidden

---
 autoload/ale/highlight.vim | 35 +++++++++++++++++++++++------------
 1 file changed, 23 insertions(+), 12 deletions(-)

(limited to 'autoload')

diff --git a/autoload/ale/highlight.vim b/autoload/ale/highlight.vim
index 8f6b54de..6cb1c3ae 100644
--- a/autoload/ale/highlight.vim
+++ b/autoload/ale/highlight.vim
@@ -65,11 +65,6 @@ endfunction
 
 function! ale#highlight#UpdateHighlights() abort
     let l:buffer = bufnr('%')
-
-    if has_key(s:buffer_restore_map, l:buffer)
-        call setmatches(s:buffer_restore_map[l:buffer])
-    endif
-
     let l:has_new_items = has_key(s:buffer_highlights, l:buffer)
     let l:loclist = l:has_new_items ? remove(s:buffer_highlights, l:buffer) : []
 
@@ -80,7 +75,13 @@ function! ale#highlight#UpdateHighlights() abort
     " Remove anything with a current match_id
     call filter(l:loclist, '!has_key(v:val, ''match_id'')')
 
-    if l:has_new_items
+    " Restore items from the map of hidden items,
+    " if we don't have some new items to set already.
+    if empty(l:loclist) && has_key(s:buffer_restore_map, l:buffer)
+        let l:loclist = s:buffer_restore_map[l:buffer]
+    endif
+
+    if g:ale_enabled
         for l:item in l:loclist
             let l:col = l:item.col
             let l:group = l:item.type ==# 'E' ? 'ALEError' : 'ALEWarning'
@@ -96,12 +97,22 @@ function! ale#highlight#UpdateHighlights() abort
 endfunction
 
 function! ale#highlight#BufferHidden(buffer) abort
-    " Remember all matches, so they can be restored later.
-    let s:buffer_restore_map[a:buffer] = filter(
-    \   getmatches(),
-    \   'get(v:val, ''group'', '''')[:2] ==# ''ALE'''
-    \)
-    call clearmatches()
+    let l:info = get(g:ale_buffer_info, a:buffer, {'loclist': []})
+    let l:loclist = deepcopy(l:info.loclist)
+
+    " Remember loclist items, so they can be restored later.
+    if !empty(l:loclist)
+        " Remove match_ids, as they must be re-calculated when buffers are
+        " shown again.
+        for l:item in l:loclist
+            if has_key(l:item, 'match_id')
+                call remove(l:item, 'match_id')
+            endif
+        endfor
+
+        let s:buffer_restore_map[a:buffer] = l:loclist
+        call clearmatches()
+    endif
 endfunction
 
 augroup ALEHighlightBufferGroup
-- 
cgit v1.2.3