averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

[Refactoring] Simplify implementation of "move members to extension" action

Browse Source
This commit is contained in:
Rintaro Ishizaki
2018-05-25 18:38:34 +09:00
parent db3020f031
commit 3c367e9227
2 changed files with 8 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
lib/IDE/Refactoring.cpp
View File

@@ -1521,36 +1521,24 @@ bool RefactoringActionExtractRepeatedExpr::performChange() {
EditConsumer).performChange();
}
// Compute a decl context that is the parent context for all decls in
// \c DeclaredDecls. Return \c nullptr if no such context exists.
DeclContext *getCommonDeclContext(ArrayRef<DeclaredDecl> DeclaredDecls) {
if (DeclaredDecls.empty())
return nullptr;
DeclContext *CommonDC = DeclaredDecls.front().VD->getDeclContext();
for (auto DD : DeclaredDecls) {
auto OtherDC = DD.VD->getDeclContext();
CommonDC = DeclContext::getCommonParentContext(CommonDC, OtherDC);
}
return CommonDC;
}
bool RefactoringActionMoveMembersToExtension::isApplicable(
ResolvedRangeInfo Info, DiagnosticEngine &Diag) {
switch (Info.Kind) {
case RangeKind::SingleDecl:
case RangeKind::MultiTypeMemberDecl: {
DeclContext *CommonDC = getCommonDeclContext(Info.DeclaredDecls);
DeclContext *DC = Info.RangeContext;
// The the common decl context is not a nomial type, we cannot create an
// extension for it
if (!CommonDC || !CommonDC->getInnermostDeclarationDeclContext() ||
!isa<NominalTypeDecl>(CommonDC->getInnermostDeclarationDeclContext()))
if (!DC || !DC->getInnermostDeclarationDeclContext() ||
!isa<NominalTypeDecl>(DC->getInnermostDeclarationDeclContext()))
return false;
// Members of types not declared at top file level cannot be extracted
// to an extension at top file level
if (CommonDC->getParent()->getContextKind() != DeclContextKind::FileUnit)
if (DC->getParent()->getContextKind() != DeclContextKind::FileUnit)
return false;
// Check if contained nodes are all allowed decls.
@@ -1570,7 +1558,7 @@ bool RefactoringActionMoveMembersToExtension::isApplicable(
if (auto ASD = dyn_cast<AbstractStorageDecl>(DD.VD)) {
// Only disallow storages in the common decl context, allow them in
// any subtypes
if (ASD->hasStorage() && ASD->getDeclContext() == CommonDC) {
if (ASD->hasStorage() && ASD->getDeclContext() == DC) {
return false;
}
}
@@ -1588,10 +1576,10 @@ bool RefactoringActionMoveMembersToExtension::isApplicable(
}
bool RefactoringActionMoveMembersToExtension::performChange() {
DeclContext *CommonDC = getCommonDeclContext(RangeInfo.DeclaredDecls);
DeclContext *DC = RangeInfo.RangeContext;
auto CommonTypeDecl =
dyn_cast<NominalTypeDecl>(CommonDC->getInnermostDeclarationDeclContext());
dyn_cast<NominalTypeDecl>(DC->getInnermostDeclarationDeclContext());
assert(CommonTypeDecl && "Not applicable if common parent is no nomial type");
SmallString<64> Buffer;