@shapeshiftoss/unchained-client 10.14.10 → 10.14.11

package/dist/cjs/tron/api.js

@@ -4,15 +4,23 @@ exports.TronApi = void 0;
 const tronweb_1 = require("tronweb");
 class TronApi {
     rpcUrl;
+    apiKey;
     tronWeb = null;
     requestQueue = Promise.resolve();
     minRequestInterval = 1_500;
     constructor(config) {
         this.rpcUrl = config.rpcUrl;
+        this.apiKey = config.apiKey ?? '';
     }
     getRpcUrl() {
         return this.rpcUrl;
     }
+    get tronGridHeaders() {
+        return {
+            'Content-Type': 'application/json',
+            ...(this.apiKey ? { 'TRON-PRO-API-KEY': this.apiKey } : {}),
+        };
+    }
     async throttle() {
         // Queue the request and wait for all previous requests to complete
         const currentRequest = this.requestQueue.then(async () => {
@@ -23,7 +31,10 @@ class TronApi {
     }
     getTronWeb() {
         if (!this.tronWeb) {
-            this.tronWeb = new tronweb_1.TronWeb({ fullHost: this.rpcUrl });
+            this.tronWeb = new tronweb_1.TronWeb({
+                fullHost: this.rpcUrl,
+                headers: this.apiKey ? { 'TRON-PRO-API-KEY': this.apiKey } : {},
+            });
         }
         return this.tronWeb;
     }
@@ -31,7 +42,7 @@ class TronApi {
         await this.throttle();
         const response = await fetch(`${this.rpcUrl}/wallet/getaccount`, {
             method: 'POST',
-            headers: { 'Content-Type': 'application/json' },
+            headers: this.tronGridHeaders,
             body: JSON.stringify({ address: params.pubkey, visible: true }),
         });
         const data = await response.json();
@@ -48,7 +59,9 @@ class TronApi {
         // Get TRC20 tokens from TronGrid API
         try {
             await this.throttle();
-            const trc20Response = await fetch(`${this.rpcUrl}/v1/accounts/${params.pubkey}`);
+            const trc20Response = await fetch(`${this.rpcUrl}/v1/accounts/${params.pubkey}`, {
+                headers: this.tronGridHeaders,
+            });
             const trc20Data = await trc20Response.json();
             if (trc20Data.data?.[0]?.trc20 && Array.isArray(trc20Data.data[0].trc20)) {
                 // trc20 is an array of objects like [{contractAddress: balance}, ...]
@@ -61,6 +74,42 @@ class TronApi {
                     }
                 });
             }
+            else if (!trc20Data.data || trc20Data.data.length === 0) {
+                // Non-activated account: /v1/accounts returns data:[] for addresses that have never sent TRX.
+                // Fall back to discovering TRC20 balances via received transactions + hardcoded top tokens.
+                try {
+                    await this.throttle();
+                    const HARDCODED_TOP_TOKENS = [
+                        'TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t',
+                        'TEkxiTehnzSmSe2XqrBj4w32RUN966rdz8', // USDC
+                    ];
+                    const MAX_FALLBACK_CONTRACTS = 20;
+                    const discoveredContracts = new Set(HARDCODED_TOP_TOKENS);
+                    const trc20TxResponse = await fetch(`${this.rpcUrl}/v1/accounts/${params.pubkey}/transactions/trc20?limit=200&only_to=true`, { headers: this.tronGridHeaders });
+                    if (trc20TxResponse.ok) {
+                        const trc20TxData = await trc20TxResponse.json();
+                        if (trc20TxData.data && Array.isArray(trc20TxData.data)) {
+                            trc20TxData.data.forEach((tx) => {
+                                const contractAddress = tx.token_info?.address;
+                                if (contractAddress)
+                                    discoveredContracts.add(contractAddress);
+                            });
+                        }
+                    }
+                    for (const contractAddress of Array.from(discoveredContracts).slice(0, MAX_FALLBACK_CONTRACTS)) {
+                        await this.throttle();
+                        const balance = await this.getTRC20Balance({ address: params.pubkey, contractAddress });
+                        if (balance !== '0')
+                            tokens.push({ contractAddress, balance });
+                    }
+                }
+                catch (fallbackErr) {
+                    console.error('Failed TRC20 fallback discovery for non-activated TRON account', {
+                        address: `${params.pubkey.slice(0, 6)}...${params.pubkey.slice(-4)}`,
+                        error: fallbackErr,
+                    });
+                }
+            }
         }
         catch (err) {
             // TRC20 fetch failed, continue with just TRC10 tokens
@@ -91,12 +140,12 @@ class TronApi {
         const [txResponse, infoResponse] = await Promise.all([
             fetch(`${this.rpcUrl}/wallet/gettransactionbyid`, {
                 method: 'POST',
-                headers: { 'Content-Type': 'application/json' },
+                headers: this.tronGridHeaders,
                 body: JSON.stringify({ value: params.txid, visible: true }),
             }),
             fetch(`${this.rpcUrl}/wallet/gettransactioninfobyid`, {
                 method: 'POST',
-                headers: { 'Content-Type': 'application/json' },
+                headers: this.tronGridHeaders,
                 body: JSON.stringify({ value: params.txid, visible: true }),
             }),
         ]);
@@ -137,7 +186,7 @@ class TronApi {
         await this.throttle();
         const response = await fetch(`${this.rpcUrl}/wallet/getblockbynum`, {
             method: 'POST',
-            headers: { 'Content-Type': 'application/json' },
+            headers: this.tronGridHeaders,
             body: JSON.stringify({ num: params.height }),
         });
         if (!response.ok) {