Syntax highlight for keywords

Cargo.toml

@@ -39,7 +39,7 @@ tokio-util = "0.7.10"
 toml = "0.8.12"
 tonic = { version = "0.11.0", optional = true }
 tui-logger = {version = "0.12", features = ["tracing-support"]}
-tui-textarea = "0.6.1"
+tui-textarea = { version = "0.6.1", features = ["search"] }
 url = { version = "2.5.2", optional = true }
 
 [dev-dependencies]

src/app/state/tabs/flightsql.rs

@@ -25,6 +25,7 @@ use ratatui::style::Style;
 use ratatui::widgets::TableState;
 use tui_textarea::TextArea;
 
+use crate::app::state::tabs::sql;
 use crate::execution::ExecutionStats;
 
 #[derive(Clone, Debug)]
@@ -115,7 +116,8 @@ impl<'app> FlightSQLTabState<'app> {
         // TODO: Enable vim mode from config?
         let mut textarea = TextArea::new(empty_text);
         textarea.set_style(Style::default().fg(tailwind::WHITE));
-
+        textarea.set_search_pattern(sql::keyword_regex()).unwrap();
+        textarea.set_search_style(sql::keyword_style());
         Self {
             editor: textarea,
             editor_editable: false,
@@ -152,6 +154,8 @@ impl<'app> FlightSQLTabState<'app> {
     pub fn clear_editor(&mut self) {
         let mut textarea = TextArea::new(vec!["".to_string()]);
         textarea.set_style(Style::default().fg(tailwind::WHITE));
+        textarea.set_search_pattern(sql::keyword_regex()).unwrap();
+        textarea.set_search_style(sql::keyword_style());
         self.editor = textarea;
     }
 

src/app/state/tabs/sql.rs

@@ -19,9 +19,10 @@ use core::cell::RefCell;
 use std::time::Duration;
 
 use datafusion::arrow::array::RecordBatch;
+use datafusion::sql::sqlparser::keywords;
 use ratatui::crossterm::event::KeyEvent;
 use ratatui::style::palette::tailwind;
-use ratatui::style::Style;
+use ratatui::style::{Modifier, Style};
 use ratatui::widgets::TableState;
 use tui_textarea::TextArea;
 
@@ -101,6 +102,27 @@ impl Query {
     }
 }
 
+pub fn get_keywords() -> Vec<String> {
+    keywords::ALL_KEYWORDS
+        .iter()
+        .map(|k| k.to_string())
+        .collect()
+}
+
+pub fn keyword_regex() -> String {
+    format!(
+        "(?i)(^|[^a-zA-Z0-9\'\"`._]*?)({})($|[^a-zA-Z0-9\'\"`._]*)",
+        get_keywords().join("|")
+    )
+}
+
+pub fn keyword_style() -> Style {
+    Style::default()
+        .bg(tailwind::BLACK)
+        .fg(tailwind::YELLOW.c100)
+        .add_modifier(Modifier::BOLD)
+}
+
 #[derive(Debug, Default)]
 pub struct SQLTabState<'app> {
     editor: TextArea<'app>,
@@ -115,6 +137,8 @@ impl<'app> SQLTabState<'app> {
         // TODO: Enable vim mode from config?
         let mut textarea = TextArea::new(empty_text);
         textarea.set_style(Style::default().fg(tailwind::WHITE));
+        textarea.set_search_pattern(keyword_regex()).unwrap();
+        textarea.set_search_style(keyword_style());
         Self {
             editor: textarea,
             editor_editable: false,
@@ -151,6 +175,8 @@ impl<'app> SQLTabState<'app> {
     pub fn clear_editor(&mut self) {
         let mut textarea = TextArea::new(vec!["".to_string()]);
         textarea.set_style(Style::default().fg(tailwind::WHITE));
+        textarea.set_search_pattern(keyword_regex()).unwrap();
+        textarea.set_search_style(keyword_style());
         self.editor = textarea;
     }