rhs_first_assign 0.1.0

//! An attribute macro to hack compound assignment.
//!
//! ## Motivation
//!
//! ```compile_fail
//! use std::num::Wrapping;
//!
//! fn main() {
//!     let mut xs = vec![Wrapping(1), Wrapping(2)];
//!
//!     // OK
//!     xs[1] = xs[0] + xs[1];
//!
//!     // Error
//!     xs[1] += xs[0];
//! }
//! ```
//!
//! ```text
//! error[E0502]: cannot borrow `xs` as immutable because it is also borrowed as mutable
//!   --> src/main.rs:10:14
//!    |
//! 10 |     xs[1] += xs[0];
//!    |     ---------^^---
//!    |     |        |
//!    |     |        immutable borrow occurs here
//!    |     mutable borrow occurs here
//!    |     mutable borrow later used here
//! ```
//!
//! ## Usage
//!
//! ```
//! use rhs_first_assign::rhs_first_assign;
//!
//! use std::num::Wrapping;
//!
//! #[rhs_first_assign]
//! fn main() {
//!     let mut xs = vec![Wrapping(1), Wrapping(2)];
//!
//!     xs[1] = xs[0] + xs[1];
//!
//!     xs[1] += xs[0];
//! }
//! ```
//!
//! ↓
//!
//! ```
//! use std::num::Wrapping;
//!
//! fn main() {
//!     let mut xs = vec![Wrapping(1), Wrapping(2)];
//!
//!     xs[1] = xs[0] + xs[1];
//!
//!     {
//!         let __rhs_first_assign_rhs_l11_c10 = xs[0];
//!         xs[1] += __rhs_first_assign_rhs_l11_c10;
//!     };
//! }
//! ```

#![allow(clippy::needless_doctest_main)]

extern crate proc_macro;

use if_chain::if_chain;
use proc_macro2::{LineColumn, Span};
use quote::quote;
use syn::spanned::Spanned as _;
use syn::visit_mut::{self, VisitMut};
use syn::{parse_macro_input, parse_quote, Block, Expr, ExprAssignOp, Ident, ItemFn, Stmt};

#[proc_macro_attribute]
pub fn rhs_first_assign(
    attr: proc_macro::TokenStream,
    input: proc_macro::TokenStream,
) -> proc_macro::TokenStream {
    let attr = proc_macro2::TokenStream::from(attr);
    let mut input = parse_macro_input!(input as ItemFn);
    let mut stmts = input.block.stmts.clone();
    for stmt in &mut stmts {
        Visitor.visit_stmt_mut(stmt);
    }
    input.block = Box::new(Block {
        brace_token: input.block.brace_token,
        stmts,
    });
    quote!(#attr #input).into()
}

struct Visitor;

impl VisitMut for Visitor {
    fn visit_stmt_mut(&mut self, stmt: &mut Stmt) {
        if_chain! {
            if let Stmt::Expr(expr) | Stmt::Semi(expr, _) = stmt;
            if let Expr::AssignOp(ExprAssignOp {
                attrs,
                left,
                op,
                right,
            }) = expr;
            then {
                let LineColumn { line, column } = op.span().start();
                let rhs = format!("__rhs_first_assign_rhs_l{}_c{}", line, column);
                let rhs = Ident::new(&rhs, Span::call_site());
                *expr = parse_quote!(
                    #(#attrs)*
                    {
                        let #rhs = #right;
                        #left #op #rhs;
                    }
                );
            } else {
                visit_mut::visit_stmt_mut(self, stmt);
            }
        }
    }
}