3 use ast::edit::IndentLevel;
4 use ide_db::base_db::AnchoredPathBuf;
5 use itertools::Itertools;
8 ast::{self, edit::AstNodeEdit, HasName},
9 AstNode, SmolStr, TextRange,
12 use crate::{AssistContext, AssistId, AssistKind, Assists};
14 // Assist: move_module_to_file
16 // Moves inline module's contents to a separate file.
27 pub(crate) fn move_module_to_file(acc: &mut Assists, ctx: &AssistContext<'_>) -> Option<()> {
28 let module_ast = ctx.find_node_at_offset::<ast::Module>()?;
29 let module_items = module_ast.item_list()?;
31 let l_curly_offset = module_items.syntax().text_range().start();
32 if l_curly_offset <= ctx.offset() {
33 cov_mark::hit!(available_before_curly);
36 let target = TextRange::new(module_ast.syntax().text_range().start(), l_curly_offset);
38 let module_name = module_ast.name()?;
40 // get to the outermost module syntax so we can grab the module of file we are in
41 let outermost_mod_decl =
42 iter::successors(Some(module_ast.clone()), |module| module.parent()).last()?;
43 let module_def = ctx.sema.to_def(&outermost_mod_decl)?;
44 let parent_module = module_def.parent(ctx.db())?;
47 AssistId("move_module_to_file", AssistKind::RefactorExtract),
48 "Extract module to file",
52 let mut buf = String::from("./");
53 match parent_module.name(ctx.db()) {
54 Some(name) if !parent_module.is_mod_rs(ctx.db()) => {
55 format_to!(buf, "{name}/")
59 let segments = iter::successors(Some(module_ast.clone()), |module| module.parent())
60 .filter_map(|it| it.name())
61 .map(|name| SmolStr::from(name.text().trim_start_matches("r#")))
64 format_to!(buf, "{}", segments.into_iter().rev().format("/"));
66 // We need to special case mod named `r#mod` and place the file in a
67 // subdirectory as "mod.rs" would be of its parent module otherwise.
68 if module_name.text() == "r#mod" {
69 format_to!(buf, "/mod.rs");
71 format_to!(buf, ".rs");
76 let items = module_items.dedent(IndentLevel(1)).to_string();
78 items.trim_start_matches('{').trim_end_matches('}').trim().to_string();
79 if !items.is_empty() {
85 let buf = format!("mod {module_name};");
87 let replacement_start = match module_ast.mod_token() {
88 Some(mod_token) => mod_token.text_range(),
89 None => module_ast.syntax().text_range(),
94 TextRange::new(replacement_start, module_ast.syntax().text_range().end()),
98 let dst = AnchoredPathBuf { anchor: ctx.file_id(), path };
99 builder.create_file(dst, contents);
106 use crate::tests::{check_assist, check_assist_not_applicable};
111 fn extract_from_root() {
129 fn extract_from_submodule() {
152 fn extract_from_mod_rs() {
158 //- /submodule/mod.rs
165 //- /submodule/mod.rs
168 //- /submodule/inner.rs
175 fn extract_public() {
193 fn extract_public_crate() {
197 pub(crate) mod $0tests {
203 pub(crate) mod tests;
211 fn available_before_curly() {
212 cov_mark::check!(available_before_curly);
213 check_assist_not_applicable(move_module_to_file, r#"mod m { $0 }"#);
217 fn keep_outer_comments_and_attributes() {
239 fn extract_nested() {
259 //- /foo/bar/baz/qux.rs
265 fn extract_mod_with_raw_ident() {
297 fn extract_r_mod_from_mod_rs() {
315 fn extract_nested_r_mod() {
333 //- /mod/foo/mod/mod.rs