function_call_info.hpp

Go to the documentation of this file.

#pragma once
 
#include <string>
#include <vector>
 
#include "porytiles/utilities/c_parser/token.hpp"
 
namespace porytiles {
 
class FunctionCallInfo {
  public:
    FunctionCallInfo(std::string function_name, std::vector<std::vector<Token>> arguments, std::size_t start_index)
        : function_name_{std::move(function_name)}, arguments_{std::move(arguments)}, start_index_{start_index}
    {
    }
 
    [[nodiscard]] const std::string &function_name() const
    {
        return function_name_;
    }
 
    [[nodiscard]] const std::vector<std::vector<Token>> &arguments() const
    {
        return arguments_;
    }
 
    [[nodiscard]] std::size_t argument_count() const
    {
        return arguments_.size();
    }
 
    [[nodiscard]] const std::vector<Token> &argument_at(std::size_t index) const
    {
        return arguments_.at(index);
    }
 
    [[nodiscard]] std::size_t start_index() const
    {
        return start_index_;
    }
 
    [[nodiscard]] std::string reconstruct_call_text() const
    {
        std::string result = function_name_ + "(";
        for (std::size_t arg_idx = 0; arg_idx < arguments_.size(); ++arg_idx) {
            if (arg_idx > 0) {
                result += ", ";
            }
            const auto &arg_tokens = arguments_[arg_idx];
            for (std::size_t tok_idx = 0; tok_idx < arg_tokens.size(); ++tok_idx) {
                if (tok_idx > 0) {
                    bool suppress_space =
                        arg_tokens[tok_idx - 1].is_any_of(TokenType::left_paren, TokenType::left_bracket) ||
                        arg_tokens[tok_idx].is_any_of(TokenType::right_paren, TokenType::right_bracket);
                    if (!suppress_space) {
                        result += " ";
                    }
                }
                result += arg_tokens[tok_idx].text();
            }
        }
        result += ")";
        return result;
    }
 
  private:
    std::string function_name_;
    std::vector<std::vector<Token>> arguments_;
    std::size_t start_index_;
};
 
[[nodiscard]] std::vector<FunctionCallInfo>
find_function_calls(const std::vector<Token> &tokens, const std::string &target_function_name);
 
[[nodiscard]] std::vector<FunctionCallInfo> find_all_function_calls(const std::vector<Token> &tokens);
 
} // namespace porytiles