From 7f3d6287b5a6a50d05243e43b3de1914beb473ca Mon Sep 17 00:00:00 2001
From: Christian Brabandt <cb@256bit.org>
Date: Tue, 20 Jun 2017 21:20:24 +0200
Subject: [PATCH] ale ext: simplify logic

---
 autoload/airline/extensions/ale.vim | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/autoload/airline/extensions/ale.vim b/autoload/airline/extensions/ale.vim
index 84b1b36e..41846c1b 100644
--- a/autoload/airline/extensions/ale.vim
+++ b/autoload/airline/extensions/ale.vim
@@ -20,17 +20,17 @@ function! s:count(index)
   return l:count
 endfunction
 
-function! airline#extensions#ale#get_errors()
-  let l:count = s:count(0)
-  return l:count ? s:error_symbol . l:count : ''
-endfunction
-
-function! airline#extensions#ale#get_warnings()
-  let l:count = s:count(1)
-  return l:count ? s:warning_symbol . l:count : ''
+function! airline#extensions#ale#get(type)
+  let is_err = a:type is# 'error'
+  let cnt = s:count(is_err)
+  if cnt == 0
+    return ''
+  else
+    return (is_err ? s:error_symbol : s:warning_symbol) . cnt
+  endif
 endfunction
 
 function! airline#extensions#ale#init(ext)
-  call airline#parts#define_function('ale_error_count', 'airline#extensions#ale#get_errors')
-  call airline#parts#define_function('ale_warning_count', 'airline#extensions#ale#get_warnings')
+  call airline#parts#define_function('ale_error_count', 'airline#extensions#ale#get("error")')
+  call airline#parts#define_function('ale_warning_count', 'airline#extensions#ale#get("warning")')
 endfunction