rust_analyzer/cli/
ssr.rs

1//! Applies structured search replace rules from the command line.
2
3use anyhow::Context;
4use ide_db::{EditionedFileId, base_db::SourceDatabase};
5use ide_ssr::MatchFinder;
6use load_cargo::{LoadCargoConfig, ProcMacroServerChoice, load_workspace_at};
7use project_model::{CargoConfig, RustLibSource};
8
9use crate::cli::flags;
10
11impl flags::Ssr {
12    pub fn run(self) -> anyhow::Result<()> {
13        let cargo_config = CargoConfig {
14            sysroot: Some(RustLibSource::Discover),
15            all_targets: true,
16            set_test: true,
17            ..Default::default()
18        };
19        let load_cargo_config = LoadCargoConfig {
20            load_out_dirs_from_check: true,
21            with_proc_macro_server: ProcMacroServerChoice::Sysroot,
22            prefill_caches: false,
23        };
24        let (ref db, vfs, _proc_macro) = load_workspace_at(
25            &std::env::current_dir()?,
26            &cargo_config,
27            &load_cargo_config,
28            &|_| {},
29        )?;
30        let mut match_finder = MatchFinder::at_first_file(db)?;
31        for rule in self.rule {
32            match_finder.add_rule(rule)?;
33        }
34        let edits = match_finder.edits();
35        for (file_id, edit) in edits {
36            if let Some(path) = vfs.file_path(file_id).as_path() {
37                let mut contents = db.file_text(file_id).text(db).to_string();
38                edit.apply(&mut contents);
39                std::fs::write(path, contents)
40                    .with_context(|| format!("failed to write {path}"))?;
41            }
42        }
43        Ok(())
44    }
45}
46
47impl flags::Search {
48    /// Searches for `patterns`, printing debug information for any nodes whose text exactly matches
49    /// `debug_snippet`. This is intended for debugging and probably isn't in it's current form useful
50    /// for much else.
51    pub fn run(self) -> anyhow::Result<()> {
52        use ide_db::base_db::SourceDatabase;
53        let cargo_config =
54            CargoConfig { all_targets: true, set_test: true, ..CargoConfig::default() };
55        let load_cargo_config = LoadCargoConfig {
56            load_out_dirs_from_check: true,
57            with_proc_macro_server: ProcMacroServerChoice::Sysroot,
58            prefill_caches: false,
59        };
60        let (ref db, _vfs, _proc_macro) = load_workspace_at(
61            &std::env::current_dir()?,
62            &cargo_config,
63            &load_cargo_config,
64            &|_| {},
65        )?;
66        let mut match_finder = MatchFinder::at_first_file(db)?;
67        for pattern in self.pattern {
68            match_finder.add_search_pattern(pattern)?;
69        }
70        if let Some(debug_snippet) = &self.debug {
71            for &root in ide_db::symbol_index::LocalRoots::get(db).roots(db).iter() {
72                let sr = db.source_root(root).source_root(db);
73                for file_id in sr.iter() {
74                    for debug_info in match_finder.debug_where_text_equal(
75                        EditionedFileId::current_edition_guess_origin(db, file_id),
76                        debug_snippet,
77                    ) {
78                        println!("{debug_info:#?}");
79                    }
80                }
81            }
82        } else {
83            for m in match_finder.matches().flattened().matches {
84                // We could possibly at some point do something more useful than just printing
85                // the matched text. For now though, that's the easiest thing to do.
86                println!("{}", m.matched_text());
87            }
88        }
89        Ok(())
90    }
91}
rust_analyzer/cli/ssr.rs

rust_analyzer/cli/
ssr.rs
