feat(dashboard): 接入汇率引擎，实现多资产交叉汇率折算与 CNY 统一计价看板

src/actions/portfolio.ts

@@ -1,11 +1,80 @@
 'use server';
 
 import { db } from '@/db';
-import { transactions, assets } from '@/db/schema';
+import { transactions, assets, exchangeRates } from '@/db/schema';
 import Big from 'big.js';
 import { desc, eq } from 'drizzle-orm';
 
-export async function getPortfolioPositions() {
+interface Position {
+  assetId: string;
+  symbol: string;
+  type: string;
+  quantity: string;
+  baseCurrency: string;
+  cnyValue: string;
+}
+
+interface RawRate {
+  fromCurrency: string;
+  toCurrency: string;
+  rate: string;
+}
+
+function buildRateMap(rates: RawRate[]): Map<string, string> {
+  const map = new Map<string, string>();
+  for (const r of rates) {
+    map.set(`${r.fromCurrency}_${r.toCurrency}`, r.rate);
+  }
+  return map;
+}
+
+function getRate(
+  rateMap: Map<string, string>,
+  from: string,
+  to: string
+): string | null {
+  const direct = rateMap.get(`${from}_${to}`);
+  if (direct) return direct;
+  return null;
+}
+
+function calculateCnyValue(
+  quantity: Big,
+  baseCurrency: string,
+  rateMap: Map<string, string>,
+  cryptoPrices: Map<string, string>
+): Big {
+  if (baseCurrency === 'CNY') {
+    return quantity;
+  }
+
+  const directRate = getRate(rateMap, baseCurrency, 'CNY');
+  if (directRate) {
+    return quantity.times(directRate);
+  }
+
+  const usdToCny = getRate(rateMap, 'USD', 'CNY');
+  if (!usdToCny) {
+    return new Big('0');
+  }
+
+  const priceKey = `${baseCurrency}_USD`;
+  const cryptoPrice = cryptoPrices.get(priceKey);
+  if (cryptoPrice) {
+    const usdValue = quantity.times(cryptoPrice);
+    return usdValue.times(usdToCny);
+  }
+
+  const usdRate = getRate(rateMap, baseCurrency, 'USD');
+  if (usdRate) {
+    const usdValue = quantity.times(usdRate);
+    return usdValue.times(usdToCny);
+  }
+
+  return new Big('0');
+}
+
+export async function getPortfolioPositions(): Promise<Position[]> {
   const allTransactions = await db
     .select({
       txType: transactions.txType,
@@ -14,6 +83,7 @@ export async function getPortfolioPositions() {
       assetSymbol: assets.symbol,
       assetType: assets.type,
       assetBaseCurrency: assets.baseCurrency,
+      assetPrice: transactions.price,
     })
     .from(transactions)
     .leftJoin(assets, eq(assets.id, transactions.assetId))
@@ -25,6 +95,7 @@ export async function getPortfolioPositions() {
     type: string;
     quantity: Big;
     baseCurrency: string;
+    latestPrice: string;
   }>();
 
   for (const tx of allTransactions) {
@@ -38,6 +109,7 @@ export async function getPortfolioPositions() {
         type: tx.assetType || 'CASH',
         quantity: new Big('0'),
         baseCurrency: tx.assetBaseCurrency || '',
+        latestPrice: tx.assetPrice || '0',
       });
     }
 
@@ -50,19 +122,107 @@ export async function getPortfolioPositions() {
     } else if (tx.txType === 'DIVIDEND') {
       holding.quantity = holding.quantity.plus(tx.quantity);
     }
+
+    if (tx.assetPrice) {
+      holding.latestPrice = tx.assetPrice;
+    }
   }
 
-  const result = [];
+  const rates = await db.select({
+    fromCurrency: exchangeRates.fromCurrency,
+    toCurrency: exchangeRates.toCurrency,
+    rate: exchangeRates.rate,
+  }).from(exchangeRates);
+
+  const rateMap = buildRateMap(rates);
+
+  const cryptoSymbols = new Set(['BTC', 'ETH', 'SOL', 'BNB', 'XRP', 'ADA', 'DOGE', 'AVAX', 'MATIC', 'DOT']);
+  const cryptoPrices = new Map<string, string>();
+  for (const [_, holding] of holdings) {
+    if (holding.type === 'CRYPTO' && cryptoSymbols.has(holding.symbol.toUpperCase())) {
+      const priceKey = `${holding.symbol}_USD`;
+      const usdRate = getRate(rateMap, holding.symbol, 'USD');
+      if (usdRate) {
+        cryptoPrices.set(priceKey, usdRate);
+      }
+    }
+  }
+
+  const result: Position[] = [];
+  let totalCnyValue = new Big('0');
+
   for (const [_, holding] of holdings) {
     if (holding.quantity.lte(0)) continue;
+
+    let cnyValue: Big;
+
+    if (holding.type === 'CRYPTO') {
+      const symbol = holding.symbol.toUpperCase();
+      const btcToUsd = getRate(rateMap, symbol, 'USD');
+      const usdToCny = getRate(rateMap, 'USD', 'CNY');
+
+      if (btcToUsd && usdToCny) {
+        const usdValue = holding.quantity.times(holding.latestPrice || '1');
+        cnyValue = usdValue.times(usdToCny);
+      } else {
+        cnyValue = new Big('0');
+      }
+    } else if (holding.baseCurrency === 'CNY') {
+      cnyValue = holding.quantity.times(holding.latestPrice || '1');
+    } else {
+      const directRate = getRate(rateMap, holding.baseCurrency, 'CNY');
+      if (directRate) {
+        cnyValue = holding.quantity.times(holding.latestPrice || '1').times(directRate);
+      } else {
+        const usdRate = getRate(rateMap, holding.baseCurrency, 'USD');
+        const usdToCny = getRate(rateMap, 'USD', 'CNY');
+        if (usdRate && usdToCny) {
+          cnyValue = holding.quantity.times(holding.latestPrice || '1').times(usdRate).times(usdToCny);
+        } else {
+          cnyValue = new Big('0');
+        }
+      }
+    }
+
+    totalCnyValue = totalCnyValue.plus(cnyValue);
+
     result.push({
       assetId: holding.assetId,
       symbol: holding.symbol,
       type: holding.type,
       quantity: holding.quantity.toString(),
       baseCurrency: holding.baseCurrency,
+      cnyValue: cnyValue.toString(),
     });
   }
 
   return result;
 }
+
+export async function getPortfolioSummary() {
+  const positions = await getPortfolioPositions();
+
+  const totalCnyValue = positions.reduce(
+    (sum, pos) => sum.plus(new Big(pos.cnyValue)),
+    new Big('0')
+  );
+
+  const chartData = positions.map((pos, index) => ({
+    name: pos.symbol,
+    value: new Big(pos.cnyValue),
+    fill: [
+      '#3b82f6',
+      '#8b5cf6',
+      '#10b981',
+      '#f59e0b',
+      '#ef4444',
+      '#06b6d4',
+    ][index % 6],
+  }));
+
+  return {
+    positions,
+    totalCnyValue: totalCnyValue.toString(),
+    chartData,
+  };
+}