cargo-minimize/src/passes/everybody_loops.rs

use quote::ToTokens;
use syn::{parse_quote, visit_mut::VisitMut};

use crate::processor::{Pass, PassController, ProcessState, SourceFile, tracking};

struct Visitor<'a> {
    current_path: Vec<String>,
    checker: &'a mut PassController,

    loop_expr: syn::Block,
    process_state: ProcessState,
}

impl<'a> Visitor<'a> {
    fn new(checker: &'a mut PassController) -> Self {
        Self {
            current_path: Vec::new(),
            checker,
            process_state: ProcessState::NoChange,
            loop_expr: parse_quote! { { loop {} } },
        }
    }
}

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 {
                    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;
            }
            _ => {}
        }
    }

    tracking!();
}

#[derive(Default)]
pub struct EverybodyLoops;

impl Pass for EverybodyLoops {
    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 {
        "everybody-loops"
    }
}