polserver/pol-core/bscript/compiler/ast/Function.cpp

#include "bscript/compiler/ast/Function.h"

#include "bscript/compiler/ast/FunctionBody.h"
#include "bscript/compiler/ast/FunctionParameterDeclaration.h"
#include "bscript/compiler/ast/FunctionParameterList.h"

namespace Pol::Bscript::Compiler
{
Function::Function( const SourceLocation& source_location, std::string scope, std::string name,
                    std::unique_ptr<FunctionParameterList> parameter_list,
                    std::unique_ptr<FunctionBody> body )
    : Node( source_location ), scope( std::move( scope ) ), name( std::move( name ) )
{
  children.reserve( 2 );
  children.push_back( std::move( parameter_list ) );
  children.push_back( std::move( body ) );
}

Function::Function( const SourceLocation& source_location, std::string scope, std::string name,
                    std::unique_ptr<FunctionParameterList> parameter_list )
    : Node( source_location, std::move( parameter_list ) ),
      scope( std::move( scope ) ),
      name( std::move( name ) )
{
}

unsigned Function::parameter_count() const
{
  return static_cast<unsigned>( children.at( 0 )->children.size() );
}

bool Function::is_variadic() const
{
  // The variadic argument, if present, is the last one
  if ( parameter_count() > 0 )
  {
    const auto& parameter_list = children.at( 0 );

    auto parameter =
        static_cast<FunctionParameterDeclaration*>( parameter_list->children.back().get() );

    return parameter->rest;
  }

  return false;
}

std::string Function::scoped_name() const
{
  if ( scope.empty() )
  {
    return name;
  }

  return fmt::format( "{}::{}", scope, name );
}

std::vector<std::reference_wrapper<FunctionParameterDeclaration>> Function::parameters()
{
  std::vector<std::reference_wrapper<FunctionParameterDeclaration>> params;
  auto& param_list = child<FunctionParameterList>( 0 );
  for ( auto& param : param_list.children )
  {
    params.emplace_back( *static_cast<FunctionParameterDeclaration*>( param.get() ) );
  }
  return params;
}

}  // namespace Pol::Bscript::Compiler