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

[ConstraintLocator] Identify whether locator is a result of keypath dynamic member lookup or belongs to keypath subscript component

Browse Source
This commit is contained in:
Pavel Yaskevich
2019-04-07 17:51:23 -07:00
parent c3460f825d
commit 93f9cb1d2a
4 changed files with 39 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
lib/Sema/ConstraintSystem.cpp
View File

@@ -1695,36 +1695,6 @@ isInvalidPartialApplication(ConstraintSystem &cs, const ValueDecl *member,
return {true, level};
}
static bool isResultOfKeyPathDynamicMemberLookup(ConstraintLocator *locator) {
auto path = locator->getPath();
return !path.empty() && path.back().isKeyPathDynamicMember();
}
/// If given anchor + path represents a key path with subscript
/// component at some index.
static bool isKeyPathSubscriptComponent(ConstraintLocator *locator) {
auto *anchor = locator->getAnchor();
auto *KPE = dyn_cast_or_null<KeyPathExpr>(anchor);
if (!KPE)
return false;
using ComponentKind = KeyPathExpr::Component::Kind;
auto path = locator->getPath();
while (!path.empty()) {
auto &last = path.back();
if (!last.isKeyPathComponent()) {
path = path.drop_back();
continue;
}
auto index = last.getValue();
auto &component = KPE->getComponents()[index];
return component.getKind() == ComponentKind::Subscript ||
component.getKind() == ComponentKind::UnresolvedSubscript;
}
return false;
}
void ConstraintSystem::resolveOverload(ConstraintLocator *locator,
Type boundType,
OverloadChoice choice,
@@ -2122,8 +2092,8 @@ void ConstraintSystem::resolveOverload(ConstraintLocator *locator,
}
if (auto *SD = dyn_cast<SubscriptDecl>(decl)) {
if (isResultOfKeyPathDynamicMemberLookup(locator) ||
isKeyPathSubscriptComponent(locator)) {
if (locator->isResultOfKeyPathDynamicMemberLookup() ||
locator->isKeyPathSubscriptComponent()) {
// Subscript type has a format of (Self[.Type) -> (Arg...) -> Result
auto declTy = openedFullType->castTo<FunctionType>();
auto subscriptTy = declTy->getResult()->castTo<FunctionType>();