tron.rb 1.0.0 → 1.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cb06efba2988905e0c66fa13f78960c58b91b08646a1822ad1878182ad359f90
-  data.tar.gz: f3f3b0b7b88280f7fcbc5733b57d752abef9a8623ea7a05a360e91c790131f33
+  metadata.gz: 68e2df783fb208cc34d8e404c6c93b7fd27aae0066b2f6efec8650c739b5c623
+  data.tar.gz: a13fae572eaebef060a24485db75d70d4d8a71a5952e62f18624b98926225258
 SHA512:
-  metadata.gz: 6fcef166d0c652dcb4deaf3040eb71f9647009e315d53486adbfd505bc322848d9516f6d60d72f3623bc4fc11b11fe6d58798544df7b566f9782828add54d501
-  data.tar.gz: deb26afaafb9c7dc395dabcb17ac31a4ff01c1458f61d6e3db394659d1c1ad2daf6e2a7a82ed4f2e59d6efb61776d178e1e38254aa2e5b978da6b77064f400fe
+  metadata.gz: f74287a0bd02186a0b8b2b7c4faf62a25fe4ba5e78168da5388855e21d9b3c36e6b5eaeda8d7ea2a787837729971fb033d3174d81352417f3839d043ccc5b275
+  data.tar.gz: 21779f5695517650257c97719bb8903e4de98d4a0c69157816d2cf71a25420385cdfd9d3c8225f6c037a00cce41fe201f4ecbe75dd1231705d4424fe05f0cd79

data/README.md

@@ -90,6 +90,20 @@ wallet_info = client.get_wallet_balance('TWd4WrZ9wn84f5x1hZhL4DHvk738ns5jwb')
 full_info = client.get_full_account_info('TWd4WrZ9wn84f5x1hZhL4DHvk738ns5jwb')
 ```
 
+#### Get Wallet Portfolio with USD Values:
+```ruby
+# Get wallet portfolio with balances, USD prices, and total value
+portfolio = client.get_wallet_portfolio('TWd4WrZ9wn84f5x1hZhL4DHvk738ns5jwb')
+
+# Include tokens with zero balance
+portfolio = client.get_wallet_portfolio('TWd4WrZ9wn84f5x1hZhL4DHvk738ns5jwb', include_zero_balances: true)
+
+puts "Total Portfolio Value: $#{portfolio[:total_value_usd]}"
+portfolio[:tokens].each do |token|
+  puts "#{token[:symbol]}: #{token[:token_balance]} ($#{token[:usd_value]})"
+end
+```
+
 ## Features
 
 - ✓ TRX balance

data/lib/tron/client.rb

@@ -65,6 +65,61 @@ module Tron
       }
     end
 
+    def get_wallet_portfolio(address, include_zero_balances: false)
+      validate_address!(address)
+
+      # Step 1: Get all balances (TRX + TRC20)
+      wallet_data = get_wallet_balance(address)
+
+      tokens = []
+
+      # Step 2: Process TRX (native token)
+      trx_balance = wallet_data[:trx_balance].to_f
+      if trx_balance > 0 || include_zero_balances
+        trx_price = price_service.get_token_price_usd('trx')
+        tokens << {
+          symbol: 'TRX',
+          name: 'Tronix',
+          token_balance: trx_balance,
+          decimals: 6,
+          address: nil,
+          price_usd: trx_price,
+          usd_value: trx_price ? (trx_balance * trx_price).round(2) : nil
+        }
+      end
+
+      # Step 3: Process TRC20 tokens
+      wallet_data[:trc20_tokens].each do |token|
+        next if token[:balance] <= 0 && !include_zero_balances
+
+        # Get USD price for this token
+        price_usd = price_service.get_token_price_usd(token[:symbol].downcase)
+        usd_value = price_usd ? (token[:balance] * price_usd).round(2) : nil
+
+        tokens << {
+          symbol: token[:symbol],
+          name: token[:name],
+          token_balance: token[:balance],
+          decimals: token[:decimals],
+          address: token[:address],
+          price_usd: price_usd,
+          usd_value: usd_value
+        }
+      end
+
+      # Step 4: Calculate total portfolio value
+      total_value_usd = tokens.sum { |t| t[:usd_value] || 0 }
+
+      # Step 5: Sort by value (highest first)
+      tokens.sort_by! { |t| -(t[:usd_value] || 0) }
+
+      {
+        address: address,
+        total_value_usd: total_value_usd.round(2),
+        tokens: tokens
+      }
+    end
+
     private
 
     def validate_address!(address)

data/lib/tron/version.rb

@@ -2,5 +2,5 @@
 
 # lib/tron/version.rb
 module Tron
-  VERSION = "1.0.0".freeze
+  VERSION = "1.0.2".freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tron.rb
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.2
 platform: ruby
 authors:
 - Your Name