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
|
//===--- AfterPoundExprCompletion.cpp -------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 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/AfterPoundExprCompletion.h"
#include "swift/IDE/CodeCompletion.h"
#include "swift/IDE/CompletionLookup.h"
#include "swift/Sema/CompletionContextFinder.h"
#include "swift/Sema/ConstraintSystem.h"
#include "swift/Sema/IDETypeChecking.h"
using namespace swift;
using namespace swift::constraints;
using namespace swift::ide;
void AfterPoundExprCompletion::sawSolutionImpl(const constraints::Solution &S) {
Type ExpectedTy = getTypeForCompletion(S, CompletionExpr);
bool IsAsync = isContextAsync(S, DC);
// If ExpectedTy is a duplicate of any other result, ignore this solution.
auto IsEqual = [&](const Result &R) {
return R.ExpectedTy->isEqual(ExpectedTy);
};
if (!llvm::any_of(Results, IsEqual)) {
bool IsImpliedResult = isImpliedResult(S, CompletionExpr);
Results.push_back({ExpectedTy, IsImpliedResult, IsAsync});
}
}
void AfterPoundExprCompletion::collectResults(
ide::CodeCompletionContext &CompletionCtx) {
ASTContext &Ctx = DC->getASTContext();
CompletionLookup Lookup(CompletionCtx.getResultSink(), Ctx, DC,
&CompletionCtx);
Lookup.shouldCheckForDuplicates(Results.size() > 1);
// The type context that is being used for global results.
ExpectedTypeContext UnifiedTypeContext;
UnifiedTypeContext.setPreferNonVoid(true);
for (auto &Result : Results) {
Lookup.setExpectedTypes({Result.ExpectedTy}, Result.IsImpliedResult,
/*expectsNonVoid=*/true);
Lookup.addPoundAvailable(ParentStmtKind);
Lookup.addObjCPoundKeywordCompletions(/*needPound=*/false);
Lookup.getMacroCompletions(CodeCompletionMacroRole::Expression);
UnifiedTypeContext.merge(*Lookup.getExpectedTypeContext());
}
collectCompletionResults(CompletionCtx, Lookup, DC, UnifiedTypeContext,
/*CanCurrDeclContextHandleAsync=*/false);
}
|
