#include <graphene/chain/sidechain_transaction_evaluator.hpp>

#include <graphene/chain/database.hpp>
#include <graphene/chain/sidechain_transaction_object.hpp>
#include <graphene/chain/son_object.hpp>
#include <graphene/chain/proposal_object.hpp>
#include <graphene/chain/hardfork.hpp>

namespace graphene { namespace chain {

void_result bitcoin_transaction_send_evaluator::do_evaluate(const bitcoin_transaction_send_operation& op)
{ try{
   FC_ASSERT(db().head_block_time() >= HARDFORK_SON_TIME, "Not allowed until SON HARDFORK"); // can be removed after HF date pass
   FC_ASSERT(op.payer == db().get_global_properties().parameters.get_son_btc_account_id(), "Payer should be the son btc account");
   return void_result();
} FC_CAPTURE_AND_RETHROW( (op) ) }

object_id_type bitcoin_transaction_send_evaluator::do_apply(const bitcoin_transaction_send_operation& op)
{ try {
   const auto& new_bitcoin_transaction_object = db().create<bitcoin_transaction_object>( [&]( bitcoin_transaction_object& obj ){
      obj.processed = false;
      obj.signatures = op.signatures;
   });
    return new_bitcoin_transaction_object.id;
} FC_CAPTURE_AND_RETHROW( (op) ) }

void_result bitcoin_transaction_sign_evaluator::do_evaluate(const bitcoin_transaction_sign_operation& op)
{ try {
    FC_ASSERT(db().head_block_time() >= HARDFORK_SON_TIME, "Not allowed until SON HARDFORK"); // can be removed after HF date pass
    const auto& proposal_idx = db().get_index_type<proposal_index>().indices().get<by_id>();
    const auto& proposal_itr = proposal_idx.find( op.proposal_id );
    FC_ASSERT( proposal_idx.end() != proposal_itr, "proposal not found");
    // Checks can this SON approve this proposal
    auto can_this_son_approve_this_proposal = [ & ]() {
       const auto& sidx = db().get_index_type<son_index>().indices().get<graphene::chain::by_account>();
       auto son_obj = sidx.find( op.payer );
       if( son_obj == sidx.end() ) {
          return false;    
       }
       // TODO: Check if the SON is included in the PW script.
       return true;
    };

    FC_ASSERT(can_this_son_approve_this_proposal(), "Invalid approval received");
    return void_result();
} FC_CAPTURE_AND_RETHROW( (op) ) }

object_id_type bitcoin_transaction_sign_evaluator::do_apply(const bitcoin_transaction_sign_operation& op)
{ try {
    const auto& proposal = op.proposal_id( db() );
    const auto& sidx = db().get_index_type<son_index>().indices().get<graphene::chain::by_account>();
    auto son_obj = sidx.find( op.payer );

    db().modify( proposal, [&]( proposal_object& po ) {
      auto bitcoin_transaction_send_op = po.proposed_transaction.operations[0].get<bitcoin_transaction_send_operation>();
      bitcoin_transaction_send_op.signatures[son_obj->id] = op.signatures;
      po.proposed_transaction.operations[0] = bitcoin_transaction_send_op;
   });

   update_proposal( op );
} FC_CAPTURE_AND_RETHROW( (op) ) }

void bitcoin_transaction_sign_evaluator::update_proposal( const bitcoin_transaction_sign_operation& op )
{
   database& d = db();
   proposal_update_operation update_op;
   
   update_op.fee_paying_account = op.payer;
   update_op.proposal = op.proposal_id;
   update_op.active_approvals_to_add = { op.payer };

   bool skip_fee_old = trx_state->skip_fee;
   bool skip_fee_schedule_check_old = trx_state->skip_fee_schedule_check;
   trx_state->skip_fee = true;
   trx_state->skip_fee_schedule_check = true;

   d.apply_operation( *trx_state, update_op );

   trx_state->skip_fee = skip_fee_old;
   trx_state->skip_fee_schedule_check = skip_fee_schedule_check_old;
}

} } // namespace graphene::chain