@@ -1207,10 +1207,12 @@ bool SwiftASTManipulator::AddExternalVariables(
1207
1207
wrapper_element_iterator = wrapper_elements_copy.insert (
1208
1208
wrapper_element_iterator, swift::ASTNode (redirected_var_decl));
1209
1209
1210
- m_wrapper_decl-> setBody ( swift::BraceStmt::create (
1210
+ auto *new_wrapper_body = swift::BraceStmt::create (
1211
1211
ast_context, wrapper_body->getLBraceLoc (),
1212
1212
ast_context.AllocateCopy (wrapper_elements_copy),
1213
- wrapper_body->getRBraceLoc ()));
1213
+ wrapper_body->getRBraceLoc ());
1214
+ m_wrapper_decl->setBody (new_wrapper_body,
1215
+ m_wrapper_decl->getBodyKind ());
1214
1216
} else {
1215
1217
element_iterator =
1216
1218
elements.insert (element_iterator, swift::ASTNode (pattern_binding));
@@ -1234,9 +1236,10 @@ bool SwiftASTManipulator::AddExternalVariables(
1234
1236
m_variables.push_back (variable);
1235
1237
}
1236
1238
1237
- m_function_decl-> setBody ( swift::BraceStmt::create (
1239
+ auto *new_function_body = swift::BraceStmt::create (
1238
1240
ast_context, body->getLBraceLoc (), ast_context.AllocateCopy (elements),
1239
- body->getRBraceLoc ()));
1241
+ body->getRBraceLoc ());
1242
+ m_function_decl->setBody (new_function_body, m_function_decl->getBodyKind ());
1240
1243
}
1241
1244
1242
1245
return true ;
0 commit comments