fix(pricing): replace currentConnectionInfo with jitContext.nodePrices in pricingContext and enhance logging in priceFeedApi

Author: Ansonhkg

packages/lit-client/src/lib/LitClient/createLitClient.ts

@@ -219,7 +219,7 @@ export const _createNagaLitClient = async (
       pricingContext: {
         product: 'SIGN',
         userMaxPrice: params.userMaxPrice,
-        nodePrices: currentConnectionInfo.priceFeedInfo.networkPrices,
+        nodePrices: jitContext.nodePrices,
         threshold: currentHandshakeResult.threshold,
       },
       authContext: params.authContext,
@@ -383,7 +383,7 @@ export const _createNagaLitClient = async (
       pricingContext: {
         product: 'LIT_ACTION',
         userMaxPrice: params.userMaxPrice,
-        nodePrices: currentConnectionInfo.priceFeedInfo.networkPrices,
+        nodePrices: jitContext.nodePrices,
         threshold: currentHandshakeResult.threshold,
       },
       authContext: params.authContext,
@@ -650,7 +650,7 @@ export const _createNagaLitClient = async (
       pricingContext: {
         product: 'DECRYPTION',
         userMaxPrice: params.userMaxPrice,
-        nodePrices: currentConnectionInfo.priceFeedInfo.networkPrices,
+        nodePrices: jitContext.nodePrices,
         threshold: currentHandshakeResult.threshold,
       },
       authContext: params.authContext,

packages/networks/src/networks/vNaga/LitChainClient/apis/highLevelApis/priceFeed/priceFeedApi.ts

@@ -23,13 +23,18 @@
  * ```
  */
 
+const _logger = getChildLogger({
+  module: 'priceFeedApi',
+});
+
 import { NodePrices } from '@lit-protocol/types';
 import { ExpectedAccountOrWalletClient } from '../../../../LitChainClient/contract-manager/createContractsManager';
 import { INetworkConfig } from '../../../../interfaces/NetworkContext';
 import {
   getNodesForRequest,
   PRODUCT_IDS,
 } from '../../../apis/rawContractApis/pricing/getNodesForRequest';
+import { getChildLogger } from '@lit-protocol/logger';
 
 // Configuration constants
 const STALE_PRICES_SECONDS = 3 * 1000; // Update prices if > X seconds old
@@ -152,6 +157,7 @@ export async function getPriceFeedInfo(
 ): Promise<PriceFeedInfo> {
   // If there's a local promise, an update is in progress; wait for that
   if (fetchingPriceFeedInfo) {
+    _logger.info('💲 Local promise is already fetching price feed info. Returning that instead.');
     return fetchingPriceFeedInfo;
   }
 
@@ -160,9 +166,13 @@ export async function getPriceFeedInfo(
     priceFeedInfo &&
     Date.now() - lastUpdatedTimestamp < STALE_PRICES_SECONDS
   ) {
+    _logger.info(
+      `💲 Returning stale price feed info. Remaining stale time: ${STALE_PRICES_SECONDS - (Date.now() - lastUpdatedTimestamp)
+      }ms`
+    );
     return priceFeedInfo;
   }
-
+  _logger.info('💲 Fetching new price feed info');
   // Fetch new prices, update local cache values, and return them
   return fetchPriceFeedInfoWithLocalPromise(params, accountOrWalletClient);
 }

packages/networks/src/networks/vNaga/envs/naga-dev/naga-dev.module.ts

@@ -47,7 +47,7 @@ import type { PKPStorageProvider } from '../../../../storage/types';
 import { createRequestId } from '../../../shared/helpers/createRequestId';
 import { handleAuthServerRequest } from '../../../shared/helpers/handleAuthServerRequest';
 import { composeLitUrl } from '../../endpoints-manager/composeLitUrl';
-import { PKPPermissionsManager } from '../../LitChainClient/apis/highLevelApis';
+import { getNodePrices, PKPPermissionsManager } from '../../LitChainClient/apis/highLevelApis';
 import { PaymentManager } from '../../LitChainClient/apis/highLevelApis/PaymentManager/PaymentManager';
 import { MintWithMultiAuthsRequest } from '../../LitChainClient/apis/highLevelApis/mintPKP/mintWithMultiAuths';
 import { PkpIdentifierRaw } from '../../LitChainClient/apis/rawContractApis/permissions/utils/resolvePkpTokenId';
@@ -80,6 +80,7 @@ import {
 } from './chain-manager/createChainManager';
 import { getMaxPricesForNodeProduct } from './pricing-manager/getMaxPricesForNodeProduct';
 import { getUserMaxPrice } from './pricing-manager/getUserMaxPrice';
+import { privateKeyToAccount } from 'viem/accounts';
 
 const MODULE_NAME = 'naga-dev';
 
@@ -484,6 +485,7 @@ const networkModuleObject = {
     ): Promise<NagaJitContext> => {
       const keySet: KeySet = {};
 
+      // 1. Generate a key set for the JIT context
       for (const url of connectionInfo.bootstrapUrls) {
         keySet[url] = {
           publicKey: hexToBytes(
@@ -495,7 +497,18 @@ const networkModuleObject = {
         };
       }
 
-      return { keySet };
+      // Use read-only account for viewing PKPs
+      const account = privateKeyToAccount(
+        '0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80'
+      );
+
+      // 2. Fetch the price feed info
+      const nodePrices = await getNodePrices({
+        realmId: 1,
+        networkCtx: networkConfig,
+      }, account);
+
+      return { keySet, nodePrices };
     },
     handshake: {
       schemas: {

packages/networks/src/networks/vNaga/envs/naga-test/naga-test.module.ts

@@ -508,7 +508,7 @@ const networkModuleObject = {
         networkCtx: networkConfig,
       }, account);
 
-      return { keySet, nodePrices: [] };
+      return { keySet, nodePrices };
     },
     handshake: {
       schemas: {