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
|
//===--- KeyPathCompletion.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/KeyPathCompletion.h"
#include "swift/IDE/CodeCompletion.h"
#include "swift/IDE/CompletionLookup.h"
#include "swift/Sema/ConstraintSystem.h"
using namespace swift;
using namespace swift::constraints;
using namespace swift::ide;
void KeyPathTypeCheckCompletionCallback::sawSolutionImpl(
const constraints::Solution &S) {
// Determine the code completion.
size_t ComponentIndex = 0;
for (auto &Component : KeyPath->getComponents()) {
if (Component.getKind() == KeyPathExpr::Component::Kind::CodeCompletion) {
break;
} else {
ComponentIndex++;
}
}
assert(ComponentIndex < KeyPath->getComponents().size() &&
"Didn't find a code compleiton component?");
Type BaseType;
if (ComponentIndex == 0) {
// We are completing on the root and need to extract the key path's root
// type.
if (auto *rootTy = KeyPath->getExplicitRootType()) {
BaseType = S.getResolvedType(rootTy);
} else {
// The key path doesn't have a root TypeRepr set, so we can't look the key
// path's root up through it. Build a constraint locator and look the
// root type up through it.
// FIXME: Improve the linear search over S.typeBindings when it's possible
// to look up type variables by their locators.
auto RootLocator =
S.getConstraintLocator(KeyPath, {ConstraintLocator::KeyPathRoot});
auto BaseVariableTypeBinding =
llvm::find_if(S.typeBindings, [&RootLocator](const auto &Entry) {
return Entry.first->getImpl().getLocator() == RootLocator;
});
if (BaseVariableTypeBinding != S.typeBindings.end()) {
BaseType = S.simplifyType(BaseVariableTypeBinding->getSecond());
}
}
} else {
// We are completing after a component. Get the previous component's result
// type.
BaseType = S.simplifyType(S.getType(KeyPath, ComponentIndex - 1));
}
if (BaseType.isNull()) {
return;
}
// If ExpectedTy is a duplicate of any other result, ignore this solution.
if (llvm::any_of(Results, [&](const Result &R) {
return R.BaseType->isEqual(BaseType);
})) {
return;
}
Results.push_back({BaseType, /*OnRoot=*/(ComponentIndex == 0)});
}
void KeyPathTypeCheckCompletionCallback::collectResults(
DeclContext *DC, SourceLoc DotLoc,
ide::CodeCompletionContext &CompletionCtx) {
ASTContext &Ctx = DC->getASTContext();
CompletionLookup Lookup(CompletionCtx.getResultSink(), Ctx, DC,
&CompletionCtx);
if (DotLoc.isValid()) {
Lookup.setHaveDot(DotLoc);
}
Lookup.shouldCheckForDuplicates(Results.size() > 1);
for (auto &Result : Results) {
Lookup.setIsSwiftKeyPathExpr(Result.OnRoot);
Lookup.getValueExprCompletions(Result.BaseType);
}
collectCompletionResults(CompletionCtx, Lookup, DC, ExpectedTypeContext(),
/*CanCurrDeclContextHandleAsync=*/false);
}
|
