refactor(ast_tools): shorten code (oxc-project#6924)

Pure refactor.
Orenbek · Oct 28, 2024 · 2818a10 · 2818a10
1 parent 6a8b0ae
commit 2818a10
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/tasks/ast_tools/src/derives/mod.rs b/tasks/ast_tools/src/derives/mod.rs
@@ -81,26 +81,20 @@ pub trait Derive {
                 |mut acc, (def, stream)| {
                     let module_path = def.module_path();
                     let krate = module_path.split("::").next().unwrap();
-                    if !acc.contains_key(krate) {
-                        acc.insert(krate, Default::default());
-                    }
-                    let streams = acc.get_mut(krate).expect("We checked this right above!");
+                    let streams = acc.entry(krate).or_default();
                     streams.0.insert(module_path);
                     streams.1.push(stream);
                     acc
                 },
             )
             .into_iter()
             .sorted_by(|lhs, rhs| lhs.0.cmp(rhs.0))
-            .fold(Vec::new(), |mut acc, (path, (modules, streams))| {
+            .fold(Vec::new(), |mut acc, (krate, (modules, streams))| {
                 let mut modules = Vec::from_iter(modules);
                 modules.sort_unstable();
 
                 let output = Output::Rust {
-                    path: output_path(
-                        format!("crates/{}", path.split("::").next().unwrap()).as_str(),
-                        &filename,
-                    ),
+                    path: output_path(&format!("crates/{krate}"), &filename),
                     tokens: Self::template(
                         modules,
                         streams.into_iter().fold(TokenStream::new(), |mut acc, it| {