mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
72cadecf21
This will allow us to run two different completion kinds and deliver results from both of them. Also: Compute a unified type context for global lookup. Previously, we always used the expected type context of the last lookup. But really, we should be considering all possible types from all constraint system solutions when computing code completion results from the cache.
66 lines
2.4 KiB
C++
66 lines
2.4 KiB
C++
//===--- 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) {
|
|
auto &CS = S.getConstraintSystem();
|
|
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 SingleExprBody = isImplicitSingleExpressionReturn(CS, CompletionExpr);
|
|
Results.push_back({ExpectedTy, SingleExprBody, 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.IsImplicitSingleExpressionReturn,
|
|
/*expectsNonVoid=*/true);
|
|
Lookup.addPoundAvailable(ParentStmtKind);
|
|
Lookup.addObjCPoundKeywordCompletions(/*needPound=*/false);
|
|
Lookup.getMacroCompletions(CodeCompletionMacroRole::Expression);
|
|
|
|
UnifiedTypeContext.merge(*Lookup.getExpectedTypeContext());
|
|
}
|
|
|
|
collectCompletionResults(CompletionCtx, Lookup, DC, UnifiedTypeContext,
|
|
/*CanCurrDeclContextHandleAsync=*/false);
|
|
}
|