summaryrefslogtreecommitdiffstats
path: root/src/ui/widgets/tui_dirlist.rs
blob: 8840bda4d4173c76bb86be685f32ee5fff3685e5 (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
101
102
103
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 draw(&mut 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.contents[skip_dist..]
            .iter()
            .enumerate()
            .take(area.height as usize)
        {
            let name = entry.file_name();
            let name_width = name.width();

            let style = if i == screen_index {
                entry.get_style().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(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,
                            );
                        }
                    }
                }
            }
        }
    }
}