1 files changed, 121 insertions, 39 deletions
diff --git a/src/widgets/luaeditor/theme.rs b/src/widgets/luaeditor/theme.rs
index 38e4498..f0520f4 100644
--- a/src/widgets/luaeditor/theme.rs
+++ b/src/widgets/luaeditor/theme.rs
@@ -1,45 +1,127 @@
-use ratatui::{
-    style::{Color, Stylize},
-    text::Span,
-};
+use mlua::{FromLua, IntoLua};
+use ratatui::{style::Color, text::Span};
 use tree_sitter_highlight::Highlight;
 
-static HIGHLIGHT_THEME: [Color; 29] = [
-    Color::White,       // attribute
-    Color::White,       // boolean
-    Color::DarkGray,    // comment
-    Color::LightYellow, // conditional
-    Color::White,       // constant
-    Color::White,       // constant.builtin
-    Color::White,       // constructor
-    Color::White,       // field
-    Color::White,       // function
-    Color::White,       // function.builtin
-    Color::White,       // function.call
-    Color::Yellow,      // keyword
-    Color::LightYellow, // keyword.function
-    Color::White,       // keyword.operator
-    Color::LightYellow, // keyword.return
-    Color::White,       // label
-    Color::White,       // method
-    Color::White,       // method.call
-    Color::Blue,        // number
-    Color::Gray,        // operator
-    Color::White,       // parameter
-    Color::White,       // preproc
-    Color::Gray,        // punctuation.bracket
-    Color::Gray,        // punctuation.delimiter
-    Color::LightYellow, // repeat
-    Color::LightGreen,  // string
-    Color::Yellow,      // string.escape
-    Color::Gray,        // variable
-    Color::White,       // variable.builtin
+use crate::config::theme::style::Style;
+
+pub(super) static HIGHLIGHT_NAMES: [&str; 29] = [
+    "attribute",
+    "boolean",
+    "comment",
+    "conditional",
+    "constant",
+    "constant.builtin",
+    "constructor",
+    "field",
+    "function",
+    "function.builtin",
+    "function.call",
+    "keyword",
+    "keyword.function",
+    "keyword.operator",
+    "keyword.return",
+    "label",
+    "method",
+    "method.call",
+    "number",
+    "operator",
+    "parameter",
+    "preproc",
+    "punctuation.bracket",
+    "punctuation.delimiter",
+    "repeat",
+    "string",
+    "string.escape",
+    "variable",
+    "variable.builtin",
 ];
 
-pub fn theme_highlight_group(hl: Option<Highlight>, span: Span) -> Span {
-    if let Some(hl) = hl {
-        span.fg(HIGHLIGHT_THEME[hl.0])
-    } else {
-        span
+#[derive(Default, Debug, Clone)]
+pub struct HighlightTheme {
+    theme: [Style; 29],
+}
+
+impl HighlightTheme {
+    pub const fn new() -> Self {
+        Self {
+            theme: [
+                Style::new().fg(Color::White),  // attribute
+                Style::new().fg(Color::White),  // boolean
+                Style::new().fg(Color::White),  // comment
+                Style::new().fg(Color::White),  // conditional
+                Style::new().fg(Color::White),  // constant
+                Style::new().fg(Color::White),  // constant.builtin
+                Style::new().fg(Color::White),  // constructor
+                Style::new().fg(Color::White),  // field
+                Style::new().fg(Color::White),  // function
+                Style::new().fg(Color::White),  // function.builtin
+                Style::new().fg(Color::White),  // function.call
+                Style::new().fg(Color::White),  // keyword
+                Style::new().fg(Color::White),  // keyword.function
+                Style::new().fg(Color::White),  // keyword.operator
+                Style::new().fg(Color::White),  // keyword.return
+                Style::new().fg(Color::White),  // label
+                Style::new().fg(Color::White),  // method
+                Style::new().fg(Color::White),  // method.call
+                Style::new().fg(Color::White),  // number
+                Style::new().fg(Color::White),  // operator
+                Style::new().fg(Color::White),  // parameter
+                Style::new().fg(Color::White),  // preproc
+                Style::new().fg(Color::White),  // punctuation.bracket
+                Style::new().fg(Color::White),  // punctuation.delimiter
+                Style::new().fg(Color::White),  // repeat
+                Style::new().fg(Color::White),  // string
+                Style::new().fg(Color::White),  // string.escape
+                Style::new().fg(Color::White),  // variable
+                Style::new().fg(Color::White),  // variable.builtin
+            ],
+        }
+    }
+
+    pub fn highlight<'a>(&'a self, hl: Option<Highlight>, span: Span<'a>) -> Span<'a> {
+        if let Some(hl) = hl {
+            self.theme[hl.0].apply(span)
+        } else {
+            span
+        }
+    }
+}
+
+impl<'lua> IntoLua<'lua> for HighlightTheme {
+    fn into_lua(
+        self,
+        lua: &'lua mlua::prelude::Lua,
+    ) -> mlua::prelude::LuaResult<mlua::prelude::LuaValue<'lua>> {
+        let table = lua.create_table()?;
+
+        for (i, names) in HIGHLIGHT_NAMES.iter().enumerate() {
+            table.set(names.replace('.', "_"), self.theme[i])?
+        }
+
+        table.into_lua(lua)
+    }
+}
+
+impl<'lua> FromLua<'lua> for HighlightTheme {
+    fn from_lua(
+        value: mlua::prelude::LuaValue<'lua>,
+        _lua: &'lua mlua::prelude::Lua,
+    ) -> mlua::prelude::LuaResult<Self> {
+        let mut theme = HighlightTheme::default();
+
+        if value.is_table() {
+            let table = value.as_table().unwrap();
+            for (i, names) in HIGHLIGHT_NAMES.iter().enumerate() {
+                if let Ok(style) = table.get(names.replace('.', "_")) {
+                    theme.theme[i] = style;
+                }
+            }
+
+            Ok(theme)
+        } else {
+            Err(mlua::prelude::LuaError::runtime(
+                "failed to parse HighlightTheme",
+            ))
+        }
     }
 }