codecache 0.1.1

A terminal-based code snippet manager with syntax highlighting.
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141

use ratatui::{
    layout::Constraint::{Fill, Length, Min},
    prelude::*,
    widgets::{Block, Paragraph},
};
use serde::{Deserialize, Serialize};
use tui_widget_list::{ListBuilder, ListState, ListView};

use crate::app::{highlight::Highlighter, language::get_lang};

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SaveSnippet {
    pub title: String,
    pub desc: String,
    pub code: String,
    pub lang: String,
}

#[derive(Debug, Clone)]
pub struct CodeSnippet {
    title: String,
    text: String,
    code: String,
    text_style: Style,
    code_style: Style,
    code_frame_style: Style,
    border_style: Style,
    highlighted_code: Option<Text<'static>>,
    lang: String,
}

pub struct SnippetList<'a> {
    pub state: &'a mut ListState,
    pub items: Vec<CodeSnippet>,
    pub highlighter: &'a Highlighter,
}

impl CodeSnippet {
    pub fn new<T: Into<String>>(title: T, text: T, code: T, lang: T) -> Self {
        Self {
            text: text.into(),
            code: code.into(),
            text_style: Style::default(),
            border_style: Style::default(),
            code_style: Style::default(),
            code_frame_style: Style::default(),
            title: title.into(),
            highlighted_code: None,
            lang: lang.into(),
        }
    }

    pub fn height(&self) -> u16 {
        let description_lines = (self.text.lines().count() as u16).max(1);
        let code_lines = (self.code.lines().count() as u16).max(1);
        let border_height = 4;
        description_lines + code_lines + border_height
    }
}

impl Widget for CodeSnippet {
    fn render(self, area: Rect, buf: &mut Buffer) {
        // Center the card horizontally
        let [_, block_area, _] = Layout::horizontal([Fill(1), Min(70), Fill(1)]).areas(area);

        // Outer block with title
        let block = Block::bordered()
            .title(self.title)
            .title_alignment(Alignment::Center)
            .border_style(self.border_style);

        let inner_area = block.inner(block_area);
        block.render(block_area, buf);

        let desc_lines = self.text.lines().count().max(1) as u16;
        let code_lines = self.code.lines().count().max(1) as u16 + 2;

        let [desc_area, code_area] =
            Layout::vertical([Length(desc_lines), Length(code_lines)]).areas(inner_area);

        Paragraph::new(self.text)
            .style(self.text_style)
            .render(desc_area, buf);

        let code_block = Block::bordered()
            .border_style(self.code_frame_style)
            .title(get_lang(&self.lang))
            .title_alignment(Alignment::Center);

        let code_inner = code_block.inner(code_area);
        code_block.render(code_area, buf);

        let code_text = match self.highlighted_code {
            Some(highlighted) => highlighted,
            None => Text::raw(&self.code),
        };

        Paragraph::new(code_text)
            .style(self.code_style)
            .render(code_inner, buf);
    }
}

impl<'a> ratatui::prelude::Widget for SnippetList<'a> {
    fn render(self, area: Rect, buf: &mut Buffer) {
        let items = self.items;
        let item_count = items.len();
        let highlighter = self.highlighter;

        let builder = ListBuilder::new(move |context| {
            let mut item = items[context.index].clone();

            // TODO: implement error message later
            item.highlighted_code = match highlighter.highlight(&item.code, &item.lang) {
                Ok(s) => Some(s),
                Err(s) => Some(s),
            };

            item.text_style = Style::default().fg(Color::Rgb(120, 112, 108));
            item.border_style = Style::default().fg(Color::Rgb(124, 111, 100));
            item.code_frame_style = Style::default().fg(Color::Rgb(124, 111, 100));

            if context.is_selected {
                item.text_style = Style::default()
                    .bg(Color::Rgb(254, 128, 25))
                    .fg(Color::Rgb(28, 28, 32));
                item.border_style = Style::default().fg(Color::Rgb(250, 189, 47));
                item.code_frame_style = Style::default().fg(Color::Rgb(180, 119, 0));
            }

            let item_height = item.height();

            (item, item_height)
        });

        let list = ListView::new(builder, item_count)
            .scroll_padding(4)
            .infinite_scrolling(false);
        list.render(area, buf, self.state);
    }
}