Make epilogue a provided method

It should be the same for all colors, but can be overwritten this way.
rust-osdev · Aug 17, 2020 · ed81de8 · ed81de8
1 parent 67c2cf8
commit ed81de8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/color.rs b/src/color.rs
@@ -3,5 +3,7 @@ use core::fmt;
 
 pub trait Color {
     fn prelude(&self, f: &mut fmt::Formatter, canvas: Canvas) -> fmt::Result;
-    fn epilogue(&self, f: &mut fmt::Formatter, canvas: Canvas) -> fmt::Result;
+    fn epilogue(&self, f: &mut core::fmt::Formatter, _canvas: crate::Canvas) -> core::fmt::Result {
+        f.write_str("\x1B[0m")
+    }
 }
diff --git a/src/colors.rs b/src/colors.rs
@@ -9,9 +9,6 @@ impl Color for RGB8 {
             Canvas::Background => write!(f, "\x1B[48;2;{};{};{}m", self.r, self.g, self.b),
         }
     }
-    fn epilogue(&self, f: &mut core::fmt::Formatter, _canvas: crate::Canvas) -> core::fmt::Result {
-        write!(f, "\x1B[0m")
-    }
 }
 
 /// Makes <code style="color: black; background: #FFFFFF">white</code>