item deleter pass

This commit is contained in:
nora 2023-01-22 13:19:18 +01:00
parent 22804c3065
commit fdfde615f6
8 changed files with 162 additions and 5 deletions

View file

@ -28,6 +28,8 @@ impl VisitMut for Visitor<'_> {
[syn::Stmt::Expr(syn::Expr::Loop(syn::ExprLoop {
body: loop_body, ..
}))] if loop_body.stmts.is_empty() => {}
// Empty bodies are empty already, no need to loopify them.
[] => {}
_ if self.checker.can_process(&self.current_path) => {
*block = self.loop_expr.clone();
self.process_state = ProcessState::Changed;

111
src/passes/item_deleter.rs Normal file
View file

@ -0,0 +1,111 @@
use quote::ToTokens;
use syn::{
visit_mut::VisitMut, Item, ItemConst, ItemEnum, ItemMacro, ItemMacro2, ItemMod, ItemStatic,
ItemStruct, ItemTrait, ItemType, ItemUnion,
};
use crate::processor::{tracking, Pass, PassController, ProcessState, SourceFile};
struct Visitor<'a> {
current_path: Vec<String>,
checker: &'a mut PassController,
process_state: ProcessState,
}
impl<'a> Visitor<'a> {
fn new(checker: &'a mut PassController) -> Self {
Self {
current_path: Vec::new(),
checker,
process_state: ProcessState::NoChange,
}
}
fn should_retain_item(&mut self) -> bool {
let can_process = self.checker.can_process(&self.current_path);
if can_process {
self.process_state = ProcessState::Changed;
}
!can_process
}
fn consider_deleting_item(&mut self, item: &Item) -> bool {
match item {
// N.B. Do not delete ItemFn because that makes testing way harder
// and also the dead_lint should cover it all.
Item::Impl(impl_) => {
self.current_path
.push(impl_.self_ty.clone().into_token_stream().to_string());
let should_retain = self.should_retain_item();
self.current_path.pop();
should_retain
}
Item::Struct(ItemStruct { ident, .. })
| Item::Enum(ItemEnum { ident, .. })
| Item::Union(ItemUnion { ident, .. })
| Item::Const(ItemConst { ident, .. })
| Item::Type(ItemType { ident, .. })
| Item::Trait(ItemTrait { ident, .. })
| Item::Macro(ItemMacro {
ident: Some(ident), ..
})
| Item::Macro2(ItemMacro2 { ident, .. })
| Item::Static(ItemStatic { ident, .. })
| Item::Mod(ItemMod { ident, .. }) => {
self.current_path.push(ident.to_string());
let should_retain = self.should_retain_item();
self.current_path.pop();
should_retain
}
_ => true,
}
}
}
impl VisitMut for Visitor<'_> {
fn visit_file_mut(&mut self, file: &mut syn::File) {
file.items
.retain_mut(|item| self.consider_deleting_item(item));
syn::visit_mut::visit_file_mut(self, file);
}
fn visit_item_mod_mut(&mut self, module: &mut syn::ItemMod) {
self.current_path.push(module.ident.to_string());
if let Some((_, items)) = &mut module.content {
items.retain(|item| self.consider_deleting_item(item));
}
syn::visit_mut::visit_item_mod_mut(self, module);
self.current_path.pop();
}
tracking!(visit_item_fn_mut);
tracking!(visit_impl_item_method_mut);
tracking!(visit_item_impl_mut);
}
#[derive(Default)]
pub struct ItemDeleter;
impl Pass for ItemDeleter {
fn process_file(
&mut self,
krate: &mut syn::File,
_: &SourceFile,
checker: &mut PassController,
) -> ProcessState {
let mut visitor = Visitor::new(checker);
visitor.visit_file_mut(krate);
visitor.process_state
}
fn name(&self) -> &'static str {
"item-deleter"
}
}

View file

@ -1,4 +1,5 @@
mod everybody_loops;
mod item_deleter;
mod privatize;
pub use self::{everybody_loops::EverybodyLoops, privatize::Privatize};
pub use self::{everybody_loops::EverybodyLoops, item_deleter::ItemDeleter, privatize::Privatize};