Skip to main content

vtcode_tui/core_tui/session/render/
mod.rs

1use ratatui::prelude::*;
2use std::borrow::Cow;
3
4use super::{Session, message::MessageLine, message_renderer, text_utils};
5
6mod modal_renderer;
7
8pub(crate) use modal_renderer::floating_modal_area;
9pub(crate) use modal_renderer::modal_render_styles;
10pub use modal_renderer::{render_modal, split_inline_modal_area};
11
12pub(super) fn render_message_spans(session: &Session, index: usize) -> Vec<Span<'static>> {
13    let Some(line) = session.lines.get(index) else {
14        return vec![Span::raw(String::new())];
15    };
16    message_renderer::render_message_spans(
17        line,
18        &session.theme,
19        &session.labels,
20        |kind| session.prefix_text(kind),
21        |line| session.prefix_style(line),
22        |kind| session.text_fallback(kind),
23    )
24}
25
26pub(super) fn agent_prefix_spans(session: &Session, line: &MessageLine) -> Vec<Span<'static>> {
27    let prefix_style = |line: &MessageLine| session.prefix_style(line);
28    message_renderer::agent_prefix_spans(line, &session.theme, &session.labels, &prefix_style)
29}
30
31pub(super) fn strip_ansi_codes(text: &str) -> Cow<'_, str> {
32    text_utils::strip_ansi_codes(text)
33}
34
35pub(super) fn render_tool_segments(session: &Session, line: &MessageLine) -> Vec<Span<'static>> {
36    message_renderer::render_tool_segments(line, &session.theme)
37}
38
39#[allow(dead_code)]
40pub fn render(session: &mut Session, frame: &mut Frame<'_>) {
41    session.render(frame);
42}
43
44fn modal_list_highlight_style(session: &Session) -> Style {
45    session.styles.modal_list_highlight_style()
46}
47
48pub fn apply_view_rows(session: &mut Session, rows: u16) {
49    session.apply_view_rows(rows);
50}
51
52pub fn apply_transcript_rows(session: &mut Session, rows: u16) {
53    session.apply_transcript_rows(rows);
54}
55
56pub fn apply_transcript_width(session: &mut Session, width: u16) {
57    session.apply_transcript_width(width);
58}
59
60pub fn recalculate_transcript_rows(session: &mut Session) {
61    session.recalculate_transcript_rows();
62}