mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
115 lines
4.4 KiB
C++
115 lines
4.4 KiB
C++
//===--- UnresolvedMemberCodeCompletion.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/UnresolvedMemberCompletion.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 UnresolvedMemberTypeCheckCompletionCallback::sawSolutionImpl(
|
|
const constraints::Solution &S) {
|
|
auto &CS = S.getConstraintSystem();
|
|
Type ExpectedTy = getTypeForCompletion(S, CompletionExpr);
|
|
|
|
bool IsAsync = isContextAsync(S, DC);
|
|
|
|
// If the type couldn't be determined (e.g. because there isn't any context
|
|
// to derive it from), let's not attempt to do a lookup since it wouldn't
|
|
// produce any useful results anyway.
|
|
if (ExpectedTy) {
|
|
// 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(ExprResults, IsEqual)) {
|
|
bool SingleExprBody =
|
|
isImplicitSingleExpressionReturn(CS, CompletionExpr);
|
|
ExprResults.push_back({ExpectedTy, SingleExprBody, IsAsync});
|
|
}
|
|
}
|
|
|
|
if (auto PatternType = getPatternMatchType(S, CompletionExpr)) {
|
|
auto IsEqual = [&](const Result &R) {
|
|
return R.ExpectedTy->isEqual(PatternType);
|
|
};
|
|
if (!llvm::any_of(EnumPatternTypes, IsEqual)) {
|
|
EnumPatternTypes.push_back({PatternType,
|
|
/*IsImplicitSingleExpressionReturn=*/false,
|
|
IsAsync});
|
|
}
|
|
}
|
|
}
|
|
|
|
void UnresolvedMemberTypeCheckCompletionCallback::deliverResults(
|
|
DeclContext *DC, SourceLoc DotLoc,
|
|
ide::CodeCompletionContext &CompletionCtx,
|
|
CodeCompletionConsumer &Consumer) {
|
|
ASTContext &Ctx = DC->getASTContext();
|
|
CompletionLookup Lookup(CompletionCtx.getResultSink(), Ctx, DC,
|
|
&CompletionCtx);
|
|
|
|
assert(DotLoc.isValid());
|
|
Lookup.setHaveDot(DotLoc);
|
|
Lookup.shouldCheckForDuplicates(ExprResults.size() + EnumPatternTypes.size() >
|
|
1);
|
|
|
|
// Get the canonical versions of the top-level types
|
|
SmallPtrSet<CanType, 4> originalTypes;
|
|
for (auto &Result : ExprResults)
|
|
originalTypes.insert(Result.ExpectedTy->getCanonicalType());
|
|
|
|
for (auto &Result : ExprResults) {
|
|
Lookup.setExpectedTypes({Result.ExpectedTy},
|
|
Result.IsImplicitSingleExpressionReturn,
|
|
/*expectsNonVoid*/ true);
|
|
Lookup.setIdealExpectedType(Result.ExpectedTy);
|
|
Lookup.setCanCurrDeclContextHandleAsync(Result.IsInAsyncContext);
|
|
|
|
// For optional types, also get members of the unwrapped type if it's not
|
|
// already equivalent to one of the top-level types. Handling it via the top
|
|
// level type and not here ensures we give the correct type relation
|
|
// (identical, rather than convertible).
|
|
if (Result.ExpectedTy->getOptionalObjectType()) {
|
|
Type Unwrapped = Result.ExpectedTy->lookThroughAllOptionalTypes();
|
|
if (originalTypes.insert(Unwrapped->getCanonicalType()).second)
|
|
Lookup.getUnresolvedMemberCompletions(Unwrapped);
|
|
}
|
|
Lookup.getUnresolvedMemberCompletions(Result.ExpectedTy);
|
|
}
|
|
|
|
// Offer completions when interpreting the pattern match as an
|
|
// EnumElementPattern.
|
|
for (auto &Result : EnumPatternTypes) {
|
|
Type Ty = Result.ExpectedTy;
|
|
Lookup.setExpectedTypes({Ty}, /*IsImplicitSingleExpressionReturn=*/false,
|
|
/*expectsNonVoid=*/true);
|
|
Lookup.setIdealExpectedType(Ty);
|
|
Lookup.setCanCurrDeclContextHandleAsync(Result.IsInAsyncContext);
|
|
|
|
// We can pattern match MyEnum against Optional<MyEnum>
|
|
if (Ty->getOptionalObjectType()) {
|
|
Type Unwrapped = Ty->lookThroughAllOptionalTypes();
|
|
Lookup.getEnumElementPatternCompletions(Unwrapped);
|
|
}
|
|
|
|
Lookup.getEnumElementPatternCompletions(Ty);
|
|
}
|
|
|
|
deliverCompletionResults(CompletionCtx, Lookup, DC, Consumer);
|
|
}
|