crea-ruby 0.0.1

data/lib/crea/transaction_builder.rb

@@ -0,0 +1,393 @@
+module Crea
+  # {TransactionBuilder} can be used to create a transaction that the
+  # {NetworkBroadcastApi} can broadcast to the rest of the platform.  The main
+  # feature of this class is the ability to cryptographically sign the
+  # transaction so that it conforms to the consensus rules that are required by
+  # the blockchain.
+  #
+  #     wif = '5JrvPrQeBBvCRdjv29iDvkwn3EQYZ9jqfAHzrCyUvfbEbRkrYFC'
+  #     builder = Crea::TransactionBuilder.new(wif: wif)
+  #     builder.put(vote: {
+  #       voter: 'alice',
+  #       author: 'bob',
+  #       permlink: 'my-burgers',
+  #       weight: 10000
+  #     })
+  #     
+  #     trx = builder.transaction
+  #     network_broadcast_api = Crea::CondenserApi.new
+  #     network_broadcast_api.broadcast_transaction_synchronous(trx: trx)
+  #
+  #
+  # The `wif` value may also be an array, when signing with multiple signatures
+  # (multisig).
+  class TransactionBuilder
+    include Retriable
+    include ChainConfig
+    include Utils
+    
+    attr_accessor :app_base, :database_api, :block_api, :expiration, :operations
+    attr_writer :wif
+    attr_reader :signed, :testnet, :force_serialize
+    
+    alias app_base? app_base
+    alias testnet? testnet
+    alias force_serialize? force_serialize
+    
+    def initialize(options = {})
+      @app_base = !!options[:app_base] # default false
+      @database_api = options[:database_api]
+      @block_api = options[:block_api]
+      
+      if app_base?
+        @database_api ||= Crea::DatabaseApi.new(options)
+        @block_api ||= Crea::BlockApi.new(options)
+      else
+        @database_api ||= Crea::CondenserApi.new(options)
+        @block_api ||= Crea::CondenserApi.new(options)
+      end
+      
+      @wif = [options[:wif]].flatten
+      @signed = false
+      @testnet = !!options[:testnet]
+      @force_serialize = !!options[:force_serialize]
+      
+      if !!(trx = options[:trx])
+        trx = case trx
+        when String then JSON[trx]
+        else; trx
+        end
+        
+        @trx = Transaction.new(trx)
+      end
+      
+      @trx ||= Transaction.new
+      @chain = options[:chain] || :crea
+      @error_pipe = options[:error_pipe] || STDERR
+      @chain_id = options[:chain_id]
+      @chain_id ||= case @chain
+      when :crea then NETWORKS_CREA_CHAIN_ID
+      when :test then NETWORKS_TEST_CHAIN_ID
+      else; raise UnsupportedChainError, "Unsupported chain: #{@chain}"
+      end
+      
+      if testnet? && @chain_id == NETWORKS_CREA_CHAIN_ID
+        raise UnsupportedChainError, "Unsupported testnet chain id: #{@chain_id}"
+      end
+    end
+    
+    def inspect
+      properties = %w(trx).map do |prop|
+        if !!(v = instance_variable_get("@#{prop}"))
+          "@#{prop}=#{v.inspect}" 
+        end
+      end.compact.join(', ')
+      
+      "#<#{self.class.name} [#{properties}]>"
+    end
+    
+    def reset
+      @trx = Transaction.new
+      @signed = false
+      
+      self
+    end
+    
+    # If the transaction can be prepared, this method will do so and set the
+    # expiration.  Once the expiration is set, it will not re-prepare.  If you
+    # call {#put}, the expiration is set {::Nil} so that it can be re-prepared.
+    #
+    # Usually, this method is called automatically by {#put} and/or {#transaction}.
+    #
+    # @return {TransactionBuilder}
+    def prepare
+      if @trx.expired?
+        catch :prepare_header do; begin
+          @database_api.get_dynamic_global_properties do |properties|
+            block_number = properties.last_irreversible_block_num
+            block_header_args = if app_base?
+              {block_num: block_number}
+            else
+              block_number
+            end
+          
+            @block_api.get_block_header(block_header_args) do |result|
+              header = if app_base?
+                result.header
+              else
+                result
+              end
+              
+              @trx.ref_block_num = (block_number - 1) & 0xFFFF
+              @trx.ref_block_prefix = unhexlify(header.previous[8..-1]).unpack('V*')[0]
+              @trx.expiration ||= (Time.parse(properties.time + 'Z') + EXPIRE_IN_SECS).utc
+            end
+          end
+        rescue => e
+          if can_retry? e
+            @error_pipe.puts "#{e} ... retrying."
+            throw :prepare_header
+          else
+            raise e
+          end
+        end; end
+      end
+      
+      self
+    end
+    
+    # Sets operations all at once, then prepares.
+    def operations=(operations)
+      @trx.operations = operations.map{ |op| normalize_operation(op) }
+      prepare
+      @trx.operations
+    end
+    
+    # A quick and flexible way to append a new operation to the transaction.
+    # This method uses ducktyping to figure out how to form the operation.
+    #
+    # There are three main ways you can call this method.  These assume that
+    # `op_type` is a {::Symbol} (or {::String}) representing the type of operation and `op` is the
+    # operation {::Hash}.
+    #
+    #     put(op_type, op)
+    #
+    # ... or ...
+    #
+    #     put(op_type => op)
+    #
+    # ... or ...
+    #
+    #     put([op_type, op])
+    #
+    # You can also chain multiple operations:
+    #
+    #     builder = Crea::TransactionBuilder.new
+    #     builder.put(vote: vote1).put(vote: vote2)
+    # @return {TransactionBuilder}
+    def put(type, op = nil)
+      @trx.expiration = nil
+      @trx.operations << normalize_operation(type, op)
+      prepare
+      self
+    end
+    
+    # If all of the required values are set, this returns a fully formed
+    # transaction that is ready to broadcast.
+    # 
+    # @return
+    #     {
+    #            :ref_block_num => 18912,
+    #         :ref_block_prefix => 575781536,
+    #               :expiration => "2018-04-26T15:26:12",
+    #               :extensions => [],
+    #               :operations => [[:vote, {
+    #                    :voter => "alice",
+    #                   :author => "bob",
+    #                 :permlink => "my-burgers",
+    #                   :weight => 10000
+    #                 }
+    #             ]],
+    #               :signatures => ["1c45b65740b4b2c17c4bcf6bcc3f8d90ddab827d50532729fc3b8f163f2c465a532b0112ae4bf388ccc97b7c2e0bc570caadda78af48cf3c261037e65eefcd941e"]
+    #     }
+    def transaction(options = {prepare: true, sign: true})
+      options[:prepare] = true unless options.has_key? :prepare
+      options[:sign] = true unless options.has_key? :sign
+      
+      prepare if !!options[:prepare]
+      
+      if !!options[:sign]
+        sign
+      else
+        @trx
+      end
+    end
+    
+    # Appends to the `signatures` array of the transaction, built from a
+    # serialized digest.
+    #
+    # @return {Hash | TransactionBuilder} The fully signed transaction if a `wif` is provided or the instance of the {TransactionBuilder} if a `wif` has not yet been provided.
+    def sign
+      return self if @wif.empty?
+      return self if @trx.expired?
+      
+      unless @signed
+        catch :serialize do; begin
+          transaction_hex.tap do |result|
+            hex = if app_base?
+              result.hex
+            else
+              result
+            end
+
+            unless force_serialize?
+              derrived_trx = Transaction.new(hex: hex)
+              derrived_ops = derrived_trx.operations
+              derrived_trx.operations = derrived_ops.map do |op|
+                op_name = if app_base?
+                  op[:type].to_sym
+                else
+                  op[:type].to_s.sub(/_operation$/, '').to_sym
+                end
+                
+                normalize_operation op_name, JSON[op[:value].to_json]
+              end
+              
+              raise SerializationMismatchError unless @trx == derrived_trx
+            end
+            
+            hex = hex[0..-4] # drop empty signature array
+            @trx.id = Digest::SHA256.hexdigest(unhexlify(hex))[0..39]
+            
+            hex = @chain_id + hex
+            digest = unhexlify(hex)
+            digest_hex = Digest::SHA256.digest(digest)
+            private_keys = @wif.map{ |wif| Bitcoin::Key.from_base58 wif }
+            ec = Bitcoin::OpenSSL_EC
+            count = 0
+            
+            private_keys.each do |private_key|
+              sig = nil
+              
+              loop do
+                count += 1
+                @error_pipe.puts "#{count} attempts to find canonical signature" if count % 40 == 0
+                public_key_hex = private_key.pub
+                sig = ec.sign_compact(digest_hex, private_key.priv, public_key_hex, false)
+                
+                next if public_key_hex != ec.recover_compact(digest_hex, sig)
+                break if canonical? sig
+              end
+              
+              @trx.signatures << hexlify(sig)
+            end
+            
+            @signed = true
+          end
+        rescue => e
+          if can_retry? e
+            @error_pipe.puts "#{e} ... retrying."
+            throw :serialize
+          else
+            raise e
+          end
+        end; end
+      end
+        
+      @trx
+    end
+    
+    def transaction_hex
+      trx = transaction(prepare: true, sign: false)
+      
+      transaction_hex_args = if app_base?
+        {trx: trx}
+      else
+        trx
+      end
+      
+      @database_api.get_transaction_hex(transaction_hex_args) do |result|
+        if app_base?
+          result[:hex]
+        else
+          result
+        end
+      end
+    end
+    
+    # @return [Array] All public keys that could possibly sign for a given transaction.
+    def potential_signatures
+      potential_signatures_args = if app_base?
+        {trx: transaction}
+      else
+        transaction
+      end
+      
+      @database_api.get_potential_signatures(potential_signatures_args) do |result|
+        if app_base?
+          result[:keys]
+        else
+          result
+        end
+      end
+    end
+    
+    # This API will take a partially signed transaction and a set of public keys
+    # that the owner has the ability to sign for and return the minimal subset
+    # of public keys that should add signatures to the transaction.
+    #
+    # @return [Array] The minimal subset of public keys that should add signatures to the transaction.
+    def required_signatures
+      required_signatures_args = if app_base?
+        {trx: transaction}
+      else
+        [transaction, []]
+      end
+      
+      @database_api.get_required_signatures(*required_signatures_args) do |result|
+        if app_base?
+          result[:keys]
+        else
+          result
+        end
+      end
+    end
+    
+    # @return [Boolean] True if the transaction has all of the required signatures.
+    def valid?
+      verify_authority_args = if app_base?
+        {trx: transaction}
+      else
+        transaction
+      end
+      
+      @database_api.verify_authority(verify_authority_args) do |result|
+        if app_base?
+          result.valid
+        else
+          result
+        end
+      end
+    end
+  private
+    # See: https://github.com/creary/crea/pull/2500
+    # @private
+    def canonical?(sig)
+      sig = sig.unpack('C*')
+      
+      !(
+        ((sig[0] & 0x80 ) != 0) || ( sig[0] == 0 ) ||
+        ((sig[1] & 0x80 ) != 0) ||
+        ((sig[32] & 0x80 ) != 0) || ( sig[32] == 0 ) ||
+        ((sig[33] & 0x80 ) != 0)
+      )
+    end
+    
+    def normalize_operation(type, op = nil)
+      if app_base?
+        case type
+        when Symbol, String
+          type_value = "#{type}_operation"
+          {type: type_value, value: op}
+        when Hash
+          type_value = "#{type.keys.first}_operation"
+          {type: type_value, value: type.values.first}
+        when Array
+          type_value = "#{type[0]}_operation"
+          {type: type_value, value: type[1]}
+        else
+          raise Crea::ArgumentError, "Don't know what to do with operation type #{type.class}: #{type} (#{op})"
+        end
+      else
+        case type
+        when Symbol then [type, op]
+        when String then [type.to_sym, op]
+        when Hash then [type.keys.first.to_sym, type.values.first]
+        when Array then type
+        else
+          raise Crea::ArgumentError, "Don't know what to do with operation type #{type.class}: #{type} (#{op})"
+        end
+      end
+    end
+  end
+end

data/lib/crea/type/amount.rb

@@ -0,0 +1,107 @@
+module Crea
+  module Type
+    
+    # See: https://github.com/xeroc/piston-lib/blob/34a7525cee119ec9b24a99577ede2d54466fca0e/creabase/operations.py
+    class Amount < BaseType
+      attr_reader :amount, :precision, :nai, :asset
+      
+      def self.to_h(amount)
+        new(amount).to_h
+      end
+      
+      def self.to_s(amount)
+        new(amount).to_s
+      end
+      
+      def self.to_bytes(amount)
+        new(amount).to_bytes
+      end
+      
+      def initialize(value)
+        super(:amount, value)
+        
+        case value
+        when Array
+          @amount, @precision, @nai = value
+          @asset = case @nai
+          when '@@000000013' then 'CBD'
+          when '@@000000021' then 'CREA'
+          when '@@000000037' then 'VESTS'
+          else; raise TypeError, "Asset #{@nai} unknown."
+          end
+          
+          @amount = "%.#{@precision}f" % (@amount.to_f / 10 ** @precision)
+        when Hash
+          @amount, @precision, @nai = value.map do |k, v|
+            v if %i(amount precision nai).include? k.to_sym
+          end.compact
+          
+          @asset = case @nai
+          when '@@000000013' then 'CBD'
+          when '@@000000021' then 'CREA'
+          when '@@000000037' then 'VESTS'
+          else; raise TypeError, "Asset #{@nai} unknown."
+          end
+          
+          @amount = "%.#{@precision}f" % (@amount.to_f / 10 ** @precision)
+        when Amount
+          @precision = value.precision
+          @nai = value.nai
+          @asset = value.asset
+          @amount = value.amount
+        else
+          @amount, @asset = value.strip.split(' ') rescue ['', '']
+          @precision = case @asset
+          when 'CREA' then 3
+          when 'VESTS' then 6
+          when 'CBD' then 3
+          when 'TESTS' then 3
+          when 'TBD' then 3
+          else; raise TypeError, "Asset #{@asset} unknown."
+          end
+        end
+      end
+      
+      def to_bytes
+        asset = @asset.ljust(7, "\x00")
+        amount = (@amount.to_f * 10 ** @precision).round
+        
+        [amount].pack('q') +
+        [@precision].pack('c') +
+        asset
+      end
+      
+      def to_a
+        case @asset
+        when 'CREA' then [(@amount.to_f * 1000).to_i.to_s, 3, '@@000000021']
+        when 'VESTS' then [(@amount.to_f * 1000000).to_i.to_s, 6, '@@000000037']
+        when 'CBD' then [(@amount.to_f * 1000).to_i.to_s, 3, '@@000000013']
+        end
+      end
+      
+      def to_h
+        case @asset
+        when 'CREA' then {
+          amount: (@amount.to_f * 1000).to_i.to_s,
+          precision: 3,
+          nai: '@@000000021'
+        }
+        when 'VESTS' then {
+          amount: (@amount.to_f * 1000000).to_i.to_s,
+          precision: 6,
+          nai: '@@000000037'
+        }
+        when 'CBD' then {
+          amount: (@amount.to_f * 1000).to_i.to_s,
+          precision: 3,
+          nai: '@@000000013'
+        }
+        end
+      end
+      
+      def to_s
+        "#{@amount} #{@asset}"
+      end
+    end
+  end
+end