guise-ui 1.5.3

//! `AISources` — the list a reply's citations point at.
//!
//! Citations are only checkable if the reader can get from the number to the
//! thing. This is that list: numbered to match, each row clickable, each
//! carrying the excerpt the model actually used when the provider returned
//! one — because "it says so on this page" and "it says so in this sentence"
//! are different claims.

use std::rc::Rc;

use gpui::prelude::*;
use gpui::{div, px, App, IntoElement, SharedString, Window};

use super::AISource;
use crate::devtools::Probed;
use crate::icon::{Icon, IconName};
use crate::theme::{theme, ColorName, Size};

/// Called with the 0-based index of the source that was clicked.
type SourceHandler = Rc<dyn Fn(usize, &mut Window, &mut App) + 'static>;

/// A numbered list of sources.
#[derive(IntoElement)]
pub struct AISources {
  sources: Vec<AISource>,
  title: Option<SharedString>,
  size: Size,
  /// Show each source's excerpt under its title.
  excerpts: bool,
  on_open: Option<SourceHandler>,
}

impl AISources {
  pub fn new(sources: impl IntoIterator<Item = AISource>) -> Self {
    AISources {
      sources: sources.into_iter().collect(),
      title: None,
      size: Size::Xs,
      excerpts: false,
      on_open: None,
    }
  }

  /// A heading above the list. Defaults to none, since a reply that ends in
  /// a numbered list of links rarely needs one.
  pub fn title(mut self, title: impl Into<SharedString>) -> Self {
    self.title = Some(title.into());
    self
  }

  pub fn size(mut self, size: Size) -> Self {
    self.size = size;
    self
  }

  /// Show the passage each source contributed, where there is one.
  pub fn excerpts(mut self, excerpts: bool) -> Self {
    self.excerpts = excerpts;
    self
  }

  /// Handle a click on a row. Without one the list is not interactive —
  /// guise has no opinion on how a host opens a URL.
  pub fn on_open(mut self, handler: impl Fn(usize, &mut Window, &mut App) + 'static) -> Self {
    self.on_open = Some(Rc::new(handler));
    self
  }
}

impl RenderOnce for AISources {
  fn render(self, _window: &mut Window, cx: &mut App) -> impl IntoElement {
    let t = theme(cx);
    let font = t.font_size(self.size);
    let dimmed = t.dimmed().hsla();
    let text_color = t.text().hsla();
    let border = t.border().hsla();
    let surface = t.surface_hover().alpha(0.5);
    let radius = t.radius(Size::Sm);

    let mut list = div().flex().flex_col().gap(px(4.0)).w_full();
    if let Some(title) = self.title {
      list = list.child(
        div()
          .flex()
          .items_center()
          .gap(px(6.0))
          .text_size(px(font))
          .text_color(dimmed)
          .child(
            Icon::new(IconName::BookOpen)
              .size(Size::Xs)
              .color(ColorName::Gray),
          )
          .child(title),
      );
    }

    for (index, source) in self.sources.into_iter().enumerate() {
      let handler = self.on_open.clone();
      list = list.child(
        div()
          .id(("guise-ai-source", index))
          .flex()
          .flex_row()
          .items_start()
          .gap(px(8.0))
          .w_full()
          .px(px(8.0))
          .py(px(6.0))
          .rounded(px(radius))
          .border_1()
          .border_color(border)
          .text_size(px(font))
          .when(handler.is_some(), |row| {
            row.cursor_pointer().hover(move |row| row.bg(surface))
          })
          .child(
            div()
              .flex_none()
              .min_w(px(font * 1.4))
              .text_color(dimmed)
              .child(SharedString::from((index + 1).to_string())),
          )
          .child(
            div()
              .flex()
              .flex_col()
              .gap(px(1.0))
              .flex_1()
              .min_w(px(0.0))
              .child(
                div()
                  .text_color(text_color)
                  .child(SharedString::from(source.title)),
              )
              .child(
                div()
                  .text_color(dimmed)
                  .child(SharedString::from(source.location)),
              )
              .when_some(
                source.excerpt.filter(|_| self.excerpts),
                |column, excerpt| {
                  column.child(
                    div()
                      .mt(px(4.0))
                      .pl(px(8.0))
                      .border_l_2()
                      .border_color(border)
                      .text_color(dimmed)
                      .child(SharedString::from(excerpt)),
                  )
                },
              ),
          )
          .when_some(handler, |row, handler| {
            row.on_click(move |_event, window, cx| handler(index, window, cx))
          }),
      );
    }
    list.probe("AISources")
  }
}