ouroboros-peatio-bitgo 3.1.0

data/lib/peatio/bitgo/client.rb

@@ -0,0 +1,54 @@
+require 'faraday'
+require 'better-faraday'
+
+module Peatio
+  module Bitgo
+    class Client
+      Error = Class.new(StandardError)
+      ConnectionError = Class.new(Error)
+
+      class ResponseError < Error
+        def initialize(msg)
+          super "#{msg}"
+        end
+      end
+
+      def initialize(endpoint, access_token)
+        @endpoint = URI.parse(endpoint)
+        @access_token = access_token
+      end
+
+      def rest_api(verb, path, data = nil)
+        args = [@endpoint.to_s + path]
+
+        if data
+          if %i[post put patch].include?(verb)
+            args << data.compact.to_json
+            args << { 'Content-Type' => 'application/json' }
+          else
+            args << data.compact
+            args << {}
+          end
+        else
+          args << nil
+          args << {}
+        end
+
+        args.last['Accept']        = 'application/json'
+        args.last['Authorization'] = 'Bearer ' + @access_token
+
+        response = Faraday.send(verb, *args)
+        response.assert_2xx!
+        response = JSON.parse(response.body)
+        response['error'].tap { |error| raise ResponseError.new(error) if error }
+        response
+      rescue Faraday::Error => e
+        if e.is_a?(Faraday::ConnectionFailed) || e.is_a?(Faraday::TimeoutError)
+          raise ConnectionError, e
+        else
+          raise ConnectionError, JSON.parse(e.response.body)['message']
+        end
+      end
+    end
+  end
+end

data/lib/peatio/bitgo/hooks.rb

@@ -0,0 +1,42 @@
+module Peatio
+  module Bitgo
+    module Hooks
+      BLOCKCHAIN_VERSION_REQUIREMENT = "~> 1.0.0"
+      WALLET_VERSION_REQUIREMENT = "~> 1.0.0"
+
+      class << self
+        def check_compatibility
+          unless Gem::Requirement.new(BLOCKCHAIN_VERSION_REQUIREMENT)
+                                 .satisfied_by?(Gem::Version.new(Peatio::Blockchain::VERSION))
+            [
+              "Bitgo blockchain version requirement was not satisfied by Peatio::Blockchain.",
+              "Bitgo blockchain requires #{BLOCKCHAIN_VERSION_REQUIREMENT}.",
+              "Peatio::Blockchain version is #{Peatio::Blockchain::VERSION}"
+            ].join('\n').tap { |s| Kernel.abort s }
+          end
+
+          unless Gem::Requirement.new(WALLET_VERSION_REQUIREMENT)
+                                 .satisfied_by?(Gem::Version.new(Peatio::Wallet::VERSION))
+            [
+              "Bitgo wallet version requirement was not satisfied by Peatio::Wallet.",
+              "Bitgo wallet requires #{WALLET_VERSION_REQUIREMENT}.",
+              "Peatio::Wallet version is #{Peatio::Wallet::VERSION}"
+            ].join('\n').tap { |s| Kernel.abort s }
+          end
+        end
+
+        def register
+          Peatio::Blockchain.registry[:bitgo] = Bitgo::Blockchain
+          Peatio::Wallet.registry[:bitgo] = Bitgo::Wallet
+        end
+      end
+
+      if defined?(Rails::Railtie)
+        require "peatio/bitgo/railtie"
+      else
+        check_compatibility
+        register
+      end
+    end
+  end
+end

data/lib/peatio/bitgo/railtie.rb

@@ -0,0 +1,13 @@
+module Peatio
+  module Bitgo
+    class Railtie < Rails::Railtie
+      config.before_initialize do
+        Hooks.check_compatibility
+      end
+
+      config.after_initialize do
+        Hooks.register
+      end
+    end
+  end
+end

data/lib/peatio/bitgo/version.rb

@@ -0,0 +1,5 @@
+module Peatio
+  module Bitgo
+    VERSION = "3.1.0".freeze
+  end
+end

data/lib/peatio/bitgo/wallet.rb

@@ -0,0 +1,345 @@
+module Peatio
+  module Bitgo
+    class Wallet < Peatio::Wallet::Abstract
+      TIME_DIFFERENCE_IN_MINUTES = 10
+      XLM_MEMO_TYPES = { 'memoId': 'id', 'memoText': 'text', 'memoHash': 'hash', 'memoReturn': 'return' }
+
+      DEFAULT_FEATURES = { skip_deposit_collection: false }.freeze
+
+      def initialize(custom_features = {})
+        @features = DEFAULT_FEATURES.merge(custom_features).slice(*SUPPORTED_FEATURES)
+        @settings = {}
+      end
+
+      def configure(settings = {})
+        # Clean client state during configure.
+        @client = nil
+
+        @settings.merge!(settings.slice(*SUPPORTED_SETTINGS))
+
+        @wallet = @settings.fetch(:wallet) do
+          raise Peatio::Wallet::MissingSettingError, :wallet
+        end.slice(:uri, :address, :secret, :access_token, :wallet_id, :testnet)
+
+         @currency = @settings.fetch(:currency) do
+          raise Peatio::Wallet::MissingSettingError, :currency
+         end.slice(:id, :base_factor, :code, :options)
+      end
+
+      def create_address!(options = {})
+        currency = erc20_currency_id
+        options.deep_symbolize_keys!
+
+        if options.dig(:pa_details, :address_id).present? &&
+           options.dig(:pa_details, :updated_at).present? &&
+           time_difference_in_minutes(options.dig(:pa_details, :updated_at)) >= TIME_DIFFERENCE_IN_MINUTES
+
+          response = client.rest_api(:get, "#{currency}/wallet/#{wallet_id}/address/#{options.dig(:pa_details, :address_id)}")
+          { address: response['address'], secret: bitgo_wallet_passphrase }
+        elsif options.dig(:pa_details, :address_id).blank?
+          response = client.rest_api(:post, "#{currency}/wallet/#{wallet_id}/address")
+          { address: response['address'], secret: bitgo_wallet_passphrase, details: { address_id: response['id'] }}
+        end
+      rescue Bitgo::Client::Error => e
+        raise Peatio::Wallet::ClientError, e
+      end
+
+      def create_transaction!(transaction, options = {})
+        currency_options = @currency.fetch(:options).slice(:gas_limit, :gas_price, :erc20_contract_address)
+
+        if currency_options[:gas_limit].present? && currency_options[:gas_price].present?
+          options.merge!(currency_options)
+          create_eth_transaction(transaction, options)
+        else
+          amount = convert_to_base_unit(transaction.amount)
+
+          if options[:subtract_fee].to_s == 'true'
+            fee = build_raw_transaction(transaction)
+            baseFeeInfo = fee.dig('feeInfo','fee')
+            fee = baseFeeInfo.present? ? baseFeeInfo : fee.dig('txInfo','Fee')
+            amount -= fee.to_i
+          end
+
+          response = client.rest_api(:post, "#{currency_id}/wallet/#{wallet_id}/sendcoins", {
+                                 address: normalize_address(transaction.to_address.to_s),
+                                 amount: amount.to_s,
+                                 walletPassphrase: bitgo_wallet_passphrase,
+                                 memo: xlm_memo(transaction.to_address.to_s)
+          }.compact)
+
+          if response['feeString'].present?
+            fee = convert_from_base_unit(response['feeString'])
+            transaction.fee = fee
+          end
+
+          transaction.hash = normalize_txid(response['txid'])
+          transaction.fee_currency_id = erc20_currency_id
+          transaction
+        end
+      rescue Bitgo::Client::Error => e
+        raise Peatio::Wallet::ClientError, e
+      end
+
+
+      def build_raw_transaction(transaction)
+        client.rest_api(:post, "#{currency_id}/wallet/#{wallet_id}/tx/build", {
+          recipients: [{
+            address: transaction.to_address,
+            amount: convert_to_base_unit(transaction.amount).to_s
+          }]
+        }.compact)
+      end
+
+      def create_eth_transaction(transaction, options = {})
+        amount = convert_to_base_unit(transaction.amount)
+        hop = true unless options.slice(:gas_price).present?
+
+        fee_estimate = fee_estimate(amount.to_s, hop)
+
+        if transaction.options.present? && transaction.options[:gas_price].present?
+          options[:gas_price] = transaction.options[:gas_price]
+        else
+          options[:gas_price] = fee_estimate['minGasPrice'].to_i
+        end
+
+        response = client.rest_api(:post, "#{currency_id}/wallet/#{wallet_id}/sendcoins", {
+          address: transaction.to_address.to_s,
+          amount: amount.to_s,
+          walletPassphrase: bitgo_wallet_passphrase,
+          gas: options.fetch(:gas_limit).to_i,
+          gasPrice: options.fetch(:gas_price).to_i,
+          hop: hop
+        }.compact)
+
+        if response['feeString'].present?
+          fee = convert_from_base_unit(response['feeString'])
+          transaction.fee = fee
+        end
+
+        transaction.hash = normalize_txid(response['txid'])
+        transaction.fee_currency_id = erc20_currency_id
+        transaction.options = options
+        transaction
+      end
+
+      def fee_estimate(amount, hop)
+        client.rest_api(:get, "#{erc20_currency_id}/tx/fee", { amount: amount, hop: hop }.compact)
+      end
+
+      def load_balance!
+        if @currency.fetch(:options).slice(:erc20_contract_address).present?
+          load_erc20_balance!
+        else
+          response = client.rest_api(:get, "#{currency_id}/wallet/#{wallet_id}")
+          convert_from_base_unit(response.fetch('balanceString'))
+        end
+      rescue Bitgo::Client::Error => e
+        raise Peatio::Wallet::ClientError, e
+      end
+
+      def load_erc20_balance!
+        response = client.rest_api(:get, "#{erc20_currency_id}/wallet/#{wallet_id}?allTokens=true")
+        convert_from_base_unit(response.dig('tokens', currency_id, 'balanceString'))
+      rescue Bitgo::Client::Error => e
+        raise Peatio::Wallet::ClientError, e
+      end
+
+      def trigger_webhook_event(request)
+        # Retrieve testnet status
+        testnet_status = @wallet.fetch(:testnet)
+        # Determine if the 't' prefix should be applied
+        currency = (testnet_status == false || testnet_status.nil? || (testnet_status.is_a?(String) && testnet_status.downcase == "false")) ? @currency.fetch(:id) : 't' + @currency.fetch(:id)
+        if request.params['type'] == 'transfer'
+          return unless currency == request.params['coin'] &&
+                        @wallet.fetch(:wallet_id) == request.params['wallet']
+        else
+          return unless @wallet.fetch(:wallet_id) == request.params['walletId']
+        end
+
+        if request.params['type'] == 'transfer'
+          transactions = fetch_transfer!(request.params['transfer'])
+          return transactions
+        elsif request.params['type'] == 'address_confirmation'
+          address_id = fetch_address_id(request.params['address'])
+          return { address_id: address_id, address: request.params['address'], currency_id: currency_id }
+        end
+      end
+
+      def register_webhooks!(url)
+        transfer_webhook(url)
+        address_confirmation_webhook(url)
+      end
+
+      def fetch_address_id(address)
+        currency = erc20_currency_id
+        client.rest_api(:get, "#{currency}/wallet/#{wallet_id}/address/#{address}")
+              .fetch('id')
+      rescue Bitgo::Client::Error => e
+        raise Peatio::Wallet::ClientError, e
+      end
+
+      def fetch_transfer!(id)
+        response = client.rest_api(:get, "#{currency_id}/wallet/#{wallet_id}/transfer/#{id}")
+        parse_entries(response['entries']).map do |entry|
+          to_address =  if response.dig('coinSpecific', 'memo').present?
+                          memo = response.dig('coinSpecific', 'memo')
+                          memo_type = memo.kind_of?(Array) ? memo.first  : memo
+                          build_address(entry['address'], memo_type)
+                        else
+                          entry['address']
+                        end
+          state = define_transaction_state(response['state'])
+
+          if response['outputs'].present?
+            output = response['outputs'].find { |out| out['address'] == to_address }
+            txout = output['index'] if output.present?
+          end
+
+          if response['feeString'].present?
+            fee = convert_from_base_unit(response['feeString']) / response['entries'].count
+          end
+
+          transaction = Peatio::Transaction.new(
+            currency_id: @currency.fetch(:id),
+            amount: convert_from_base_unit(entry['valueString']),
+            fee: fee,
+            fee_currency_id: erc20_currency_id,
+            hash: normalize_txid(response['txid']),
+            to_address: to_address,
+            block_number: response['height'],
+            txout: txout.to_i,
+            status: state
+          )
+
+          transaction if transaction.valid?
+        end.compact
+      rescue Bitgo::Client::Error => e
+        raise Peatio::Wallet::ClientError, e
+      end
+
+      def transfer_webhook(url)
+        client.rest_api(:post, "#{currency_id}/wallet/#{wallet_id}/webhooks", {
+          type: 'transfer',
+          allToken: true,
+          url: url,
+          label: "webhook for #{url}",
+          listenToFailureStates: false
+        })
+      end
+
+      def address_confirmation_webhook(url)
+        client.rest_api(:post, "#{currency_id}/wallet/#{wallet_id}/webhooks", {
+          type: 'address_confirmation',
+          allToken: true,
+          url: url,
+          label: "webhook for #{url}",
+          listenToFailureStates: false
+        })
+      end
+
+      def parse_entries(entries)
+        entries.map do |e|
+          e if e["valueString"].to_i.positive?
+        end.compact
+      end
+
+      private
+
+      def client
+        uri = @wallet.fetch(:uri) { raise Peatio::Wallet::MissingSettingError, :uri }
+        access_token = @wallet.fetch(:access_token) { raise Peatio::Wallet::MissingSettingError, :access_token }
+
+        currency_code_prefix = @wallet.fetch(:testnet) && @wallet.fetch(:testnet).to_s.downcase != "false" ? 't' : ''
+        uri = uri.gsub(/\/+\z/, '') + '/' + currency_code_prefix
+        @client ||= Client.new(uri, access_token)
+      end
+
+      def build_address(address, memo)
+        "#{address}?memoId=#{memo['value']}"
+      end
+
+      # All these functions will have to be done with the coin set to eth or teth
+      # since that is the actual coin type being used.
+      def erc20_currency_id
+        return 'eth' if @currency.fetch(:options).slice(:erc20_contract_address).present?
+
+        currency_id
+      end
+
+      def xlm_memo(address)
+        build_xlm_memo(address) if @currency.fetch(:id) == 'xlm'
+      end
+
+      def build_xlm_memo(address)
+        case address.split('?').last.split('=').first
+        when 'memoId'
+          memo_value_from(address, 'memoId')
+        when 'memoText'
+          memo_value_from(address, 'memoText')
+        when 'memoHash'
+          memo_value_from(address, 'memoHash')
+        when 'memoReturn'
+          memo_value_from(address, 'memoReturn')
+        end
+      end
+
+      def memo_value_from(address, type)
+        memo_value = address.partition(type + '=').last
+        return { type: XLM_MEMO_TYPES[type.to_sym], value: memo_value } if memo_value.present?
+      end
+
+      def currency_id
+        @currency.fetch(:id) { raise Peatio::Wallet::MissingSettingError, :id }
+      end
+
+      def bitgo_wallet_passphrase
+        @wallet.fetch(:secret)
+      end
+
+      def wallet_id
+        @wallet.fetch(:wallet_id)
+      end
+
+      def normalize_address(address)
+        if @currency.fetch(:id) == 'xlm'
+          address.split('?').first
+        else
+          address
+        end
+      end
+
+      def normalize_txid(txid)
+        txid.downcase
+      end
+
+      def convert_from_base_unit(value)
+        value.to_d / @currency.fetch(:base_factor)
+      end
+
+      def convert_to_base_unit(value)
+        x = value.to_d * @currency.fetch(:base_factor)
+        unless (x % 1).zero?
+          raise Peatio::WalletClient::Error,
+                "Failed to convert value to base (smallest) unit because it exceeds the maximum precision: " \
+                "#{value.to_d} - #{x.to_d} must be equal to zero."
+        end
+        x.to_i
+      end
+
+      def time_difference_in_minutes(updated_at)
+        (Time.now - updated_at)/60
+      end
+
+      def define_transaction_state(state)
+        case state
+        when 'unconfirmed'
+          'pending'
+        when 'confirmed'
+          'success'
+        when 'failed','rejected'
+          'failed'
+        end
+      end
+    end
+  end
+end

data/lib/peatio/bitgo.rb

@@ -0,0 +1,16 @@
+require "peatio"
+
+module Peatio
+  module Bitgo
+    require "bigdecimal"
+    require "bigdecimal/util"
+
+    require "peatio/bitgo/blockchain"
+    require "peatio/bitgo/client"
+    require "peatio/bitgo/wallet"
+
+    require "peatio/bitgo/hooks"
+
+    require "peatio/bitgo/version"
+  end
+end

data/peatio-bitgo.gemspec

@@ -0,0 +1,39 @@
+
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "peatio/bitgo/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "ouroboros-peatio-bitgo"
+  spec.version       = Peatio::Bitgo::VERSION
+  spec.authors       = ["ISPOS"]
+  spec.email         = ["contacto@ispos.mx"]
+
+  spec.summary       = %q{Gem for extending Peatio plugable system with Bitgo implementation.}
+  spec.description   = %q{Bitgo Peatio gem which implements Peatio::Blockchain::Abstract & Peatio::Wallet::Abstract.}
+  spec.homepage      = "https://ispos.mx/"
+  spec.license       = "MIT"
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files         = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "activesupport", "~> 6.1.0"
+  spec.add_dependency "better-faraday", "~> 2.0.1"
+  spec.add_dependency "faraday", "~> 1.10"
+  spec.add_dependency "memoist", "~> 0.16.0"
+  spec.add_dependency "peatio", ">= 3.1.1"
+  spec.add_dependency 'net-http-persistent', '~> 4.0.1'
+
+  spec.add_development_dependency "bundler", "~> 2.4.7"
+  spec.add_development_dependency "mocha", "~> 1.8"
+  spec.add_development_dependency "pry-byebug"
+  spec.add_development_dependency "rake", "~> 13.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency "webmock", "~> 3.5"
+end