Change settings

src/lib/prices.ts

@@ -1,8 +1,8 @@
 // src/lib/prices.ts
 type PriceMap = Record<string, number>
 
-// Map your asset code -> provider + symbol
-const MAP: Record<string, { p: 'binance'|'stooq'; s: string }> = {
+// Map asset code -> provider + symbol
+const MAP: Record<string, { p: 'binance'|'yahoo'; s: string }> = {
   // crypto (USDT pairs)
   BTC: { p: 'binance', s: 'BTCUSDT' },
   ETH: { p: 'binance', s: 'ETHUSDT' },
@@ -11,15 +11,14 @@ const MAP: Record<string, { p: 'binance'|'stooq'; s: string }> = {
   DOGE:{ p: 'binance', s: 'DOGEUSDT' },
   XRP: { p: 'binance', s: 'XRPUSDT' },
 
-  // equities (stooq wants .us)
-  AAPL:{ p: 'stooq',   s: 'aapl.us' },
-  MSFT:{ p: 'stooq',   s: 'msft.us' },
-  BMRN:{ p: 'stooq',   s: 'bmrn.us' },
-  MRNA:{ p: 'stooq',   s: 'mrna.us' },
-  TSLA:{ p: 'stooq',   s: 'tsla.us' },
+  // equities (Yahoo uses plain tickers)
+  AAPL:{ p: 'yahoo',   s: 'AAPL' },
+  MSFT:{ p: 'yahoo',   s: 'MSFT' },
+  BMRN:{ p: 'yahoo',   s: 'BMRN' },
+  MRNA:{ p: 'yahoo',   s: 'MRNA' },
+  TSLA:{ p: 'yahoo',   s: 'TSLA' },
 };
 
-// --- simple cache + pub/sub
 let latest: PriceMap = {};
 const subs = new Set<(m: PriceMap)=>void>();
 const push = () => subs.forEach(fn => fn({ ...latest }));
@@ -31,30 +30,53 @@ export function subscribePrices(fn: (m: PriceMap)=>void) {
 
 async function fetchBinance(symbol: string) {
   const r = await fetch(`https://api.binance.com/api/v3/ticker/price?symbol=${encodeURIComponent(symbol)}`);
+  if (!r.ok) throw new Error(`binance ${symbol} ${r.status}`);
   const j = await r.json(); const n = Number(j?.price);
   return Number.isFinite(n) ? n : undefined;
 }
 
-async function fetchStooq(symbol: string) {
-  const url = `https://stooq.com/q/l/?s=${encodeURIComponent(symbol)}&f=sd2t2ohlcv&h&e=csv`;
+async function fetchYahoo(symbolsCsv: string) {
+  const url = `https://query1.finance.yahoo.com/v7/finance/quote?symbols=${encodeURIComponent(symbolsCsv)}`
   const r = await fetch(url);
-  const txt = await r.text(); // first line is header; second line has data, column 'Close'
-  const line = txt.split('\n')[1] || '';
-  const cols = line.split(','); // Symbol,Date,Time,Open,High,Low,Close,Volume
-  const n = Number(cols[6]);
-  return Number.isFinite(n) ? n : undefined;
+  if (!r.ok) throw new Error(`yahoo ${symbolsCsv} ${r.status}`);
+  const j = await r.json();
+  const out: Record<string, number> = {};
+  for (const it of (j?.quoteResponse?.result ?? [])) {
+    const sym = it?.symbol;
+    const px = Number(it?.regularMarketPrice ?? it?.postMarketPrice ?? it?.preMarketPrice);
+    if (sym && Number.isFinite(px)) out[sym] = px;
+  }
+  return out;
 }
 
 export async function pollPrices(codes: string[], intervalMs = 30000) {
   const uniq = [...new Set(codes)].filter(c => MAP[c]);
+
+  // split by provider for efficient requests
+  const binanceSyms = uniq.filter(c => MAP[c].p === 'binance').map(c => MAP[c].s);
+  const yahooSyms   = uniq.filter(c => MAP[c].p === 'yahoo').map(c => MAP[c].s);
+
   async function tick() {
-    await Promise.all(uniq.map(async code => {
-      const { p, s } = MAP[code];
-      const px = p === 'binance' ? await fetchBinance(s) : await fetchStooq(s);
-      if (px !== undefined) latest[code] = px;
-    }));
+    try {
+      // Binance: per-symbol (cheap for a handful)
+      await Promise.all(binanceSyms.map(async s => {
+        const px = await fetchBinance(s);
+        const code = Object.entries(MAP).find(([k,v]) => v.s === s)?.[0];
+        if (code && px !== undefined) latest[code] = px;
+      }));
+
+      // Yahoo: batch in one call
+      if (yahooSyms.length) {
+        const batch = await fetchYahoo(yahooSyms.join(','));
+        for (const [sym, px] of Object.entries(batch)) {
+          const code = Object.entries(MAP).find(([k,v]) => v.s === sym)?.[0];
+          if (code && px !== undefined) latest[code] = px;
+        }
+      }
+    } catch { /* ignore transient errors */ }
     push();
   }
+
   await tick();
   setInterval(tick, intervalMs);
 }