// Copyright 2021 xensik. All rights reserved.
//
// Use of this source code is governed by a GNU GPLv3 license
// that can be found in the LICENSE file.

#pragma once

namespace xsk::gsc::iw5
{

class decompiler : public gsc::decompiler
{
    std::string filename_;
    ast::program::ptr program_;
    ast::decl_thread::ptr func_;
    std::unordered_map<std::uint32_t, std::string> labels_;
    std::vector<std::string> expr_labels_;
    std::vector<std::string> tern_labels_;
    std::stack<ast::node::ptr> stack_;
    std::vector<block> blocks_;
    bool in_waittill_;

public:
    auto output() -> std::vector<std::uint8_t>;
    void decompile(const std::string& file, std::vector<function::ptr>& funcs);

private:
    void decompile_function(const function::ptr& func);
    void decompile_instruction(const instruction::ptr& inst);
    void decompile_expressions(const instruction::ptr& inst);
    void decompile_statements(const ast::stmt_list::ptr& stmt);
    void decompile_infinites(const ast::stmt_list::ptr& stmt);
    void decompile_loops(const ast::stmt_list::ptr& stmt);
    void decompile_switches(const ast::stmt_list::ptr& stmt);
    void decompile_ifelses(const ast::stmt_list::ptr& stmt);
    void decompile_aborts(const ast::stmt_list::ptr& stmt);
    void decompile_if(const ast::stmt_list::ptr& stmt, std::uint32_t begin, std::uint32_t end);
    void decompile_ifelse(const ast::stmt_list::ptr& stmt, std::uint32_t begin, std::uint32_t end);
    void decompile_last_ifelse(const ast::stmt_list::ptr& stmt, std::uint32_t begin, std::uint32_t end);
    void decompile_infinite(const ast::stmt_list::ptr& stmt, std::uint32_t begin, std::uint32_t end);
    void decompile_loop(const ast::stmt_list::ptr& stmt, std::uint32_t begin, std::uint32_t end);
    void decompile_while(const ast::stmt_list::ptr& stmt, std::uint32_t begin, std::uint32_t end);
    void decompile_for(const ast::stmt_list::ptr& stmt, std::uint32_t begin, std::uint32_t end);
    void decompile_foreach(const ast::stmt_list::ptr& stmt, std::uint32_t begin, std::uint32_t end);
    void decompile_switch(const ast::stmt_list::ptr& stmt, std::uint32_t begin);
    auto find_location_reference(const ast::stmt_list::ptr& stmt, std::uint32_t begin, std::uint32_t end, const std::string& location) -> bool;
    auto find_location_index(const ast::stmt_list::ptr& stmt, const std::string& location) -> std::uint32_t;
    auto last_location_index(const ast::stmt_list::ptr& stmt, std::uint32_t index) -> bool;
    void process_stack(const ast::decl_thread::ptr& thread);
    void process_parameters(const ast::expr_parameters::ptr& params, const block::ptr& blk);
    void process_stmt(const ast::stmt& stmt, const block::ptr& blk);
    void process_stmt_list(const ast::stmt_list::ptr& stmt, const block::ptr& blk);
    void process_stmt_expr(const ast::stmt_expr::ptr& stmt, const block::ptr& blk);
    void process_stmt_call(const ast::stmt_call::ptr& stmt, const block::ptr& blk);
    void process_stmt_assign(const ast::stmt_assign::ptr& stmt, const block::ptr& blk);
    void process_stmt_endon(const ast::stmt_endon::ptr& stmt, const block::ptr& blk);
    void process_stmt_notify(const ast::stmt_notify::ptr& stmt, const block::ptr& blk);
    void process_stmt_wait(const ast::stmt_wait::ptr& stmt, const block::ptr& blk);
    void process_stmt_waittill(const ast::stmt_waittill::ptr& stmt, const block::ptr& blk);
    void process_stmt_waittillmatch(const ast::stmt_waittillmatch::ptr& stmt, const block::ptr& blk);
    void process_stmt_if(const ast::stmt_if::ptr& stmt, const block::ptr& blk);
    void process_stmt_ifelse(const ast::stmt_ifelse::ptr& stmt, const block::ptr& blk);
    void process_stmt_while(const ast::stmt_while::ptr& stmt, const block::ptr& blk);
    void process_stmt_dowhile(const ast::stmt_dowhile::ptr& stmt, const block::ptr& blk);
    void process_stmt_for(const ast::stmt_for::ptr& stmt, const block::ptr& blk);
    void process_stmt_foreach(const ast::stmt_foreach::ptr& stmt, const block::ptr& blk);
    void process_stmt_switch(const ast::stmt_switch::ptr& stmt, const block::ptr& blk);
    void process_stmt_cases(const ast::stmt_list::ptr& stmt, const block::ptr& blk);
    void process_stmt_break(const ast::stmt_break::ptr& stmt, const block::ptr& blk);
    void process_stmt_continue(const ast::stmt_continue::ptr& stmt, const block::ptr& blk);
    void process_stmt_return(const ast::stmt_return::ptr& stmt, const block::ptr& blk);
    void process_expr(ast::expr& expr, const block::ptr& blk);
    void process_expr_assign(ast::expr_assign::ptr& expr, const block::ptr& blk);
    void process_expr_increment(const ast::expr_increment::ptr& expr, const block::ptr& blk);
    void process_expr_decrement(const ast::expr_decrement::ptr& expr, const block::ptr& blk);
    void process_expr_ternary(const ast::expr_ternary::ptr& expr, const block::ptr& blk);
    void process_expr_binary(const ast::expr_binary::ptr& expr, const block::ptr& blk);
    void process_expr_and(const ast::expr_and::ptr& expr, const block::ptr& blk);
    void process_expr_or(const ast::expr_or::ptr& expr, const block::ptr& blk);
    void process_expr_complement(const ast::expr_complement::ptr& expr, const block::ptr& blk);
    void process_expr_not(const ast::expr_not::ptr& expr, const block::ptr& blk);
    void process_expr_call(const ast::expr_call::ptr& expr, const block::ptr& blk);
    void process_expr_method(const ast::expr_method::ptr& expr, const block::ptr& blk);
    void process_expr_call_pointer(const ast::expr_pointer::ptr& expr, const block::ptr& blk);
    void process_expr_call_function(const ast::expr_function::ptr& expr, const block::ptr& blk);
    void process_expr_method_pointer(const ast::expr_pointer::ptr& expr, ast::expr& obj, const block::ptr& blk);
    void process_expr_method_function(const ast::expr_function::ptr& expr, ast::expr& obj, const block::ptr& blk);
    void process_expr_arguments(const ast::expr_arguments::ptr& expr, const block::ptr& blk);
    void process_expr_add_array(const ast::expr_add_array::ptr& expr, const block::ptr& blk);
    void process_expr_size(const ast::expr_size::ptr& expr, const block::ptr& blk);
    void process_array_variable(const ast::expr_array::ptr& expr, const block::ptr& blk);
    void process_field_variable(const ast::expr_field::ptr& expr, const block::ptr& blk);
    void process_local_variable(const ast::expr_identifier::ptr& expr, const block::ptr& blk);
    void process_expr_vector(const ast::expr_vector::ptr& vec, const block::ptr& blk);
    void process_var_create(ast::expr& expr, const block::ptr& blk, bool fromstmt = false);
    void process_var_access(ast::expr& expr, const block::ptr& blk);
    void process_var_remove(const ast::asm_remove::ptr& expr, const block::ptr& blk);
};

} // namespace xsk::gsc::iw5