style fix

Author: coord-e

thrust-macros/src/formula.rs

@@ -33,6 +33,24 @@ pub fn expand(input: TokenStream) -> TokenStream {
         Ok(expr) => expr,
         Err(e) => return e.to_compile_error(),
     };
+
+    // Rewrites each assignment `lhs = rhs` (produced by [`desugar_arrows`] from
+    // `lhs ==> rhs`) into the boolean expression `(!(lhs)) || (rhs)`. Visiting
+    // post-order means nested implications are rewritten innermost-first, so the
+    // right-associative chain `a ==> b ==> c` becomes `!a || (!b || c)`.
+    struct ImplicationRewriter;
+
+    impl VisitMut for ImplicationRewriter {
+        fn visit_expr_mut(&mut self, expr: &mut syn::Expr) {
+            syn::visit_mut::visit_expr_mut(self, expr);
+            if let syn::Expr::Assign(assign) = expr {
+                let left = &assign.left;
+                let right = &assign.right;
+                *expr = syn::parse_quote!((!(#left)) || (#right));
+            }
+        }
+    }
+
     ImplicationRewriter.visit_expr_mut(&mut expr);
 
     expr.into_token_stream()
@@ -105,23 +123,6 @@ fn desugar_arrows(input: TokenStream) -> TokenStream {
     out
 }
 
-/// Rewrites each assignment `lhs = rhs` (produced by [`desugar_arrows`] from
-/// `lhs ==> rhs`) into the boolean expression `(!(lhs)) || (rhs)`. Visiting
-/// post-order means nested implications are rewritten innermost-first, so the
-/// right-associative chain `a ==> b ==> c` becomes `!a || (!b || c)`.
-struct ImplicationRewriter;
-
-impl VisitMut for ImplicationRewriter {
-    fn visit_expr_mut(&mut self, expr: &mut syn::Expr) {
-        syn::visit_mut::visit_expr_mut(self, expr);
-        if let syn::Expr::Assign(assign) = expr {
-            let left = &assign.left;
-            let right = &assign.right;
-            *expr = syn::parse_quote!((!(#left)) || (#right));
-        }
-    }
-}
-
 #[cfg(test)]
 mod tests {
     use super::*;

thrust-macros/src/spec.rs

@@ -554,7 +554,7 @@ fn rewrite_self_in_expr(expr: &mut syn::Expr) {
 }
 
 /// Replaces every `self` identifier with `self_`, recursing into groups.
-pub(crate) fn rewrite_self_in_tokens(tokens: TokenStream2) -> TokenStream2 {
+pub fn rewrite_self_in_tokens(tokens: TokenStream2) -> TokenStream2 {
     use proc_macro2::TokenTree;
     tokens
         .into_iter()