// Copyright 2015-2020 Parity Technologies (UK) Ltd.
// This file is part of Tetsy Vapory.

// Tetsy Vapory is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Tetsy Vapory is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Tetsy Vapory.  If not, see <http://www.gnu.org/licenses/>.

//! Vapory transaction

use vvm::Schedule;
use common_types::transaction::{self, Action};

/// Extends transaction with gas verification method.
pub trait Transaction {
	/// Get the transaction cost in gas for this transaction.
	fn gas_required(&self, schedule: &Schedule) -> u64;
}

impl Transaction for transaction::Transaction {
	fn gas_required(&self, schedule: &Schedule) -> u64 {
		gas_required_for(match self.action {
			Action::Create => true,
			Action::Call(_) => false
		}, &self.data, schedule)
	}
}

/// Get the transaction cost in gas for the given params.
fn gas_required_for(is_create: bool, data: &[u8], schedule: &Schedule) -> u64 {
	data.iter().fold(
		(if is_create {schedule.tx_create_gas} else {schedule.tx_gas}) as u64,
		|g, b| g + (match *b { 0 => schedule.tx_data_zero_gas, _ => schedule.tx_data_non_zero_gas }) as u64
	)
}
