aster/aster-lsp/src/rename.rs

use std::collections::HashMap;

use aster_core::analysis;
use aster_core::lexer::Lexer;
use lsp_server::{Connection, Message, Request};
use lsp_types::{Position, Range, RenameParams, TextEdit, Uri, WorkspaceEdit};

use crate::completion;

pub fn handle_rename(
    documents: &HashMap<Uri, String>,
    connection: &Connection,
    req: Request,
) {
    let id = req.id;
    let params: RenameParams = match serde_json::from_value(req.params) {
        Ok(p) => p,
        Err(e) => {
            eprintln!("Invalid rename params: {}", e);
            return;
        }
    };

    let uri = params.text_document_position.text_document.uri;
    let position = params.text_document_position.position;
    let new_name = params.new_name;

    let text = documents.get(&uri).map(|s| s.as_str()).unwrap_or("");
    let word = completion::get_word_at_position(text, position);

    if word.is_empty() || word == new_name {
        let resp = lsp_server::Response::new_ok::<Option<WorkspaceEdit>>(id, None);
        let _ = connection.sender.send(Message::Response(resp));
        return;
    }

    let (tokens, _) = Lexer::new(text).tokenize();
    let positions = analysis::find_all_references(&tokens, &word);

    let edits: Vec<TextEdit> = positions
        .into_iter()
        .map(|(line, col)| {
            let l = (line as u32).saturating_sub(1);
            let c = (col as u32).saturating_sub(1);
            TextEdit {
                range: Range {
                    start: Position {
                        line: l,
                        character: c,
                    },
                    end: Position {
                        line: l,
                        character: c + word.len() as u32,
                    },
                },
                new_text: new_name.clone(),
            }
        })
        .collect();

    let mut changes = HashMap::new();
    changes.insert(uri, edits);

    let edit = WorkspaceEdit {
        changes: Some(changes),
        ..Default::default()
    };

    let resp = lsp_server::Response::new_ok(id, edit);
    let _ = connection.sender.send(Message::Response(resp));
}