gered/llm-ls
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

fix: nameless file crash (#84)

Browse source
This commit is contained in:
Luc Georges 2024-02-19 12:08:32 +01:00 committed by GitHub
parent 50b62bd107
commit 0d035901f0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 22 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
crates/llm-ls/src/document.rs
View file

@ -176,7 +176,7 @@ impl Document {
let rope = Rope::from_str(text);
let text_len = rope.len_chars();
let end_idx = idx + text_len;
self.text.insert(idx, text);
self.text.try_insert(idx, text)?;
(
end_idx,
Point {
@ -189,7 +189,7 @@ impl Document {
+ (range.end.character as usize);
let slice_size = removal_idx - start_idx;
self.text.try_remove(start_idx..removal_idx)?;
self.text.insert(start_idx, text);
self.text.try_insert(start_idx, text)?;
let rope = Rope::from_str(text);
let text_len = rope.len_chars();
let character_difference = text_len as isize - slice_size as isize;

24
crates/llm-ls/src/main.rs
View file

@ -429,12 +429,22 @@ impl LlmService {
) -> LspResult<GetCompletionsResult> {
let request_id = Uuid::new_v4();
let span = info_span!("completion_request", %request_id);
async move {
let document_map = self.document_map.read().await;
let document = document_map
.get(params.text_document_position.text_document.uri.as_str())
.ok_or_else(|| internal_error("failed to find document"))?;
let document =
match document_map.get(params.text_document_position.text_document.uri.as_str()) {
Some(doc) => doc,
None => {
debug!("failed to find document");
return Ok(GetCompletionsResult {
request_id,
completions: vec![],
});
}
};
info!(
document_url = %params.text_document_position.text_document.uri,
cursor_line = ?params.text_document_position.position.line,
@ -545,6 +555,9 @@ impl LanguageServer for LlmService {
async fn did_open(&self, params: DidOpenTextDocumentParams) {
let uri = params.text_document.uri.to_string();
if uri == "file:///" {
return;
}
match Document::open(
&params.text_document.language_id,
&params.text_document.text,
@ -567,6 +580,9 @@ impl LanguageServer for LlmService {
async fn did_change(&self, params: DidChangeTextDocumentParams) {
let uri = params.text_document.uri.to_string();
if uri == "file:///" {
return;
}
if params.content_changes.is_empty() {
return;
}
@ -593,7 +609,7 @@ impl LanguageServer for LlmService {
}
}
} else {
warn!("textDocument/didChange {uri}: document not found");
debug!("textDocument/didChange {uri}: document not found");
}
}