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        let this = cx.view().clone();
 434
 435        let mentioning_you = message
 436            .mentions
 437            .iter()
 438            .any(|m| Some(m.1) == self.client.user_id());
 439
 440        let message_id = match message.id {
 441            ChannelMessageId::Saved(id) => Some(id),
 442            ChannelMessageId::Pending(_) => None,
 443        };
 444
 445        let reply_to_message = message
 446            .reply_to_message_id
 447            .and_then(|id| active_chat.read(cx).find_loaded_message(id))
 448            .cloned();
 449
 450        let replied_to_you =
 451            reply_to_message.as_ref().map(|m| m.sender.id) == self.client.user_id();
 452
 453        let is_highlighted_message = self
 454            .highlighted_message
 455            .as_ref()
 456            .is_some_and(|(id, _)| Some(id) == message_id.as_ref());
 457        let background = if is_highlighted_message {
 458            cx.theme().status().info_background
 459        } else if mentioning_you || replied_to_you {
 460            cx.theme().colors().background
 461        } else {
 462            cx.theme().colors().panel_background
 463        };
 464
 465        v_flex()
 466            .w_full()
 467            .relative()
 468            .child(
 469                div()
 470                    .bg(background)
 471                    .rounded_md()
 472                    .overflow_hidden()
 473                    .px_1()
 474                    .py_0p5()
 475                    .when(!is_continuation_from_previous, |this| {
 476                        this.mt_2().child(
 477                            h_flex()
 478                                .text_ui_sm()
 479                                .child(div().absolute().child(
 480                                    Avatar::new(message.sender.avatar_uri.clone()).size(rems(1.)),
 481                                ))
 482                                .child(
 483                                    div()
 484                                        .pl(cx.rem_size() + px(6.0))
 485                                        .pr(px(8.0))
 486                                        .font_weight(FontWeight::BOLD)
 487                                        .child(Label::new(message.sender.github_login.clone())),
 488                                )
 489                                .child(
 490                                    Label::new(time_format::format_localized_timestamp(
 491                                        OffsetDateTime::now_utc(),
 492                                        message.timestamp,
 493                                        self.local_timezone,
 494                                    ))
 495                                    .size(LabelSize::Small)
 496                                    .color(Color::Muted),
 497                                ),
 498                        )
 499                    })
 500                    .when(
 501                        message.reply_to_message_id.is_some() && reply_to_message.is_none(),
 502                        |this| {
 503                            const MESSAGE_DELETED: &str = "Message has been deleted";
 504
 505                            let body_text = StyledText::new(MESSAGE_DELETED).with_highlights(
 506                                &cx.text_style(),
 507                                vec![(
 508                                    0..MESSAGE_DELETED.len(),
 509                                    HighlightStyle {
 510                                        font_style: Some(FontStyle::Italic),
 511                                        ..Default::default()
 512                                    },
 513                                )],
 514                            );
 515
 516                            this.child(
 517                                div()
 518                                    .border_l_2()
 519                                    .text_ui_xs()
 520                                    .border_color(cx.theme().colors().border)
 521                                    .px_1()
 522                                    .py_0p5()
 523                                    .child(body_text),
 524                            )
 525                        },
 526                    )
 527                    .when_some(reply_to_message, |el, reply_to_message| {
 528                        el.child(self.render_replied_to_message(
 529                            Some(message.id),
 530                            &reply_to_message,
 531                            cx,
 532                        ))
 533                    })
 534                    .when(mentioning_you || replied_to_you, |this| this.my_0p5())
 535                    .map(|el| {
 536                        let text = self.markdown_data.entry(message.id).or_insert_with(|| {
 537                            Self::render_markdown_with_mentions(
 538                                &self.languages,
 539                                self.client.id(),
 540                                &message,
 541                            )
 542                        });
 543                        el.child(
 544                            v_flex()
 545                                .w_full()
 546                                .text_ui_sm()
 547                                .id(element_id)
 548                                .group("")
 549                                .child(text.element("body".into(), cx))
 550                                .child(
 551                                    div()
 552                                        .absolute()
 553                                        .z_index(1)
 554                                        .right_0()
 555                                        .w_6()
 556                                        .bg(background)
 557                                        .when(!self.has_open_menu(message_id), |el| {
 558                                            el.visible_on_hover("")
 559                                        })
 560                                        .when_some(message_id, |el, message_id| {
 561                                            el.child(
 562                                                popover_menu(("menu", message_id))
 563                                                    .trigger(IconButton::new(
 564                                                        ("trigger", message_id),
 565                                                        IconName::Ellipsis,
 566                                                    ))
 567                                                    .menu(move |cx| {
 568                                                        Some(Self::render_message_menu(
 569                                                            &this,
 570                                                            message_id,
 571                                                            can_delete_message,
 572                                                            cx,
 573                                                        ))
 574                                                    }),
 575                                            )
 576                                        }),
 577                                ),
 578                        )
 579                    }),
 580            )
 581            .when(
 582                self.last_acknowledged_message_id
 583                    .is_some_and(|l| Some(l) == message_id),
 584                |this| {
 585                    this.child(
 586                        h_flex()
 587                            .py_2()
 588                            .gap_1()
 589                            .items_center()
 590                            .child(div().w_full().h_0p5().bg(cx.theme().colors().border))
 591                            .child(
 592                                div()
 593                                    .px_1()
 594                                    .rounded_md()
 595                                    .text_ui_xs()
 596                                    .bg(cx.theme().colors().background)
 597                                    .child("New messages"),
 598                            )
 599                            .child(div().w_full().h_0p5().bg(cx.theme().colors().border)),
 600                    )
 601                },
 602            )
 603    }
 604
 605    fn has_open_menu(&self, message_id: Option<u64>) -> bool {
 606        match self.open_context_menu.as_ref() {
 607            Some((id, _)) => Some(*id) == message_id,
 608            None => false,
 609        }
 610    }
 611
 612    fn render_message_menu(
 613        this: &View<Self>,
 614        message_id: u64,
 615        can_delete_message: bool,
 616        cx: &mut WindowContext,
 617    ) -> View<ContextMenu> {
 618        let menu = {
 619            ContextMenu::build(cx, move |menu, cx| {
 620                menu.entry(
 621                    "Reply to message",
 622                    None,
 623                    cx.handler_for(&this, move |this, cx| {
 624                        this.message_editor.update(cx, |editor, cx| {
 625                            editor.set_reply_to_message_id(message_id);
 626                            editor.focus_handle(cx).focus(cx);
 627                        })
 628                    }),
 629                )
 630                .entry(
 631                    "Copy message text",
 632                    None,
 633                    cx.handler_for(&this, move |this, cx| {
 634                        if let Some(message) = this.active_chat().and_then(|active_chat| {
 635                            active_chat.read(cx).find_loaded_message(message_id)
 636                        }) {
 637                            let text = message.body.clone();
 638                            cx.write_to_clipboard(ClipboardItem::new(text))
 639                        }
 640                    }),
 641                )
 642                .when(can_delete_message, move |menu| {
 643                    menu.entry(
 644                        "Delete message",
 645                        None,
 646                        cx.handler_for(&this, move |this, cx| this.remove_message(message_id, cx)),
 647                    )
 648                })
 649            })
 650        };
 651        this.update(cx, |this, cx| {
 652            let subscription = cx.subscribe(&menu, |this: &mut Self, _, _: &DismissEvent, _| {
 653                this.open_context_menu = None;
 654            });
 655            this.open_context_menu = Some((message_id, subscription));
 656        });
 657        menu
 658    }
 659
 660    fn render_markdown_with_mentions(
 661        language_registry: &Arc<LanguageRegistry>,
 662        current_user_id: u64,
 663        message: &channel::ChannelMessage,
 664    ) -> RichText {
 665        let mentions = message
 666            .mentions
 667            .iter()
 668            .map(|(range, user_id)| rich_text::Mention {
 669                range: range.clone(),
 670                is_self_mention: *user_id == current_user_id,
 671            })
 672            .collect::<Vec<_>>();
 673
 674        rich_text::render_rich_text(message.body.clone(), &mentions, language_registry, None)
 675    }
 676
 677    fn send(&mut self, _: &Confirm, cx: &mut ViewContext<Self>) {
 678        if let Some((chat, _)) = self.active_chat.as_ref() {
 679            let message = self
 680                .message_editor
 681                .update(cx, |editor, cx| editor.take_message(cx));
 682
 683            if let Some(task) = chat
 684                .update(cx, |chat, cx| chat.send_message(message, cx))
 685                .log_err()
 686            {
 687                task.detach();
 688            }
 689        }
 690    }
 691
 692    fn remove_message(&mut self, id: u64, cx: &mut ViewContext<Self>) {
 693        if let Some((chat, _)) = self.active_chat.as_ref() {
 694            chat.update(cx, |chat, cx| chat.remove_message(id, cx).detach())
 695        }
 696    }
 697
 698    fn load_more_messages(&mut self, cx: &mut ViewContext<Self>) {
 699        if let Some((chat, _)) = self.active_chat.as_ref() {
 700            chat.update(cx, |channel, cx| {
 701                if let Some(task) = channel.load_more_messages(cx) {
 702                    task.detach();
 703                }
 704            })
 705        }
 706    }
 707
 708    pub fn select_channel(
 709        &mut self,
 710        selected_channel_id: ChannelId,
 711        scroll_to_message_id: Option<u64>,
 712        cx: &mut ViewContext<ChatPanel>,
 713    ) -> Task<Result<()>> {
 714        let open_chat = self
 715            .active_chat
 716            .as_ref()
 717            .and_then(|(chat, _)| {
 718                (chat.read(cx).channel_id == selected_channel_id)
 719                    .then(|| Task::ready(anyhow::Ok(chat.clone())))
 720            })
 721            .unwrap_or_else(|| {
 722                self.channel_store.update(cx, |store, cx| {
 723                    store.open_channel_chat(selected_channel_id, cx)
 724                })
 725            });
 726
 727        cx.spawn(|this, mut cx| async move {
 728            let chat = open_chat.await?;
 729            let highlight_message_id = scroll_to_message_id;
 730            let scroll_to_message_id = this.update(&mut cx, |this, cx| {
 731                this.set_active_chat(chat.clone(), cx);
 732
 733                scroll_to_message_id.or_else(|| this.last_acknowledged_message_id)
 734            })?;
 735
 736            if let Some(message_id) = scroll_to_message_id {
 737                if let Some(item_ix) =
 738                    ChannelChat::load_history_since_message(chat.clone(), message_id, (*cx).clone())
 739                        .await
 740                {
 741                    this.update(&mut cx, |this, cx| {
 742                        if let Some(highlight_message_id) = highlight_message_id {
 743                            let task = cx.spawn({
 744                                |this, mut cx| async move {
 745                                    cx.background_executor().timer(Duration::from_secs(2)).await;
 746                                    this.update(&mut cx, |this, cx| {
 747                                        this.highlighted_message.take();
 748                                        cx.notify();
 749                                    })
 750                                    .ok();
 751                                }
 752                            });
 753
 754                            this.highlighted_message = Some((highlight_message_id, task));
 755                        }
 756
 757                        if this.active_chat.as_ref().map_or(false, |(c, _)| *c == chat) {
 758                            this.message_list.scroll_to(ListOffset {
 759                                item_ix,
 760                                offset_in_item: px(0.0),
 761                            });
 762                            cx.notify();
 763                        }
 764                    })?;
 765                }
 766            }
 767
 768            Ok(())
 769        })
 770    }
 771
 772    fn close_reply_preview(&mut self, _: &CloseReplyPreview, cx: &mut ViewContext<Self>) {
 773        self.message_editor
 774            .update(cx, |editor, _| editor.clear_reply_to_message_id());
 775    }
 776}
 777
 778impl Render for ChatPanel {
 779    fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
 780        let reply_to_message_id = self.message_editor.read(cx).reply_to_message_id();
 781
 782        v_flex()
 783            .key_context("ChatPanel")
 784            .track_focus(&self.focus_handle)
 785            .size_full()
 786            .on_action(cx.listener(Self::send))
 787            .child(
 788                h_flex().z_index(1).child(
 789                    TabBar::new("chat_header").child(
 790                        h_flex()
 791                            .w_full()
 792                            .h(rems(ui::Tab::CONTAINER_HEIGHT_IN_REMS))
 793                            .px_2()
 794                            .child(Label::new(
 795                                self.active_chat
 796                                    .as_ref()
 797                                    .and_then(|c| {
 798                                        Some(format!("#{}", c.0.read(cx).channel(cx)?.name))
 799                                    })
 800                                    .unwrap_or("Chat".to_string()),
 801                            )),
 802                    ),
 803                ),
 804            )
 805            .child(div().flex_grow().px_2().map(|this| {
 806                if self.active_chat.is_some() {
 807                    this.child(list(self.message_list.clone()).size_full())
 808                } else {
 809                    this.child(
 810                        div()
 811                            .size_full()
 812                            .p_4()
 813                            .child(
 814                                Label::new("Select a channel to chat in.")
 815                                    .size(LabelSize::Small)
 816                                    .color(Color::Muted),
 817                            )
 818                            .child(
 819                                div().pt_1().w_full().items_center().child(
 820                                    Button::new("toggle-collab", "Open")
 821                                        .full_width()
 822                                        .key_binding(KeyBinding::for_action(
 823                                            &collab_panel::ToggleFocus,
 824                                            cx,
 825                                        ))
 826                                        .on_click(|_, cx| {
 827                                            cx.dispatch_action(
 828                                                collab_panel::ToggleFocus.boxed_clone(),
 829                                            )
 830                                        }),
 831                                ),
 832                            ),
 833                    )
 834                }
 835            }))
 836            .when_some(reply_to_message_id, |el, reply_to_message_id| {
 837                let reply_message = self
 838                    .active_chat()
 839                    .and_then(|active_chat| {
 840                        active_chat.read(cx).messages().iter().find(|message| {
 841                            message.id == ChannelMessageId::Saved(reply_to_message_id)
 842                        })
 843                    })
 844                    .cloned();
 845
 846                el.when_some(reply_message, |el, reply_message| {
 847                    el.child(
 848                        h_flex()
 849                            .when(!self.is_scrolled_to_bottom, |el| {
 850                                el.border_t_1().border_color(cx.theme().colors().border)
 851                            })
 852                            .justify_between()
 853                            .overflow_hidden()
 854                            .items_start()
 855                            .py_1()
 856                            .px_2()
 857                            .bg(cx.theme().colors().background)
 858                            .child(
 859                                div().flex_shrink().overflow_hidden().child(
 860                                    self.render_replied_to_message(None, &reply_message, cx),
 861                                ),
 862                            )
 863                            .child(
 864                                IconButton::new("close-reply-preview", IconName::Close)
 865                                    .shape(ui::IconButtonShape::Square)
 866                                    .tooltip(|cx| {
 867                                        Tooltip::for_action(
 868                                            "Close reply preview",
 869                                            &CloseReplyPreview,
 870                                            cx,
 871                                        )
 872                                    })
 873                                    .on_click(cx.listener(move |_, _, cx| {
 874                                        cx.dispatch_action(CloseReplyPreview.boxed_clone())
 875                                    })),
 876                            ),
 877                    )
 878                })
 879            })
 880            .children(
 881                Some(
 882                    h_flex()
 883                        .key_context("MessageEditor")
 884                        .on_action(cx.listener(ChatPanel::close_reply_preview))
 885                        .when(
 886                            !self.is_scrolled_to_bottom && reply_to_message_id.is_none(),
 887                            |el| el.border_t_1().border_color(cx.theme().colors().border),
 888                        )
 889                        .p_2()
 890                        .map(|el| el.child(self.message_editor.clone())),
 891                )
 892                .filter(|_| self.active_chat.is_some()),
 893            )
 894            .into_any()
 895    }
 896}
 897
 898impl FocusableView for ChatPanel {
 899    fn focus_handle(&self, cx: &AppContext) -> gpui::FocusHandle {
 900        if self.active_chat.is_some() {
 901            self.message_editor.read(cx).focus_handle(cx)
 902        } else {
 903            self.focus_handle.clone()
 904        }
 905    }
 906}
 907
 908impl Panel for ChatPanel {
 909    fn position(&self, cx: &gpui::WindowContext) -> DockPosition {
 910        ChatPanelSettings::get_global(cx).dock
 911    }
 912
 913    fn position_is_valid(&self, position: DockPosition) -> bool {
 914        matches!(position, DockPosition::Left | DockPosition::Right)
 915    }
 916
 917    fn set_position(&mut self, position: DockPosition, cx: &mut ViewContext<Self>) {
 918        settings::update_settings_file::<ChatPanelSettings>(self.fs.clone(), cx, move |settings| {
 919            settings.dock = Some(position)
 920        });
 921    }
 922
 923    fn size(&self, cx: &gpui::WindowContext) -> Pixels {
 924        self.width
 925            .unwrap_or_else(|| ChatPanelSettings::get_global(cx).default_width)
 926    }
 927
 928    fn set_size(&mut self, size: Option<Pixels>, cx: &mut ViewContext<Self>) {
 929        self.width = size;
 930        self.serialize(cx);
 931        cx.notify();
 932    }
 933
 934    fn set_active(&mut self, active: bool, cx: &mut ViewContext<Self>) {
 935        self.active = active;
 936        if active {
 937            self.acknowledge_last_message(cx);
 938        }
 939    }
 940
 941    fn persistent_name() -> &'static str {
 942        "ChatPanel"
 943    }
 944
 945    fn icon(&self, cx: &WindowContext) -> Option<ui::IconName> {
 946        Some(ui::IconName::MessageBubbles).filter(|_| ChatPanelSettings::get_global(cx).button)
 947    }
 948
 949    fn icon_tooltip(&self, _cx: &WindowContext) -> Option<&'static str> {
 950        Some("Chat Panel")
 951    }
 952
 953    fn toggle_action(&self) -> Box<dyn gpui::Action> {
 954        Box::new(ToggleFocus)
 955    }
 956
 957    fn starts_open(&self, cx: &WindowContext) -> bool {
 958        ActiveCall::global(cx)
 959            .read(cx)
 960            .room()
 961            .is_some_and(|room| room.read(cx).contains_guests())
 962    }
 963}
 964
 965impl EventEmitter<PanelEvent> for ChatPanel {}
 966
 967#[cfg(test)]
 968mod tests {
 969    use super::*;
 970    use gpui::HighlightStyle;
 971    use pretty_assertions::assert_eq;
 972    use rich_text::Highlight;
 973    use time::OffsetDateTime;
 974    use util::test::marked_text_ranges;
 975
 976    #[gpui::test]
 977    fn test_render_markdown_with_mentions() {
 978        let language_registry = Arc::new(LanguageRegistry::test());
 979        let (body, ranges) = marked_text_ranges("*hi*, «@abc», let's **call** «@fgh»", false);
 980        let message = channel::ChannelMessage {
 981            id: ChannelMessageId::Saved(0),
 982            body,
 983            timestamp: OffsetDateTime::now_utc(),
 984            sender: Arc::new(client::User {
 985                github_login: "fgh".into(),
 986                avatar_uri: "avatar_fgh".into(),
 987                id: 103,
 988            }),
 989            nonce: 5,
 990            mentions: vec![(ranges[0].clone(), 101), (ranges[1].clone(), 102)],
 991            reply_to_message_id: None,
 992        };
 993
 994        let message = ChatPanel::render_markdown_with_mentions(&language_registry, 102, &message);
 995
 996        // Note that the "'" was replaced with ’ due to smart punctuation.
 997        let (body, ranges) = marked_text_ranges("«hi», «@abc», let’s «call» «@fgh»", false);
 998        assert_eq!(message.text, body);
 999        assert_eq!(
1000            message.highlights,
1001            vec![
1002                (
1003                    ranges[0].clone(),
1004                    HighlightStyle {
1005                        font_style: Some(gpui::FontStyle::Italic),
1006                        ..Default::default()
1007                    }
1008                    .into()
1009                ),
1010                (ranges[1].clone(), Highlight::Mention),
1011                (
1012                    ranges[2].clone(),
1013                    HighlightStyle {
1014                        font_weight: Some(gpui::FontWeight::BOLD),
1015                        ..Default::default()
1016                    }
1017                    .into()
1018                ),
1019                (ranges[3].clone(), Highlight::SelfMention)
1020            ]
1021        );
1022    }
1023
1024    #[gpui::test]
1025    fn test_render_markdown_with_auto_detect_links() {
1026        let language_registry = Arc::new(LanguageRegistry::test());
1027        let message = channel::ChannelMessage {
1028            id: ChannelMessageId::Saved(0),
1029            body: "Here is a link https://zed.dev to zeds website".to_string(),
1030            timestamp: OffsetDateTime::now_utc(),
1031            sender: Arc::new(client::User {
1032                github_login: "fgh".into(),
1033                avatar_uri: "avatar_fgh".into(),
1034                id: 103,
1035            }),
1036            nonce: 5,
1037            mentions: Vec::new(),
1038            reply_to_message_id: None,
1039        };
1040
1041        let message = ChatPanel::render_markdown_with_mentions(&language_registry, 102, &message);
1042
1043        // Note that the "'" was replaced with ’ due to smart punctuation.
1044        let (body, ranges) =
1045            marked_text_ranges("Here is a link «https://zed.dev» to zeds website", false);
1046        assert_eq!(message.text, body);
1047        assert_eq!(1, ranges.len());
1048        assert_eq!(
1049            message.highlights,
1050            vec![(
1051                ranges[0].clone(),
1052                HighlightStyle {
1053                    underline: Some(gpui::UnderlineStyle {
1054                        thickness: 1.0.into(),
1055                        ..Default::default()
1056                    }),
1057                    ..Default::default()
1058                }
1059                .into()
1060            ),]
1061        );
1062    }
1063
1064    #[gpui::test]
1065    fn test_render_markdown_with_auto_detect_links_and_additional_formatting() {
1066        let language_registry = Arc::new(LanguageRegistry::test());
1067        let message = channel::ChannelMessage {
1068            id: ChannelMessageId::Saved(0),
1069            body: "**Here is a link https://zed.dev to zeds website**".to_string(),
1070            timestamp: OffsetDateTime::now_utc(),
1071            sender: Arc::new(client::User {
1072                github_login: "fgh".into(),
1073                avatar_uri: "avatar_fgh".into(),
1074                id: 103,
1075            }),
1076            nonce: 5,
1077            mentions: Vec::new(),
1078            reply_to_message_id: None,
1079        };
1080
1081        let message = ChatPanel::render_markdown_with_mentions(&language_registry, 102, &message);
1082
1083        // Note that the "'" was replaced with ’ due to smart punctuation.
1084        let (body, ranges) = marked_text_ranges(
1085            "«Here is a link »«https://zed.dev»« to zeds website»",
1086            false,
1087        );
1088        assert_eq!(message.text, body);
1089        assert_eq!(3, ranges.len());
1090        assert_eq!(
1091            message.highlights,
1092            vec![
1093                (
1094                    ranges[0].clone(),
1095                    HighlightStyle {
1096                        font_weight: Some(gpui::FontWeight::BOLD),
1097                        ..Default::default()
1098                    }
1099                    .into()
1100                ),
1101                (
1102                    ranges[1].clone(),
1103                    HighlightStyle {
1104                        font_weight: Some(gpui::FontWeight::BOLD),
1105                        underline: Some(gpui::UnderlineStyle {
1106                            thickness: 1.0.into(),
1107                            ..Default::default()
1108                        }),
1109                        ..Default::default()
1110                    }
1111                    .into()
1112                ),
1113                (
1114                    ranges[2].clone(),
1115                    HighlightStyle {
1116                        font_weight: Some(gpui::FontWeight::BOLD),
1117                        ..Default::default()
1118                    }
1119                    .into()
1120                ),
1121            ]
1122        );
1123    }
1124}