// Rust Bitcoin Library
// Written in 2014 by
//   Andrew Poelstra <apoelstra@wpsoftware.net>
//
// To the extent possible under law, the author(s) have dedicated all
// copyright and related and neighboring rights to this software to
// the public domain worldwide. This software is distributed without
// any warranty.
//
// You should have received a copy of the CC0 Public Domain Dedication
// along with this software.
// If not, see <http://creativecommons.org/publicdomain/zero/1.0/>.
//

//! # Bitcoin Transaction
//!
//! A transaction describes a transfer of money. It consumes previously-unspent
//! transaction outputs and produces new ones, satisfying the condition to spend
//! the old outputs (typically a digital signature with a specific key must be
//! provided) and defining the condition to spend the new ones. The use of digital
//! signatures ensures that coins cannot be spent by unauthorized parties.
//!
//! This module provides the structures and functions needed to support transactions.
//!

use std::default::Default;
use serialize::hex::ToHex;
use serialize::json;

use util::hash::Sha256dHash;
use blockdata::script::{mod, Script, ScriptError, TraceIteration, read_scriptbool};
use blockdata::utxoset::UtxoSet;
use network::encodable::ConsensusEncodable;
use network::serialize::BitcoinHash;

/// A transaction input, which defines old coins to be consumed
#[deriving(Clone, PartialEq, Eq, Show)]
pub struct TxIn {
  /// The hash of the transaction whose output is being used an an input
  pub prev_hash: Sha256dHash,
  /// The index of the output in the previous transaction, which may have several
  pub prev_index: u32,
  /// The script which pushes values on the stack which will cause
  /// the referenced output's script to accept
  pub script_sig: Script,
  /// The sequence number, which suggests to miners which of two
  /// conflicting transactions should be preferred, or 0xFFFFFFFF
  /// to ignore this feature. This is generally never used since
  /// the miner behaviour cannot be enforced.
  pub sequence: u32,
}

/// A transaction output, which defines new coins to be created from old ones.
#[deriving(Clone, PartialEq, Eq, Show)]
pub struct TxOut {
  /// The value of the output, in satoshis
  pub value: u64,
  /// The script which must satisfy for the output to be spent
  pub script_pubkey: Script
}

// This is used as a "null txout" in consensus signing code
impl Default for TxOut {
  fn default() -> TxOut {
    TxOut { value: 0xffffffffffffffff, script_pubkey: Script::new() }
  }
}

/// A Bitcoin transaction, which describes an authenticated movement of coins
#[deriving(Clone, PartialEq, Eq, Show)]
pub struct Transaction {
  /// The protocol version, should always be 1.
  pub version: u32,
  /// Block number before which this transaction is valid, or 0 for
  /// valid immediately.
  pub lock_time: u32,
  /// List of inputs
  pub input: Vec<TxIn>,
  /// List of outputs
  pub output: Vec<TxOut>
}

/// A transaction error
#[deriving(PartialEq, Eq, Clone, Show)]
pub enum TransactionError {
  /// Concatenated script failed in the input half (script error)
  InputScriptFailure(ScriptError),
  /// Concatenated script failed in the output half (script error)
  OutputScriptFailure(ScriptError),
  /// P2SH serialized script failed (script error)
  P2shScriptFailure(ScriptError),
  /// P2SH serialized script ended with false at the top of the stack 
  P2shScriptReturnedFalse,
  /// P2SH serialized script ended with nothing in the stack
  P2shScriptReturnedEmptyStack,
  /// Script ended with false at the top of the stack 
  ScriptReturnedFalse,
  /// Script ended with nothing in the stack
  ScriptReturnedEmptyStack,
  /// Script ended with nothing in the stack (input txid, input vout)
  InputNotFound(Sha256dHash, u32),
}

impl json::ToJson for TransactionError {
  fn to_json(&self) -> json::Json {
    json::String(self.to_string())
  }
}

/// A trace of a script execution
#[deriving(PartialEq, Eq, Show, Clone)]
pub struct ScriptTrace {
  script: Script,
  initial_stack: Vec<String>,
  iterations: Vec<TraceIteration>,
  error: Option<ScriptError>
}

/// A trace of a transaction input's script execution
#[deriving(PartialEq, Eq, Clone, Show)]
pub struct InputTrace {
  input_txid: Sha256dHash,
  input_vout: uint,
  sig_trace: ScriptTrace,
  pubkey_trace: Option<ScriptTrace>,
  p2sh_trace: Option<ScriptTrace>,
  error: Option<TransactionError>
}

impl_json!(ScriptTrace, script, initial_stack, iterations, error)
impl_json!(InputTrace, input_txid, input_vout, sig_trace,
                       pubkey_trace, p2sh_trace, error)

/// A trace of a transaction's execution
#[deriving(PartialEq, Eq, Clone, Show)]
pub struct TransactionTrace {
  txid: Sha256dHash,
  inputs: Vec<InputTrace>
}

impl_json!(TransactionTrace, txid, inputs)

impl Transaction {
  /// Check a transaction for validity
  pub fn validate(&self, utxoset: &UtxoSet) -> Result<(), TransactionError> {
    for (n, input) in self.input.iter().enumerate() {
      let txo = utxoset.get_utxo(input.prev_hash, input.prev_index);
      match txo {
        Some(txo) => {
          let mut p2sh_stack = Vec::new();
          let mut p2sh_script = Script::new();

          let mut stack = Vec::with_capacity(6);
          match input.script_sig.evaluate(&mut stack, Some((self, n)), None) {
            Ok(_) => {}
            Err(e) => { return Err(InputScriptFailure(e)); }
          }
          if txo.script_pubkey.is_p2sh() && stack.len() > 0 {
            p2sh_stack = stack.clone();
            p2sh_script = match p2sh_stack.pop() {
              Some(script::Owned(v)) => Script::from_vec(v),
              Some(script::Slice(s)) => Script::from_vec(Vec::from_slice(s)),
              None => unreachable!()
            };
          }
          match txo.script_pubkey.evaluate(&mut stack, Some((self, n)), None) {
            Ok(_) => {}
            Err(e) => { return Err(OutputScriptFailure(e)); }
          }
          match stack.pop() {
            Some(v) => {
              if !read_scriptbool(v.as_slice()) {
                return Err(ScriptReturnedFalse);
              }
            }
            None => { return Err(ScriptReturnedEmptyStack); }
          }
          if txo.script_pubkey.is_p2sh() {
            match p2sh_script.evaluate(&mut p2sh_stack, Some((self, n)), None) {
              Ok(_) => {}
              Err(e) => { return Err(P2shScriptFailure(e)); }
            }
            match p2sh_stack.pop() {
              Some(v) => {
                if !read_scriptbool(v.as_slice()) {
                  return Err(P2shScriptReturnedFalse);
                }
              }
              None => { return Err(P2shScriptReturnedEmptyStack); }
            }
          }
        }
        None => { return Err(InputNotFound(input.prev_hash, input.prev_index)); }
      }
    }
    Ok(())
  }

  /// Produce a trace of a transaction's execution
  pub fn trace(&self, utxoset: &UtxoSet) -> TransactionTrace {
    let mut ret = TransactionTrace { txid: self.bitcoin_hash(),
                                     inputs: Vec::with_capacity(self.input.len()) };
    for (n, input) in self.input.iter().enumerate() {
      // Setup trace
      let mut trace = InputTrace {
        input_txid: input.prev_hash,
        input_vout: input.prev_index as uint,
        sig_trace: ScriptTrace {
          script: Script::new(),
          initial_stack: vec![],
          iterations: vec![],
          error: None
        },
        pubkey_trace: None,
        p2sh_trace: None,
        error: None
      };
      // Run through the input
      let txo = utxoset.get_utxo(input.prev_hash, input.prev_index);
      match txo {
        Some(txo) => {
          let mut p2sh_stack = Vec::new();
          let mut p2sh_script = Script::new();

          let mut stack = Vec::with_capacity(6);
          trace.sig_trace.script = input.script_sig.clone();
          match input.script_sig.evaluate(&mut stack,
                                          Some((self, n)),
                                          Some(&mut trace.sig_trace.iterations)) {
            Ok(_) => {}
            Err(e) => {
              trace.sig_trace.error = Some(e.clone());
              trace.error = Some(InputScriptFailure(e));
            }
          }
          if txo.script_pubkey.is_p2sh() && stack.len() > 0 {
            p2sh_stack = stack.clone();
            p2sh_script = match p2sh_stack.pop() {
              Some(script::Owned(v)) => Script::from_vec(v),
              Some(script::Slice(s)) => Script::from_vec(Vec::from_slice(s)),
              None => unreachable!()
            };
          }
          if trace.error.is_none() {
            let mut pk_trace = ScriptTrace {
              script: txo.script_pubkey.clone(),
              initial_stack: stack.iter().map(|elem| elem.as_slice().to_hex()).collect(),
              iterations: vec![],
              error: None
            };
            match txo.script_pubkey.evaluate(&mut stack,
                                             Some((self, n)),
                                             Some(&mut pk_trace.iterations)) {
              Ok(_) => {}
              Err(e) => {
                pk_trace.error = Some(e.clone());
                trace.error = Some(OutputScriptFailure(e));
              }
            }
            match stack.pop() {
              Some(v) => {
                if !read_scriptbool(v.as_slice()) {
                  trace.error = Some(ScriptReturnedFalse);
                }
              }
              None => { trace.error = Some(ScriptReturnedEmptyStack); }
            }
            trace.pubkey_trace = Some(pk_trace);
            if trace.error.is_none() && txo.script_pubkey.is_p2sh() {
              let mut p2sh_trace = ScriptTrace {
                script: p2sh_script.clone(),
                initial_stack: p2sh_stack.iter().map(|elem| elem.as_slice().to_hex()).collect(),
                iterations: vec![],
                error: None
              };
              match p2sh_script.evaluate(&mut p2sh_stack,
                                         Some((self, n)),
                                         Some(&mut p2sh_trace.iterations)) {
                Ok(_) => {}
                Err(e) => {
                  p2sh_trace.error = Some(e.clone());
                  trace.error = Some(P2shScriptFailure(e));
                }
              }
              match p2sh_stack.pop() {
                Some(v) => {
                  if !read_scriptbool(v.as_slice()) {
                    trace.error = Some(P2shScriptReturnedFalse);
                  }
                }
                None => { trace.error = Some(P2shScriptReturnedEmptyStack); }
              }
              trace.p2sh_trace = Some(p2sh_trace);
            }
          }
        }
        None => {
          trace.error = Some(InputNotFound(input.prev_hash, input.prev_index));
        }
      }
      ret.inputs.push(trace);
    }
    ret
  }
}

impl BitcoinHash for Transaction {
  fn bitcoin_hash(&self) -> Sha256dHash {
    use network::serialize::serialize;
    Sha256dHash::from_data(serialize(self).unwrap().as_slice())
  }
}

impl_consensus_encoding!(TxIn, prev_hash, prev_index, script_sig, sequence)
impl_json!(TxIn, prev_hash, prev_index, script_sig, sequence)
impl_consensus_encoding!(TxOut, value, script_pubkey)
impl_json!(TxOut, value, script_pubkey)
impl_consensus_encoding!(Transaction, version, input, output, lock_time)
impl_json!(Transaction, version, input, output, lock_time)


#[cfg(test)]
mod tests {
  use super::{Transaction, TxIn};

  use std::io::IoResult;

  use network::serialize::BitcoinHash;
  use network::serialize::deserialize;
  use util::misc::hex_bytes;

  #[test]
  fn test_txin() {
    let txin: IoResult<TxIn> = deserialize(hex_bytes("a15d57094aa7a21a28cb20b59aab8fc7d1149a3bdbcddba9c622e4f5f6a99ece010000006c493046022100f93bb0e7d8db7bd46e40132d1f8242026e045f03a0efe71bbb8e3f475e970d790221009337cd7f1f929f00cc6ff01f03729b069a7c21b59b1736ddfee5db5946c5da8c0121033b9b137ee87d5a812d6f506efdd37f0affa7ffc310711c06c7f3e097c9447c52ffffffff").unwrap());
    assert!(txin.is_ok());
  }

  #[test]
  fn test_transaction() {
    let hex_tx = hex_bytes("0100000001a15d57094aa7a21a28cb20b59aab8fc7d1149a3bdbcddba9c622e4f5f6a99ece010000006c493046022100f93bb0e7d8db7bd46e40132d1f8242026e045f03a0efe71bbb8e3f475e970d790221009337cd7f1f929f00cc6ff01f03729b069a7c21b59b1736ddfee5db5946c5da8c0121033b9b137ee87d5a812d6f506efdd37f0affa7ffc310711c06c7f3e097c9447c52ffffffff0100e1f505000000001976a9140389035a9225b3839e2bbf32d826a1e222031fd888ac00000000").unwrap();
    let tx: IoResult<Transaction> = deserialize(hex_tx);
    assert!(tx.is_ok());
    let realtx = tx.unwrap();
    // All these tests aren't really needed because if they fail, the hash check at the end
    // will also fail. But these will show you where the failure is so I'll leave them in.
    assert_eq!(realtx.version, 1);
    assert_eq!(realtx.input.len(), 1);
    // In particular this one is easy to get backward -- in bitcoin hashes are encoded
    // as little-endian 256-bit numbers rather than as data strings.
    assert_eq!(realtx.input[0].prev_hash.be_hex_string(),
               "ce9ea9f6f5e422c6a9dbcddb3b9a14d1c78fab9ab520cb281aa2a74a09575da1".to_string());
    assert_eq!(realtx.input[0].prev_index, 1);
    assert_eq!(realtx.output.len(), 1);
    assert_eq!(realtx.lock_time, 0);

    assert_eq!(realtx.bitcoin_hash().be_hex_string(),
               "a6eab3c14ab5272a58a5ba91505ba1a4b6d7a3a9fcbd187b6cd99a7b6d548cb7".to_string());
  }
}