1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
//===--- SelectedOverloadInfo.cpp -----------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2022 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
#include "swift/IDE/SelectedOverloadInfo.h"
using namespace swift::ide;
SelectedOverloadInfo
swift::ide::getSelectedOverloadInfo(const Solution &S,
ConstraintLocator *CalleeLocator) {
auto &CS = S.getConstraintSystem();
SelectedOverloadInfo Result;
auto SelectedOverload = S.getOverloadChoiceIfAvailable(CalleeLocator);
if (!SelectedOverload) {
return Result;
}
switch (SelectedOverload->choice.getKind()) {
case OverloadChoiceKind::KeyPathApplication:
case OverloadChoiceKind::Decl:
case OverloadChoiceKind::DeclViaDynamic:
case OverloadChoiceKind::DeclViaBridge:
case OverloadChoiceKind::DeclViaUnwrappedOptional: {
Result.BaseTy = SelectedOverload->choice.getBaseType();
if (Result.BaseTy) {
Result.BaseTy = S.simplifyType(Result.BaseTy)->getRValueType();
}
if (Result.BaseTy && Result.BaseTy->is<ModuleType>()) {
Result.BaseTy = nullptr;
}
if (auto ReferencedDecl = SelectedOverload->choice.getDeclOrNull()) {
Result.ValueRef = S.resolveConcreteDeclRef(ReferencedDecl, CalleeLocator);
}
Result.ValueTy =
S.simplifyTypeForCodeCompletion(SelectedOverload->adjustedOpenedType);
// For completion as the arg in a call to the implicit [keypath: _]
// subscript the solver can't know what kind of keypath is expected without
// an actual argument (e.g. a KeyPath vs WritableKeyPath) so it ends up as a
// hole. Just assume KeyPath so we show the expected keypath's root type to
// users rather than '_'.
if (SelectedOverload->choice.getKind() ==
OverloadChoiceKind::KeyPathApplication) {
auto Params = Result.ValueTy->getAs<AnyFunctionType>()->getParams();
if (Params.size() == 1 &&
Params[0].getPlainType()->is<UnresolvedType>()) {
auto *KPDecl = CS.getASTContext().getKeyPathDecl();
Type KPTy =
KPDecl->mapTypeIntoContext(KPDecl->getDeclaredInterfaceType());
Type KPValueTy = KPTy->castTo<BoundGenericType>()->getGenericArgs()[1];
KPTy =
BoundGenericType::get(KPDecl, Type(), {Result.BaseTy, KPValueTy});
Result.ValueTy =
FunctionType::get({Params[0].withType(KPTy)}, KPValueTy);
}
}
break;
}
case OverloadChoiceKind::KeyPathDynamicMemberLookup: {
auto *fnType = SelectedOverload->adjustedOpenedType->castTo<FunctionType>();
assert(fnType->getParams().size() == 1 &&
"subscript always has one argument");
// Parameter type is KeyPath<T, U> where `T` is a root type
// and U is a leaf type (aka member type).
auto keyPathTy =
fnType->getParams()[0].getPlainType()->castTo<BoundGenericType>();
auto *keyPathDecl = keyPathTy->getAnyNominal();
assert(isKnownKeyPathType(keyPathTy) &&
"parameter is supposed to be a keypath");
auto KeyPathDynamicLocator = CS.getConstraintLocator(
CalleeLocator, LocatorPathElt::KeyPathDynamicMember(keyPathDecl));
Result = getSelectedOverloadInfo(S, KeyPathDynamicLocator);
break;
}
case OverloadChoiceKind::DynamicMemberLookup:
case OverloadChoiceKind::TupleIndex:
case OverloadChoiceKind::MaterializePack:
case OverloadChoiceKind::ExtractFunctionIsolation:
// If it's DynamicMemberLookup, we don't know which function is being
// called, so we can't extract any information from it.
// TupleIndex isn't a function call and is not relevant for argument
// completion because it doesn't take arguments.
break;
}
return Result;
}
|
