Skip to content

Commit

Permalink
flip conditional 🥞
Browse files Browse the repository at this point in the history
  • Loading branch information
pvdrz committed Sep 27, 2022
1 parent 7160f2e commit c2df74d
Showing 1 changed file with 19 additions and 17 deletions.
36 changes: 19 additions & 17 deletions src/codegen/postprocessing/unsafe_blocks.rs
Expand Up @@ -16,24 +16,26 @@ struct Visitor;

impl Visitor {
fn add_unsafe_block(&self, sig: &syn::Signature, block: &mut syn::Block) {
if sig.unsafety.is_some() {
let inner_block = std::mem::replace(
block,
syn::Block {
brace_token: Default::default(),
stmts: Default::default(),
},
);

block.stmts.push(Stmt::Expr(
ExprUnsafe {
attrs: Default::default(),
block: inner_block,
unsafe_token: Default::default(),
}
.into(),
));
if sig.unsafety.is_none() {
return;
}

let inner_block = std::mem::replace(
block,
syn::Block {
brace_token: Default::default(),
stmts: Default::default(),
},
);

block.stmts.push(Stmt::Expr(
ExprUnsafe {
attrs: Default::default(),
block: inner_block,
unsafe_token: Default::default(),
}
.into(),
));
}
}

Expand Down

0 comments on commit c2df74d

Please sign in to comment.