@@ -1219,10 +1219,12 @@ bool SwiftASTManipulator::AddExternalVariables(
1219
1219
wrapper_element_iterator = wrapper_elements_copy.insert (
1220
1220
wrapper_element_iterator, swift::ASTNode (redirected_var_decl));
1221
1221
1222
- m_wrapper_decl-> setBody ( swift::BraceStmt::create (
1222
+ auto *new_wrapper_body = swift::BraceStmt::create (
1223
1223
ast_context, wrapper_body->getLBraceLoc (),
1224
1224
ast_context.AllocateCopy (wrapper_elements_copy),
1225
- wrapper_body->getRBraceLoc ()));
1225
+ wrapper_body->getRBraceLoc ());
1226
+ m_wrapper_decl->setBody (new_wrapper_body,
1227
+ m_wrapper_decl->getBodyKind ());
1226
1228
} else {
1227
1229
element_iterator =
1228
1230
elements.insert (element_iterator, swift::ASTNode (pattern_binding));
@@ -1246,9 +1248,10 @@ bool SwiftASTManipulator::AddExternalVariables(
1246
1248
m_variables.push_back (variable);
1247
1249
}
1248
1250
1249
- m_function_decl-> setBody ( swift::BraceStmt::create (
1251
+ auto *new_function_body = swift::BraceStmt::create (
1250
1252
ast_context, body->getLBraceLoc (), ast_context.AllocateCopy (elements),
1251
- body->getRBraceLoc ()));
1253
+ body->getRBraceLoc ());
1254
+ m_function_decl->setBody (new_function_body, m_function_decl->getBodyKind ());
1252
1255
}
1253
1256
1254
1257
return true ;
0 commit comments