#include "RemoveInit.h"
#include "SynTree/Declaration.h"
#include "SynTree/Type.h"
#include "SynTree/Expression.h"
#include "SynTree/Statement.h"
#include "SynTree/Initializer.h"
#include "SynTree/Mutator.h"

namespace InitTweak {

namespace {
const std::list<Label> noLabels;
}

void tweak( std::list< Declaration * > translationUnit ) {
  RemoveInit remover;
  mutateAll( translationUnit, remover );
}

void RemoveInit::mutateStatementList( std::list< Statement* > &statements ) {
  for( std::list< Statement* >::iterator i = statements.begin(); i != statements.end(); ++i ) {
    if( !stmtsToAddAfter.empty() ) {
      statements.splice( i, stmtsToAddAfter );
    }
    *i = (*i)->acceptMutator( *this );
  }
  if( !stmtsToAddAfter.empty() ) {
    statements.splice( statements.end(), stmtsToAddAfter );
  }
}

CompoundStmt *RemoveInit::mutate(CompoundStmt *compoundStmt) {
  mutateStatementList( compoundStmt->get_kids() );
  return compoundStmt;
}

// in the case where an object has an initializer and a polymorphic type, insert an assignment
// immediately after the declaration. This will (seemingly) cause the later phases to do the right
// thing with the assignment
ObjectDecl *RemoveInit::mutate( ObjectDecl *objDecl ) {
  if (objDecl->get_init() && dynamic_cast<TypeInstType*>(objDecl->get_type())) {
    if (SingleInit * single = dynamic_cast<SingleInit*>(objDecl->get_init())) {
      UntypedExpr *assign = new UntypedExpr( new NameExpr( "?=?" ) );
      assign->get_args().push_back( new AddressExpr (new NameExpr( objDecl->get_name() ) ) );
      assign->get_args().push_back( single->get_value()->clone() );
      stmtsToAddAfter.push_back(new ExprStmt(noLabels, assign));
    }
  }
  return objDecl;
}
} // namespace InitTweak