blob: e0a4e472f3496130c19a1331e218e3900b09af57 [file] [log] [blame]
//===--- ExpectedTypes.cpp ---------------------------------------*- C++-*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "ExpectedTypes.h"
#include "clang/AST/ASTContext.h"
#include "clang/AST/DeclTemplate.h"
#include "clang/AST/Type.h"
#include "clang/Index/USRGeneration.h"
#include "clang/Sema/CodeCompleteConsumer.h"
#include "llvm/ADT/None.h"
#include "llvm/ADT/STLExtras.h"
namespace clang {
namespace clangd {
namespace {
static const Type *toEquivClass(ASTContext &Ctx, QualType T) {
if (T.isNull() || T->isDependentType())
return nullptr;
// Drop references, we do not handle reference inits properly anyway.
T = T.getCanonicalType().getNonReferenceType();
// Numeric types are the simplest case.
if (T->isBooleanType())
return Ctx.BoolTy.getTypePtr();
if (T->isIntegerType() && !T->isEnumeralType())
return Ctx.IntTy.getTypePtr(); // All integers are equivalent.
if (T->isFloatingType() && !T->isComplexType())
return Ctx.FloatTy.getTypePtr(); // All floats are equivalent.
// Do some simple transformations.
if (T->isArrayType()) // Decay arrays to pointers.
return Ctx.getPointerType(QualType(T->getArrayElementTypeNoTypeQual(), 0))
.getTypePtr();
// Drop the qualifiers and return the resulting type.
// FIXME: also drop qualifiers from pointer types, e.g. 'const T* => T*'
return T.getTypePtr();
}
static llvm::Optional<QualType>
typeOfCompletion(const CodeCompletionResult &R) {
const NamedDecl *D = R.Declaration;
// Templates do not have a type on their own, look at the templated decl.
if (auto *Template = dyn_cast_or_null<TemplateDecl>(D))
D = Template->getTemplatedDecl();
auto *VD = dyn_cast_or_null<ValueDecl>(D);
if (!VD)
return llvm::None; // We handle only variables and functions below.
auto T = VD->getType();
if (T.isNull())
return llvm::None;
if (auto FuncT = T->getAs<FunctionType>()) {
// Functions are a special case. They are completed as 'foo()' and we want
// to match their return type rather than the function type itself.
// FIXME(ibiryukov): in some cases, we might want to avoid completing `()`
// after the function name, e.g. `std::cout << std::endl`.
return FuncT->getReturnType();
}
return T;
}
} // namespace
llvm::Optional<OpaqueType> OpaqueType::encode(ASTContext &Ctx, QualType T) {
if (T.isNull())
return None;
const Type *C = toEquivClass(Ctx, T);
if (!C)
return None;
llvm::SmallString<128> Encoded;
if (index::generateUSRForType(QualType(C, 0), Ctx, Encoded))
return None;
return OpaqueType(std::string(Encoded.str()));
}
OpaqueType::OpaqueType(std::string Data) : Data(std::move(Data)) {}
llvm::Optional<OpaqueType> OpaqueType::fromType(ASTContext &Ctx,
QualType Type) {
return encode(Ctx, Type);
}
llvm::Optional<OpaqueType>
OpaqueType::fromCompletionResult(ASTContext &Ctx,
const CodeCompletionResult &R) {
auto T = typeOfCompletion(R);
if (!T)
return None;
return encode(Ctx, *T);
}
} // namespace clangd
} // namespace clang