summaryrefslogtreecommitdiffstats
path: root/src/ui/widgets/tui_dirlist.rs
blob: 32ded90cf6e38c1e9aec026262c92160360ec602 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
use tui::buffer::Buffer;
use tui::layout::Rect;
use tui::style::{Color, Modifier, Style};
use tui::widgets::Widget;
use unicode_width::UnicodeWidthStr;

use crate::fs::JoshutoDirList;

pub struct TuiDirList<'a> {
    dirlist: &'a JoshutoDirList,
}

impl<'a> TuiDirList<'a> {
    pub fn new(dirlist: &'a JoshutoDirList) -> Self {
        Self { dirlist }
    }
}

impl<'a> Widget for TuiDirList<'a> {
    fn render(self, area: Rect, buf: &mut Buffer) {
        if area.width < 1 || area.height < 1 {
            return;
        }
        if area.width < 4 {
            return;
        }

        let x = area.left();
        let y = area.top();

        let dir_len = self.dirlist.contents.len();
        if dir_len == 0 {
            let style = Style::default().bg(Color::Red).fg(Color::White);
            buf.set_stringn(x, y, "empty", area.width as usize, style);
            return;
        }

        let curr_index = self.dirlist.index.unwrap();
        let skip_dist = curr_index / area.height as usize * area.height as usize;

        let screen_index = if skip_dist > 0 {
            curr_index % skip_dist
        } else {
            curr_index
        };

        let area_width = area.width as usize - 1;
        for (i, entry) in self
            .dirlist
            .iter()
            .skip(skip_dist)
            .enumerate()
            .take(area.height as usize)
        {
            let name = entry.label();
            let name_width = name.width();

            let style = if i == screen_index {
                entry.get_style().add_modifier(Modifier::REVERSED)
            } else {
                entry.get_style()
            };

            let file_type = &entry.metadata.file_type;
            if file_type.is_dir() {
                if name_width <= area_width {
                    buf.set_stringn(x, y + i as u16, name, area_width, style);
                } else {
                    buf.set_stringn(x, y + i as u16, name, area_width - 1, style);
                    buf.set_string(x + area_width as u16 - 1, y + i as u16, "…", style);
                }
            } else if name_width < area_width {
                buf.set_stringn(x, y + i as u16, name, area_width, style);
            } else {
                match name.rfind('.') {
                    None => {
                        buf.set_stringn(x, y + i as u16, name, area_width, style);
                    }
                    Some(0) => {
                        let file_name_width = area_width;
                        buf.set_stringn(x, y + i as u16, &name, file_name_width, style);
                    }
                    Some(p_ind) => {
                        let ext_width = name[p_ind..].width();
                        let file_name_width = area_width - ext_width - 1;

                        buf.set_stringn(x, y + i as u16, &name[..p_ind], file_name_width, style);
                        buf.set_string(x + file_name_width as u16, y + i as u16, "…", style);
                        buf.set_string(
                            x + file_name_width as u16 + 1,
                            y + i as u16,
                            &name[p_ind..],
                            style,
                        );
                    }
                }
            }
        }
    }
}