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.
101 lines
3.9 KiB
C++
101 lines
3.9 KiB
C++
//===--- ExprCompletion.h -------------------------------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef SWIFT_IDE_EXPRCOMPLETION_H
|
|
#define SWIFT_IDE_EXPRCOMPLETION_H
|
|
|
|
#include "swift/IDE/CodeCompletionConsumer.h"
|
|
#include "swift/IDE/CodeCompletionContext.h"
|
|
#include "swift/IDE/TypeCheckCompletionCallback.h"
|
|
|
|
namespace swift {
|
|
namespace ide {
|
|
|
|
class ExprTypeCheckCompletionCallback : public TypeCheckCompletionCallback {
|
|
public:
|
|
struct Result {
|
|
/// If the code completion expression is an implied result, e.g in a
|
|
/// single-expression closure.
|
|
bool IsImpliedResult;
|
|
|
|
/// Whether the surrounding context is async and thus calling async
|
|
/// functions is supported.
|
|
bool IsInAsyncContext;
|
|
|
|
/// If unresolved member lookup is enabled in the completion callback,
|
|
/// the type that the unresolved member lookup should be performed on,
|
|
/// otherwise a null type.
|
|
Type UnresolvedMemberBaseType;
|
|
|
|
/// Types of variables that were determined in the solution that produced
|
|
/// this result. This in particular includes parameters of closures that
|
|
/// were type-checked with the code completion expression.
|
|
llvm::SmallDenseMap<const VarDecl *, Type> SolutionSpecificVarTypes;
|
|
|
|
bool operator==(const Result &Other) const;
|
|
};
|
|
|
|
private:
|
|
CodeCompletionExpr *CompletionExpr;
|
|
DeclContext *DC;
|
|
|
|
/// The contextual types to which the code completion results should be
|
|
/// convertible.
|
|
/// Technically, each result should have its own expected type because some
|
|
/// expected types may only be available e.g. for certain
|
|
/// \c SolutionSpecificVarTypes. But that means that we need to do a separate
|
|
/// completion lookup for each expected type and de-duplicate the results,
|
|
/// which can have huge performance implications (>5mins instead of <2secs).
|
|
/// In practice sharing ExpectedTypes between results yields identical results
|
|
/// in almost all cases and acceptable results in the other cases.
|
|
SmallVector<Type, 4> ExpectedTypes;
|
|
|
|
/// When \c true, unresolved member completions of the expected types are
|
|
/// also reported when delivering results. Used in places that frequently
|
|
/// use unresolved members such as enum cases.
|
|
bool AddUnresolvedMemberCompletions;
|
|
|
|
SmallVector<Result, 4> Results;
|
|
|
|
/// Adds the given type to \c ExpectedTypes unless \c ExpectedTypes already
|
|
/// contains the type.
|
|
void addExpectedType(Type ExpectedType);
|
|
|
|
/// Adds the result with the given parameters to \c Results unless \c Results
|
|
/// already contains an entry with exactly the same values.
|
|
/// If \c AddUnresolvedMemberCompletions is false, the
|
|
/// \p UnresolvedMemberBaseType is ignored.
|
|
void addResult(
|
|
bool IsImpliedResult, bool IsInAsyncContext,
|
|
Type UnresolvedMemberBaseType,
|
|
llvm::SmallDenseMap<const VarDecl *, Type> SolutionSpecificVarTypes);
|
|
|
|
void sawSolutionImpl(const constraints::Solution &solution) override;
|
|
|
|
public:
|
|
/// \param DC The decl context in which the \p CompletionExpr occurs.
|
|
ExprTypeCheckCompletionCallback(CodeCompletionExpr *CompletionExpr,
|
|
DeclContext *DC,
|
|
bool AddUnresolvedMemberCompletions)
|
|
: CompletionExpr(CompletionExpr), DC(DC),
|
|
AddUnresolvedMemberCompletions(AddUnresolvedMemberCompletions) {}
|
|
|
|
/// \param CCLoc The location of the code completion token.
|
|
void collectResults(SourceLoc CCLoc,
|
|
ide::CodeCompletionContext &CompletionCtx);
|
|
};
|
|
|
|
} // end namespace ide
|
|
} // end namespace swift
|
|
|
|
#endif // SWIFT_IDE_EXPRCOMPLETION_H
|