/****************************************************************************
**
** Copyright (C) 2017 The Qt Company Ltd.
** Contact: https://www.qt.io/licensing/
**
** This file is part of Qt Creator.
**
** Commercial License Usage
** Licensees holding valid commercial Qt licenses may use this file in
** accordance with the commercial license agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and The Qt Company. For licensing terms
** and conditions see https://www.qt.io/terms-conditions. For further
** information use the contact form at https://www.qt.io/contact-us.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU
** General Public License version 3 as published by the Free Software
** Foundation with exceptions as appearing in the file LICENSE.GPL3-EXCEPT
** included in the packaging of this file. Please review the following
** information to ensure the GNU General Public License requirements will
** be met: https://www.gnu.org/licenses/gpl-3.0.html.
**
****************************************************************************/

#pragma once

#include "clangrefactoringbackend_global.h"
#include "filestatus.h"
#include "sourcelocationentry.h"
#include "symbolentry.h"
#include "symbolsvisitorbase.h"

#include <filepathcachingfwd.h>

#include <clang/Index/IndexDataConsumer.h>

namespace ClangBackEnd {

class IndexDataConsumer : public clang::index::IndexDataConsumer,
                          public SymbolsVisitorBase
{
public:
    IndexDataConsumer(SymbolEntries &symbolEntries,
                      SourceLocationEntries &sourceLocationEntries,
                      FileStatuses &fileStatuses,
                      FilePathCachingInterface &filePathCache,
                      SourcesManager &symbolSourcesManager,
                      SourcesManager &macroSourcesManager)
        : SymbolsVisitorBase(filePathCache, nullptr, m_filePathIndices)
        , m_symbolEntries(symbolEntries)
        , m_sourceLocationEntries(sourceLocationEntries)
        , m_fileStatuses(fileStatuses)
        , m_symbolSourcesManager(symbolSourcesManager)
        , m_macroSourcesManager(macroSourcesManager)

    {}

    IndexDataConsumer(const IndexDataConsumer &) = delete;
    IndexDataConsumer &operator=(const IndexDataConsumer &) = delete;

    void setPreprocessor(std::shared_ptr<clang::Preprocessor> preprocessor) override;

    bool handleDeclOccurence(const clang::Decl *declaration,
                             clang::index::SymbolRoleSet symbolRoles,
                             llvm::ArrayRef<clang::index::SymbolRelation> symbolRelations,
                             clang::SourceLocation sourceLocation,
                             ASTNodeInfo astNodeInfo) override;

    bool handleMacroOccurence(const clang::IdentifierInfo *identifierInfo,
                              const clang::MacroInfo *macroInfo,
                              clang::index::SymbolRoleSet roles,
                              clang::SourceLocation sourceLocation) override;

    void finish() override;

private:
    bool skipSymbol(clang::FileID fileId, clang::index::SymbolRoleSet symbolRoles);
    bool isAlreadyParsed(clang::FileID fileId, SourcesManager &sourcesManager);

private:
    FilePathIds m_filePathIndices;
    SymbolEntries &m_symbolEntries;
    SourceLocationEntries &m_sourceLocationEntries;
    FileStatuses &m_fileStatuses;
    SourcesManager &m_symbolSourcesManager;
    SourcesManager &m_macroSourcesManager;
};

} // namespace ClangBackEnd
