/*
    SPDX-FileCopyrightText: 2015 Klarälvdalens Datakonsult AB a KDAB Group company info@kdab.com
    SPDX-FileCopyrightText: 2015-2016 Sergio Martins <smartins@kde.org>

    SPDX-License-Identifier: LGPL-2.0-or-later
*/

#ifndef MOREWARNINGS_UTILS_H
#define MOREWARNINGS_UTILS_H

#include <clang/AST/DeclCXX.h>
#include <clang/AST/DeclTemplate.h>
#include <clang/AST/Expr.h>
#include <clang/AST/ExprCXX.h>
#include <clang/AST/Stmt.h>
#include <clang/Basic/SourceLocation.h>
#include <clang/Basic/SourceManager.h>
#include <llvm/ADT/ArrayRef.h>
#include <llvm/ADT/StringRef.h>
#include <llvm/Config/llvm-config.h>

#include <string>
#include <vector>

// TODO: this is a dumping ground, most of these functions should be moved to the other *Utils classes

namespace clang
{
class CXXNamedCastExpr;
class CXXRecordDecl;
class CXXMemberCallExpr;
class CXXConstructExpr;
class CompilerInstance;
class ClassTemplateSpecializationDecl;
class Decl;
class ParentMap;
class SourceManager;
class Stmt;
class SourceLocation;
class ExprWithCleanups;
class ValueDecl;
class ConditionalOperator;
class CXXMethodDecl;
class BinaryOperator;
class CXXOperatorCallExpr;
class CallExpr;
class DeclStmt;
class Expr;
class FunctionDecl;
class LangOptions;
class ParmVarDecl;
class StringLiteral;
class UserDefinedLiteral;
class VarDecl;
}

struct StmtBodyRange;

namespace Utils
{
/// Returns true if the class has at least one constexpr ctor
bool hasConstexprCtor(clang::CXXRecordDecl *decl);

/// Returns the type we're casting *from*
clang::CXXRecordDecl *namedCastInnerDecl(clang::CXXNamedCastExpr *staticOrDynamicCast);

/// Returns the type we're casting *to*
clang::CXXRecordDecl *namedCastOuterDecl(clang::CXXNamedCastExpr *staticOrDynamicCast);

/// Returns the class declaration from a variable declaration
// So, if the var decl is "Foo f"; it returns the declaration of Foo
clang::CXXRecordDecl *recordFromVarDecl(clang::Decl *);

/// Returns the template specialization from a variable declaration
// So, if the var decl is "QList<Foo> f;", returns the template specialization QList<Foo>
clang::ClassTemplateSpecializationDecl *templateSpecializationFromVarDecl(clang::Decl *);

/// Returns true if all the child member function calls are const functions.
bool allChildrenMemberCallsConst(clang::Stmt *stm);

/// Returns true if at least a UnaryOperator, BinaryOperator or non-const function call is found
bool childsHaveSideEffects(clang::Stmt *stm);

/// Receives a member call, such as "list.reserve()" and returns the declaration of the variable list
// such as "QList<F> list"
clang::ValueDecl *valueDeclForMemberCall(clang::CXXMemberCallExpr *);

/// Receives an operator call, such as "list << fooo" and returns the declaration of the variable list
// such as "QList<F> list"
clang::ValueDecl *valueDeclForOperatorCall(clang::CXXOperatorCallExpr *);

// overload
clang::ValueDecl *valueDeclForCallExpr(clang::CallExpr *);

// Returns true of this value decl is a member variable of a class or struct
// returns null if not
clang::CXXRecordDecl *isMemberVariable(clang::ValueDecl *);

// Returns true if a body of statements contains a non const member call on object declared by varDecl
// For example:
// Foo foo; // this is the varDecl
// while (bar) { foo.setValue(); // non-const call }
bool containsNonConstMemberCall(clang::ParentMap *map, clang::Stmt *body, const clang::VarDecl *varDecl);

// Returns true if there's an assignment to varDecl in body
// Example: our_var = something_else
bool isAssignedFrom(clang::Stmt *body, const clang::VarDecl *varDecl);

// Returns true if varDecl is assigned to something
// Example: something_else = our_var
bool isAssignedTo(clang::Stmt *body, const clang::VarDecl *varDecl);

// Returns whether the variable is returned in body
bool isReturned(clang::Stmt *body, const clang::VarDecl *varDecl);

// Returns true if a body of statements contains a function call that takes our variable (varDecl)
// By ref or pointer
bool isPassedToFunction(const StmtBodyRange &bodyRange, const clang::VarDecl *varDecl, bool byRefOrPtrOnly);

// Returns true if we take the address of varDecl, such as: &foo
bool addressIsTaken(clang::Stmt *body, const clang::ValueDecl *valDecl);

// QString::fromLatin1("foo")    -> true
// QString::fromLatin1("foo", 1) -> false
bool callHasDefaultArguments(clang::CallExpr *expr);

// If there's a child of type StringLiteral, returns true
// if allowEmpty is false, "" will be ignored
bool containsStringLiteral(clang::Stmt *, bool allowEmpty = true, int depth = -1);

bool isInsideOperatorCall(clang::ParentMap *map, clang::Stmt *s, const std::vector<llvm::StringRef> &anyOf);

bool insideCTORCall(clang::ParentMap *map, clang::Stmt *s, const std::vector<llvm::StringRef> &anyOf);

// returns true if the ternary operator has two string literal arguments, such as:
// foo ? "bar" : "baz"
bool ternaryOperatorIsOfStringLiteral(clang::ConditionalOperator *);

bool isAssignOperator(clang::CXXOperatorCallExpr *op, llvm::StringRef className, llvm::StringRef argumentType, const clang::LangOptions &lo);

bool isImplicitCastTo(clang::Stmt *, const std::string &);

bool presumedLocationsEqual(const clang::PresumedLoc &l1, const clang::PresumedLoc &l2);

// Returns the list of methods with name methodName that the class/struct record contains
std::vector<clang::CXXMethodDecl *> methodsFromString(const clang::CXXRecordDecl *record, const std::string &methodName);

// Returns the most derived class. (CXXMemberCallExpr::getRecordDecl() return the first base class with the method)
// The returned callee is the name of the variable on which the member call was made:
// o1->foo() => "o1"
// foo() => "this"
const clang::CXXRecordDecl *recordForMemberCall(clang::CXXMemberCallExpr *call, std::string &implicitCallee);

bool isAscii(clang::StringLiteral *lt);

// Checks if Statement s inside an operator* call
bool isInDerefExpression(clang::Stmt *s, clang::ParentMap *map);

// For a a chain called expression like foo().bar().baz() returns a list of calls
// {baz(), bar(), foo()}

// the parameter lastCallExpr to pass would be baz()
// No need to specify the other callexprs, they are children of the first one, since the AST looks like:
// - baz
// -- bar
// --- foo
std::vector<clang::CallExpr *> callListForChain(clang::CallExpr *lastCallExpr);

// Returns the first base class
clang::CXXRecordDecl *rootBaseClass(clang::CXXRecordDecl *derived);

// Returns the copy ctor for this class
clang::CXXConstructorDecl *copyCtor(const clang::CXXRecordDecl *);

// Returns the copy-assignment operator for this class
clang::CXXMethodDecl *copyAssign(const clang::CXXRecordDecl *);

bool hasMember(clang::CXXRecordDecl *record, const std::string &memberTypeName);

/**
 * Returns true if varDecl is initialized externally.
 * Example:
 *     QList<Foo> list = getList(); // true
 *     QList<int> list = list2;     // true
 *     QList<int> list = {1, 2, 3}; // false
 *     QList<int> list;             // false
 */
bool isInitializedExternally(clang::VarDecl *varDecl);

/**
 * Returns true if declStmt refers to varDecl
 */
bool referencesVarDecl(clang::DeclStmt *declStmt, clang::VarDecl *varDecl);

/**
 * Returns true if the body of a function is empty.
 * Returns false if either function or it's body are null.
 */
bool functionHasEmptyBody(clang::FunctionDecl *func);

/**
 * If stm is an UnaryOperator or BinaryOperator that writes to the variable it returns the expression
 * that represents the variable (Usually a MemberExpr or DeclRefExpr for local variables).
 *
 * Otherwise returns nullptr.
 *
 * The operators that write to the variable are operator=, operator+=, operator++, etc.
 */
clang::Expr *isWriteOperator(clang::Stmt *stm);

/**
 * Gets the UserDefinedLiteral of type @p type which is somewhere in the ast of @p stm.
 * Returns nullptr if there's no such UserDefinedLiteral.
 */
clang::UserDefinedLiteral *userDefinedLiteral(clang::Stmt *stm, const std::string &type, const clang::LangOptions &lo);

/**
 * Returns the function parameters from @p func
 * This should be used instead of calling FunctionDecl::params() since it changed signature in
 * clang 3.9.
 */
clang::ArrayRef<clang::ParmVarDecl *> functionParameters(clang::FunctionDecl *func);

/**
 * For the given ctor, and ctor param, returns the ctor member initializers that used that param.
 * Example:
 * MyCtor(int a, int b) : c(a), d(b) {}
 * auto result = Utils::ctorInitializer(MyCtor, b); // Result is the statement "d(b)"
 */
std::vector<clang::CXXCtorInitializer *> ctorInitializer(clang::CXXConstructorDecl *ctor, clang::ParmVarDecl *param);

/**
 * Returns true if a ctor initializer contains a std::move()
 * Example
 * MyCtor(Foo a) : c(move(a)) {} // Would return true for this init list
 */
bool ctorInitializerContainsMove(clang::CXXCtorInitializer *);

// Overload that receives a vector and returns true if any ctor initializer contains a move()
bool ctorInitializerContainsMove(const std::vector<clang::CXXCtorInitializer *> &);

/**
 * Returns the filename for the source location loc
 */
std::string filenameForLoc(clang::SourceLocation loc, const clang::SourceManager &sm);

/**
 * Returns the location after the lexer token that is at loc.
 * For example:
 *     emit sig();
 * If loc refers to the location of 'emit', then this function will return the source location if
 * the sig() call.
 */
clang::SourceLocation locForNextToken(clang::SourceLocation loc, const clang::SourceManager &sm, const clang::LangOptions &lo);

inline bool isMainFile(const clang::SourceManager &sm, clang::SourceLocation loc)
{
    if (loc.isMacroID()) {
        loc = sm.getExpansionLoc(loc);
    }

    return sm.isInFileID(loc, sm.getMainFileID());
}

/**
 * Returns true if the string literal contains escaped bytes, such as \x12, \123, \u00F6.
 */
bool literalContainsEscapedBytes(clang::StringLiteral *lt, const clang::SourceManager &sm, const clang::LangOptions &lo);

/**
 * Returns true if this method overrides one from the base class
 */
inline bool methodOverrides(clang::CXXMethodDecl *method)
{
    return method && method->isVirtual() && method->size_overridden_methods() > 0;
}
}

#endif
