exchange-tron 2.6.1

data/lib/peatio/tron/client.rb

@@ -0,0 +1,62 @@
+require 'memoist'
+require 'faraday'
+require 'better-faraday'
+
+module Peatio
+  module Tron
+    class Client
+      Error = Class.new(StandardError)
+
+      class ConnectionError < Error; end
+
+      class ResponseError < Error
+        def initialize(code, msg)
+          @code = code
+          @msg = msg
+        end
+
+        def message
+          "#{@msg} (#{@code})"
+        end
+      end
+
+      extend Memoist
+
+      def initialize(endpoint)
+        @json_rpc_endpoint = URI.parse(endpoint)
+      end
+
+      def json_rpc(path:, params: {})
+        response = connection.post do |req|
+          req.url path
+          req.body = params.to_json
+        end
+        response.assert_2xx!
+        response = JSON.parse(response.body)
+        response['Error'].tap { |error| raise ResponseError.new(200, error) if error }
+        response
+      rescue => e
+        if e.is_a?(Error)
+          raise e
+        elsif e.is_a?(Faraday::Error)
+          raise ConnectionError, e
+        else
+          raise Error, e
+        end
+      end
+
+      private
+
+      def connection
+        @connection ||= Faraday.new(@json_rpc_endpoint) do |f|
+          f.adapter :net_http_persistent, pool_size: 5, idle_timeout: @idle_timeout
+        end.tap do |connection|
+          unless @json_rpc_endpoint.user.blank?
+            connection.basic_auth(@json_rpc_endpoint.user, @json_rpc_endpoint.password)
+          end
+        end
+      end
+    end
+  end
+end
+

data/lib/peatio/tron/concerns/encryption.rb

@@ -0,0 +1,57 @@
+require 'digest/sha3'
+module Encryption
+  extend ActiveSupport::Concern
+  
+  ALGO = '123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz'
+  
+  def reformat_decode_address(address)
+    base58_to_binary(address)
+  end
+
+  def reformat_encode_address(hexstring)
+    sha256 = sha256_encode(sha256_encode(hexstring))
+    (hexstring + sha256[0, 8]).hex.digits(58).reverse.map {|i| ALGO[i]}.join
+  end
+
+  def reformat_txid(txid)
+    txid
+  end
+
+  def encode_hex(str)
+    str.unpack('H*')[0]
+  end
+
+  def decode_hex(str)
+    [str].pack('H*')
+  end
+
+  def abi_encode(*args)
+    args.each_with_object('') do |arg, data|
+      data.concat(arg.gsub(/\A0x/, '').rjust(64, '0'))
+    end
+  end
+  
+  private
+
+  def sha256_encode(hexstring)
+    Digest::SHA256.hexdigest([hexstring].pack('H*'))
+  end
+
+  def base58_to_int(base58_val)
+    int_val = 0
+    base58_val.reverse.split(//).each_with_index do |char, index|
+      char_index = ALGO.index(char)
+      int_val += (char_index) * (ALGO.length ** (index))
+    end
+    int_val
+  end
+
+  def base58_to_binary(base58_val)
+    int_encode_hex(base58_to_int(base58_val))[0, 42]
+  end
+
+  def int_encode_hex int
+    hex = int.to_s(16)
+    (hex.length % 2 == 0) ? hex : ('0' + hex)
+  end
+end

data/lib/peatio/tron/hooks.rb

@@ -0,0 +1,42 @@
+module Peatio
+  module Tron
+    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))
+            [
+              "Tron blockchain version requirement was not satisfied by Peatio::Blockchain.",
+              "Tron 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))
+            [
+              "Tron wallet version requirement was not satisfied by Peatio::Wallet.",
+              "Tron 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[:tron] = Tron::Blockchain
+          Peatio::Wallet.registry[:tron] = Tron::Wallet
+        end
+      end
+
+      if defined?(Rails::Railtie)
+        require "peatio/tron/railtie"
+      else
+        check_compatibility
+        register
+      end
+    end
+  end
+end

data/lib/peatio/tron/railtie.rb

@@ -0,0 +1,13 @@
+module Peatio
+  module Tron
+    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/tron/version.rb

@@ -0,0 +1,5 @@
+module Peatio
+  module Tron
+    VERSION = "2.6.1"
+  end
+end

data/lib/peatio/tron/wallet.rb

@@ -0,0 +1,198 @@
+module Peatio
+  module Tron
+    class Wallet < Peatio::Wallet::Abstract
+      include Encryption
+
+      DEFAULT_FEE = { fee_limit: 1000000 }
+      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)
+
+        @currency = @settings.fetch(:currency) do
+          raise Peatio::Wallet::MissingSettingError, :currency
+        end.slice(:id, :base_factor, :options)
+      end
+
+      def create_address!(options = {})
+        client.json_rpc(path: 'wallet/generateaddress')
+              .yield_self { |r| { address: r.fetch('address'), secret: r.fetch('privateKey') } }
+      rescue Tron::Client::Error => e
+        raise Peatio::Wallet::ClientError, e
+      end
+
+      def create_transaction!(transaction, options = {})
+        if @currency.dig(:options, :trc10_token_id).present?
+          create_trc10_transaction!(transaction)
+        elsif @currency.dig(:options, :trc20_contract_address).present?
+          create_trc20_transaction!(transaction, options)
+        else
+          create_coin_transaction!(transaction, options)
+        end
+      rescue Tron::Client::Error => e
+        raise Peatio::Wallet::ClientError, e
+      end
+
+      def prepare_deposit_collection!(transaction, deposit_spread, deposit_currency)
+        # Don't prepare for deposit_collection in case of coin(tron) deposit.
+        return [] if is_coin?(deposit_currency)
+        return [] if deposit_spread.blank?
+
+        options = DEFAULT_FEE.merge(deposit_currency.fetch(:options).slice(:fee_limit))
+
+        # We collect fees depending on the number of spread deposit size
+        # Example: if deposit spreads on three wallets need to collect tron fee for 3 transactions
+        fees = convert_from_base_unit(options.fetch(:fee_limit).to_i)
+        transaction.amount = fees * deposit_spread.size
+
+        [create_coin_transaction!(transaction)]
+      rescue Tron::Client::Error => e
+        raise Peatio::Wallet::ClientError, e
+      end
+
+      def load_balance!
+        if @currency.dig(:options, :trc10_token_id).present?
+          client.json_rpc(path: 'wallet/getaccount',
+                          params: { address: reformat_decode_address(@wallet.fetch(:address)) }
+          ).fetch('assetV2', [])
+                .find { |a| a['key'] == @currency[:options][:trc10_token_id] }
+                .try(:fetch, 'value', 0)
+        elsif @currency.dig(:options, :trc20_contract_address).present?
+          client.json_rpc(path: 'wallet/triggersmartcontract',
+                          params: {
+                            owner_address: reformat_decode_address(@wallet.fetch(:address)),
+                            contract_address: reformat_decode_address(@currency.dig(:options, :trc20_contract_address)),
+                            function_selector: 'balanceOf(address)',
+                            parameter: abi_encode(reformat_decode_address(@wallet.fetch(:address))[2..42]) }
+          ).fetch('constant_result')[0].hex
+        else
+          client.json_rpc(path: 'wallet/getaccount',
+                          params: { address: reformat_decode_address(@wallet.fetch(:address)) }
+          ).fetch('balance', nil)
+        end.yield_self { |amount| convert_from_base_unit(amount.to_i) }
+      rescue Tron::Client::Error => e
+        raise Peatio::Wallet::ClientError, e
+      end
+
+      private
+
+      def create_trc10_transaction!(transaction, options = {})
+        currency_options = @currency.fetch(:options).slice(:trc10_token_id)
+        options.merge!(currency_options)
+
+        amount = convert_to_base_unit(transaction.amount)
+
+        txid = client.json_rpc(path: 'wallet/easytransferassetbyprivate',
+                               params: {
+                                 privateKey: @wallet.fetch(:secret),
+                                 toAddress: reformat_decode_address(transaction.to_address),
+                                 assetId: currency_options.fetch(:trc10_token_id),
+                                 amount: amount
+                               }).dig('transaction', 'txID')
+                     .yield_self { |txid| reformat_txid(txid) }
+
+        unless txid
+          raise Peatio::Wallet::ClientError, \
+              "Withdrawal from #{@wallet.fetch(:address)} to #{transaction.to_address} failed."
+        end
+        transaction.hash = reformat_txid(txid)
+        transaction
+      end
+
+      def create_trc20_transaction!(transaction, options = {})
+        currency_options = @currency.fetch(:options).slice(:trc20_contract_address, :fee_limit)
+        options.merge!(DEFAULT_FEE, currency_options)
+
+        amount = convert_to_base_unit(transaction.amount)
+
+        signed_txn = sign_transaction(transaction, amount, options)
+
+        # broadcast txn
+        response = client.json_rpc(path: 'wallet/broadcasttransaction',
+                                   params: signed_txn)
+
+        txid = response.fetch('result', false) ? signed_txn.fetch('txID') : nil
+
+        unless txid
+          raise Peatio::Wallet::ClientError, \
+              "Withdrawal from #{@wallet.fetch(:address)} to #{transaction.to_address} failed."
+        end
+        transaction.hash = reformat_txid(txid)
+        transaction
+      end
+
+      def create_coin_transaction!(transaction, options = {})
+        amount = convert_to_base_unit(transaction.amount)
+        txid = client.json_rpc(path: 'wallet/easytransferbyprivate',
+                               params: {
+                                 privateKey: @wallet.fetch(:secret),
+                                 toAddress: reformat_decode_address(transaction.to_address),
+                                 amount: amount
+                               }).dig('transaction', 'txID')
+                     .yield_self { |txid| reformat_txid(txid) }
+
+        unless txid
+          raise Peatio::Wallet::ClientError, \
+              "Withdrawal from #{@wallet.fetch(:address)} to #{transaction.to_address} failed."
+        end
+        transaction.amount = convert_from_base_unit(amount)
+        transaction.hash = reformat_txid(txid)
+        transaction
+      end
+
+      def sign_transaction(transaction, amount, options)
+        client.json_rpc(path: 'wallet/gettransactionsign',
+                        params: {
+                          transaction: trigger_smart_contract(transaction, amount, options),
+                          privateKey: @wallet.fetch(:secret)
+                        })
+      end
+
+      def trigger_smart_contract(transaction, amount, options)
+        client.json_rpc(path: 'wallet/triggersmartcontract',
+                        params: {
+                          contract_address: reformat_decode_address(options.fetch(:trc20_contract_address)),
+                          function_selector: 'transfer(address,uint256)',
+                          parameter: abi_encode(reformat_decode_address(transaction.to_address)[2..42], amount.to_s(16)),
+                          fee_limit: options.fetch(:fee_limit),
+                          owner_address: reformat_decode_address(@wallet.fetch(:address))
+                        }).fetch('transaction')
+      end
+
+      def is_coin?(deposit_currency)
+        deposit_currency.dig(:options, :trc20_contract_address).blank?
+      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::Wallet::ClientError,
+                "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 client
+        uri = @wallet.fetch(:uri) { raise Peatio::Wallet::MissingSettingError, :uri }
+        @client ||= Client.new(uri)
+      end
+    end
+  end
+end

data/lib/peatio/tron.rb

@@ -0,0 +1,19 @@
+require "active_support/core_ext/object/blank"
+require "active_support/core_ext/enumerable"
+require "active_support/core_ext/string/inquiry"
+require "active_support/core_ext/object/try"
+require "peatio"
+
+module Peatio
+  module Tron
+    require "bigdecimal"
+    require "bigdecimal/util"
+    require "digest/sha3"
+    require "peatio/tron/concerns/encryption"
+    require "peatio/tron/blockchain"
+    require "peatio/tron/client"
+    require "peatio/tron/wallet"
+    require "peatio/tron/hooks"
+    require "peatio/tron/version"
+  end
+end

metadata

@@ -0,0 +1,276 @@
+--- !ruby/object:Gem::Specification
+name: exchange-tron
+version: !ruby/object:Gem::Version
+  version: 2.6.1
+platform: ruby
+authors:
+- ABC
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2021-12-11 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.2.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 5.2.3
+- !ruby/object:Gem::Dependency
+  name: better-faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.5
+- !ruby/object:Gem::Dependency
+  name: faraday
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.17'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.17'
+- !ruby/object:Gem::Dependency
+  name: memoist
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.16.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.16.0
+- !ruby/object:Gem::Dependency
+  name: digest-sha3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+- !ruby/object:Gem::Dependency
+  name: ed25519
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: ffi
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: peatio
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.6.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.6.5
+- !ruby/object:Gem::Dependency
+  name: net-http-persistent
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.0.1
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.3.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.3.0
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.8'
+- !ruby/object:Gem::Dependency
+  name: pry-byebug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.5'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.5'
+description: Tron gem which implements Peatio::Blockchain::Abstract & Peatio::Wallet::Abstract.
+email:
+- ''
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitattributes"
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- config/blockchains.yml
+- config/currencies.yml
+- config/wallets.yml
+- exchange-tron.gemspec
+- lib/peatio/tron.rb
+- lib/peatio/tron/blockchain.rb
+- lib/peatio/tron/client.rb
+- lib/peatio/tron/concerns/encryption.rb
+- lib/peatio/tron/hooks.rb
+- lib/peatio/tron/railtie.rb
+- lib/peatio/tron/version.rb
+- lib/peatio/tron/wallet.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.9
+signing_key: 
+specification_version: 4
+summary: Gem for extending Peatio plugable system with Tron implementation.
+test_files: []