mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
d7fc22aaca
Use the generalized implied result logic, and rename to `isImpliedResult` since that's really what we're querying here, and it needs to handle implicit-last-exprs if enabled.
64 lines
2.3 KiB
C++
64 lines
2.3 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) {
|
|
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);
|
|
}
|