From f9207a326c7f0e5861ee9489313861fdcd7bbff0 Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Sun, 15 Oct 2023 22:00:43 +0200
Subject: Add lines style

---
 src/registers.rs | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

(limited to 'src/registers.rs')

diff --git a/src/registers.rs b/src/registers.rs
index 80b4a9b..d73424d 100644
--- a/src/registers.rs
+++ b/src/registers.rs
@@ -1,4 +1,9 @@
 use nix::libc::user_regs_struct;
+use owo_colors::OwoColorize;
+use ratatui::{
+    prelude::{Line, Span, Style},
+    style::Modifier,
+};
 
 /// Struct used to manipulate registers data from https://docs.rs/libc/0.2.147/libc/struct.user_regs_struct.html
 pub struct RegistersData {
@@ -25,7 +30,28 @@ impl RegistersData {
     pub fn output(&self) -> String {
         format!(
             "{}({:x}, {:x}, {:x}, ...) = {:x}",
-            self.orig_rax, self.rdi, self.rsi, self.rdx, self.rax
+            self.orig_rax.bold(),
+            self.rdi,
+            self.rsi,
+            self.rdx,
+            self.rax
         )
     }
+
+    /// Returns a good line for TUI
+    pub fn output_ui(&self) -> Line {
+        Line::from(vec![
+            Span::styled(
+                format!("{}", self.orig_rax),
+                Style::default().add_modifier(Modifier::BOLD),
+            ),
+            Span::styled(
+                format!(
+                    "({:x}, {:x}, {:x}, ...) = {:x}",
+                    self.rdi, self.rsi, self.rdx, self.rax
+                ),
+                Style::default(),
+            ),
+        ])
+    }
 }
-- 
cgit v1.2.3-18-g5258