/**
 * src/sequence_diagram/visitor/call_expression_context.h
 *
 * Copyright (c) 2021-2022 Bartek Kryza <bkryza@gmail.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#pragma once
//
//#include "common/visitor/translation_unit_visitor.h"
//#include "config/config.h"
//#include "sequence_diagram/model/diagram.h"

#include "util/util.h"

#include <clang/AST/Expr.h>
#include <clang/AST/RecursiveASTVisitor.h>
#include <clang/Basic/SourceManager.h>

#include <stack>

namespace clanguml::sequence_diagram::visitor {

struct call_expression_context {
    call_expression_context();

    void reset();

    void dump();

    bool valid() const;

    clang::ASTContext *get_ast_context();

    void update(clang::CXXRecordDecl *cls);

    void update(clang::ClassTemplateSpecializationDecl *clst);

    void update(clang::ClassTemplateDecl *clst);

    void update(clang::CXXMethodDecl *method);

    void update(clang::FunctionDecl *function);

    void update(clang::FunctionTemplateDecl *function_template);

    bool in_class_method() const;

    bool in_function() const;

    bool in_function_template() const;

    std::int64_t caller_id() const;

    std::int64_t lambda_caller_id() const;

    void set_caller_id(std::int64_t id);

    void enter_lambda_expression(std::int64_t id);

    void leave_lambda_expression();

    clang::CXXRecordDecl *current_class_decl_;
    clang::ClassTemplateDecl *current_class_template_decl_;
    clang::ClassTemplateSpecializationDecl
        *current_class_template_specialization_decl_;
    clang::CXXMethodDecl *current_method_decl_;
    clang::FunctionDecl *current_function_decl_;
    clang::FunctionTemplateDecl *current_function_template_decl_;

private:
    std::int64_t current_caller_id_;
    std::stack<std::int64_t> current_lambda_caller_id_;
};

}