wow

src/everybody_loops.rs

@@ -1,22 +1,28 @@
+use quote::ToTokens;
 use syn::{parse_quote, visit_mut::VisitMut};
 
-use crate::processor::Processor;
+use crate::processor::{ProcessChecker, Processor};
+
+struct Visitor<'a> {
+    current_path: Vec<String>,
+    checker: &'a mut ProcessChecker,
 
-struct Visitor {
     loop_expr: syn::Block,
     has_made_change: bool,
 }
 
-impl Visitor {
-    fn new() -> Self {
+impl<'a> Visitor<'a> {
+    fn new(checker: &'a mut ProcessChecker) -> Self {
         Self {
+            current_path: Vec::new(),
+            checker,
             has_made_change: false,
             loop_expr: parse_quote! { { loop {} } },
         }
     }
 }
 
-impl VisitMut for Visitor {
+impl VisitMut for Visitor<'_> {
     fn visit_block_mut(&mut self, block: &mut syn::Block) {
         match block.stmts.as_slice() {
             [syn::Stmt::Expr(syn::Expr::Loop(syn::ExprLoop {
@@ -28,14 +34,39 @@ impl VisitMut for Visitor {
             }
         }
     }
+
+    fn visit_item_fn_mut(&mut self, func: &mut syn::ItemFn) {
+        self.current_path.push(func.sig.ident.to_string());
+        syn::visit_mut::visit_item_fn_mut(self, func);
+        self.current_path.pop();
+    }
+
+    fn visit_impl_item_method_mut(&mut self, method: &mut syn::ImplItemMethod) {
+        self.current_path.push(method.sig.ident.to_string());
+        syn::visit_mut::visit_impl_item_method_mut(self, method);
+        self.current_path.pop();
+    }
+
+    fn visit_item_impl_mut(&mut self, item: &mut syn::ItemImpl) {
+        self.current_path
+            .push(item.self_ty.clone().into_token_stream().to_string());
+        syn::visit_mut::visit_item_impl_mut(self, item);
+        self.current_path.pop();
+    }
+
+    fn visit_item_mod_mut(&mut self, module: &mut syn::ItemMod) {
+        self.current_path.push(module.ident.to_string());
+        syn::visit_mut::visit_item_mod_mut(self, module);
+        self.current_path.pop();
+    }
 }
 
 #[derive(Default)]
 pub struct EverybodyLoops;
 
 impl Processor for EverybodyLoops {
-    fn process_file(&mut self, krate: &mut syn::File) -> bool {
-        let mut visitor = Visitor::new();
+    fn process_file(&mut self, krate: &mut syn::File, checker: &mut ProcessChecker) -> bool {
+        let mut visitor = Visitor::new(checker);
         visitor.visit_file_mut(krate);
         visitor.has_made_change
     }