ui: load syntax theme based on the number of supported colors

The theme to use can be overriden via the $VIS_THEME environment
variable.

 $ VIS_THEME=solarized vis

3 files changed, 3 insertions, 1 deletions

diff --git a/lexers/themes/default.lua b/lexers/themes/default-256.lua
index 8479b1b..8479b1b 120000
--- a/lexers/themes/default.lua
+++ b/lexers/themes/default-256.lua
diff --git a/ui-curses.c b/ui-curses.c
index 79e75e5..313f3d6 100644
--- a/ui-curses.c
+++ b/ui-curses.c
@@ -987,6 +987,9 @@ static void ui_prompt_hide(Ui *ui) {
 static bool ui_init(Ui *ui, Vis *vis) {
 	UiCurses *uic = (UiCurses*)ui;
 	uic->vis = vis;
+	const char *theme = getenv("VIS_THEME");
+	if (!theme || !theme[0] || !vis_theme_load(vis, theme))
+		vis_theme_load(vis, COLORS <= 16 ? "default-16" : "default-256");
 	return true;
 }
 
diff --git a/vis.c b/vis.c
index dc56fb3..31e4ecf 100644
--- a/vis.c
+++ b/vis.c
@@ -365,7 +365,6 @@ Vis *vis_new(Ui *ui) {
 		lua_pushvalue(L, -2); /* require return value */
 		lua_setfield(L, -2, "lexers");
 		lua_setglobal(L, "vis");
-		vis_theme_load(vis, "default");
 	}
 
 	vis->ui = ui;