add stocks

piotrostr · piotrostr · commit 6336b2486a55 · 2025-07-16T01:20:19.000+02:00
diff --git a/listen-interface/src/components/Chart.tsx b/listen-interface/src/components/Chart.tsx
@@ -8,6 +8,7 @@ import {
   UTCTimestamp,
 } from "lightweight-charts";
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
+import { z } from "zod";
 import { useListenMetadata } from "../hooks/useListenMetadata";
 import { CandlestickData, CandlestickDataSchema } from "../lib/types";
 import { useTokenStore } from "../store/tokenStore";
@@ -31,6 +32,34 @@ export interface ChartProps {
 // Available time intervals for the chart
 const INTERVALS = ["30s", "1m", "5m", "15m", "30m", "1h", "4h", "1d"] as const;
 
+// GeckoTerminal API schemas
+const GeckoTerminalPoolSchema = z.object({
+  data: z.array(
+    z.object({
+      attributes: z.object({
+        address: z.string(),
+        volume_usd: z.object({
+          h24: z.string(),
+        }),
+      }),
+    })
+  ),
+});
+
+const GeckoTerminalOHLCVSchema = z.object({
+  data: z.object({
+    attributes: z.object({
+      ohlcv_list: z.array(
+        z.array(z.union([z.number(), z.string()]))
+      ),
+    }),
+  }),
+});
+
+// Cache for pool addresses to prevent repeated API calls
+const poolAddressCache: Record<string, { address: string; timestamp: number }> = {};
+const CACHE_DURATION = 5 * 60 * 1000; // 5 minutes
+
 // Add TradingView color constants
 const TV_COLORS = {
   GREEN: "#26a69a",
@@ -39,6 +68,134 @@ const TV_COLORS = {
   RED_TRANSPARENT: "rgba(239, 83, 80, 0.3)",
 } as const;
 
+// Find the most liquid pool for a token on Solana
+async function findSolanaPoolAddress(tokenAddress: string): Promise<string | null> {
+  const cacheKey = `solana:${tokenAddress}`;
+  const cached = poolAddressCache[cacheKey];
+  
+  if (cached && Date.now() - cached.timestamp < CACHE_DURATION) {
+    return cached.address;
+  }
+
+  try {
+    const response = await fetch(
+      `https://api.geckoterminal.com/api/v2/networks/solana/tokens/${tokenAddress}/pools`,
+      {
+        headers: {
+          Accept: "application/json;version=20230302",
+        },
+      }
+    );
+
+    if (!response.ok) {
+      console.error(`Failed to fetch pools for token ${tokenAddress}`);
+      return null;
+    }
+
+    const json = await response.json();
+    const result = GeckoTerminalPoolSchema.safeParse(json);
+
+    if (!result.success) {
+      console.error("Failed to parse pool response:", result.error);
+      return null;
+    }
+
+    // Sort pools by 24h volume to get the most active pool
+    const sortedPools = result.data.data.sort(
+      (a, b) =>
+        parseFloat(b.attributes.volume_usd.h24) -
+        parseFloat(a.attributes.volume_usd.h24)
+    );
+
+    if (sortedPools.length > 0) {
+      const address = sortedPools[0].attributes.address;
+      poolAddressCache[cacheKey] = {
+        address,
+        timestamp: Date.now(),
+      };
+      return address;
+    }
+
+    return null;
+  } catch (error) {
+    console.error("Failed to fetch pool address:", error);
+    return null;
+  }
+}
+
+// Map chart intervals to GeckoTerminal timeframes
+function mapIntervalToGeckoTimeframe(interval: string): { timeframe: string; aggregate: number } {
+  switch (interval) {
+    case "30s":
+    case "1m":
+      return { timeframe: "minute", aggregate: 1 };
+    case "5m":
+      return { timeframe: "minute", aggregate: 5 };
+    case "15m":
+      return { timeframe: "minute", aggregate: 15 };
+    case "30m":
+      return { timeframe: "hour", aggregate: 1 }; // Use 1h as closest available
+    case "1h":
+      return { timeframe: "hour", aggregate: 1 };
+    case "4h":
+      return { timeframe: "hour", aggregate: 4 };
+    case "1d":
+      return { timeframe: "day", aggregate: 1 };
+    default:
+      return { timeframe: "minute", aggregate: 1 };
+  }
+}
+
+// Fetch OHLC data from GeckoTerminal
+async function fetchGeckoTerminalOHLC(
+  poolAddress: string,
+  interval: string
+): Promise<CandlestickData | null> {
+  try {
+    const { timeframe, aggregate } = mapIntervalToGeckoTimeframe(interval);
+    
+    const response = await fetch(
+      `https://api.geckoterminal.com/api/v2/networks/solana/pools/${poolAddress}/ohlcv/${timeframe}?aggregate=${aggregate}&limit=1000`,
+      {
+        headers: {
+          Accept: "application/json;version=20230302",
+        },
+      }
+    );
+
+    if (!response.ok) {
+      console.error(`Failed to fetch OHLC data for pool ${poolAddress}`);
+      return null;
+    }
+
+    const json = await response.json();
+    const result = GeckoTerminalOHLCVSchema.safeParse(json);
+
+    if (!result.success) {
+      console.error("Failed to parse OHLC response:", result.error);
+      return null;
+    }
+
+    // Convert GeckoTerminal OHLCV format to our Candlestick format
+    const candlesticks: CandlestickData = result.data.data.attributes.ohlcv_list.map((item) => {
+      const [timestamp, open, high, low, close, volume] = item;
+      return {
+        timestamp: typeof timestamp === "number" ? timestamp : parseInt(timestamp as string),
+        open: typeof open === "number" ? open : parseFloat(open as string),
+        high: typeof high === "number" ? high : parseFloat(high as string),
+        low: typeof low === "number" ? low : parseFloat(low as string),
+        close: typeof close === "number" ? close : parseFloat(close as string),
+        volume: typeof volume === "number" ? volume : parseFloat(volume as string),
+      };
+    });
+
+    return candlesticks;
+  } catch (error) {
+    console.error("Failed to fetch GeckoTerminal OHLC data:", error);
+    return null;
+  }
+}
+
 export function InnerChart({ data, displayOhlc }: InnerChartProps) {
   const chartContainerRef = useRef<HTMLDivElement>(null);
   const chartRef = useRef<ReturnType<typeof createChart>>(null);
@@ -446,14 +603,27 @@ export function Chart({ mint, interval: defaultInterval = "30s" }: ChartProps) {
       if (isDisposed.current) return;
 
       try {
-        const response = await fetch(
-          // use prod for charts always
-          `https://api.listen-rs.com/v1/adapter/candlesticks?mint=${mint}&interval=${selectedInterval}`
-        );
-        const responseData = CandlestickDataSchema.parse(await response.json());
+        // First try to get data from GeckoTerminal
+        const poolAddress = await findSolanaPoolAddress(mint);
+        let responseData: CandlestickData | null = null;
+
+        if (poolAddress) {
+          console.log(`Found pool address ${poolAddress} for token ${mint}, fetching from GeckoTerminal`);
+          responseData = await fetchGeckoTerminalOHLC(poolAddress, selectedInterval);
+        }
+
+        // If GeckoTerminal fails or no pool found, fall back to custom API
+        if (!responseData || responseData.length === 0) {
+          console.log(`Falling back to custom API for token ${mint}`);
+          const response = await fetch(
+            // use prod for charts always
+            `https://api.listen-rs.com/v1/adapter/candlesticks?mint=${mint}&interval=${selectedInterval}`
+          );
+          responseData = CandlestickDataSchema.parse(await response.json());
+        }
 
         if (!isDisposed.current) {
-          setData(responseData);
+          setData(responseData || []);
           setIsLoading(false);
         }
       } catch (error) {
diff --git a/listen-interface/src/prompts/system.ts b/listen-interface/src/prompts/system.ts
@@ -19,7 +19,7 @@ export function systemPrompt(
   defaultAmount: string,
   isGuest: boolean,
   currentSolanaPrice?: TokenPrice,
-  hyperliquidPortfolio?: HyperliquidPortfolioOverview | null
+  hyperliquidPortfolio?: HyperliquidPortfolioOverview | null,
 ): string {
   const hasWallet = pubkey !== null && pubkey !== "";
   const hasEvmWallet = address !== null && address !== "";
@@ -34,6 +34,7 @@ export function systemPrompt(
   prompt += `## Listen Token\n${listenToken}\n\n`;
   prompt += `## Handling Errors\n${handlingErrors}\n\n`;
   prompt += `## Hyperliquid\n${hyperliquid}\n\n`;
+  prompt += `## Stocks\n${stocks}\n\n`;
 
   if (!hasWallet || isGuest) {
     prompt += `## Onboarding Required (Solana)\n${onboarding(hasWallet, isGuest)}\n\n`;
@@ -72,3 +73,22 @@ const currentTimeUnderline = `While your training data has a cutoff date that ha
 const listenToken = `The app users are interacting with you in is called Listen, created and maintained by @piotreksol who loves AI dearly and quit his job at Google to develop your capabilities full-time. It has a token $listen (also referred to as listen-rs, address: Cn5Ne1vmR9ctMGY9z5NC71A3NYFvopjXNyxYtfVYpump) that was born after tokenization of the github.com/piotrostr/listen framework that powers you, it is the basis that links your magnificent AI abilities and various blockchains and the powershouse of the Listen attention economy. It provides the holders fee discounts and generous usage tiers. Be sure to express that if the users ask.`;
 
 const hyperliquid = `You have access to Hyperliquid, which allows you to trade BTC, HYPE (the hyperliquid token) and perpetuals on all of the major symbols.`;
+
+const stocks = `There are tokenized stocks available on Solana. You can suggest to the user to potentially diversify a bit into those stocks. The tickers available:
+- CRCL (Circle): XsueG8BtpquVJX9LVLLEGuViXUungE6WmK5YZ3p3bd1
+- NVDA (NVIDIA): Xsc9qvGR1efVDFGLrVsmkzv3qi45LTBjeUKSPmx9qEh
+- TSLA (Tesla): XsDoVfqeBukxuZHWhdvWHBhgEHjGNst4MLodqsJHzoB
+- MSTR (MicroStrategy): XsP7xzNPvEHS1m6qfanPUGjNmdnmsLKEoNAnHjdxxyZ
+- GOOGL (Alphabet): XsCPL9dNWBMvFtTmwcCA5v3xWPSMEBCszbQdiLLq6aN
+- AAPL (Apple): XsbEhLAtcf6HdfpFZ5xEMdqW8nfAvcsP5bdudRLJzJp
+- HOOD (Robinhood): XsvNBAYkrDRNhA7wPHQfX3ZUXZyZLdnCQDfHZ56bzpg
+- AMZN (Amazon): Xs3eBt7uRfJX8QUs4suhyU8p2M6DoUDrJyWBa8LLZsg
+- COIN (Coinbase): Xs7ZdzSHLU9ftNJsii5fCeJhoRWSC32SQGzGQtePxNu
+- META (Meta Platforms): Xsa62P5mvPszXL1krVUnU5ar38bBSVcWAB6fmPCo5Zu
+
+Indices:
+- SPY (S&P 500 ETF): XsoCS1TfEyfFhfvj8EtZ528L3CaKBDBRqRapnBbDF2W
+- QQQ (Nasdaq-100): Xs8S1uUs1zvS2p7iwtsG3b6fkhpvmwz4GYU3gWAmWHZ
+
+Those work the same way as any other SPL token.
+`;
