1 files changed, 86 insertions, 0 deletions
diff --git a/vendor/tabled/src/settings/format/format_content.rs b/vendor/tabled/src/settings/format/format_content.rs
new file mode 100644
index 000000000..c14eee64f
--- /dev/null
+++ b/vendor/tabled/src/settings/format/format_content.rs
@@ -0,0 +1,86 @@
+use crate::{
+    grid::config::Entity,
+    grid::records::{ExactRecords, PeekableRecords, Records, RecordsMut},
+    settings::{CellOption, TableOption},
+};
+
+/// A lambda which formats cell content.
+#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
+pub struct FormatContent<F> {
+    f: F,
+    multiline: bool,
+}
+
+impl<F> FormatContent<F> {
+    pub(crate) fn new(f: F) -> Self {
+        Self {
+            f,
+            multiline: false,
+        }
+    }
+}
+
+impl<F> FormatContent<F> {
+    /// Multiline a helper function for changing multiline content of cell.
+    /// Using this formatting applied for all rows not to a string as a whole.
+    ///
+    /// ```rust,no_run
+    /// use tabled::{Table, settings::{Format, object::Segment, Modify}};
+    ///
+    /// let data: Vec<&'static str> = Vec::new();
+    /// let table = Table::new(&data)
+    ///     .with(Modify::new(Segment::all()).with(Format::content(|s| s.to_string()).multiline()))
+    ///     .to_string();
+    /// ```
+    pub fn multiline(mut self) -> Self {
+        self.multiline = true;
+        self
+    }
+}
+
+impl<F, R, D, C> TableOption<R, D, C> for FormatContent<F>
+where
+    F: FnMut(&str) -> String + Clone,
+    R: Records + ExactRecords + PeekableRecords + RecordsMut<String>,
+{
+    fn change(self, records: &mut R, cfg: &mut C, _: &mut D) {
+        CellOption::change(self, records, cfg, Entity::Global);
+    }
+}
+
+impl<F, R, C> CellOption<R, C> for FormatContent<F>
+where
+    F: FnMut(&str) -> String + Clone,
+    R: Records + ExactRecords + PeekableRecords + RecordsMut<String>,
+{
+    fn change(mut self, records: &mut R, _: &mut C, entity: Entity) {
+        let count_rows = records.count_rows();
+        let count_cols = records.count_columns();
+
+        for pos in entity.iter(count_rows, count_cols) {
+            let is_valid_pos = pos.0 < count_rows && pos.1 < count_cols;
+            if !is_valid_pos {
+                continue;
+            }
+
+            let content = records.get_text(pos);
+            let content = if self.multiline {
+                multiline(self.f.clone())(content)
+            } else {
+                (self.f)(content)
+            };
+            records.set(pos, content);
+        }
+    }
+}
+
+fn multiline<F: FnMut(&str) -> String>(mut f: F) -> impl FnMut(&str) -> String {
+    move |s: &str| {
+        let mut v = Vec::new();
+        for line in s.lines() {
+            v.push(f(line));
+        }
+
+        v.join("\n")
+    }
+}