use std::ops::Deref;
use std::ops::DerefMut;
pub struct StringBuffer(Vec<Vec<char>>);
impl Deref for StringBuffer {
type Target = Vec<Vec<char>>;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl DerefMut for StringBuffer {
fn deref_mut(&mut self) -> &mut Self::Target {
&mut self.0
}
}
impl StringBuffer {
pub(crate) fn new() -> Self {
StringBuffer(vec![])
}
fn add_rows(&mut self, n: i32) {
for _i in 0..n {
self.push(vec![]);
}
}
fn add_column(&mut self, row: usize, n: i32) {
for _i in 0..n {
self[row].push(' ');
}
}
pub(crate) fn add_char(&mut self, x: i32, y: i32, ch: char) {
if x >= 0 && y >= 0 {
let row_index = y as usize;
let column_index = x as usize;
let row_diff = y - self.len() as i32;
if row_diff >= 0 {
self.add_rows(row_diff + 1);
}
let column = &self[y as usize];
let column_diff = x - column.len() as i32;
if column_diff >= 0 {
self.add_column(row_index as usize, column_diff + 1);
}
self[row_index][column_index] = ch;
}
}
}
impl ToString for StringBuffer {
fn to_string(&self) -> String {
let mut lines = vec![];
for row in self.iter() {
let row_contents: Vec<String> = row
.iter()
.filter(|ch| **ch != '\0')
.map(ToString::to_string)
.collect();
let line = row_contents.join("").trim_end().to_string();
lines.push(line);
}
lines.join("\n")
}
}