1 | use crate::{ |
2 | grid::{ |
3 | config::{ColoredConfig, Entity}, |
4 | dimension::CompleteDimensionVecRecords, |
5 | records::{ExactRecords, PeekableRecords, Records, RecordsMut}, |
6 | util::string::{count_lines, get_lines}, |
7 | }, |
8 | settings::{measurement::Measurement, peaker::Peaker, CellOption, Height, TableOption}, |
9 | }; |
10 | |
11 | use super::table_height_limit::TableHeightLimit; |
12 | |
13 | /// A modification for cell/table to increase its height. |
14 | /// |
15 | /// If used for a [`Table`] [`PriorityNone`] is used. |
16 | /// |
17 | /// [`PriorityNone`]: crate::settings::peaker::PriorityNone |
18 | /// [`Table`]: crate::Table |
19 | #[derive (Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord)] |
20 | pub struct CellHeightLimit<W = usize> { |
21 | height: W, |
22 | } |
23 | |
24 | impl<W> CellHeightLimit<W> { |
25 | /// Constructs a new object. |
26 | pub fn new(height: W) -> Self |
27 | where |
28 | W: Measurement<Height>, |
29 | { |
30 | Self { height } |
31 | } |
32 | |
33 | /// Set's a priority by which the limit logic will be applied. |
34 | pub fn priority<P>(self) -> TableHeightLimit<W, P> |
35 | where |
36 | P: Peaker, |
37 | W: Measurement<Height>, |
38 | { |
39 | TableHeightLimit::new(self.height).priority::<P>() |
40 | } |
41 | } |
42 | |
43 | impl<W, R> CellOption<R, ColoredConfig> for CellHeightLimit<W> |
44 | where |
45 | W: Measurement<Height>, |
46 | R: Records + ExactRecords + PeekableRecords + RecordsMut<String>, |
47 | for<'a> &'a R: Records, |
48 | { |
49 | fn change(self, records: &mut R, cfg: &mut ColoredConfig, entity: Entity) { |
50 | let height = self.height.measure(&*records, cfg); |
51 | |
52 | let count_rows = records.count_rows(); |
53 | let count_columns = records.count_columns(); |
54 | |
55 | for pos in entity.iter(count_rows, count_columns) { |
56 | let is_valid_pos = pos.0 < count_rows && pos.1 < count_columns; |
57 | if !is_valid_pos { |
58 | continue; |
59 | } |
60 | |
61 | let text = records.get_text(pos); |
62 | let count_lines = count_lines(text); |
63 | |
64 | if count_lines <= height { |
65 | continue; |
66 | } |
67 | |
68 | let content = limit_lines(text, height); |
69 | records.set(pos, content); |
70 | } |
71 | } |
72 | } |
73 | |
74 | impl<R, W> TableOption<R, CompleteDimensionVecRecords<'static>, ColoredConfig> |
75 | for CellHeightLimit<W> |
76 | where |
77 | W: Measurement<Height>, |
78 | R: Records + ExactRecords + PeekableRecords + RecordsMut<String>, |
79 | for<'a> &'a R: Records, |
80 | { |
81 | fn change( |
82 | self, |
83 | records: &mut R, |
84 | cfg: &mut ColoredConfig, |
85 | dims: &mut CompleteDimensionVecRecords<'static>, |
86 | ) { |
87 | let height: usize = self.height.measure(&*records, cfg); |
88 | TableHeightLimit::new(height).change(records, cfg, dimension:dims) |
89 | } |
90 | } |
91 | |
92 | fn limit_lines(s: &str, n: usize) -> String { |
93 | let mut text: String = String::new(); |
94 | for (i: usize, line: Cow<'_, str>) in get_lines(text:s).take(n).enumerate() { |
95 | if i > 0 { |
96 | text.push(ch:' \n' ); |
97 | } |
98 | |
99 | text.push_str(&line); |
100 | } |
101 | |
102 | text |
103 | } |
104 | |