copilot: Remove an unwrap in URI parsing code (#32698)

Closes #32630

Release Notes:

- Fixed a potential crash when opening active modules in a debugger
session (with Copilot enabled).
This commit is contained in:
Piotr Osiewicz 2025-06-13 19:17:35 +02:00 committed by GitHub
parent cf129aa19d
commit e59fb2e16a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -26,6 +26,7 @@ use parking_lot::Mutex;
use request::StatusNotification;
use settings::SettingsStore;
use sign_in::{reinstall_and_sign_in_within_workspace, sign_out_within_workspace};
use std::collections::hash_map::Entry;
use std::{
any::TypeId,
env,
@ -731,10 +732,11 @@ impl Copilot {
return;
}
registered_buffers
.entry(buffer.entity_id())
.or_insert_with(|| {
let uri: lsp::Url = uri_for_buffer(buffer, cx);
let entry = registered_buffers.entry(buffer.entity_id());
if let Entry::Vacant(e) = entry {
let Ok(uri) = uri_for_buffer(buffer, cx) else {
return;
};
let language_id = id_for_language(buffer.read(cx).language());
let snapshot = buffer.read(cx).snapshot();
server
@ -750,7 +752,7 @@ impl Copilot {
)
.ok();
RegisteredBuffer {
e.insert(RegisteredBuffer {
uri,
language_id,
snapshot,
@ -765,10 +767,10 @@ impl Copilot {
this.unregister_buffer(&weak_buffer);
}),
],
}
});
}
}
}
fn handle_buffer_event(
&mut self,
@ -798,7 +800,9 @@ impl Copilot {
language::BufferEvent::FileHandleChanged
| language::BufferEvent::LanguageChanged => {
let new_language_id = id_for_language(buffer.read(cx).language());
let new_uri = uri_for_buffer(&buffer, cx);
let Ok(new_uri) = uri_for_buffer(&buffer, cx) else {
return Ok(());
};
if new_uri != registered_buffer.uri
|| new_language_id != registered_buffer.language_id
{
@ -1068,11 +1072,13 @@ fn id_for_language(language: Option<&Arc<Language>>) -> String {
.unwrap_or_else(|| "plaintext".to_string())
}
fn uri_for_buffer(buffer: &Entity<Buffer>, cx: &App) -> lsp::Url {
fn uri_for_buffer(buffer: &Entity<Buffer>, cx: &App) -> Result<lsp::Url, ()> {
if let Some(file) = buffer.read(cx).file().and_then(|file| file.as_local()) {
lsp::Url::from_file_path(file.abs_path(cx)).unwrap()
lsp::Url::from_file_path(file.abs_path(cx))
} else {
format!("buffer://{}", buffer.entity_id()).parse().unwrap()
format!("buffer://{}", buffer.entity_id())
.parse()
.map_err(|_| ())
}
}