module Bitcoin
  module Protocol

    class Version
      attr_reader :fields
      def initialize(opts={})
        @fields = {
          :version    => Bitcoin::Protocol::VERSION,
          :services   => 1,
          :time       => Time.now.tv_sec,
          :from       => "127.0.0.1:8333",
          :to         => "127.0.0.1:8333",
          :nonce      => Bitcoin::Protocol::Uniq,
          :user_agent => "/bitcoin-ruby:#{Bitcoin::VERSION}/",
          :last_block => 0 # 188617
        }.merge( opts.reject{|k,v| v == nil } )
      end

      def to_payload
        payload = [
          @fields.values_at(:version, :services, :time).pack("IQQ"),
          pack_address_field(@fields[:from]),
          pack_address_field(@fields[:to]),
          @fields.values_at(:nonce).pack("Q"),
          Protocol.pack_var_string(@fields[:user_agent]),
          @fields.values_at(:last_block).pack("I")
        ].join
      end

      def to_pkt
        Bitcoin::Protocol.pkt("version", to_payload)
      end

      def parse(payload)
        version, services, timestamp, to, from, nonce, payload = payload.unpack("Ia8Qa26a26Qa*")
        to, from = unpack_address_field(to), unpack_address_field(from)
        user_agent, payload = Protocol.unpack_var_string(payload)
        last_block = payload.unpack("I")[0]

        @fields = {
         :version => version, :services => services, :time => timestamp,
         :from => from, :to => to, :nonce => nonce,
         :user_agent => user_agent, :last_block => last_block
        }
        self
      end

      def unpack_address_field(payload)
        ip, port = payload.unpack("x8x12a4n")
        "#{ip.unpack("C*").join(".")}:#{port}"
      end

      def pack_address_field(addr_str)
        host, port = addr_str.split(":")
        port = port ? port.to_i : 8333
        sockaddr = Socket.pack_sockaddr_in(port, host)
        #raise "invalid IPv4 Address: #{addr}" unless sockaddr[0...2] == "\x02\x00"
        port, host = sockaddr[2...4], sockaddr[4...8]
        [[1].pack("Q"), "\x00"*10, "\xFF\xFF",  host, port].join
      end

      def uptime
        @fields[:time] - Time.now.tv_sec
      end

      def self.parse(payload); new.parse(payload); end
    end

  end
end