cairo-language-server 2.18.0

use std::{
    collections::HashMap,
    path::{Path, PathBuf},
};

use cairo_lang_defs::{
    db::DefsGroup,
    ids::{ModuleId, NamedLanguageElementId},
};
use cairo_lang_filesystem::{
    db::{CrateConfiguration, FilesGroup},
    ids::Directory,
};
use lsp_types::{FileRename, RenameFilesParams, TextEdit, Url, WorkspaceEdit};
use tracing::error;

use super::db::AnalysisDatabase;
use crate::{
    lang::{
        defs::SymbolSearch,
        lsp::{LsProtoGroup, ToLsp},
    },
    server::is_cairo_file_path,
};

pub fn rename_files(db: &AnalysisDatabase, params: RenameFilesParams) -> Option<WorkspaceEdit> {
    let mut changes: HashMap<Url, Vec<TextEdit>> = Default::default();

    for rename in params.files {
        handle_rename(db, rename, &mut changes);
    }

    Some(WorkspaceEdit { changes: Some(changes), ..Default::default() })
}

fn handle_rename(
    db: &AnalysisDatabase,
    rename: FileRename,
    changes: &mut HashMap<Url, Vec<TextEdit>>,
) -> Option<()> {
    let old_uri = Url::parse(&rename.old_uri).ok()?;
    let new_uri = Url::parse(&rename.new_uri).ok()?;

    if !is_cairo_file_path(&old_uri) || !is_cairo_file_path(&new_uri) {
        return None;
    }

    let file = db.file_for_url(&old_uri)?;

    let first = *db.file_modules(file).ok()?.first()?;

    let submodule = match first {
        ModuleId::CrateRoot(_) | ModuleId::MacroCall { .. } => {
            // If renamed file was src/lib.cairo there is nothing we can do.
            return None;
        }
        ModuleId::Submodule(submodule) => submodule,
    };

    let CrateConfiguration { root: Directory::Real(root), .. } =
        db.crate_config(first.owning_crate(db))?
    else {
        error!("got unexpected virtual directory while renaming files");
        return None;
    };

    let old_path = old_uri.to_file_path().ok()?;
    let new_path = new_uri.to_file_path().ok()?;

    assert!(old_path.starts_with(root));

    if !new_path.starts_with(root) {
        return None;
    }

    let mut prefix = PathBuf::new();
    for (comp1, comp2) in old_path.components().zip(new_path.components()) {
        if comp1 == comp2 {
            prefix.push(comp1);
        } else {
            break;
        }
    }

    let new_file_name = assert_single_component(new_path.strip_prefix(&prefix).unwrap())?;

    let mod_name = submodule.name_identifier(db);

    for usage in
        SymbolSearch::find_definition(db, &mod_name)?.usages(db).include_declaration(true).collect()
    {
        let usage_location = usage.location();
        let file = db.url_for_file(usage_location.file_id)?;
        let range = usage_location.span.position_in_file(db, usage_location.file_id)?;

        changes
            .entry(file)
            .or_default()
            .push(TextEdit { range: range.to_lsp(), new_text: new_file_name.clone() });
    }

    Some(())
}

fn assert_single_component(path: &Path) -> Option<String> {
    let mut components = path.components();
    let result = components
        .next()
        .and_then(|c| c.as_os_str().to_str()?.split(".").next())
        .map(ToString::to_string);

    if components.next().is_some() { None } else { result }
}