From 2706bdc94942a8630724a010c2453fadee5559b9 Mon Sep 17 00:00:00 2001
From: patricknraanes <patrick.n.raanes@gmail.com>
Date: Mon, 28 Feb 2022 17:29:47 +0100
Subject: [PATCH] Enable display country flag instead of spelllang

---
 autoload/airline/parts.vim | 88 ++++++++++++++++++++++++++++++++++++++
 doc/airline.txt            |  3 ++
 2 files changed, 91 insertions(+)

diff --git a/autoload/airline/parts.vim b/autoload/airline/parts.vim
index 9af5b612..1cc136d0 100644
--- a/autoload/airline/parts.vim
+++ b/autoload/airline/parts.vim
@@ -64,9 +64,97 @@ function! airline#parts#paste()
   return g:airline_detect_paste && &paste ? g:airline_symbols.paste : ''
 endfunction
 
+" Sources:
+" https://ftp.nluug.nl/pub/vim/runtime/spell/
+" https://en.wikipedia.org/wiki/Regional_indicator_symbol
+let s:flags = {
+                  \ 'af_za': '🇿🇦[af]',
+                  \ 'am_et': '🇭🇺[am]',
+                  \ 'bg_bg': '🇧🇬',
+                  \ 'br_fr': '🇫🇷[br]',
+                  \ 'ca_es': '🇪🇸[ca]',
+                  \ 'cs_cz': '🇨🇿',
+                  \ 'cy_gb': '🇬🇧[cy]',
+                  \ 'da_dk': '🇩🇰',
+                  \ 'de'   : '🇩🇪',
+                  \ 'de_19': '🇩🇪[19]',
+                  \ 'de_20': '🇩🇪[20]',
+                  \ 'de_at': '🇩🇪[at]',
+                  \ 'de_ch': '🇩🇪[ch]',
+                  \ 'de_de': '🇩🇪',
+                  \ 'el_gr': '🇬🇷',
+                  \ 'en':    '🇬🇧',
+                  \ 'en_au': '🇦🇺',
+                  \ 'en_ca': '🇨🇦',
+                  \ 'en_gb': '🇬🇧',
+                  \ 'en_nz': '🇳🇿',
+                  \ 'en_us': '🇺🇸',
+                  \ 'es':    '🇪🇸',
+                  \ 'es_es': '🇪🇸',
+                  \ 'es_mx': '🇲🇽',
+                  \ 'fo_fo': '🇫🇴',
+                  \ 'fr_fr': '🇫🇷',
+                  \ 'ga_ie': '🇮🇪',
+                  \ 'gd_gb': '🇬🇧[gd]',
+                  \ 'gl_es': '🇪🇸[gl]',
+                  \ 'he_il': '🇮🇱',
+                  \ 'hr_hr': '🇭🇷',
+                  \ 'hu_hu': '🇭🇺',
+                  \ 'id_id': '🇮🇩',
+                  \ 'it_it': '🇮🇹',
+                  \ 'ku_tr': '🇹🇷[ku]',
+                  \ 'la'   : '🇮🇹[la]',
+                  \ 'lt_lt': '🇱🇹',
+                  \ 'lv_lv': '🇱🇻',
+                  \ 'mg_mg': '🇲🇬',
+                  \ 'mi_nz': '🇳🇿[mi]',
+                  \ 'ms_my': '🇲🇾',
+                  \ 'nb_no': '🇳🇴',
+                  \ 'nl_nl': '🇳🇱',
+                  \ 'nn_no': '🇳🇴[ny]',
+                  \ 'ny_mw': '🇲🇼',
+                  \ 'pl_pl': '🇵🇱',
+                  \ 'pt':    '🇵🇹',
+                  \ 'pt_br': '🇧🇷',
+                  \ 'pt_pt': '🇵🇹',
+                  \ 'ro_ro': '🇷🇴',
+                  \ 'ru'   : '🇷🇺',
+                  \ 'ru_ru': '🇷🇺',
+                  \ 'ru_yo': '🇷🇺[yo]',
+                  \ 'rw_rw': '🇷🇼',
+                  \ 'sk_sk': '🇸🇰',
+                  \ 'sl_si': '🇸🇮',
+                  \ 'sr_rs': '🇷🇸',
+                  \ 'sv_se': '🇸🇪',
+                  \ 'sw_ke': '🇰🇪',
+                  \ 'tet_id': '🇮🇩[tet]',
+                  \ 'th'   : '🇹🇭',
+                  \ 'tl_ph': '🇵🇭',
+                  \ 'tn_za': '🇿🇦[tn]',
+                  \ 'uk_ua': '🇺🇦',
+                  \ 'yi'   : '🇻🇮',
+                  \ 'yi_tr': '🇹🇷',
+                  \ 'zu_za': '🇿🇦[zu]',
+      \ }
+" Also support spelllang without region codes
+let s:flags_noregion = {}
+for s:key in keys(s:flags)
+  let s:flags_noregion[split(s:key, '_')[0]] = s:flags[s:key]
+endfor
+
 function! airline#parts#spell()
   let spelllang = g:airline_detect_spelllang ? printf(" [%s]", toupper(substitute(&spelllang, ',', '/', 'g'))) : ''
   if g:airline_detect_spell && (&spell || (exists('g:airline_spell_check_command') && eval(g:airline_spell_check_command)))
+
+    if g:airline_detect_spelllang ==? 'flag'
+      let spelllang = tolower(&spelllang)
+      if has_key(s:flags, spelllang)
+        return s:flags[spelllang]
+      elseif has_key(s:flags_noregion, spelllang)
+        return s:flags_noregion[spelllang]
+      endif
+    endif
+
     let winwidth = airline#util#winwidth()
     if winwidth >= 90
       return g:airline_symbols.spell . spelllang
diff --git a/doc/airline.txt b/doc/airline.txt
index 73b58f27..285d53be 100644
--- a/doc/airline.txt
+++ b/doc/airline.txt
@@ -146,6 +146,9 @@ values):
   (if enough space is available) >
   let g:airline_detect_spelllang=1
 <
+  Set to 'flag' to get a unicode icon of the relavant country flag instead of
+  the 'spelllang' itself
+
 * enable iminsert detection >
   let g:airline_detect_iminsert=0
 <