coinbase-sdk 0.0.16 → 0.1.0

data/lib/coinbase/staking_operation.rb

@@ -11,22 +11,24 @@ module Coinbase
 
     # Builds an ephemeral staking operation this is intended to be called via an Address or Wallet.
     # @param amount [BigDecimal] The amount to stake, in the primary denomination of the asset
-    # @param network_id [Symbol] The Network ID
+    # @param network [Coinbase::Network, Symbol] The Network or Network ID
     # @param asset_id [Symbol] The Asset ID
     # @param address_id [String] The Address ID
     # @param action [Symbol] The action to perform
     # @param mode [Symbol] The staking mode
     # @param options [Hash] Additional options
     # @return [Coinbase::StakingOperation] The staking operation
-    def self.build(amount, network_id, asset_id, address_id, action, mode, options)
+    def self.build(amount, network, asset_id, address_id, action, mode, options)
+      network = Coinbase::Network.from_id(network)
+      asset = network.get_asset(asset_id)
+
       model = Coinbase.call_api do
-        asset = Coinbase::Asset.fetch(network_id, asset_id)
         stake_api.build_staking_operation(
           {
             asset_id: asset.primary_denomination.to_s,
             address_id: address_id,
             action: action,
-            network_id: Coinbase.normalize_network(network_id),
+            network_id: Coinbase.normalize_network(network),
             options: {
               amount: asset.to_atomic_amount(amount).to_i.to_s,
               mode: mode
@@ -40,7 +42,7 @@ module Coinbase
 
     # Creates a persisted staking operation this is intended to be called via an Address or Wallet.
     # @param amount [BigDecimal] The amount to stake, in the primary denomination of the asset
-    # @param network_id [Symbol] The Network ID
+    # @param network [Coinbase::Network, Symbol] The Network or Network ID
     # @param asset_id [Symbol] The Asset ID
     # @param address_id [String] The Address ID
     # @param wallet_id [String] The Wallet ID
@@ -48,17 +50,19 @@ module Coinbase
     # @param mode [Symbol] The staking mode
     # @param options [Hash] Additional options
     # @return [Coinbase::StakingOperation] The staking operation
-    def self.create(amount, network_id, asset_id, address_id, wallet_id, action, mode, options)
+    def self.create(amount, network, asset_id, address_id, wallet_id, action, mode, options)
+      network = Coinbase::Network.from_id(network)
+      asset = network.get_asset(asset_id)
+
       model = Coinbase.call_api do
-        asset = Coinbase::Asset.fetch(network_id, asset_id)
-        stake_api.create_staking_operation(
+        wallet_stake_api.create_staking_operation(
           wallet_id,
           address_id,
           {
             asset_id: asset.primary_denomination.to_s,
             address_id: address_id,
             action: action,
-            network_id: Coinbase.normalize_network(network_id),
+            network_id: Coinbase.normalize_network(network),
             options: {
               amount: asset.to_atomic_amount(amount).to_i.to_s,
               mode: mode
@@ -70,10 +74,24 @@ module Coinbase
       new(model)
     end
 
+    def self.stake_api
+      Coinbase::Client::StakeApi.new(Coinbase.configuration.api_client)
+    end
+
+    private_class_method :stake_api
+
+    def self.wallet_stake_api
+      Coinbase::Client::WalletStakeApi.new(Coinbase.configuration.api_client)
+    end
+
+    private_class_method :wallet_stake_api
+
     # Returns a new StakingOperation object.
     # @param model [Coinbase::Client::StakingOperation] The underlying StakingOperation object
     def initialize(model)
-      from_model(model)
+      @model = model
+      @transactions ||= []
+      update_transactions(model.transactions)
     end
 
     # Returns the Staking Operation ID.
@@ -82,10 +100,10 @@ module Coinbase
       @model.id
     end
 
-    # Returns the Network ID of the Staking Operation.
-    # @return [Symbol] The Network ID
-    def network_id
-      Coinbase.to_sym(@model.network_id)
+    # Returns the Network of the Staking Operation.
+    # @return [Coinbase::Network] The Network
+    def network
+      @network ||= Coinbase::Network.from_id(@model.network_id)
     end
 
     # Returns the Address ID of the Staking Operation.
@@ -100,18 +118,48 @@ module Coinbase
       @model.status
     end
 
+    # Returns whether the Staking Operation is in a terminal state.
+    # @return [Boolean] Whether the Staking Operation is in a terminal state
+    def terminal_state?
+      failed? || completed?
+    end
+
+    # Returns whether the Staking Operation is in a failed state.
+    # @return [Boolean] Whether the Staking Operation is in a failed state
+    def failed?
+      status == 'failed'
+    end
+
+    # Returns whether the Staking Operation is in a complete state.
+    # @return [Boolean] Whether the Staking Operation is in a complete state
+    def completed?
+      status == 'complete'
+    end
+
+    # Returns a String representation of the Staking Operation.
+    # @return [String] a String representation of the Staking Operation
+    def to_s
+      Coinbase.pretty_print_object(
+        self.class,
+        id: id,
+        status: status,
+        network_id: network.id,
+        address_id: address_id
+      )
+    end
+
     # Returns the Wallet ID of the Staking Operation.
     # @return [String] The Wallet ID
     def wallet_id
       @model.wallet_id
     end
 
-    # Waits until the Staking Operation is completed or failed by polling its status at the given interval. Raises a
-    # Timeout::Error if the Staking Operation takes longer than the given timeout.
+    # Waits until the Staking Operation is completed or failed by polling its status at the given interval.
     # @param interval_seconds [Integer] The interval at which to poll, in seconds
     # @param timeout_seconds [Integer] The maximum amount of time
     #   to wait for the StakingOperation to complete, in seconds
     # @return [StakingOperation] The completed StakingOperation object
+    # @raise [Timeout::Error] if the Staking Operation takes longer than the given timeout.
     def wait!(interval_seconds = 5, timeout_seconds = 3600)
       start_time = Time.now
 
@@ -119,7 +167,7 @@ module Coinbase
         reload
 
         # Wait for the Staking Operation to be in a terminal state.
-        break if status == 'complete'
+        break if terminal_state?
 
         raise Timeout::Error, 'Staking Operation timed out' if Time.now - start_time > timeout_seconds
 
@@ -129,18 +177,58 @@ module Coinbase
       self
     end
 
+    # Complete helps the Staking Operation reach complete state, by polling its status at the given interval, signing
+    # and broadcasting any available transaction.
+    # @param key [Eth::Key] The key to sign the Staking Operation transactions with
+    # @param interval_seconds [Integer] The interval at which to poll, in seconds
+    # @param timeout_seconds [Integer] The maximum amount of time
+    #   to wait for the StakingOperation to complete, in seconds
+    # @return [StakingOperation] The completed StakingOperation object
+    # @raise [Timeout::Error] if the Staking Operation takes longer than the given timeout.
+    def complete(key, interval_seconds: 5, timeout_seconds: 600)
+      start_time = Time.now
+
+      loop do
+        @transactions.each_with_index do |transaction, i|
+          next if transaction.signed?
+
+          transaction.sign(key)
+          @model = Coinbase.call_api do
+            stake_api.broadcast_staking_operation(
+              wallet_id,
+              address_id,
+              id,
+              { signed_payload: transaction.raw.hex, transaction_index: i }
+            )
+          end
+        end
+
+        return self if terminal_state?
+
+        reload
+
+        raise Timeout::Error, 'Staking Operation timed out' if Time.now - start_time > timeout_seconds
+
+        sleep interval_seconds
+      end
+
+      self
+    end
+
     # Fetch the StakingOperation with the provided network, address and staking operation ID.
-    # @param network_id [Symbol] The Network ID
+    # @param network [Coinbase::Network, Symbol] The Network or Network ID
     # @param address_id [Symbol] The Address ID
     # @param id [String] The ID of the StakingOperation
     # @param wallet_id [String] The optional Wallet ID
     # @return [Coinbase::StakingOperation] The staking operation
-    def self.fetch(network_id, address_id, id, wallet_id: nil)
+    def self.fetch(network, address_id, id, wallet_id: nil)
+      network = Coinbase::Network.from_id(network)
+
       staking_operation_model = Coinbase.call_api do
         if wallet_id.nil?
-          stake_api.get_external_staking_operation(network_id, address_id, id)
+          stake_api.get_external_staking_operation(network.id, address_id, id)
         else
-          stake_api.get_staking_operation(wallet_id, address_id, id)
+          wallet_stake_api.get_staking_operation(wallet_id, address_id, id)
         end
       end
 
@@ -160,13 +248,15 @@ module Coinbase
     def reload
       @model = Coinbase.call_api do
         if wallet_id.nil?
-          stake_api.get_external_staking_operation(network_id, address_id, id)
+          stake_api.get_external_staking_operation(network.id, address_id, id)
         else
-          stake_api.get_staking_operation(wallet_id, address_id, id)
+          wallet_stake_api.get_staking_operation(wallet_id, address_id, id)
         end
       end
 
-      from_model(@model)
+      update_transactions(@model.transactions)
+
+      self
     end
 
     # Fetches the presigned_voluntary exit messages for the staking operation
@@ -191,7 +281,7 @@ module Coinbase
         raise TransactionNotSignedError unless transaction.signed?
 
         Coinbase.call_api do
-          stake_api.broadcast_staking_operation(
+          wallet_stake_api.broadcast_staking_operation(
             wallet_id,
             address_id,
             id,
@@ -203,24 +293,32 @@ module Coinbase
       self
     end
 
-    def self.stake_api
-      Coinbase::Client::StakeApi.new(Coinbase.configuration.api_client)
-    end
-
     private
 
     def stake_api
-      @stake_api ||= self.class.stake_api
+      @stake_api ||= Coinbase::Client::StakeApi.new(Coinbase.configuration.api_client)
     end
 
-    def from_model(model)
-      @model = model
-      @status = model.status
-      @transactions = model.transactions.map do |transaction_model|
-        Transaction.new(transaction_model)
+    def wallet_stake_api
+      @wallet_stake_api ||= Coinbase::Client::WalletStakeApi.new(Coinbase.configuration.api_client)
+    end
+
+    def update_transactions(transactions)
+      # Only overwrite the transactions if the response is populated.
+      return unless transactions && !transactions.empty?
+
+      # Create a set of existing unsigned payloads to avoid duplicates.
+      existing_unsigned_payloads = Set.new
+      @transactions.each do |transaction|
+        existing_unsigned_payloads.add(transaction.unsigned_payload)
       end
 
-      self
+      # Add transactions that are not already in the transactions array.
+      transactions.each do |transaction_model|
+        unless existing_unsigned_payloads.include?(transaction_model.unsigned_payload)
+          @transactions << Transaction.new(transaction_model)
+        end
+      end
     end
   end
 end

data/lib/coinbase/staking_reward.rb

@@ -62,7 +62,7 @@ module Coinbase
     # Returns the USD conversion price of the StakingReward.
     # @return [BigDecimal] The USD conversion price
     def usd_conversion_price
-      BigDecimal(@model.usd_value.conversion_price.to_i) / BigDecimal(100)
+      BigDecimal(@model.usd_value.conversion_price.to_i)
     end
 
     # Returns the USD conversion time of the StakingReward.

data/lib/coinbase/trade.rb

@@ -15,12 +15,13 @@ module Coinbase
       # @param from_asset_id [Symbol] The Asset ID of the Asset to trade from
       # @param to_asset_id [Symbol] The Asset ID of the Asset to trade to
       # @param amount [BigDecimal] The amount of the Asset to send
-      # @param network_id [Symbol] The Network ID of the Asset
+      # @param network [Coinbase::Network, Symbol] The Network or Network ID of the Asset
       # @param wallet_id [String] The Wallet ID of the sending Wallet
       # @return [Send] The new pending Send object
-      def create(address_id:, from_asset_id:, to_asset_id:, amount:, network_id:, wallet_id:)
-        from_asset = Asset.fetch(network_id, from_asset_id)
-        to_asset = Asset.fetch(network_id, to_asset_id)
+      def create(address_id:, from_asset_id:, to_asset_id:, amount:, network:, wallet_id:)
+        network = Coinbase::Network.from_id(network)
+        from_asset = network.get_asset(from_asset_id)
+        to_asset = network.get_asset(to_asset_id)
 
         model = Coinbase.call_api do
           trades_api.create_trade(
@@ -75,10 +76,10 @@ module Coinbase
       @model.trade_id
     end
 
-    # Returns the Network ID of the Trade.
-    # @return [Symbol] The Network ID
-    def network_id
-      Coinbase.to_sym(@model.network_id)
+    # Returns the Network of the Trade.
+    # @return [Coinbase::Network] The Network the Trade is on
+    def network
+      @network ||= Coinbase::Network.from_id(@model.network_id)
     end
 
     # Returns the Wallet ID of the Trade.
@@ -198,7 +199,7 @@ module Coinbase
     # Returns a String representation of the Trade.
     # @return [String] a String representation of the Trade
     def to_s
-      "Coinbase::Trade{transfer_id: '#{id}', network_id: '#{network_id}', " \
+      "Coinbase::Trade{transfer_id: '#{id}', network_id: '#{network.id}', " \
         "address_id: '#{address_id}', from_asset_id: '#{from_asset_id}', " \
         "to_asset_id: '#{to_asset_id}', from_amount: '#{from_amount}', " \
         "to_amount: '#{to_amount}' status: '#{status}'}"

data/lib/coinbase/transaction.rb

@@ -68,6 +68,12 @@ module Coinbase
       @model.from_address_id
     end
 
+    # Returns the to address for the Transaction.
+    # @return [String] The to address
+    def to_address_id
+      @model.to_address_id
+    end
+
     # Returns whether the Transaction is in a terminal state.
     # @return [Boolean] Whether the Transaction is in a terminal state
     def terminal_state?

data/lib/coinbase/transfer.rb

@@ -20,12 +20,13 @@ module Coinbase
       #   If the destination is a Wallet, it uses the default Address of the Wallet.
       #   If the destination is an Address, it uses the Address's ID.
       #   If the destination is a String, it uses it as the Address ID.
-      # @param network_id [Symbol] The Network ID of the Asset
+      # @param network [Coinbase::Network, Symbol] The Network or Network ID of the Asset
       # @param wallet_id [String] The Wallet ID of the sending Wallet
       # @return [Transfer] The new pending Transfer object
       # @raise [Coinbase::ApiError] If the Transfer fails
-      def create(address_id:, amount:, asset_id:, destination:, network_id:, wallet_id:, gasless: false)
-        asset = Asset.fetch(network_id, asset_id)
+      def create(address_id:, amount:, asset_id:, destination:, network:, wallet_id:, gasless: false)
+        network = Coinbase::Network.from_id(network)
+        asset = network.get_asset(asset_id)
 
         model = Coinbase.call_api do
           transfers_api.create_transfer(
@@ -34,8 +35,8 @@ module Coinbase
             {
               amount: asset.to_atomic_amount(amount).to_i.to_s,
               asset_id: asset.primary_denomination.to_s,
-              destination: Coinbase::Destination.new(destination, network_id: network_id).address_id,
-              network_id: Coinbase.normalize_network(network_id),
+              destination: Coinbase::Destination.new(destination, network: network).address_id,
+              network_id: network.normalized_id,
               gasless: gasless
             }
           )
@@ -82,10 +83,10 @@ module Coinbase
       @model.transfer_id
     end
 
-    # Returns the Network ID of the Transfer.
-    # @return [Symbol] The Network ID
-    def network_id
-      Coinbase.to_sym(@model.network_id)
+    # Returns the Network of the Transfer.
+    # @return [Symbol] The Network
+    def network
+      @network ||= Coinbase::Network.from_id(@model.network_id)
     end
 
     # Returns the Wallet ID of the Transfer.
@@ -229,7 +230,7 @@ module Coinbase
     # Returns a String representation of the Transfer.
     # @return [String] a String representation of the Transfer
     def to_s
-      "Coinbase::Transfer{transfer_id: '#{id}', network_id: '#{network_id}', " \
+      "Coinbase::Transfer{transfer_id: '#{id}', network_id: '#{network.id}', " \
         "from_address_id: '#{from_address_id}', destination_address_id: '#{destination_address_id}', " \
         "asset_id: '#{asset_id}', amount: '#{amount}', transaction_link: '#{transaction_link}', " \
         "status: '#{status}'}"

data/lib/coinbase/validator.rb

@@ -10,27 +10,31 @@ module Coinbase
     end
 
     # Returns a list of Validators for the provided network and asset.
-    # @param network_id [Symbol] The network ID
+    # @param network [Coinbase::Nework, Symbol] The Network or Network ID
     # @param asset_id [Symbol] The asset ID
     # @param status [Symbol] The status of the validator. Defaults to nil.
     # @return [Enumerable<Coinbase::Validator>] The validators
-    def self.list(network_id, asset_id, status: nil)
-      Coinbase::Pagination.enumerate(
-        ->(page) { list_page(network_id, asset_id, status, page) }
-      ) do |validator|
+    def self.list(network, asset_id, status: nil)
+      network = Coinbase::Network.from_id(network)
+
+      Coinbase::Pagination.enumerate(lambda { |page|
+        list_page(network, asset_id, status, page)
+      }) do |validator|
         new(validator)
       end
     end
 
     # Returns a Validator for the provided network, asset, and validator.
-    # @param network_id [Symbol] The network ID
+    # @param network [Coinbase::Network, Symbol] The Network or Network ID
     # @param asset_id [Symbol] The asset ID
     # @param validator_id [String] The validator ID
     # @return [Coinbase::Validator] The validator
-    def self.fetch(network_id, asset_id, validator_id)
+    def self.fetch(network, asset_id, validator_id)
+      network = Coinbase::Network.from_id(network)
+
       validator = Coinbase.call_api do
         validators_api.get_validator(
-          network_id,
+          network.normalized_id,
           asset_id,
           validator_id
         )
@@ -62,10 +66,10 @@ module Coinbase
       to_s
     end
 
-    def self.list_page(network_id, asset_id, status, page)
+    def self.list_page(network, asset_id, status, page)
       Coinbase.call_api do
         validators_api.list_validators(
-          network_id,
+          network.normalized_id,
           asset_id,
           {
             status: status,
@@ -75,8 +79,12 @@ module Coinbase
       end
     end
 
+    private_class_method :list_page
+
     def self.validators_api
       Coinbase::Client::ValidatorsApi.new(Coinbase.configuration.api_client)
     end
+
+    private_class_method :validators_api
   end
 end

data/lib/coinbase/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Coinbase
+  VERSION = '0.1.0'
+end