chat_panel.rs

   1use crate::{collab_panel, ChatPanelSettings};
   2use anyhow::Result;
   3use call::{room, ActiveCall};
   4use channel::{ChannelChat, ChannelChatEvent, ChannelMessage, ChannelMessageId, ChannelStore};
   5use client::{ChannelId, Client};
   6use collections::HashMap;
   7use db::kvp::KEY_VALUE_STORE;
   8use editor::Editor;
   9use gpui::{
  10    actions, div, list, prelude::*, px, Action, AppContext, AsyncWindowContext, ClipboardItem,
  11    CursorStyle, DismissEvent, ElementId, EventEmitter, FocusHandle, FocusableView, FontStyle,
  12    FontWeight, HighlightStyle, ListOffset, ListScrollEvent, ListState, Model, Render, StyledText,
  13    Subscription, Task, View, ViewContext, VisualContext, WeakView,
  14};
  15use language::LanguageRegistry;
  16use menu::Confirm;
  17use message_editor::MessageEditor;
  18use project::Fs;
  19use rich_text::RichText;
  20use serde::{Deserialize, Serialize};
  21use settings::Settings;
  22use std::{sync::Arc, time::Duration};
  23use time::{OffsetDateTime, UtcOffset};
  24use ui::{
  25    popover_menu, prelude::*, Avatar, Button, ContextMenu, IconButton, IconName, KeyBinding, Label,
  26    TabBar, Tooltip,
  27};
  28use util::{ResultExt, TryFutureExt};
  29use workspace::{
  30    dock::{DockPosition, Panel, PanelEvent},
  31    Workspace,
  32};
  33
  34mod message_editor;
  35
  36const MESSAGE_LOADING_THRESHOLD: usize = 50;
  37const CHAT_PANEL_KEY: &str = "ChatPanel";
  38
  39pub fn init(cx: &mut AppContext) {
  40    cx.observe_new_views(|workspace: &mut Workspace, _| {
  41        workspace.register_action(|workspace, _: &ToggleFocus, cx| {
  42            workspace.toggle_panel_focus::<ChatPanel>(cx);
  43        });
  44    })
  45    .detach();
  46}
  47
  48pub struct ChatPanel {
  49    client: Arc<Client>,
  50    channel_store: Model<ChannelStore>,
  51    languages: Arc<LanguageRegistry>,
  52    message_list: ListState,
  53    active_chat: Option<(Model<ChannelChat>, Subscription)>,
  54    message_editor: View<MessageEditor>,
  55    local_timezone: UtcOffset,
  56    fs: Arc<dyn Fs>,
  57    width: Option<Pixels>,
  58    active: bool,
  59    pending_serialization: Task<Option<()>>,
  60    subscriptions: Vec<gpui::Subscription>,
  61    is_scrolled_to_bottom: bool,
  62    markdown_data: HashMap<ChannelMessageId, RichText>,
  63    focus_handle: FocusHandle,
  64    open_context_menu: Option<(u64, Subscription)>,
  65    highlighted_message: Option<(u64, Task<()>)>,
  66    last_acknowledged_message_id: Option<u64>,
  67}
  68
  69#[derive(Serialize, Deserialize)]
  70struct SerializedChatPanel {
  71    width: Option<Pixels>,
  72}
  73
  74actions!(chat_panel, [ToggleFocus, CloseReplyPreview]);
  75
  76impl ChatPanel {
  77    pub fn new(workspace: &mut Workspace, cx: &mut ViewContext<Workspace>) -> View<Self> {
  78        let fs = workspace.app_state().fs.clone();
  79        let client = workspace.app_state().client.clone();
  80        let channel_store = ChannelStore::global(cx);
  81        let languages = workspace.app_state().languages.clone();
  82
  83        let input_editor = cx.new_view(|cx| {
  84            MessageEditor::new(
  85                languages.clone(),
  86                channel_store.clone(),
  87                cx.new_view(|cx| Editor::auto_height(4, cx)),
  88                cx,
  89            )
  90        });
  91
  92        cx.new_view(|cx: &mut ViewContext<Self>| {
  93            let view = cx.view().downgrade();
  94            let message_list =
  95                ListState::new(0, gpui::ListAlignment::Bottom, px(1000.), move |ix, cx| {
  96                    if let Some(view) = view.upgrade() {
  97                        view.update(cx, |view, cx| {
  98                            view.render_message(ix, cx).into_any_element()
  99                        })
 100                    } else {
 101                        div().into_any()
 102                    }
 103                });
 104
 105            message_list.set_scroll_handler(cx.listener(|this, event: &ListScrollEvent, cx| {
 106                if event.visible_range.start < MESSAGE_LOADING_THRESHOLD {
 107                    this.load_more_messages(cx);
 108                }
 109                this.is_scrolled_to_bottom = !event.is_scrolled;
 110            }));
 111
 112            let mut this = Self {
 113                fs,
 114                client,
 115                channel_store,
 116                languages,
 117                message_list,
 118                active_chat: Default::default(),
 119                pending_serialization: Task::ready(None),
 120                message_editor: input_editor,
 121                local_timezone: cx.local_timezone(),
 122                subscriptions: Vec::new(),
 123                is_scrolled_to_bottom: true,
 124                active: false,
 125                width: None,
 126                markdown_data: Default::default(),
 127                focus_handle: cx.focus_handle(),
 128                open_context_menu: None,
 129                highlighted_message: None,
 130                last_acknowledged_message_id: None,
 131            };
 132
 133            if let Some(channel_id) = ActiveCall::global(cx)
 134                .read(cx)
 135                .room()
 136                .and_then(|room| room.read(cx).channel_id())
 137            {
 138                this.select_channel(channel_id, None, cx)
 139                    .detach_and_log_err(cx);
 140            }
 141
 142            this.subscriptions.push(cx.subscribe(
 143                &ActiveCall::global(cx),
 144                move |this: &mut Self, call, event: &room::Event, cx| match event {
 145                    room::Event::RoomJoined { channel_id } => {
 146                        if let Some(channel_id) = channel_id {
 147                            this.select_channel(*channel_id, None, cx)
 148                                .detach_and_log_err(cx);
 149
 150                            if call
 151                                .read(cx)
 152                                .room()
 153                                .is_some_and(|room| room.read(cx).contains_guests())
 154                            {
 155                                cx.emit(PanelEvent::Activate)
 156                            }
 157                        }
 158                    }
 159                    room::Event::Left { channel_id } => {
 160                        if channel_id == &this.channel_id(cx) {
 161                            cx.emit(PanelEvent::Close)
 162                        }
 163                    }
 164                    _ => {}
 165                },
 166            ));
 167
 168            this
 169        })
 170    }
 171
 172    pub fn channel_id(&self, cx: &AppContext) -> Option<ChannelId> {
 173        self.active_chat
 174            .as_ref()
 175            .map(|(chat, _)| chat.read(cx).channel_id)
 176    }
 177
 178    pub fn is_scrolled_to_bottom(&self) -> bool {
 179        self.is_scrolled_to_bottom
 180    }
 181
 182    pub fn active_chat(&self) -> Option<Model<ChannelChat>> {
 183        self.active_chat.as_ref().map(|(chat, _)| chat.clone())
 184    }
 185
 186    pub fn load(
 187        workspace: WeakView<Workspace>,
 188        cx: AsyncWindowContext,
 189    ) -> Task<Result<View<Self>>> {
 190        cx.spawn(|mut cx| async move {
 191            let serialized_panel = if let Some(panel) = cx
 192                .background_executor()
 193                .spawn(async move { KEY_VALUE_STORE.read_kvp(CHAT_PANEL_KEY) })
 194                .await
 195                .log_err()
 196                .flatten()
 197            {
 198                Some(serde_json::from_str::<SerializedChatPanel>(&panel)?)
 199            } else {
 200                None
 201            };
 202
 203            workspace.update(&mut cx, |workspace, cx| {
 204                let panel = Self::new(workspace, cx);
 205                if let Some(serialized_panel) = serialized_panel {
 206                    panel.update(cx, |panel, cx| {
 207                        panel.width = serialized_panel.width.map(|r| r.round());
 208                        cx.notify();
 209                    });
 210                }
 211                panel
 212            })
 213        })
 214    }
 215
 216    fn serialize(&mut self, cx: &mut ViewContext<Self>) {
 217        let width = self.width;
 218        self.pending_serialization = cx.background_executor().spawn(
 219            async move {
 220                KEY_VALUE_STORE
 221                    .write_kvp(
 222                        CHAT_PANEL_KEY.into(),
 223                        serde_json::to_string(&SerializedChatPanel { width })?,
 224                    )
 225                    .await?;
 226                anyhow::Ok(())
 227            }
 228            .log_err(),
 229        );
 230    }
 231
 232    fn set_active_chat(&mut self, chat: Model<ChannelChat>, cx: &mut ViewContext<Self>) {
 233        if self.active_chat.as_ref().map(|e| &e.0) != Some(&chat) {
 234            let channel_id = chat.read(cx).channel_id;
 235            {
 236                self.markdown_data.clear();
 237                let chat = chat.read(cx);
 238                self.message_list.reset(chat.message_count());
 239
 240                let channel_name = chat.channel(cx).map(|channel| channel.name.clone());
 241                self.message_editor.update(cx, |editor, cx| {
 242                    editor.set_channel(channel_id, channel_name, cx);
 243                    editor.clear_reply_to_message_id();
 244                });
 245            };
 246            let subscription = cx.subscribe(&chat, Self::channel_did_change);
 247            self.active_chat = Some((chat, subscription));
 248            self.acknowledge_last_message(cx);
 249            cx.notify();
 250        }
 251    }
 252
 253    fn channel_did_change(
 254        &mut self,
 255        _: Model<ChannelChat>,
 256        event: &ChannelChatEvent,
 257        cx: &mut ViewContext<Self>,
 258    ) {
 259        match event {
 260            ChannelChatEvent::MessagesUpdated {
 261                old_range,
 262                new_count,
 263            } => {
 264                self.message_list.splice(old_range.clone(), *new_count);
 265                if self.active {
 266                    self.acknowledge_last_message(cx);
 267                }
 268            }
 269            ChannelChatEvent::NewMessage {
 270                channel_id,
 271                message_id,
 272            } => {
 273                if !self.active {
 274                    self.channel_store.update(cx, |store, cx| {
 275                        store.update_latest_message_id(*channel_id, *message_id, cx)
 276                    })
 277                }
 278            }
 279        }
 280        cx.notify();
 281    }
 282
 283    fn acknowledge_last_message(&mut self, cx: &mut ViewContext<Self>) {
 284        if self.active && self.is_scrolled_to_bottom {
 285            if let Some((chat, _)) = &self.active_chat {
 286                if let Some(channel_id) = self.channel_id(cx) {
 287                    self.last_acknowledged_message_id = self
 288                        .channel_store
 289                        .read(cx)
 290                        .last_acknowledge_message_id(channel_id);
 291                }
 292
 293                chat.update(cx, |chat, cx| {
 294                    chat.acknowledge_last_message(cx);
 295                });
 296            }
 297        }
 298    }
 299
 300    fn render_replied_to_message(
 301        &mut self,
 302        message_id: Option<ChannelMessageId>,
 303        reply_to_message: &ChannelMessage,
 304        cx: &mut ViewContext<Self>,
 305    ) -> impl IntoElement {
 306        let body_element_id: ElementId = match message_id {
 307            Some(ChannelMessageId::Saved(id)) => ("reply-to-saved-message", id).into(),
 308            Some(ChannelMessageId::Pending(id)) => ("reply-to-pending-message", id).into(), // This should never happen
 309            None => ("composing-reply").into(),
 310        };
 311
 312        let message_element_id: ElementId = match message_id {
 313            Some(ChannelMessageId::Saved(id)) => ("reply-to-saved-message-container", id).into(),
 314            Some(ChannelMessageId::Pending(id)) => {
 315                ("reply-to-pending-message-container", id).into()
 316            } // This should never happen
 317            None => ("composing-reply-container").into(),
 318        };
 319
 320        let current_channel_id = self.channel_id(cx);
 321        let reply_to_message_id = reply_to_message.id;
 322
 323        let reply_to_message_body = self
 324            .markdown_data
 325            .entry(reply_to_message.id)
 326            .or_insert_with(|| {
 327                Self::render_markdown_with_mentions(
 328                    &self.languages,
 329                    self.client.id(),
 330                    reply_to_message,
 331                )
 332            });
 333
 334        const REPLY_TO_PREFIX: &str = "Reply to @";
 335
 336        div().flex_grow().child(
 337            v_flex()
 338                .id(message_element_id)
 339                .text_ui_xs()
 340                .child(
 341                    h_flex()
 342                        .gap_x_1()
 343                        .items_center()
 344                        .justify_start()
 345                        .overflow_x_hidden()
 346                        .whitespace_nowrap()
 347                        .child(
 348                            StyledText::new(format!(
 349                                "{}{}",
 350                                REPLY_TO_PREFIX,
 351                                reply_to_message.sender.github_login.clone()
 352                            ))
 353                            .with_highlights(
 354                                &cx.text_style(),
 355                                vec![(
 356                                    (REPLY_TO_PREFIX.len() - 1)
 357                                        ..(reply_to_message.sender.github_login.len()
 358                                            + REPLY_TO_PREFIX.len()),
 359                                    HighlightStyle {
 360                                        font_weight: Some(FontWeight::BOLD),
 361                                        ..Default::default()
 362                                    },
 363                                )],
 364                            ),
 365                        ),
 366                )
 367                .child(
 368                    div()
 369                        .border_l_2()
 370                        .border_color(cx.theme().colors().border)
 371                        .px_1()
 372                        .py_0p5()
 373                        .mb_1()
 374                        .child(
 375                            div()
 376                                .overflow_hidden()
 377                                .max_h_12()
 378                                .child(reply_to_message_body.element(body_element_id, cx)),
 379                        ),
 380                )
 381                .cursor(CursorStyle::PointingHand)
 382                .tooltip(|cx| Tooltip::text("Go to message", cx))
 383                .on_click(cx.listener(move |chat_panel, _, cx| {
 384                    if let Some(channel_id) = current_channel_id {
 385                        chat_panel
 386                            .select_channel(channel_id, reply_to_message_id.into(), cx)
 387                            .detach_and_log_err(cx)
 388                    }
 389                })),
 390        )
 391    }
 392
 393    fn render_message(&mut self, ix: usize, cx: &mut ViewContext<Self>) -> impl IntoElement {
 394        let active_chat = &self.active_chat.as_ref().unwrap().0;
 395        let (message, is_continuation_from_previous, is_admin) =
 396            active_chat.update(cx, |active_chat, cx| {
 397                let is_admin = self
 398                    .channel_store
 399                    .read(cx)
 400                    .is_channel_admin(active_chat.channel_id);
 401
 402                let last_message = active_chat.message(ix.saturating_sub(1));
 403                let this_message = active_chat.message(ix).clone();
 404
 405                let duration_since_last_message = this_message.timestamp - last_message.timestamp;
 406                let is_continuation_from_previous = last_message.sender.id
 407                    == this_message.sender.id
 408                    && last_message.id != this_message.id
 409                    && duration_since_last_message < Duration::from_secs(5 * 60);
 410
 411                if let ChannelMessageId::Saved(id) = this_message.id {
 412                    if this_message
 413                        .mentions
 414                        .iter()
 415                        .any(|(_, user_id)| Some(*user_id) == self.client.user_id())
 416                    {
 417                        active_chat.acknowledge_message(id);
 418                    }
 419                }
 420
 421                (this_message, is_continuation_from_previous, is_admin)
 422            });
 423
 424        let _is_pending = message.is_pending();
 425
 426        let belongs_to_user = Some(message.sender.id) == self.client.user_id();
 427        let can_delete_message = belongs_to_user || is_admin;
 428
 429        let element_id: ElementId = match message.id {
 430            ChannelMessageId::Saved(id) => ("saved-message", id).into(),
 431            ChannelMessageId::Pending(id) => ("pending-message", id).into(),
 432        };
 433
 434        let mentioning_you = message
 435            .mentions
 436            .iter()
 437            .any(|m| Some(m.1) == self.client.user_id());
 438
 439        let message_id = match message.id {
 440            ChannelMessageId::Saved(id) => Some(id),
 441            ChannelMessageId::Pending(_) => None,
 442        };
 443
 444        let reply_to_message = message
 445            .reply_to_message_id
 446            .and_then(|id| active_chat.read(cx).find_loaded_message(id))
 447            .cloned();
 448
 449        let replied_to_you =
 450            reply_to_message.as_ref().map(|m| m.sender.id) == self.client.user_id();
 451
 452        let is_highlighted_message = self
 453            .highlighted_message
 454            .as_ref()
 455            .is_some_and(|(id, _)| Some(id) == message_id.as_ref());
 456        let background = if is_highlighted_message {
 457            cx.theme().status().info_background
 458        } else if mentioning_you || replied_to_you {
 459            cx.theme().colors().background
 460        } else {
 461            cx.theme().colors().panel_background
 462        };
 463
 464        v_flex()
 465            .w_full()
 466            .relative()
 467            .group("")
 468            .when(!is_continuation_from_previous, |this| this.pt_2())
 469            .child(
 470                div()
 471                    .group("")
 472                    .bg(background)
 473                    .rounded_md()
 474                    .overflow_hidden()
 475                    .px_1p5()
 476                    .py_0p5()
 477                    .when(!self.has_open_menu(message_id), |this| {
 478                        this.hover(|style| style.bg(cx.theme().colors().element_hover))
 479                    })
 480                    .when(!is_continuation_from_previous, |this| {
 481                        this.child(
 482                            h_flex()
 483                                .text_ui_sm()
 484                                .child(div().absolute().child(
 485                                    Avatar::new(message.sender.avatar_uri.clone()).size(rems(1.)),
 486                                ))
 487                                .child(
 488                                    div()
 489                                        .pl(cx.rem_size() + px(6.0))
 490                                        .pr(px(8.0))
 491                                        .font_weight(FontWeight::BOLD)
 492                                        .child(Label::new(message.sender.github_login.clone())),
 493                                )
 494                                .child(
 495                                    Label::new(time_format::format_localized_timestamp(
 496                                        OffsetDateTime::now_utc(),
 497                                        message.timestamp,
 498                                        self.local_timezone,
 499                                    ))
 500                                    .size(LabelSize::Small)
 501                                    .color(Color::Muted),
 502                                ),
 503                        )
 504                    })
 505                    .when(
 506                        message.reply_to_message_id.is_some() && reply_to_message.is_none(),
 507                        |this| {
 508                            const MESSAGE_DELETED: &str = "Message has been deleted";
 509
 510                            let body_text = StyledText::new(MESSAGE_DELETED).with_highlights(
 511                                &cx.text_style(),
 512                                vec![(
 513                                    0..MESSAGE_DELETED.len(),
 514                                    HighlightStyle {
 515                                        font_style: Some(FontStyle::Italic),
 516                                        ..Default::default()
 517                                    },
 518                                )],
 519                            );
 520
 521                            this.child(
 522                                div()
 523                                    .border_l_2()
 524                                    .text_ui_xs()
 525                                    .border_color(cx.theme().colors().border)
 526                                    .px_1()
 527                                    .py_0p5()
 528                                    .child(body_text),
 529                            )
 530                        },
 531                    )
 532                    .when_some(reply_to_message, |el, reply_to_message| {
 533                        el.child(self.render_replied_to_message(
 534                            Some(message.id),
 535                            &reply_to_message,
 536                            cx,
 537                        ))
 538                    })
 539                    .when(mentioning_you || replied_to_you, |this| this.my_0p5())
 540                    .map(|el| {
 541                        let text = self.markdown_data.entry(message.id).or_insert_with(|| {
 542                            Self::render_markdown_with_mentions(
 543                                &self.languages,
 544                                self.client.id(),
 545                                &message,
 546                            )
 547                        });
 548                        el.child(
 549                            v_flex()
 550                                .w_full()
 551                                .text_ui_sm()
 552                                .id(element_id)
 553                                .child(text.element("body".into(), cx)),
 554                        )
 555                        .when(self.has_open_menu(message_id), |el| {
 556                            el.bg(cx.theme().colors().element_selected)
 557                        })
 558                    }),
 559            )
 560            .when(
 561                self.last_acknowledged_message_id
 562                    .is_some_and(|l| Some(l) == message_id),
 563                |this| {
 564                    this.child(
 565                        h_flex()
 566                            .py_2()
 567                            .gap_1()
 568                            .items_center()
 569                            .child(div().w_full().h_0p5().bg(cx.theme().colors().border))
 570                            .child(
 571                                div()
 572                                    .px_1()
 573                                    .rounded_md()
 574                                    .text_ui_xs()
 575                                    .bg(cx.theme().colors().background)
 576                                    .child("New messages"),
 577                            )
 578                            .child(div().w_full().h_0p5().bg(cx.theme().colors().border)),
 579                    )
 580                },
 581            )
 582            .child(
 583                self.render_popover_buttons(&cx, message_id, can_delete_message)
 584                    .neg_mt_2p5(),
 585            )
 586    }
 587
 588    fn has_open_menu(&self, message_id: Option<u64>) -> bool {
 589        match self.open_context_menu.as_ref() {
 590            Some((id, _)) => Some(*id) == message_id,
 591            None => false,
 592        }
 593    }
 594
 595    fn render_popover_buttons(
 596        &self,
 597        cx: &ViewContext<Self>,
 598        message_id: Option<u64>,
 599        can_delete_message: bool,
 600    ) -> Div {
 601        div()
 602            .absolute()
 603            .child(
 604                div()
 605                    .absolute()
 606                    .right_8()
 607                    .w_6()
 608                    .rounded_tl_md()
 609                    .rounded_bl_md()
 610                    .border_l_1()
 611                    .border_t_1()
 612                    .border_b_1()
 613                    .border_color(cx.theme().colors().element_selected)
 614                    .bg(cx.theme().colors().element_background)
 615                    .hover(|style| style.bg(cx.theme().colors().element_hover))
 616                    .when(!self.has_open_menu(message_id), |el| {
 617                        el.visible_on_hover("")
 618                    })
 619                    .when_some(message_id, |el, message_id| {
 620                        el.child(
 621                            div()
 622                                .id("reply")
 623                                .child(
 624                                    IconButton::new(("reply", message_id), IconName::ReplyArrow)
 625                                        .on_click(cx.listener(move |this, _, cx| {
 626                                            this.message_editor.update(cx, |editor, cx| {
 627                                                editor.set_reply_to_message_id(message_id);
 628                                                editor.focus_handle(cx).focus(cx);
 629                                            })
 630                                        })),
 631                                )
 632                                .tooltip(|cx| Tooltip::text("Reply", cx)),
 633                        )
 634                    }),
 635            )
 636            .child(
 637                div()
 638                    .absolute()
 639                    .right_2()
 640                    .w_6()
 641                    .rounded_tr_md()
 642                    .rounded_br_md()
 643                    .border_r_1()
 644                    .border_t_1()
 645                    .border_b_1()
 646                    .border_color(cx.theme().colors().element_selected)
 647                    .bg(cx.theme().colors().element_background)
 648                    .hover(|style| style.bg(cx.theme().colors().element_hover))
 649                    .when(!self.has_open_menu(message_id), |el| {
 650                        el.visible_on_hover("")
 651                    })
 652                    .when_some(message_id, |el, message_id| {
 653                        let this = cx.view().clone();
 654
 655                        el.child(
 656                            div()
 657                                .id("more")
 658                                .child(
 659                                    popover_menu(("menu", message_id))
 660                                        .trigger(IconButton::new(
 661                                            ("trigger", message_id),
 662                                            IconName::Ellipsis,
 663                                        ))
 664                                        .menu(move |cx| {
 665                                            Some(Self::render_message_menu(
 666                                                &this,
 667                                                message_id,
 668                                                can_delete_message,
 669                                                cx,
 670                                            ))
 671                                        }),
 672                                )
 673                                .tooltip(|cx| Tooltip::text("More", cx)),
 674                        )
 675                    }),
 676            )
 677    }
 678
 679    fn render_message_menu(
 680        this: &View<Self>,
 681        message_id: u64,
 682        can_delete_message: bool,
 683        cx: &mut WindowContext,
 684    ) -> View<ContextMenu> {
 685        let menu = {
 686            ContextMenu::build(cx, move |menu, cx| {
 687                menu.entry(
 688                    "Reply to message",
 689                    None,
 690                    cx.handler_for(&this, move |this, cx| {
 691                        this.message_editor.update(cx, |editor, cx| {
 692                            editor.set_reply_to_message_id(message_id);
 693                            editor.focus_handle(cx).focus(cx);
 694                        })
 695                    }),
 696                )
 697                .entry(
 698                    "Copy message text",
 699                    None,
 700                    cx.handler_for(&this, move |this, cx| {
 701                        if let Some(message) = this.active_chat().and_then(|active_chat| {
 702                            active_chat.read(cx).find_loaded_message(message_id)
 703                        }) {
 704                            let text = message.body.clone();
 705                            cx.write_to_clipboard(ClipboardItem::new(text))
 706                        }
 707                    }),
 708                )
 709                .when(can_delete_message, move |menu| {
 710                    menu.entry(
 711                        "Delete message",
 712                        None,
 713                        cx.handler_for(&this, move |this, cx| this.remove_message(message_id, cx)),
 714                    )
 715                })
 716            })
 717        };
 718        this.update(cx, |this, cx| {
 719            let subscription = cx.subscribe(&menu, |this: &mut Self, _, _: &DismissEvent, _| {
 720                this.open_context_menu = None;
 721            });
 722            this.open_context_menu = Some((message_id, subscription));
 723        });
 724        menu
 725    }
 726
 727    fn render_markdown_with_mentions(
 728        language_registry: &Arc<LanguageRegistry>,
 729        current_user_id: u64,
 730        message: &channel::ChannelMessage,
 731    ) -> RichText {
 732        let mentions = message
 733            .mentions
 734            .iter()
 735            .map(|(range, user_id)| rich_text::Mention {
 736                range: range.clone(),
 737                is_self_mention: *user_id == current_user_id,
 738            })
 739            .collect::<Vec<_>>();
 740
 741        rich_text::render_rich_text(message.body.clone(), &mentions, language_registry, None)
 742    }
 743
 744    fn send(&mut self, _: &Confirm, cx: &mut ViewContext<Self>) {
 745        if let Some((chat, _)) = self.active_chat.as_ref() {
 746            let message = self
 747                .message_editor
 748                .update(cx, |editor, cx| editor.take_message(cx));
 749
 750            if let Some(task) = chat
 751                .update(cx, |chat, cx| chat.send_message(message, cx))
 752                .log_err()
 753            {
 754                task.detach();
 755            }
 756        }
 757    }
 758
 759    fn remove_message(&mut self, id: u64, cx: &mut ViewContext<Self>) {
 760        if let Some((chat, _)) = self.active_chat.as_ref() {
 761            chat.update(cx, |chat, cx| chat.remove_message(id, cx).detach())
 762        }
 763    }
 764
 765    fn load_more_messages(&mut self, cx: &mut ViewContext<Self>) {
 766        if let Some((chat, _)) = self.active_chat.as_ref() {
 767            chat.update(cx, |channel, cx| {
 768                if let Some(task) = channel.load_more_messages(cx) {
 769                    task.detach();
 770                }
 771            })
 772        }
 773    }
 774
 775    pub fn select_channel(
 776        &mut self,
 777        selected_channel_id: ChannelId,
 778        scroll_to_message_id: Option<u64>,
 779        cx: &mut ViewContext<ChatPanel>,
 780    ) -> Task<Result<()>> {
 781        let open_chat = self
 782            .active_chat
 783            .as_ref()
 784            .and_then(|(chat, _)| {
 785                (chat.read(cx).channel_id == selected_channel_id)
 786                    .then(|| Task::ready(anyhow::Ok(chat.clone())))
 787            })
 788            .unwrap_or_else(|| {
 789                self.channel_store.update(cx, |store, cx| {
 790                    store.open_channel_chat(selected_channel_id, cx)
 791                })
 792            });
 793
 794        cx.spawn(|this, mut cx| async move {
 795            let chat = open_chat.await?;
 796            let highlight_message_id = scroll_to_message_id;
 797            let scroll_to_message_id = this.update(&mut cx, |this, cx| {
 798                this.set_active_chat(chat.clone(), cx);
 799
 800                scroll_to_message_id.or_else(|| this.last_acknowledged_message_id)
 801            })?;
 802
 803            if let Some(message_id) = scroll_to_message_id {
 804                if let Some(item_ix) =
 805                    ChannelChat::load_history_since_message(chat.clone(), message_id, (*cx).clone())
 806                        .await
 807                {
 808                    this.update(&mut cx, |this, cx| {
 809                        if let Some(highlight_message_id) = highlight_message_id {
 810                            let task = cx.spawn({
 811                                |this, mut cx| async move {
 812                                    cx.background_executor().timer(Duration::from_secs(2)).await;
 813                                    this.update(&mut cx, |this, cx| {
 814                                        this.highlighted_message.take();
 815                                        cx.notify();
 816                                    })
 817                                    .ok();
 818                                }
 819                            });
 820
 821                            this.highlighted_message = Some((highlight_message_id, task));
 822                        }
 823
 824                        if this.active_chat.as_ref().map_or(false, |(c, _)| *c == chat) {
 825                            this.message_list.scroll_to(ListOffset {
 826                                item_ix,
 827                                offset_in_item: px(0.0),
 828                            });
 829                            cx.notify();
 830                        }
 831                    })?;
 832                }
 833            }
 834
 835            Ok(())
 836        })
 837    }
 838
 839    fn close_reply_preview(&mut self, _: &CloseReplyPreview, cx: &mut ViewContext<Self>) {
 840        self.message_editor
 841            .update(cx, |editor, _| editor.clear_reply_to_message_id());
 842    }
 843}
 844
 845impl Render for ChatPanel {
 846    fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
 847        let reply_to_message_id = self.message_editor.read(cx).reply_to_message_id();
 848
 849        v_flex()
 850            .key_context("ChatPanel")
 851            .track_focus(&self.focus_handle)
 852            .size_full()
 853            .on_action(cx.listener(Self::send))
 854            .child(
 855                h_flex().child(
 856                    TabBar::new("chat_header").child(
 857                        h_flex()
 858                            .w_full()
 859                            .h(rems(ui::Tab::CONTAINER_HEIGHT_IN_REMS))
 860                            .px_2()
 861                            .child(Label::new(
 862                                self.active_chat
 863                                    .as_ref()
 864                                    .and_then(|c| {
 865                                        Some(format!("#{}", c.0.read(cx).channel(cx)?.name))
 866                                    })
 867                                    .unwrap_or("Chat".to_string()),
 868                            )),
 869                    ),
 870                ),
 871            )
 872            .child(div().flex_grow().px_2().map(|this| {
 873                if self.active_chat.is_some() {
 874                    this.child(list(self.message_list.clone()).size_full())
 875                } else {
 876                    this.child(
 877                        div()
 878                            .size_full()
 879                            .p_4()
 880                            .child(
 881                                Label::new("Select a channel to chat in.")
 882                                    .size(LabelSize::Small)
 883                                    .color(Color::Muted),
 884                            )
 885                            .child(
 886                                div().pt_1().w_full().items_center().child(
 887                                    Button::new("toggle-collab", "Open")
 888                                        .full_width()
 889                                        .key_binding(KeyBinding::for_action(
 890                                            &collab_panel::ToggleFocus,
 891                                            cx,
 892                                        ))
 893                                        .on_click(|_, cx| {
 894                                            cx.dispatch_action(
 895                                                collab_panel::ToggleFocus.boxed_clone(),
 896                                            )
 897                                        }),
 898                                ),
 899                            ),
 900                    )
 901                }
 902            }))
 903            .when_some(reply_to_message_id, |el, reply_to_message_id| {
 904                let reply_message = self
 905                    .active_chat()
 906                    .and_then(|active_chat| {
 907                        active_chat.read(cx).messages().iter().find(|message| {
 908                            message.id == ChannelMessageId::Saved(reply_to_message_id)
 909                        })
 910                    })
 911                    .cloned();
 912
 913                el.when_some(reply_message, |el, reply_message| {
 914                    el.child(
 915                        h_flex()
 916                            .when(!self.is_scrolled_to_bottom, |el| {
 917                                el.border_t_1().border_color(cx.theme().colors().border)
 918                            })
 919                            .justify_between()
 920                            .overflow_hidden()
 921                            .items_start()
 922                            .py_1()
 923                            .px_2()
 924                            .bg(cx.theme().colors().background)
 925                            .child(
 926                                div().flex_shrink().overflow_hidden().child(
 927                                    self.render_replied_to_message(None, &reply_message, cx),
 928                                ),
 929                            )
 930                            .child(
 931                                IconButton::new("close-reply-preview", IconName::Close)
 932                                    .shape(ui::IconButtonShape::Square)
 933                                    .tooltip(|cx| {
 934                                        Tooltip::for_action(
 935                                            "Close reply preview",
 936                                            &CloseReplyPreview,
 937                                            cx,
 938                                        )
 939                                    })
 940                                    .on_click(cx.listener(move |_, _, cx| {
 941                                        cx.dispatch_action(CloseReplyPreview.boxed_clone())
 942                                    })),
 943                            ),
 944                    )
 945                })
 946            })
 947            .children(
 948                Some(
 949                    h_flex()
 950                        .key_context("MessageEditor")
 951                        .on_action(cx.listener(ChatPanel::close_reply_preview))
 952                        .when(
 953                            !self.is_scrolled_to_bottom && reply_to_message_id.is_none(),
 954                            |el| el.border_t_1().border_color(cx.theme().colors().border),
 955                        )
 956                        .p_2()
 957                        .map(|el| el.child(self.message_editor.clone())),
 958                )
 959                .filter(|_| self.active_chat.is_some()),
 960            )
 961            .into_any()
 962    }
 963}
 964
 965impl FocusableView for ChatPanel {
 966    fn focus_handle(&self, cx: &AppContext) -> gpui::FocusHandle {
 967        if self.active_chat.is_some() {
 968            self.message_editor.read(cx).focus_handle(cx)
 969        } else {
 970            self.focus_handle.clone()
 971        }
 972    }
 973}
 974
 975impl Panel for ChatPanel {
 976    fn position(&self, cx: &gpui::WindowContext) -> DockPosition {
 977        ChatPanelSettings::get_global(cx).dock
 978    }
 979
 980    fn position_is_valid(&self, position: DockPosition) -> bool {
 981        matches!(position, DockPosition::Left | DockPosition::Right)
 982    }
 983
 984    fn set_position(&mut self, position: DockPosition, cx: &mut ViewContext<Self>) {
 985        settings::update_settings_file::<ChatPanelSettings>(self.fs.clone(), cx, move |settings| {
 986            settings.dock = Some(position)
 987        });
 988    }
 989
 990    fn size(&self, cx: &gpui::WindowContext) -> Pixels {
 991        self.width
 992            .unwrap_or_else(|| ChatPanelSettings::get_global(cx).default_width)
 993    }
 994
 995    fn set_size(&mut self, size: Option<Pixels>, cx: &mut ViewContext<Self>) {
 996        self.width = size;
 997        self.serialize(cx);
 998        cx.notify();
 999    }
1000
1001    fn set_active(&mut self, active: bool, cx: &mut ViewContext<Self>) {
1002        self.active = active;
1003        if active {
1004            self.acknowledge_last_message(cx);
1005        }
1006    }
1007
1008    fn persistent_name() -> &'static str {
1009        "ChatPanel"
1010    }
1011
1012    fn icon(&self, cx: &WindowContext) -> Option<ui::IconName> {
1013        Some(ui::IconName::MessageBubbles).filter(|_| ChatPanelSettings::get_global(cx).button)
1014    }
1015
1016    fn icon_tooltip(&self, _cx: &WindowContext) -> Option<&'static str> {
1017        Some("Chat Panel")
1018    }
1019
1020    fn toggle_action(&self) -> Box<dyn gpui::Action> {
1021        Box::new(ToggleFocus)
1022    }
1023
1024    fn starts_open(&self, cx: &WindowContext) -> bool {
1025        ActiveCall::global(cx)
1026            .read(cx)
1027            .room()
1028            .is_some_and(|room| room.read(cx).contains_guests())
1029    }
1030}
1031
1032impl EventEmitter<PanelEvent> for ChatPanel {}
1033
1034#[cfg(test)]
1035mod tests {
1036    use super::*;
1037    use gpui::HighlightStyle;
1038    use pretty_assertions::assert_eq;
1039    use rich_text::Highlight;
1040    use time::OffsetDateTime;
1041    use util::test::marked_text_ranges;
1042
1043    #[gpui::test]
1044    fn test_render_markdown_with_mentions() {
1045        let language_registry = Arc::new(LanguageRegistry::test());
1046        let (body, ranges) = marked_text_ranges("*hi*, «@abc», let's **call** «@fgh»", false);
1047        let message = channel::ChannelMessage {
1048            id: ChannelMessageId::Saved(0),
1049            body,
1050            timestamp: OffsetDateTime::now_utc(),
1051            sender: Arc::new(client::User {
1052                github_login: "fgh".into(),
1053                avatar_uri: "avatar_fgh".into(),
1054                id: 103,
1055            }),
1056            nonce: 5,
1057            mentions: vec![(ranges[0].clone(), 101), (ranges[1].clone(), 102)],
1058            reply_to_message_id: None,
1059        };
1060
1061        let message = ChatPanel::render_markdown_with_mentions(&language_registry, 102, &message);
1062
1063        // Note that the "'" was replaced with ’ due to smart punctuation.
1064        let (body, ranges) = marked_text_ranges("«hi», «@abc», let’s «call» «@fgh»", false);
1065        assert_eq!(message.text, body);
1066        assert_eq!(
1067            message.highlights,
1068            vec![
1069                (
1070                    ranges[0].clone(),
1071                    HighlightStyle {
1072                        font_style: Some(gpui::FontStyle::Italic),
1073                        ..Default::default()
1074                    }
1075                    .into()
1076                ),
1077                (ranges[1].clone(), Highlight::Mention),
1078                (
1079                    ranges[2].clone(),
1080                    HighlightStyle {
1081                        font_weight: Some(gpui::FontWeight::BOLD),
1082                        ..Default::default()
1083                    }
1084                    .into()
1085                ),
1086                (ranges[3].clone(), Highlight::SelfMention)
1087            ]
1088        );
1089    }
1090
1091    #[gpui::test]
1092    fn test_render_markdown_with_auto_detect_links() {
1093        let language_registry = Arc::new(LanguageRegistry::test());
1094        let message = channel::ChannelMessage {
1095            id: ChannelMessageId::Saved(0),
1096            body: "Here is a link https://zed.dev to zeds website".to_string(),
1097            timestamp: OffsetDateTime::now_utc(),
1098            sender: Arc::new(client::User {
1099                github_login: "fgh".into(),
1100                avatar_uri: "avatar_fgh".into(),
1101                id: 103,
1102            }),
1103            nonce: 5,
1104            mentions: Vec::new(),
1105            reply_to_message_id: None,
1106        };
1107
1108        let message = ChatPanel::render_markdown_with_mentions(&language_registry, 102, &message);
1109
1110        // Note that the "'" was replaced with ’ due to smart punctuation.
1111        let (body, ranges) =
1112            marked_text_ranges("Here is a link «https://zed.dev» to zeds website", false);
1113        assert_eq!(message.text, body);
1114        assert_eq!(1, ranges.len());
1115        assert_eq!(
1116            message.highlights,
1117            vec![(
1118                ranges[0].clone(),
1119                HighlightStyle {
1120                    underline: Some(gpui::UnderlineStyle {
1121                        thickness: 1.0.into(),
1122                        ..Default::default()
1123                    }),
1124                    ..Default::default()
1125                }
1126                .into()
1127            ),]
1128        );
1129    }
1130
1131    #[gpui::test]
1132    fn test_render_markdown_with_auto_detect_links_and_additional_formatting() {
1133        let language_registry = Arc::new(LanguageRegistry::test());
1134        let message = channel::ChannelMessage {
1135            id: ChannelMessageId::Saved(0),
1136            body: "**Here is a link https://zed.dev to zeds website**".to_string(),
1137            timestamp: OffsetDateTime::now_utc(),
1138            sender: Arc::new(client::User {
1139                github_login: "fgh".into(),
1140                avatar_uri: "avatar_fgh".into(),
1141                id: 103,
1142            }),
1143            nonce: 5,
1144            mentions: Vec::new(),
1145            reply_to_message_id: None,
1146        };
1147
1148        let message = ChatPanel::render_markdown_with_mentions(&language_registry, 102, &message);
1149
1150        // Note that the "'" was replaced with ’ due to smart punctuation.
1151        let (body, ranges) = marked_text_ranges(
1152            "«Here is a link »«https://zed.dev»« to zeds website»",
1153            false,
1154        );
1155        assert_eq!(message.text, body);
1156        assert_eq!(3, ranges.len());
1157        assert_eq!(
1158            message.highlights,
1159            vec![
1160                (
1161                    ranges[0].clone(),
1162                    HighlightStyle {
1163                        font_weight: Some(gpui::FontWeight::BOLD),
1164                        ..Default::default()
1165                    }
1166                    .into()
1167                ),
1168                (
1169                    ranges[1].clone(),
1170                    HighlightStyle {
1171                        font_weight: Some(gpui::FontWeight::BOLD),
1172                        underline: Some(gpui::UnderlineStyle {
1173                            thickness: 1.0.into(),
1174                            ..Default::default()
1175                        }),
1176                        ..Default::default()
1177                    }
1178                    .into()
1179                ),
1180                (
1181                    ranges[2].clone(),
1182                    HighlightStyle {
1183                        font_weight: Some(gpui::FontWeight::BOLD),
1184                        ..Default::default()
1185                    }
1186                    .into()
1187                ),
1188            ]
1189        );
1190    }
1191}