let imports[m[2]] = m[1]
endfor
endfor
+
+ let lines = getloclist(winnr())
+ if len(lines) == 0 | let lines = getqflist() | endif
let result = []
- for diag in lsp#ui#vim#diagnostics#get_all_buffer_diagnostics()
- if diag["message"] !~ "^undefined name" | continue | endif
- let m = diag["message"][strridx(diag["message"][:-2], "'")+1:-2]
+ for line in lines
+ if line.text !~ "^undefined name" | continue | endif
+ let m = line.text[strridx(line.text[:-2], "'")+1:-2]
if len(m) == 0 || !has_key(imports, m) | continue | endif
call insert(result, imports[m])
endfor