From 5233dea0e26268587a41ee195cf1d9ac17fd948b Mon Sep 17 00:00:00 2001
From: DarkWiiPlayer <darkwiiplayer@hotmail.com>
Date: Wed, 19 Aug 2020 17:57:36 +0200
Subject: [PATCH] Improve vim colors plugin

---
 vim/plugin/colors.vim | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/vim/plugin/colors.vim b/vim/plugin/colors.vim
index 73d8505..c405f72 100644
--- a/vim/plugin/colors.vim
+++ b/vim/plugin/colors.vim
@@ -16,11 +16,16 @@ augroup kitty
 augroup END
 
 if $TERM=="xterm-kitty"
-  augroup kitty
-    let s:kitty_bg=system("kitty @ get-colors | grep ^background")
-    let g:kitty_bg=s:kitty_bg[match(s:kitty_bg, "#"):]
-    au VimLeavePre * echom system("kitty @ set-colors background=".g:kitty_bg)
-  augroup END
+	augroup kitty
+		call system("which updatecolors")
+		if v:shell_error
+			let s:kitty_bg=system("kitty @ get-colors | grep ^background")
+			let g:kitty_bg=s:kitty_bg[match(s:kitty_bg, "#"):]
+			au VimLeavePre * echom system("kitty @ set-colors background=".g:kitty_bg)
+		else
+			au VimLeavePre * call system("updatecolors")
+		end
+	augroup END
 end
 
 com! Dark silent! let g:colors_name_bak = g:colors_name