T23a: AI chat, messages
- messages: yours in a quiet bubble, the assistant's as prose beside its mark, system notes, errors; avatars, headers with model and time, footers - MessageList over gpui's bottom-aligned list: drawn near the view, held at the newest, counting arrivals while away; a way back down; day separators - streaming text and markdown reveal toward the text as it grows, with a caret; a blinking cursor before the first words - CodeBlock with language, copy, run, apply, download and folding; MarkdownRenderer::code lets messages draw fences with it - pictures, grids, files, voice notes, video over host frames, link previews - quotes, actions, an editor that takes focus, versions, regenerate, stop and continue - errors with retry, a rate limit counting down, feedback, thinking blocks - gallery: the AI Chat page with scripted states
This commit is contained in:
ZacharyZhang-NY committed
1 parent
03f6ceb583
commit
c0afce2507
21 files changed
+3376
-34
No files matched your search
@@ -0,0 +1,192 @@
|
||||
use std::{ops::Range, rc::Rc};
|
||||
|
||||
use gpui::{
|
||||
App, ElementId, HighlightStyle, InteractiveElement, IntoElement, ParentElement, RenderOnce,
|
||||
SharedString, StatefulInteractiveElement, Styled, StyledText, Window, div, prelude::*,
|
||||
relative,
|
||||
};
|
||||
|
||||
use crate::{
|
||||
buttons::{Button, ButtonVariant, CopyButton, IconButton},
|
||||
editor::code_colors,
|
||||
forms::Run,
|
||||
primitives::{Disclosure, IconName},
|
||||
theme::{ActiveTheme, ControlSize, Radius, TextSize},
|
||||
};
|
||||
|
||||
/// Lines a block shows before it folds.
|
||||
const FOLD_AFTER: usize = 14;
|
||||
|
||||
/// Where `code` folds after `lines`: the byte end of the lines kept, when there are more.
|
||||
pub(crate) fn fold_at(code: &str, lines: usize) -> Option<usize> {
|
||||
code.match_indices('\n')
|
||||
.nth(lines.checked_sub(1)?)
|
||||
.map(|(at, _)| at)
|
||||
.filter(|at| code[at + 1..].contains(|c: char| !c.is_whitespace()))
|
||||
}
|
||||
|
||||
/// Code in a message: its language, a copy button, and the owner's download, run and apply; long code folds to a few lines until opened.
|
||||
#[derive(IntoElement)]
|
||||
pub struct CodeBlock {
|
||||
id: ElementId,
|
||||
code: SharedString,
|
||||
language: Option<SharedString>,
|
||||
on_download: Option<Run>,
|
||||
on_run: Option<Run>,
|
||||
on_apply: Option<Run>,
|
||||
}
|
||||
|
||||
impl CodeBlock {
|
||||
pub fn new(id: impl Into<ElementId>, code: impl Into<SharedString>) -> Self {
|
||||
Self {
|
||||
id: id.into(),
|
||||
code: code.into(),
|
||||
language: None,
|
||||
on_download: None,
|
||||
on_run: None,
|
||||
on_apply: None,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn language(mut self, language: impl Into<SharedString>) -> Self {
|
||||
self.language = Some(language.into());
|
||||
self
|
||||
}
|
||||
|
||||
pub fn on_download(mut self, handler: impl Fn(&mut Window, &mut App) + 'static) -> Self {
|
||||
self.on_download = Some(Rc::new(handler));
|
||||
self
|
||||
}
|
||||
|
||||
pub fn on_run(mut self, handler: impl Fn(&mut Window, &mut App) + 'static) -> Self {
|
||||
self.on_run = Some(Rc::new(handler));
|
||||
self
|
||||
}
|
||||
|
||||
/// Puts the code where it belongs, such as into the open file.
|
||||
pub fn on_apply(mut self, handler: impl Fn(&mut Window, &mut App) + 'static) -> Self {
|
||||
self.on_apply = Some(Rc::new(handler));
|
||||
self
|
||||
}
|
||||
}
|
||||
|
||||
impl RenderOnce for CodeBlock {
|
||||
fn render(self, window: &mut Window, cx: &mut App) -> impl IntoElement {
|
||||
let open = window.use_keyed_state((self.id.clone(), "open"), cx, |_, _| false);
|
||||
let opened = *open.read(cx);
|
||||
let theme = cx.theme();
|
||||
let colors = theme.colors.clone();
|
||||
let code = self.code.trim_end_matches('\n').to_string();
|
||||
let fold = fold_at(&code, FOLD_AFTER).filter(|_| !opened);
|
||||
let hidden = fold.map(|at| code[at..].lines().count() - 1);
|
||||
let shown = fold.map_or(code.as_str(), |at| &code[..at]).to_string();
|
||||
let styles: Vec<(Range<usize>, HighlightStyle)> = code_colors(&shown, cx);
|
||||
let action = |key: &str, icon: IconName, tip: &'static str, run: Option<Run>| {
|
||||
run.map(|run| {
|
||||
IconButton::new((self.id.clone(), key.to_string()), icon)
|
||||
.variant(ButtonVariant::Ghost)
|
||||
.size(ControlSize::Sm)
|
||||
.tooltip(tip)
|
||||
.on_click(move |_, window, cx| run(window, cx))
|
||||
})
|
||||
};
|
||||
let lines = code.lines().count();
|
||||
let toggle = open.clone();
|
||||
div()
|
||||
.w_full()
|
||||
.flex()
|
||||
.flex_col()
|
||||
.rounded(theme.radius(Radius::Md))
|
||||
.border_1()
|
||||
.border_color(colors.border)
|
||||
.bg(colors.sunken)
|
||||
.child(
|
||||
div()
|
||||
.flex()
|
||||
.items_center()
|
||||
.gap_1()
|
||||
.pl_3()
|
||||
.pr_1()
|
||||
.py_0p5()
|
||||
.border_b_1()
|
||||
.border_color(colors.border)
|
||||
.child(
|
||||
div()
|
||||
.flex_1()
|
||||
.text_size(theme.text_size(TextSize::Xs))
|
||||
.text_color(colors.fg_subtle)
|
||||
.child(self.language.clone().unwrap_or_else(|| "code".into())),
|
||||
)
|
||||
.children(action("run", IconName::Play, "Run", self.on_run))
|
||||
.children(action("apply", IconName::Check, "Apply", self.on_apply))
|
||||
.children(action(
|
||||
"download",
|
||||
IconName::Download,
|
||||
"Download",
|
||||
self.on_download,
|
||||
))
|
||||
.child(CopyButton::new((self.id.clone(), "copy"), code.clone())),
|
||||
)
|
||||
.child(
|
||||
div()
|
||||
.id((self.id.clone(), "code"))
|
||||
.overflow_x_scroll()
|
||||
.px_3()
|
||||
.py_2()
|
||||
.font_family(theme.mono_family.clone())
|
||||
.text_size(theme.text_size(TextSize::Sm))
|
||||
.line_height(relative(1.55))
|
||||
.text_color(colors.syntax.variable)
|
||||
.whitespace_nowrap()
|
||||
.child(StyledText::new(shown).with_highlights(styles)),
|
||||
)
|
||||
.when(lines > FOLD_AFTER, |block| {
|
||||
block.child(
|
||||
div()
|
||||
.flex()
|
||||
.items_center()
|
||||
.gap_1()
|
||||
.px_1()
|
||||
.py_0p5()
|
||||
.border_t_1()
|
||||
.border_color(colors.border)
|
||||
.child(
|
||||
Button::new(
|
||||
(self.id.clone(), "fold"),
|
||||
match hidden {
|
||||
Some(hidden) => format!("Show {hidden} more lines"),
|
||||
None => "Show less".to_string(),
|
||||
},
|
||||
)
|
||||
.variant(ButtonVariant::Ghost)
|
||||
.size(ControlSize::Sm)
|
||||
.on_click(move |_, _, cx| {
|
||||
toggle.update(cx, |open, cx| {
|
||||
*open = !*open;
|
||||
log::info!("code block: open {open}");
|
||||
cx.notify();
|
||||
})
|
||||
}),
|
||||
)
|
||||
.child(Disclosure::new((self.id.clone(), "chevron"), opened)),
|
||||
)
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::fold_at;
|
||||
|
||||
#[test]
|
||||
fn long_code_folds_after_its_lines() {
|
||||
assert_eq!(fold_at("a\nb\nc", 2), Some(3));
|
||||
assert_eq!(fold_at("a\nb", 2), None, "two lines fit");
|
||||
assert_eq!(
|
||||
fold_at("a\nb\n\n", 2),
|
||||
None,
|
||||
"blank lines past the fold do not count"
|
||||
);
|
||||
assert_eq!(fold_at("a", 0), None);
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user