From 83dc13a86d4ddc00be23e22f81843bf025517c08 Mon Sep 17 00:00:00 2001
From: sharkdp <davidpeter@web.de>
Date: Sat, 21 Mar 2020 20:31:32 +0100
Subject: [PATCH] Add fallback theme, remove BAT_THEME_DEFAULT

---
 src/assets.rs      | 16 +++++++++++++---
 src/bin/bat/app.rs |  6 +++---
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/src/assets.rs b/src/assets.rs
index 954bf615..203ae39b 100644
--- a/src/assets.rs
+++ b/src/assets.rs
@@ -11,15 +11,18 @@ use crate::errors::*;
 use crate::inputfile::{InputFile, InputFileReader};
 use crate::syntax_mapping::SyntaxMapping;
 
-pub const BAT_THEME_DEFAULT: &str = "Monokai Extended";
-
 #[derive(Debug)]
 pub struct HighlightingAssets {
     pub(crate) syntax_set: SyntaxSet,
     pub(crate) theme_set: ThemeSet,
+    fallback_theme: Option<&'static str>,
 }
 
 impl HighlightingAssets {
+    pub fn default_theme() -> &'static str {
+        "Monokai Extended"
+    }
+
     pub fn from_files(source_dir: &Path, start_empty: bool) -> Result<Self> {
         let mut theme_set = if start_empty {
             ThemeSet {
@@ -60,6 +63,7 @@ impl HighlightingAssets {
         Ok(HighlightingAssets {
             syntax_set: syntax_set_builder.build(),
             theme_set,
+            fallback_theme: None,
         })
     }
 
@@ -85,6 +89,7 @@ impl HighlightingAssets {
         Ok(HighlightingAssets {
             syntax_set,
             theme_set,
+            fallback_theme: None,
         })
     }
 
@@ -103,6 +108,7 @@ impl HighlightingAssets {
         HighlightingAssets {
             syntax_set,
             theme_set,
+            fallback_theme: None,
         }
     }
 
@@ -138,6 +144,10 @@ impl HighlightingAssets {
         Ok(())
     }
 
+    pub fn set_fallback_theme(&mut self, theme: &'static str) {
+        self.fallback_theme = Some(theme);
+    }
+
     pub fn syntaxes(&self) -> &[SyntaxReference] {
         self.syntax_set.syntaxes()
     }
@@ -156,7 +166,7 @@ impl HighlightingAssets {
                     Yellow.paint("[bat warning]"),
                     theme
                 );
-                &self.theme_set.themes[BAT_THEME_DEFAULT]
+                &self.theme_set.themes[self.fallback_theme.unwrap_or(Self::default_theme())]
             }
         }
     }
diff --git a/src/bin/bat/app.rs b/src/bin/bat/app.rs
index 367f700b..281882a7 100644
--- a/src/bin/bat/app.rs
+++ b/src/bin/bat/app.rs
@@ -17,7 +17,7 @@ use console::Term;
 use ansi_term;
 
 use bat::{
-    assets::BAT_THEME_DEFAULT,
+    assets::HighlightingAssets,
     config::{Config, PagingMode},
     errors::*,
     inputfile::InputFile,
@@ -195,12 +195,12 @@ impl App {
                 .or_else(|| env::var("BAT_THEME").ok())
                 .map(|s| {
                     if s == "default" {
-                        String::from(BAT_THEME_DEFAULT)
+                        String::from(HighlightingAssets::default_theme())
                     } else {
                         s
                     }
                 })
-                .unwrap_or_else(|| String::from(BAT_THEME_DEFAULT)),
+                .unwrap_or_else(|| String::from(HighlightingAssets::default_theme())),
             line_ranges: self
                 .matches
                 .values_of("line-range")