taco bell'ize

Author: wcampbell0x2a

src/main.rs

@@ -1,10 +1,3 @@
-use clap::Parser;
-use mi::{
-    data_disassemble, data_read_memory_bytes, parse_asm_insns_values, parse_key_value_pairs,
-    parse_register_values, register_x86_64, Asm, MIResponse, Register,
-};
-use ratatui::widgets::block::Title;
-use ratatui::widgets::{Cell, Row, Table, TableState};
 use std::fs::File;
 use std::io::{BufRead, BufReader, Read, Write};
 use std::net::{SocketAddr, TcpStream};
@@ -13,13 +6,15 @@ use std::sync::{Arc, Mutex};
 use std::thread;
 use std::time::Duration;
 use std::{error::Error, io};
-use tui_input::backend::crossterm::EventHandler;
-use tui_input::Input;
 
+use clap::Parser;
 use env_logger::{Builder, Env};
 use log::debug;
 use ratatui::layout::{Constraint, Layout};
 use ratatui::prelude::*;
+use ratatui::style::Styled;
+use ratatui::widgets::block::Title;
+use ratatui::widgets::{Cell, Row, Table, TableState};
 use ratatui::{
     crossterm::{
         event::{self, DisableMouseCapture, Event, KeyCode},
@@ -28,15 +23,27 @@ use ratatui::{
     },
     widgets::{Block, Borders, List, ListItem, Paragraph},
 };
+use tui_input::backend::crossterm::EventHandler;
+use tui_input::Input;
 use Constraint::{Fill, Length, Max, Min};
 
 mod mi;
+use mi::{
+    data_disassemble, data_read_memory_bytes, parse_asm_insns_values, parse_key_value_pairs,
+    parse_register_values, register_x86_64, Asm, MIResponse, Register,
+};
 
 enum InputMode {
     Normal,
     Editing,
 }
 
+// Taco bell colors
+const BLUE: Color = Color::Rgb(54, 57, 154);
+const PURPLE: Color = Color::Rgb(167, 123, 202);
+const PINK: Color = Color::Rgb(239, 24, 151);
+const YELLOW: Color = Color::Rgb(254, 224, 18);
+
 use std::collections::{HashMap, VecDeque};
 
 struct LimitedBuffer<T> {
@@ -484,8 +491,8 @@ fn ui(f: &mut Frame, app: &App) {
         Ok(regs) => {
             for (name, register) in regs.iter() {
                 rows.push(Row::new(vec![
-                    name.to_string(),
-                    register.value.clone().unwrap(),
+                    Cell::from(name.to_string()).style(Style::new().fg(PURPLE)),
+                    Cell::from(register.value.clone().unwrap()),
                 ]));
             }
         }
@@ -495,8 +502,8 @@ fn ui(f: &mut Frame, app: &App) {
     let widths = [Constraint::Length(5), Constraint::Length(20)];
     let table = Table::new(rows, widths).block(
         Block::default()
-            .borders(Borders::ALL)
-            .title("Registers".blue()),
+            .borders(Borders::TOP)
+            .title("Registers".fg(PINK)),
     );
 
     f.render_widget(table, register);
@@ -509,7 +516,7 @@ fn ui(f: &mut Frame, app: &App) {
             entries.sort_by(|a, b| a.0.cmp(&b.0));
             for (addr, value) in entries.iter() {
                 rows.push(Row::new(vec![
-                    Cell::from(format!("0x{:02x}", addr)).yellow(),
+                    Cell::from(format!("0x{:02x}", addr)).style(Style::new().fg(PURPLE)),
                     Cell::from(format!("0x{:02x}", value)),
                 ]));
             }
@@ -518,8 +525,11 @@ fn ui(f: &mut Frame, app: &App) {
     }
 
     let widths = [Constraint::Length(16), Fill(1)];
-    let table = Table::new(rows, widths)
-        .block(Block::default().borders(Borders::ALL).title("Stack".blue()));
+    let table = Table::new(rows, widths).block(
+        Block::default()
+            .borders(Borders::TOP)
+            .title("Stack".fg(PINK)),
+    );
 
     f.render_widget(table, stack);
 
@@ -552,7 +562,8 @@ fn ui(f: &mut Frame, app: &App) {
                         function_name = Some(func_name.clone());
                     }
                 }
-                let addr_cell = Cell::from(format!("0x{:02x}", a.address)).yellow();
+                let addr_cell =
+                    Cell::from(format!("0x{:02x}", a.address)).style(Style::default().fg(PURPLE));
                 let inst_cell = if let Some(pc_index) = pc_index {
                     if pc_index == index {
                         Cell::from(format!("{}", a.inst)).green()
@@ -570,14 +581,14 @@ fn ui(f: &mut Frame, app: &App) {
     }
 
     let tital = if let Some(function_name) = function_name {
-        Title::from(format!("Instructions ({})", function_name).blue())
+        Title::from(format!("Instructions ({})", function_name).fg(PINK))
     } else {
-        Title::from("Instructions".blue())
+        Title::from("Instructions".fg(PINK))
     };
     if let Some(pc_index) = pc_index {
         let widths = [Constraint::Length(16), Fill(1)];
         let table = Table::new(rows, widths)
-            .block(Block::default().borders(Borders::ALL).title(tital))
+            .block(Block::default().borders(Borders::TOP).title(tital))
             .row_highlight_style(Style::new().green())
             .highlight_symbol(">>");
         let start_offset = if pc_index < 5 { 0 } else { pc_index - 5 };
@@ -586,7 +597,7 @@ fn ui(f: &mut Frame, app: &App) {
             .with_selected(pc_index);
         f.render_stateful_widget(table, asm, &mut table_state);
     } else {
-        let block = Block::default().borders(Borders::ALL).title(tital);
+        let block = Block::default().borders(Borders::TOP).title(tital);
         f.render_widget(block, asm);
     }
 
@@ -603,13 +614,13 @@ fn ui(f: &mut Frame, app: &App) {
     let messages = List::new(messages).block(
         Block::default()
             .borders(Borders::ALL)
-            .title("Messages".blue()),
+            .title("Messages".fg(BLUE)),
     );
     f.render_widget(messages, other);
 
     let response_widget = Paragraph::new(response_text).block(
         Block::default()
-            .title("Parsed Responses".blue())
+            .title("Parsed Responses".fg(BLUE))
             .borders(Borders::ALL),
     );
     f.render_widget(response_widget, parsed);
@@ -624,7 +635,11 @@ fn ui(f: &mut Frame, app: &App) {
             InputMode::Editing => Style::default().fg(Color::Green),
         })
         .scroll((0, scroll as u16))
-        .block(Block::default().borders(Borders::ALL).title("Input"));
+        .block(
+            Block::default()
+                .borders(Borders::ALL)
+                .title("Input".fg(YELLOW)),
+        );
     f.render_widget(txt_input, input);
     match app.input_mode {
         InputMode::Normal =>