q9f/ruby-eth

module Eth
  class Tx

    include RLP::Sedes::Serializable
    extend Sedes

    set_serializable_fields({
      nonce: big_endian_int,
      gas_price: big_endian_int,
      gas_limit: big_endian_int,
      to: address,
      value: big_endian_int,
      data_bin: binary,
      v: big_endian_int,
      r: big_endian_int,
      s: big_endian_int
    })

    attr_writer :signature

    def self.decode(data)
      data = Utils.hex_to_bin(data) if data.match(/\A(?:0x)?\h+\Z/)
      txh = deserialize(RLP.decode data).to_h

      txh[:chain_id] = Eth.chain_id_from_signature(txh)

      self.new txh
    end

    def initialize(params)
      fields = {v: 0, r: 0, s: 0}.merge params
      fields[:to] = Utils.normalize_address(fields[:to])

      self.chain_id = (params[:chain_id]) ? params.delete(:chain_id) : Eth.chain_id

      if params[:data]
        self.data = params.delete(:data)
        fields[:data_bin] = data_bin
      end
      serializable_initialize fields

      check_transaction_validity
    end

    def unsigned_encoded
      us = unsigned
      RLP.encode(us, sedes: us.sedes)
    end

    def signing_data
      Utils.bin_to_prefixed_hex unsigned_encoded
    end

    def encoded
      RLP.encode self
    end

    def hex
      Utils.bin_to_prefixed_hex encoded
    end

    def sign(key)
      sig = key.sign(unsigned_encoded)
      vrs = Utils.v_r_s_for sig
      self.v = (self.chain_id) ? ((self.chain_id * 2) + vrs[0] + 8) : vrs[0]
      self.r = vrs[1]
      self.s = vrs[2]

      clear_signature
      self
    end

    def to_h
      hash_keys.inject({}) do |hash, field|
        hash[field] = send field
        hash
      end
    end

    def from
      if ecdsa_signature
        public_key = OpenSsl.recover_compact(signature_hash, ecdsa_signature)
        Utils.public_key_to_address(public_key) if public_key
      end
    end

    def signature
      return @signature if @signature
      @signature = { v: v, r: r, s: s } if [v, r, s].all? && (v > 0)
    end

    def ecdsa_signature
      return @ecdsa_signature if @ecdsa_signature

      if [v, r, s].all? && (v > 0)
        s_v = (self.chain_id) ? (v - (self.chain_id * 2) - 8) : v
        @ecdsa_signature = [
          Utils.int_to_base256(s_v),
          Utils.zpad_int(r),
          Utils.zpad_int(s),
        ].join
      end
    end

    def hash
      "0x#{Utils.bin_to_hex Utils.keccak256_rlp(self)}"
    end
    alias_method :id, :hash

    def data_hex
      Utils.bin_to_prefixed_hex data_bin
    end

    def data_hex=(hex)
      self.data_bin = Utils.hex_to_bin(hex)
    end

    def data
      Eth.tx_data_hex? ? data_hex : data_bin
    end

    def data=(string)
      Eth.tx_data_hex? ? self.data_hex=(string) : self.data_bin=(string)
    end

    def chain_id
      @chain_id
    end

    def chain_id=(cid)
      if cid != @chain_id
        self.v = 0
        self.r = 0
        self.s = 0

        clear_signature
      end

      @chain_id = (cid == 0) ? nil : cid
    end

    def prevent_replays?
      !self.chain_id.nil?
    end

    private

    def clear_signature
      @signature = nil
      @ecdsa_signature = nil
    end

    def hash_keys
      keys = self.class.serializable_fields.keys
      keys.delete(:data_bin)
      keys + [:data, :chain_id]
    end

    def check_transaction_validity
      if [gas_price, gas_limit, value, nonce].max > UINT_MAX
        raise InvalidTransaction, "Values way too high!"
      elsif gas_limit < intrinsic_gas_used
        raise InvalidTransaction, "Gas limit too low"
      end
    end

    def intrinsic_gas_used
      num_zero_bytes = data_bin.count(BYTE_ZERO)
      num_non_zero_bytes = data_bin.size - num_zero_bytes

      Gas::GTXCOST +
        Gas::GTXDATAZERO * num_zero_bytes +
        Gas::GTXDATANONZERO * num_non_zero_bytes
    end

    def signature_hash
      Utils.keccak256 unsigned_encoded
    end

    def unsigned
      Tx.new to_h.merge(v: (self.chain_id) ? self.chain_id : 0, r: 0, s: 0)
    end

    protected

    def sedes
      if self.prevent_replays? && !(Eth.replayable_v? v)
        self.class
      else
        UnsignedTx
      end
    end

  end

  UnsignedTx = Tx.exclude([:v, :r, :s])
end