From 84eb9b1a72fdd0cd722a752fca75ec3ffb4ba86c Mon Sep 17 00:00:00 2001
From: Yash Karandikar <nerdstep710@gmail.com>
Date: Fri, 17 Sep 2021 08:46:03 -0500
Subject: [PATCH] Add function to change writer color

---
 src/main.rs       | 2 ++
 src/vga_buffer.rs | 6 ++++++
 2 files changed, 8 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index c17e2c9..4626f90 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -36,7 +36,9 @@ pub extern "C" fn _start() {
     #[cfg(test)]
     test_main();
 
+    crate::vga_buffer::change_color(crate::vga_buffer::Color::Yellow, crate::vga_buffer::Color::Black);
     println!("It did not crash!");
+    crate::vga_buffer::change_color(crate::vga_buffer::Color::White, crate::vga_buffer::Color::Black);
     loop {
         x86_64::instructions::hlt();
     }
diff --git a/src/vga_buffer.rs b/src/vga_buffer.rs
index 38435bc..0045294 100644
--- a/src/vga_buffer.rs
+++ b/src/vga_buffer.rs
@@ -199,3 +199,9 @@ pub fn move_cursor(x: u16, y: u16) {
     let mut cursor = CURSOR.lock();
     cursor.move_cursor(pos);
 }
+
+pub fn change_color(foreground: Color, background: Color) {
+   let mut writer = WRITER.lock();
+   let color = ColorCode::new(foreground, background);
+   writer.color_code = color;
+}