fix handling x cashu for messages endpoint

routstr/balance.py

@@ -5,7 +5,6 @@ from time import monotonic
 from typing import Annotated, NoReturn
 
 from fastapi import APIRouter, Depends, Header, HTTPException
-from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 from sqlmodel import select
 
@@ -205,57 +204,19 @@ async def _refund_cache_set(authorization: str, value: dict[str, str]) -> None:
         _refund_cache[key] = (expiry, value)
 
 
-@router.post("/refund", response_model=None)
+@router.post("/refund")
 async def refund_wallet_endpoint(
-    authorization: Annotated[str | None, Header()] = None,
-    x_cashu: Annotated[str | None, Header()] = None,
+    authorization: Annotated[str, Header(...)],
     session: AsyncSession = Depends(get_session),
-) -> JSONResponse | dict[str, str]:
-    if x_cashu:
-        # Find the "in" transaction by the original payment token
-        in_tx_result = await session.exec(
-            select(CashuTransaction).where(
-                CashuTransaction.token == x_cashu,
-                CashuTransaction.type == "in",
-            )
-        )
-        in_tx = in_tx_result.first()
-        if in_tx is None:
-            raise HTTPException(status_code=404, detail="Refund not found")
-
-        # Use the request_id to find the associated "out" (refund) transaction
-        if in_tx.request_id is None:
-            raise HTTPException(status_code=404, detail="Refund not found")
-
-        out_tx_result = await session.exec(
-            select(CashuTransaction).where(
-                CashuTransaction.request_id == in_tx.request_id,
-                CashuTransaction.type == "out",
-            )
-        )
-        out_tx = out_tx_result.first()
-        if out_tx is None:
-            raise HTTPException(status_code=404, detail="Refund not found")
-        if out_tx.swept:
-            raise HTTPException(status_code=410, detail="Refund has been swept")
-
-        out_tx.collected = True
-        session.add(out_tx)
-        await session.commit()
-        body: dict[str, str] = {"token": out_tx.token}
-        if out_tx.unit == "sat":
-            body["sats"] = str(out_tx.amount)
-        else:
-            body["msats"] = str(out_tx.amount)
-        return JSONResponse(content=body, headers={"X-Cashu": out_tx.token})
-
-    if authorization is None or not authorization.startswith("Bearer "):
+) -> dict[str, str]:
+    if not authorization.startswith("Bearer "):
         raise HTTPException(
             status_code=401,
             detail="Invalid authorization. Use 'Bearer <cashu-token>' or 'Bearer <api-key>'",
         )
 
     bearer_value: str = authorization[7:]
+
     key: ApiKey = await validate_bearer_key(bearer_value, session)
 
     if key.total_balance <= 0:
@@ -268,12 +229,6 @@ async def refund_wallet_endpoint(
             detail="Cannot refund child key. Please refund the parent key instead.",
         )
 
-    if key.reserved_balance > 0:
-        raise HTTPException(
-            status_code=400,
-            detail="Cannot refund key. There are ongoing requests for this api key.",
-        )
-
     remaining_balance_msats: int = key.total_balance
 
     if key.refund_currency == "sat":
@@ -328,20 +283,11 @@ async def refund_wallet_endpoint(
 
     await _refund_cache_set(bearer_value, result)
 
-    previous_reserved_balance = key.reserved_balance
     key.balance = 0
     key.reserved_balance = 0
     session.add(key)
     await session.commit()
 
-    logger.info(
-        "refund_wallet_endpoint: refund successful",
-        extra={
-            "refunded_msats": remaining_balance_msats,
-            "previous_reserved_balance": previous_reserved_balance,
-        },
-    )
-
     return result
 
 

routstr/payment/cost_calculation.py

@@ -109,20 +109,12 @@ async def calculate_cost(  # todo: can be sync
     )
 
     usd_cost = 0.0
-    input_usd = 0.0
-    output_usd = 0.0
 
+    # Prioritize cost_details.upstream_inference_cost
     if "cost_details" in usage_data:
         usd_cost = float(
             usage_data["cost_details"].get("upstream_inference_cost", 0) or 0
         )
-        input_usd = float(
-            usage_data["cost_details"].get("upstream_inference_prompt_cost", 0) or 0
-        )
-        output_usd = float(
-            usage_data["cost_details"].get("upstream_inference_completions_cost", 0)
-            or 0
-        )
 
     # Fallback to cost field if upstream_inference_cost is 0
     if usd_cost == 0 and "cost" in usage_data:
@@ -131,34 +123,12 @@ async def calculate_cost(  # todo: can be sync
         except Exception:
             pass
 
-    MSATS_PER_1K_INPUT_TOKENS: float = (
-        float(settings.fixed_per_1k_input_tokens) * 1000.0
-    )
-    MSATS_PER_1K_OUTPUT_TOKENS: float = (
-        float(settings.fixed_per_1k_output_tokens) * 1000.0
-    )
-
     if usd_cost > 0:
         try:
             sats_per_usd = 1.0 / sats_usd_price()
             cost_in_sats = usd_cost * sats_per_usd
             cost_in_msats = math.ceil(cost_in_sats * 1000)
 
-            input_msats = 0
-            output_msats = 0
-
-            if input_usd > 0 or output_usd > 0:
-                input_msats = int((input_usd * sats_per_usd) * 1000)
-                output_msats = int((output_usd * sats_per_usd) * 1000)
-            else:
-                total_tokens = input_tokens + output_tokens
-                if total_tokens > 0:
-                    input_ratio = input_tokens / total_tokens
-                    input_msats = int(cost_in_msats * input_ratio)
-                    output_msats = cost_in_msats - input_msats
-                else:
-                    output_msats = cost_in_msats
-
             logger.info(
                 "Using cost from usage data/details",
                 extra={
@@ -170,9 +140,9 @@ async def calculate_cost(  # todo: can be sync
             )
 
             return CostData(
-                base_msats=0,
-                input_msats=input_msats,
-                output_msats=output_msats,
+                base_msats=-1,
+                input_msats=-1,  # Cost field doesn't break down by token type
+                output_msats=-1,
                 total_msats=cost_in_msats,
                 total_usd=usd_cost,
                 input_tokens=input_tokens,
@@ -189,6 +159,13 @@ async def calculate_cost(  # todo: can be sync
             )
             # Fall through to token-based calculation
 
+    MSATS_PER_1K_INPUT_TOKENS: float = (
+        float(settings.fixed_per_1k_input_tokens) * 1000.0
+    )
+    MSATS_PER_1K_OUTPUT_TOKENS: float = (
+        float(settings.fixed_per_1k_output_tokens) * 1000.0
+    )
+
     if not settings.fixed_pricing:
         response_model = response_data.get("model", "")
         logger.debug(
@@ -254,10 +231,10 @@ async def calculate_cost(  # todo: can be sync
             output_tokens=output_tokens,
         )
 
-    calc_input_msats = round(input_tokens / 1000 * MSATS_PER_1K_INPUT_TOKENS, 3)
+    input_msats = round(input_tokens / 1000 * MSATS_PER_1K_INPUT_TOKENS, 3)
 
-    calc_output_msats = round(output_tokens / 1000 * MSATS_PER_1K_OUTPUT_TOKENS, 3)
-    token_based_cost = math.ceil(calc_input_msats + calc_output_msats)
+    output_msats = round(output_tokens / 1000 * MSATS_PER_1K_OUTPUT_TOKENS, 3)
+    token_based_cost = math.ceil(input_msats + output_msats)
     total_usd = (token_based_cost / 1000.0) * sats_usd_price()
 
     logger.info(
@@ -265,8 +242,8 @@ async def calculate_cost(  # todo: can be sync
         extra={
             "input_tokens": input_tokens,
             "output_tokens": output_tokens,
-            "input_cost_msats": calc_input_msats,
-            "output_cost_msats": calc_output_msats,
+            "input_cost_msats": input_msats,
+            "output_cost_msats": output_msats,
             "total_cost_msats": token_based_cost,
             "total_usd": total_usd,
             "model": response_data.get("model", "unknown"),
@@ -275,8 +252,8 @@ async def calculate_cost(  # todo: can be sync
 
     return CostData(
         base_msats=0,
-        input_msats=int(calc_input_msats),
-        output_msats=int(calc_output_msats),
+        input_msats=int(input_msats),
+        output_msats=int(output_msats),
         total_msats=token_based_cost,
         total_usd=total_usd,
         input_tokens=input_tokens,

tests/unit/test_balance.py

@@ -1,116 +0,0 @@
-import json
-from unittest.mock import AsyncMock, MagicMock
-
-import pytest
-from fastapi.responses import JSONResponse
-
-from routstr.balance import refund_wallet_endpoint
-from routstr.core.db import CashuTransaction
-
-
-def _make_cashu_tx(
-    token: str,
-    amount: int,
-    unit: str,
-    type: str = "out",
-    request_id: str | None = "req-abc",
-    swept: bool = False,
-    collected: bool = False,
-) -> CashuTransaction:
-    tx = CashuTransaction(token=token, amount=amount, unit=unit, type=type, request_id=request_id)
-    tx.swept = swept
-    tx.collected = collected
-    return tx
-
-
-def _exec_result(tx: CashuTransaction | None) -> MagicMock:
-    result = MagicMock()
-    result.first.return_value = tx
-    return result
-
-
-@pytest.mark.asyncio
-async def test_refund_x_cashu_returns_token() -> None:
-    x_cashu_token = "cashuAtest_token_value"
-    in_tx = _make_cashu_tx(token=x_cashu_token, amount=0, unit="msat", type="in", request_id="req-abc")
-    out_tx = _make_cashu_tx(token="cashuArefund_token", amount=1000, unit="msat", type="out", request_id="req-abc")
-
-    session = MagicMock()
-    session.exec = AsyncMock(side_effect=[_exec_result(in_tx), _exec_result(out_tx)])
-    session.add = MagicMock()
-    session.commit = AsyncMock()
-
-    result = await refund_wallet_endpoint(
-        authorization="Bearer sk-somekey",
-        x_cashu=x_cashu_token,
-        session=session,
-    )
-
-    assert isinstance(result, JSONResponse)
-    body = json.loads(result.body)
-    assert body["token"] == "cashuArefund_token"
-    assert body["msats"] == "1000"
-    assert result.headers["X-Cashu"] == "cashuArefund_token"
-    assert out_tx.collected is True
-
-
-@pytest.mark.asyncio
-async def test_refund_x_cashu_sat_unit() -> None:
-    x_cashu_token = "cashuAsat_token"
-    in_tx = _make_cashu_tx(token=x_cashu_token, amount=0, unit="sat", type="in", request_id="req-sat")
-    out_tx = _make_cashu_tx(token="cashuArefund_sat", amount=500, unit="sat", type="out", request_id="req-sat")
-
-    session = MagicMock()
-    session.exec = AsyncMock(side_effect=[_exec_result(in_tx), _exec_result(out_tx)])
-    session.add = MagicMock()
-    session.commit = AsyncMock()
-
-    result = await refund_wallet_endpoint(
-        authorization="Bearer sk-somekey",
-        x_cashu=x_cashu_token,
-        session=session,
-    )
-
-    assert isinstance(result, JSONResponse)
-    body = json.loads(result.body)
-    assert body["token"] == "cashuArefund_sat"
-    assert body["sats"] == "500"
-    assert "msats" not in body
-    assert result.headers["X-Cashu"] == "cashuArefund_sat"
-
-
-@pytest.mark.asyncio
-async def test_refund_x_cashu_not_found_raises_404() -> None:
-    from fastapi import HTTPException
-
-    session = MagicMock()
-    session.exec = AsyncMock(return_value=_exec_result(None))
-
-    with pytest.raises(HTTPException) as exc_info:
-        await refund_wallet_endpoint(
-            authorization="Bearer sk-somekey",
-            x_cashu="cashuAmissing_token",
-            session=session,
-        )
-
-    assert exc_info.value.status_code == 404
-
-
-@pytest.mark.asyncio
-async def test_refund_x_cashu_swept_raises_410() -> None:
-    from fastapi import HTTPException
-
-    in_tx = _make_cashu_tx(token="cashuAswept_token", amount=0, unit="msat", type="in", request_id="req-swept")
-    out_tx = _make_cashu_tx(token="cashuAswept", amount=100, unit="msat", type="out", request_id="req-swept", swept=True)
-
-    session = MagicMock()
-    session.exec = AsyncMock(side_effect=[_exec_result(in_tx), _exec_result(out_tx)])
-
-    with pytest.raises(HTTPException) as exc_info:
-        await refund_wallet_endpoint(
-            authorization="Bearer sk-somekey",
-            x_cashu="cashuAswept_token",
-            session=session,
-        )
-
-    assert exc_info.value.status_code == 410

ui/components/api-key-input.tsx

@@ -1,39 +0,0 @@
-'use client';
-
-import { useState, useEffect } from 'react';
-import * as React from 'react';
-import { Input } from '@/components/ui/input';
-
-interface ApiKeyInputProps extends React.ComponentProps<'input'> {
-  onApiKeyChange: (apiKey: string) => void;
-}
-
-export function ApiKeyInput({
-  value,
-  onApiKeyChange,
-  ...props
-}: ApiKeyInputProps) {
-  const [internalValue, setInternalValue] = useState(value || '');
-
-  useEffect(() => {
-    setInternalValue(value || '');
-  }, [value]);
-
-  useEffect(() => {
-    const handler = setTimeout(() => {
-      onApiKeyChange(internalValue as string);
-    }, 300);
-
-    return () => clearTimeout(handler);
-  }, [internalValue, onApiKeyChange]);
-
-  return (
-    <Input
-      value={internalValue}
-      onChange={(e) => setInternalValue(e.target.value)}
-      placeholder='sk-...'
-      className='font-mono text-sm'
-      {...props}
-    />
-  );
-}

ui/components/child-key-creator.tsx

@@ -1,9 +1,7 @@
 'use client';
 
 import { useState } from 'react';
-import { useWalletInfo } from '@/hooks/use-wallet-info';
 import { WalletService } from '@/lib/api/services/wallet';
-import { ApiKeyInput } from './api-key-input';
 import { Button } from '@/components/ui/button';
 import {
   Card,
@@ -16,8 +14,17 @@ import { Alert, AlertDescription, AlertTitle } from '@/components/ui/alert';
 import { Input } from '@/components/ui/input';
 import { Textarea } from '@/components/ui/textarea';
 import { Label } from '@/components/ui/label';
-import { Key, Copy, Check, Loader2, Plus, Trash2 } from 'lucide-react';
+import {
+  Key,
+  Copy,
+  Check,
+  Loader2,
+  RotateCcw,
+  Plus,
+  Trash2,
+} from 'lucide-react';
 import { toast } from 'sonner';
+import { Badge } from '@/components/ui/badge';
 import { KeyOptions } from './key-options';
 
 interface KeyConfig {
@@ -35,14 +42,6 @@ interface ChildKeyCreatorProps {
   costPerKeyMsats?: number;
 }
 
-function formatSats(msats: number): string {
-  return new Intl.NumberFormat('en-US').format(Math.floor(msats / 1000));
-}
-
-function formatMsats(msats: number): string {
-  return new Intl.NumberFormat('en-US').format(msats);
-}
-
 export function ChildKeyCreator({
   baseUrl,
   apiKey: propApiKey,
@@ -51,7 +50,6 @@ export function ChildKeyCreator({
 }: ChildKeyCreatorProps) {
   const [internalApiKey, setInternalApiKey] = useState('');
   const [loading, setLoading] = useState(false);
-  const [error, setError] = useState<string | null>(null);
   const [configs, setConfigs] = useState<KeyConfig[]>([
     {
       id: crypto.randomUUID(),
@@ -61,15 +59,15 @@ export function ChildKeyCreator({
       validityDate: '',
     },
   ]);
-
-  const activeApiKey = propApiKey ?? internalApiKey;
-  const { data: walletInfo } = useWalletInfo(baseUrl ?? '', activeApiKey);
-
-  const handleApiKeyChange = (val: string) => {
-    setInternalApiKey(val);
-    onApiKeyChange?.(val);
-  };
-
+  const [childKeyToCheck, setChildKeyToCheck] = useState('');
+  const [checking, setChecking] = useState(false);
+  const [keyStatus, setKeyStatus] = useState<{
+    total_spent: number;
+    balance_limit: number | null;
+    validity_date: number | null;
+    is_expired: boolean;
+    is_drained: boolean;
+  } | null>(null);
   const [newKeys, setNewKeys] = useState<string[]>([]);
   const [resultInfo, setResultInfo] = useState<{
     cost_msats: number;
@@ -77,6 +75,13 @@ export function ChildKeyCreator({
   } | null>(null);
   const [copiedKey, setCopiedKey] = useState<string | null>(null);
 
+  const activeApiKey = propApiKey ?? internalApiKey;
+
+  const handleApiKeyChange = (val: string) => {
+    setInternalApiKey(val);
+    onApiKeyChange?.(val);
+  };
+
   const addConfig = () => {
     setConfigs([
       ...configs,
@@ -107,7 +112,6 @@ export function ChildKeyCreator({
     }
 
     setLoading(true);
-    setError(null);
     try {
       let allNewKeys: string[] = [];
       let totalCost = 0;
@@ -148,21 +152,55 @@ export function ChildKeyCreator({
       );
     } catch (error) {
       console.error('Failed to create child key:', error);
-      let errorMessage =
-        error instanceof Error ? error.message : 'Failed to create child key';
-      try {
-        const parsed = JSON.parse(errorMessage);
-        errorMessage =
-          parsed.detail?.error?.message ||
-          (typeof parsed.detail === 'string' ? parsed.detail : errorMessage);
-      } catch {}
-      setError(errorMessage);
-      toast.error(errorMessage);
+      toast.error(
+        error instanceof Error ? error.message : 'Failed to create child key'
+      );
     } finally {
       setLoading(false);
     }
   };
 
+  const handleCheckKey = async () => {
+    if (!childKeyToCheck) {
+      toast.error('Please provide a Child API key to check');
+      return;
+    }
+
+    setChecking(true);
+    setKeyStatus(null);
+    try {
+      const baseUrlToUse = baseUrl || '';
+      const response = await fetch(`${baseUrlToUse}/v1/balance/info`, {
+        headers: {
+          Authorization: `Bearer ${childKeyToCheck}`,
+        },
+      });
+
+      if (!response.ok) {
+        throw new Error('Failed to fetch key info');
+      }
+
+      const info = await response.json();
+      const now = Math.floor(Date.now() / 1000);
+
+      setKeyStatus({
+        total_spent: info.total_spent,
+        balance_limit: info.balance_limit,
+        validity_date: info.validity_date,
+        is_expired: info.validity_date ? now > info.validity_date : false,
+        is_drained: info.balance_limit
+          ? info.total_spent >= info.balance_limit
+          : false,
+      });
+    } catch (error) {
+      toast.error(
+        error instanceof Error ? error.message : 'Failed to check child key'
+      );
+    } finally {
+      setChecking(false);
+    }
+  };
+
   const copyToClipboard = (key: string) => {
     navigator.clipboard.writeText(key);
     setCopiedKey(key);
@@ -205,55 +243,12 @@ export function ChildKeyCreator({
                 <Label className='text-muted-foreground text-[0.7rem] tracking-wider'>
                   Parent API Key
                 </Label>
-                <div className='flex gap-2'>
-                  <div className='flex-1'>
-                    <ApiKeyInput
-                      value={activeApiKey}
-                      onApiKeyChange={handleApiKeyChange}
-                    />
-                  </div>
-                  <Button
-                    variant='outline'
-                    size='icon'
-                    onClick={() => navigator.clipboard.writeText(activeApiKey)}
-                    disabled={!activeApiKey}
-                  >
-                    <Copy className='h-4 w-4' />
-                  </Button>
-                </div>
-                {walletInfo && (
-                  <div className='bg-muted/30 mt-2 space-y-2 rounded-lg p-3'>
-                    <div className='flex items-center justify-between'>
-                      <span className='text-muted-foreground text-sm'>
-                        Spendable Balance
-                      </span>
-                      <span className='text-primary font-mono text-sm font-medium'>
-                        {formatSats(walletInfo.balanceMsats)} sats
-                      </span>
-                    </div>
-                    <div className='flex items-center justify-between'>
-                      <span className='text-muted-foreground text-sm'>
-                        Total Requests
-                      </span>
-                      <span className='font-mono text-sm font-medium'>
-                        {walletInfo.totalRequests}
-                      </span>
-                    </div>
-                    <div className='flex items-center justify-between'>
-                      <span className='text-muted-foreground text-sm'>
-                        Total Spent
-                      </span>
-                      <div className='text-right'>
-                        <p className='font-mono text-sm font-medium'>
-                          {formatSats(walletInfo.totalSpent)} sats
-                        </p>
-                        <p className='text-muted-foreground font-mono text-[0.6rem]'>
-                          {formatMsats(walletInfo.totalSpent)} msats
-                        </p>
-                      </div>
-                    </div>
-                  </div>
-                )}
+                <Input
+                  value={activeApiKey}
+                  onChange={(e) => handleApiKeyChange(e.target.value)}
+                  placeholder='sk-...'
+                  className='font-mono text-sm'
+                />
               </div>
             )}
 
@@ -367,19 +362,12 @@ export function ChildKeyCreator({
                   )}
                 </Button>
               </div>
-
-              {error && (
-                <Alert variant='destructive'>
-                  <AlertTitle>Error</AlertTitle>
-                  <AlertDescription>{error}</AlertDescription>
-                </Alert>
-              )}
-
-              <p className='text-muted-foreground text-xs'>
-                Each key creation has a small one-time fee.
-              </p>
             </div>
 
+            <p className='text-muted-foreground text-xs'>
+              Each key creation has a small one-time fee.
+            </p>
+
             {newKeys.length > 0 && (
               <div className='mt-6 space-y-4'>
                 <Alert>
@@ -470,6 +458,89 @@ export function ChildKeyCreator({
           </div>
         </CardContent>
       </Card>
+
+      <Card>
+        <CardHeader>
+          <CardTitle className='text-lg'>Check Child Key Status</CardTitle>
+          <CardDescription>
+            View the current spending, limit, and expiration status of any child
+            key.
+          </CardDescription>
+        </CardHeader>
+        <CardContent>
+          <div className='space-y-4'>
+            <div className='space-y-2'>
+              <Label className='text-muted-foreground text-[0.7rem] tracking-wider'>
+                Child API Key
+              </Label>
+              <Input
+                value={childKeyToCheck}
+                onChange={(e) => setChildKeyToCheck(e.target.value)}
+                placeholder='sk-...'
+                className='font-mono text-sm'
+              />
+            </div>
+            <Button
+              onClick={handleCheckKey}
+              disabled={checking || !childKeyToCheck}
+              variant='outline'
+              className='w-full'
+            >
+              {checking ? (
+                <>
+                  <Loader2 className='mr-2 h-4 w-4 animate-spin' />
+                  Checking...
+                </>
+              ) : (
+                <>
+                  <RotateCcw className='mr-2 h-4 w-4' />
+                  Check Status
+                </>
+              )}
+            </Button>
+
+            {keyStatus && (
+              <div className='bg-muted/30 mt-4 space-y-3 rounded-lg border p-4 text-sm'>
+                <div className='flex justify-between'>
+                  <span className='text-muted-foreground'>Total Spent:</span>
+                  <span className='font-mono font-medium'>
+                    {keyStatus.total_spent} mSats
+                  </span>
+                </div>
+                {keyStatus.balance_limit !== null && (
+                  <div className='flex justify-between'>
+                    <span className='text-muted-foreground'>Limit:</span>
+                    <span className='font-mono font-medium'>
+                      {keyStatus.balance_limit} mSats
+                    </span>
+                  </div>
+                )}
+                {keyStatus.validity_date !== null && (
+                  <div className='flex justify-between'>
+                    <span className='text-muted-foreground'>Expires:</span>
+                    <span className='font-mono font-medium'>
+                      {new Date(
+                        keyStatus.validity_date * 1000
+                      ).toLocaleDateString()}
+                    </span>
+                  </div>
+                )}
+                <div className='flex gap-2 pt-2'>
+                  {keyStatus.is_drained && (
+                    <Badge variant='destructive'>Drained</Badge>
+                  )}
+                  {keyStatus.is_expired && (
+                    <Badge variant='destructive'>Expired</Badge>
+                  )}
+                  {!keyStatus.is_drained && !keyStatus.is_expired && (
+                    <Badge>Active</Badge>
+                  )}
+                </div>
+              </div>
+            )}
+          </div>
+        </CardContent>
+      </Card>
     </div>
   );
 }

ui/components/landing/api-key-manager.tsx

@@ -239,7 +239,7 @@ export function ApiKeyManager({
                     className='gap-2'
                   >
                     <Trash2 className='h-4 w-4' />
-                    {isRefunding ? 'Processing...' : 'Refund Key'}
+                    {isRefunding ? 'Processing...' : 'Refund & Delete Key'}
                   </Button>
                   <span className='text-muted-foreground text-xs'>
                     Burns the key and returns a fresh Cashu token.

ui/components/landing/cashu-payment-workflow.tsx

@@ -1,17 +1,16 @@
 'use client';
 
 import { type JSX, useCallback, useState } from 'react';
-import { Copy, RefreshCcw } from 'lucide-react';
-import { Alert, AlertDescription, AlertTitle } from '@/components/ui/alert';
+import { Copy, RefreshCcw, Trash2 } from 'lucide-react';
 import { toast } from 'sonner';
 import { Card, CardContent, CardHeader, CardTitle } from '@/components/ui/card';
 import { Button } from '@/components/ui/button';
 import { Textarea } from '@/components/ui/textarea';
+import { Input } from '@/components/ui/input';
 import { Separator } from '@/components/ui/separator';
 import { KeyOptions } from '@/components/key-options';
-import { ApiKeyInput } from '../api-key-input';
-import type { WalletSnapshot } from './key-info-details';
-import { useWalletInfo } from '@/hooks/use-wallet-info';
+import { WalletBalanceStats } from './wallet-balance-stats';
+import type { ChildKeyInfo, WalletSnapshot } from './key-info-details';
 
 export type RefundReceipt = {
   token?: string;
@@ -30,41 +29,80 @@ interface CashuPaymentWorkflowProps {
   onRefundComplete?: (receipt: RefundReceipt) => void;
 }
 
-function formatSats(msats: number): string {
-  return new Intl.NumberFormat('en-US').format(Math.floor(msats / 1000));
+async function fetchWalletInfo(
+  baseUrl: string,
+  apiKey: string
+): Promise<WalletSnapshot> {
+  const response = await fetch(`${baseUrl}/v1/balance/info`, {
+    cache: 'no-store',
+    headers: {
+      'Content-Type': 'application/json',
+      Authorization: `Bearer ${apiKey}`,
+    },
+  });
+
+  if (!response.ok) {
+    const errorText = await response.text();
+    throw new Error(errorText || 'Unable to load wallet info');
+  }
+
+  const payload = (await response.json()) as {
+    api_key: string;
+    balance: number;
+    reserved?: number;
+    is_child: boolean;
+    parent_key: string | null;
+    total_requests: number;
+    total_spent: number;
+    balance_limit: number | null;
+    balance_limit_reset: string | null;
+    validity_date: number | null;
+    child_keys?: ChildKeyInfo[];
+  };
+
+  return {
+    apiKey: payload.api_key || apiKey,
+    balanceMsats: payload.balance ?? 0,
+    reservedMsats: payload.reserved ?? 0,
+    isChild: payload.is_child,
+    parentKey: payload.parent_key,
+    totalRequests: payload.total_requests,
+    totalSpent: payload.total_spent,
+    balanceLimit: payload.balance_limit,
+    balanceLimitReset: payload.balance_limit_reset,
+    validityDate: payload.validity_date,
+    childKeys: payload.child_keys,
+  };
 }
 
-function formatMsats(msats: number): string {
-  return new Intl.NumberFormat('en-US').format(msats);
+function formatSats(msats: number): string {
+  return new Intl.NumberFormat('en-US').format(Math.floor(msats / 1000));
 }
 
 export function CashuPaymentWorkflow({
   baseUrl,
   apiKey = '',
-  walletInfo: propWalletInfo = null,
+  walletInfo = null,
   onApiKeyCreated,
+  onApiKeyChanged,
   onWalletInfoUpdated,
+  onRefundComplete,
 }: CashuPaymentWorkflowProps): JSX.Element {
   const [initialToken, setInitialToken] = useState('');
   const [topupToken, setTopupToken] = useState('');
   const [apiKeyInput, setApiKeyInput] = useState(apiKey);
   const [isCreatingKey, setIsCreatingKey] = useState(false);
   const [isTopupLoading, setIsTopupLoading] = useState(false);
+  const [isRefunding, setIsRefunding] = useState(false);
+  const [isSyncingBalance, setIsSyncingBalance] = useState(false);
+  const [hasInteractedManage, setHasInteractedManage] = useState(false);
   const [hasInteractedTopup, setHasInteractedTopup] = useState(false);
   const [balanceLimit, setBalanceLimit] = useState<string>('');
   const [balanceLimitReset, setBalanceLimitReset] = useState<string>('');
   const [validityDate, setValidityDate] = useState<string>('');
-  const [error, setError] = useState<string | null>(null);
 
   const activeApiKey = apiKeyInput.trim();
 
-  const {
-    data: queryWalletInfo,
-    refetch,
-    isFetching,
-  } = useWalletInfo(baseUrl, activeApiKey);
-  const walletInfo = propWalletInfo ?? queryWalletInfo ?? null;
-
   const handleCopy = useCallback(async (value: string): Promise<void> => {
     if (!value) {
       return;
@@ -165,18 +203,20 @@ export function CashuPaymentWorkflow({
       return;
     }
 
-    setError(null);
+    setIsSyncingBalance(true);
     try {
-      await refetch();
+      const snapshot = await fetchWalletInfo(baseUrl, activeApiKey);
+      onWalletInfoUpdated?.(snapshot);
       toast.success('Balance synced');
     } catch (error) {
       console.error(error);
-      const message =
-        error instanceof Error ? error.message : 'Failed to sync balance';
-      setError(message);
-      toast.error(message);
+      toast.error(
+        error instanceof Error ? error.message : 'Failed to sync balance'
+      );
+    } finally {
+      setIsSyncingBalance(false);
     }
-  }, [activeApiKey, refetch]);
+  }, [activeApiKey, baseUrl, onWalletInfoUpdated]);
 
   const handleTopup = useCallback(async (): Promise<void> => {
     if (!activeApiKey) {
@@ -205,25 +245,59 @@ export function CashuPaymentWorkflow({
       const payload = (await response.json()) as { msats: number };
       toast.success(`Added ${formatSats(payload.msats)} sats`);
       setTopupToken('');
-      await refetch();
+      const snapshot = await fetchWalletInfo(baseUrl, activeApiKey);
+      onApiKeyCreated?.(snapshot.apiKey, snapshot);
     } catch (error) {
       console.error(error);
       toast.error(error instanceof Error ? error.message : 'Top-up failed');
     } finally {
       setIsTopupLoading(false);
     }
-  }, [activeApiKey, baseUrl, topupToken, refetch]);
+  }, [activeApiKey, baseUrl, topupToken, onApiKeyCreated]);
+
+  const handleRefund = useCallback(async (): Promise<void> => {
+    if (!activeApiKey) {
+      toast.error('Paste an API key first');
+      return;
+    }
+
+    setIsRefunding(true);
+    try {
+      const response = await fetch(`${baseUrl}/v1/balance/refund`, {
+        method: 'POST',
+        headers: {
+          Authorization: `Bearer ${activeApiKey}`,
+        },
+      });
+      if (!response.ok) {
+        const errorText = await response.text();
+        throw new Error(errorText || 'Refund failed');
+      }
+      const payload = (await response.json()) as RefundReceipt;
+      onRefundComplete?.(payload);
+      onWalletInfoUpdated?.(null);
+      setApiKeyInput('');
+      toast.success('Refund requested');
+    } catch (error) {
+      console.error(error);
+      toast.error(error instanceof Error ? error.message : 'Refund failed');
+    } finally {
+      setIsRefunding(false);
+    }
+  }, [activeApiKey, baseUrl, onRefundComplete, onWalletInfoUpdated]);
 
   const handleApiKeyChange = useCallback(
     (newKey: string) => {
       setApiKeyInput(newKey);
+      onApiKeyChanged?.(newKey);
       if (newKey !== apiKey) {
         onWalletInfoUpdated?.(null);
       }
     },
-    [apiKey, onWalletInfoUpdated]
+    [apiKey, onApiKeyChanged, onWalletInfoUpdated]
   );
 
+  const showManageDetails = hasInteractedManage || Boolean(walletInfo);
   const showTopupDetails = hasInteractedTopup || topupToken.trim().length > 0;
   const canTopup = Boolean(activeApiKey);
   const showCreateDetails = initialToken.trim().length > 0;
@@ -291,72 +365,40 @@ export function CashuPaymentWorkflow({
             )}
           </header>
           <div className='flex flex-col gap-2 sm:flex-row'>
-            <ApiKeyInput
+            <Input
               value={apiKeyInput}
-              onApiKeyChange={handleApiKeyChange}
+              onChange={(event) => handleApiKeyChange(event.target.value)}
+              placeholder='sk-...'
+              className='font-mono text-sm'
+              onFocus={() => setHasInteractedManage(true)}
             />
             <div className='flex gap-2'>
               <Button
                 variant='outline'
                 size='icon'
+                className='h-10 w-10'
                 onClick={() => handleCopy(activeApiKey)}
                 disabled={!activeApiKey}
               >
                 <Copy className='h-4 w-4' />
               </Button>
-
               <Button
                 variant='secondary'
                 size='sm'
                 className='gap-1'
                 onClick={handleSyncBalance}
-                disabled={isFetching || !activeApiKey}
+                disabled={isSyncingBalance || !activeApiKey}
               >
-                <RefreshCcw
-                  className={`h-4 w-4 ${isFetching ? 'animate-spin' : ''}`}
-                />
+                <RefreshCcw className='h-4 w-4' />
                 Sync
               </Button>
             </div>
           </div>
-          {walletInfo && (
-            <div className='bg-muted/30 mt-2 space-y-2 rounded-lg p-3'>
-              <div className='flex items-center justify-between'>
-                <span className='text-muted-foreground text-sm'>
-                  Spendable Balance
-                </span>
-                <span className='text-primary font-mono text-sm font-medium'>
-                  {formatSats(walletInfo.balanceMsats)} sats
-                </span>
-              </div>
-              <div className='flex items-center justify-between'>
-                <span className='text-muted-foreground text-sm'>
-                  Total Requests
-                </span>
-                <span className='font-mono text-sm font-medium'>
-                  {walletInfo.totalRequests}
-                </span>
-              </div>
-              <div className='flex items-center justify-between'>
-                <span className='text-muted-foreground text-sm'>
-                  Total Spent
-                </span>
-                <div className='text-right'>
-                  <p className='font-mono text-sm font-medium'>
-                    {formatSats(walletInfo.totalSpent)} sats
-                  </p>
-                  <p className='text-muted-foreground font-mono text-[0.6rem]'>
-                    {formatMsats(walletInfo.totalSpent)} msats
-                  </p>
-                </div>
-              </div>
-            </div>
-          )}
-          {error && (
-            <Alert variant='destructive' className='mt-2'>
-              <AlertTitle>Error</AlertTitle>
-              <AlertDescription>{error}</AlertDescription>
-            </Alert>
+          {showManageDetails && (
+            <WalletBalanceStats
+              balanceMsats={walletInfo?.balanceMsats}
+              reservedMsats={walletInfo?.reservedMsats}
+            />
           )}
         </section>
 
@@ -402,6 +444,28 @@ export function CashuPaymentWorkflow({
             </div>
           )}
         </section>
+
+        <Separator />
+
+        <section className='space-y-2'>
+          <header className='text-muted-foreground flex items-center justify-between text-[0.7rem] tracking-wider'>
+            <span>4 · Refund</span>
+          </header>
+          <div className='flex flex-wrap gap-2'>
+            <Button
+              onClick={handleRefund}
+              disabled={isRefunding || !activeApiKey}
+              variant='destructive'
+              className='gap-2'
+            >
+              <Trash2 className='h-4 w-4' />
+              {isRefunding ? 'Processing…' : 'Refund remaining balance'}
+            </Button>
+            <span className='text-muted-foreground text-xs'>
+              Burns the key and returns a fresh Cashu token.
+            </span>
+          </div>
+        </section>
       </CardContent>
     </Card>
   );

ui/components/landing/cheat-sheet.tsx

@@ -18,6 +18,7 @@ import {
   type RefundReceipt,
 } from './cashu-payment-workflow';
 import { LightningPaymentWorkflow } from './lightning-payment-workflow';
+import { ApiKeyManager } from './api-key-manager';
 import { KeyInfoDetails, type WalletSnapshot } from './key-info-details';
 import { ChildKeyCreator } from '@/components/child-key-creator';
 
@@ -134,8 +135,8 @@ export function CheatSheet(): JSX.Element {
 
   const handleRefundComplete = useCallback((receipt: RefundReceipt) => {
     setRefundReceipt(receipt);
-    // setWalletInfo(null); // Keep info
-    // setApiKeyInput(''); // Keep input
+    setWalletInfo(null);
+    setApiKeyInput('');
   }, []);
 
   const handleRefreshInfo = useCallback(async (): Promise<void> => {
@@ -390,11 +391,12 @@ export function CheatSheet(): JSX.Element {
         </section>
 
         <Tabs defaultValue='cashu' className='w-full'>
-          <TabsList className='grid w-full grid-cols-4'>
+          <TabsList className='grid w-full grid-cols-5'>
             <TabsTrigger value='cashu'>Cashu</TabsTrigger>
             <TabsTrigger value='lightning'>Lightning</TabsTrigger>
+            <TabsTrigger value='manage'>Manage Keys</TabsTrigger>
+            <TabsTrigger value='details'>Key Details</TabsTrigger>
             <TabsTrigger value='child-keys'>Child Keys</TabsTrigger>
-            <TabsTrigger value='management'>Key Management</TabsTrigger>
           </TabsList>
 
           <TabsContent value='cashu' className='space-y-4'>
@@ -411,8 +413,8 @@ export function CheatSheet(): JSX.Element {
             />
           </TabsContent>
 
-          <TabsContent value='management' className='space-y-4'>
-            <KeyInfoDetails
+          <TabsContent value='manage' className='space-y-4'>
+            <ApiKeyManager
               baseUrl={normalizedBaseUrl}
               apiKey={apiKeyInput}
               walletInfo={walletInfo}
@@ -443,7 +445,7 @@ export function CheatSheet(): JSX.Element {
                     <Textarea
                       value={refundToken}
                       readOnly
-                      rows={15}
+                      rows={4}
                       className='font-mono text-xs'
                     />
                   </div>
@@ -452,6 +454,16 @@ export function CheatSheet(): JSX.Element {
             )}
           </TabsContent>
 
+          <TabsContent value='details' className='space-y-4'>
+            <KeyInfoDetails
+              baseUrl={normalizedBaseUrl}
+              apiKey={apiKeyInput}
+              walletInfo={walletInfo}
+              onApiKeyChanged={handleApiKeyChanged}
+              onWalletInfoUpdated={handleWalletInfoUpdated}
+            />
+          </TabsContent>
+
           <TabsContent value='child-keys' className='space-y-4'>
             <ChildKeyCreator
               baseUrl={normalizedBaseUrl}

ui/components/landing/key-info-details.tsx

@@ -1,10 +1,8 @@
 'use client';
 
-import { useWalletInfo } from '@/hooks/use-wallet-info';
-import type { RefundReceipt } from './cashu-payment-workflow';
+import { type JSX, useState, useCallback, useEffect } from 'react';
+import { Copy, RefreshCcw, RotateCcw } from 'lucide-react';
 import { toast } from 'sonner';
-import { ApiKeyInput } from '../api-key-input';
-import { Alert, AlertDescription, AlertTitle } from '@/components/ui/alert';
 import {
   Card,
   CardContent,
@@ -14,9 +12,8 @@ import {
 } from '@/components/ui/card';
 import { Button } from '@/components/ui/button';
 import { Badge } from '@/components/ui/badge';
+import { Input } from '@/components/ui/input';
 import { WalletService } from '@/lib/api/services/wallet';
-import React, { useState, useCallback, useEffect } from 'react';
-import { Copy, RefreshCcw, RotateCcw, Trash2 } from 'lucide-react';
 
 export type ChildKeyInfo = {
   api_key: string;
@@ -47,44 +44,68 @@ interface KeyInfoDetailsProps {
   walletInfo?: WalletSnapshot | null;
   onApiKeyChanged?: (apiKey: string) => void;
   onWalletInfoUpdated?: (walletInfo: WalletSnapshot | null) => void;
-  onRefundComplete?: (receipt: RefundReceipt) => void;
 }
 
 export function KeyInfoDetails({
   baseUrl,
   apiKey = '',
-  walletInfo: propWalletInfo = null,
+  walletInfo = null,
   onApiKeyChanged,
   onWalletInfoUpdated,
-  onRefundComplete,
-}: KeyInfoDetailsProps): React.ReactNode {
+}: KeyInfoDetailsProps): JSX.Element {
   const [apiKeyInput, setApiKeyInput] = useState(apiKey);
+  const [isRefreshing, setIsRefreshing] = useState(false);
   const [isResetting, setIsResetting] = useState<string | null>(null);
-  const [isRefunding, setIsRefunding] = useState(false);
-  const [error, setError] = useState<string | null>(null);
-
-  const {
-    data: queryWalletInfo,
-    refetch,
-    isFetching,
-  } = useWalletInfo(baseUrl, apiKeyInput);
-  const walletInfo = propWalletInfo ?? queryWalletInfo ?? null;
 
   // Sync internal state with props if they change
   useEffect(() => {
     setApiKeyInput(apiKey);
   }, [apiKey]);
 
-  const handleRefresh = async (e: React.MouseEvent) => {
-    e.preventDefault();
-    e.stopPropagation();
+  const fetchDetails = useCallback(
+    async (keyToFetch: string) => {
+      setIsRefreshing(true);
+      try {
+        const response = await fetch(`${baseUrl}/v1/balance/info`, {
+          headers: { Authorization: `Bearer ${keyToFetch}` },
+        });
+        if (!response.ok) {
+          throw new Error('Failed to fetch key info');
+        }
+        const payload = await response.json();
+        const snapshot: WalletSnapshot = {
+          apiKey: payload.api_key || keyToFetch,
+          balanceMsats: payload.balance ?? 0,
+          reservedMsats: payload.reserved ?? 0,
+          isChild: payload.is_child,
+          parentKey: payload.parent_key,
+          totalRequests: payload.total_requests,
+          totalSpent: payload.total_spent,
+          balanceLimit: payload.balance_limit,
+          balanceLimitReset: payload.balance_limit_reset,
+          validityDate: payload.validity_date,
+          childKeys: payload.child_keys,
+        };
+        onWalletInfoUpdated?.(snapshot);
+        toast.success('Key details synced');
+      } catch (error) {
+        toast.error(
+          error instanceof Error ? error.message : 'Failed to fetch details'
+        );
+      } finally {
+        setIsRefreshing(false);
+      }
+    },
+    [baseUrl, onWalletInfoUpdated]
+  );
+
+  const handleRefresh = async () => {
     if (!apiKeyInput) return;
-    await refetch();
+    await fetchDetails(apiKeyInput);
   };
 
   const handleKeyChange = (newKey: string) => {
     setApiKeyInput(newKey);
-    setError(null);
     onApiKeyChanged?.(newKey);
     // Optionally clear info when key changes
     if (newKey !== apiKey) {
@@ -104,7 +125,7 @@ export function KeyInfoDetails({
     try {
       await WalletService.resetChildKeySpent(baseUrl, apiKeyInput, childKey);
       toast.success('Child key spent reset');
-      await refetch();
+      await fetchDetails(apiKeyInput);
     } catch (error) {
       toast.error(
         error instanceof Error ? error.message : 'Failed to reset child key'
@@ -114,36 +135,6 @@ export function KeyInfoDetails({
     }
   };
 
-  const handleRefund = useCallback(async (): Promise<void> => {
-    if (!apiKeyInput) {
-      toast.error('Paste an API key first');
-      return;
-    }
-
-    setIsRefunding(true);
-    try {
-      const response = await fetch(`${baseUrl}/v1/balance/refund`, {
-        method: 'POST',
-        headers: {
-          Authorization: `Bearer ${apiKeyInput}`,
-        },
-      });
-      if (!response.ok) {
-        const errorText = await response.text();
-        throw new Error(errorText || 'Refund failed');
-      }
-      const receipt = (await response.json()) as RefundReceipt;
-      onRefundComplete?.(receipt);
-      toast.success('Refund completed');
-      await refetch();
-    } catch (error) {
-      console.error(error);
-      toast.error(error instanceof Error ? error.message : 'Refund failed');
-    } finally {
-      setIsRefunding(false);
-    }
-  }, [apiKeyInput, baseUrl, onRefundComplete, refetch]);
-
   const formatSats = (msats: number) =>
     new Intl.NumberFormat('en-US').format(Math.floor(msats / 1000));
   const formatMsats = (msats: number) =>
@@ -162,11 +153,17 @@ export function KeyInfoDetails({
         </CardHeader>
         <CardContent>
           <div className='flex flex-col gap-2 sm:flex-row'>
-            <ApiKeyInput value={apiKeyInput} onApiKeyChange={handleKeyChange} />
+            <Input
+              value={apiKeyInput}
+              onChange={(e) => handleKeyChange(e.target.value)}
+              placeholder='sk-...'
+              className='font-mono text-sm'
+            />
             <div className='flex gap-2'>
               <Button
                 variant='outline'
                 size='icon'
+                className='h-10 w-10 shrink-0'
                 onClick={() => handleCopy(apiKeyInput)}
                 disabled={!apiKeyInput}
               >
@@ -177,22 +174,15 @@ export function KeyInfoDetails({
                 size='sm'
                 className='min-w-[80px] gap-1'
                 onClick={handleRefresh}
-                disabled={isFetching || !apiKeyInput}
-                type='button'
+                disabled={isRefreshing || !apiKeyInput}
               >
                 <RefreshCcw
-                  className={`h-8 w-4 ${isFetching ? 'animate-spin' : ''}`}
+                  className={`h-4 w-4 ${isRefreshing ? 'animate-spin' : ''}`}
                 />
-                {isFetching ? 'Syncing...' : 'Sync'}
+                {isRefreshing ? 'Syncing...' : 'Sync'}
               </Button>
             </div>
           </div>
-          {error && (
-            <Alert variant='destructive' className='mt-2'>
-              <AlertTitle>Error</AlertTitle>
-              <AlertDescription>{error}</AlertDescription>
-            </Alert>
-          )}
         </CardContent>
       </Card>
 
@@ -238,14 +228,6 @@ export function KeyInfoDetails({
                     {formatDate(walletInfo.validityDate)}
                   </span>
                 </div>
-              </CardContent>
-            </Card>
-
-            <Card>
-              <CardHeader className='pb-2'>
-                <CardTitle className='text-lg'>Infos</CardTitle>
-              </CardHeader>
-              <CardContent className='space-y-4'>
                 <div className='flex items-center justify-between'>
                   <span className='text-muted-foreground text-sm'>
                     Spendable Balance
@@ -254,6 +236,14 @@ export function KeyInfoDetails({
                     {formatSats(walletInfo.balanceMsats)} sats
                   </span>
                 </div>
+              </CardContent>
+            </Card>
+
+            <Card>
+              <CardHeader className='pb-2'>
+                <CardTitle className='text-lg'>Consumption</CardTitle>
+              </CardHeader>
+              <CardContent className='space-y-4'>
                 <div className='flex items-center justify-between'>
                   <span className='text-muted-foreground text-sm'>
                     Total Requests
@@ -400,16 +390,18 @@ export function KeyInfoDetails({
               </Card>
             )}
 
-          <div className='flex justify-center gap-4'>
+          <div className='flex justify-center'>
             <Button
-              onClick={handleRefund}
-              disabled={isRefunding || !apiKeyInput}
-              variant='destructive'
+              variant='ghost'
               size='sm'
-              className='gap-2'
+              onClick={handleRefresh}
+              disabled={isRefreshing}
+              className='text-muted-foreground'
             >
-              <Trash2 className='h-4 w-4' />
-              {isRefunding ? 'Processing...' : 'Refund Key'}
+              <RefreshCcw
+                className={`mr-2 h-3 w-3 ${isRefreshing ? 'animate-spin' : ''}`}
+              />
+              Last synced: {new Date().toLocaleTimeString()}
             </Button>
           </div>
         </>

ui/components/landing/key-info-display.tsx

@@ -1,198 +0,0 @@
-'use client';
-
-import {
-  Card,
-  CardContent,
-  CardHeader,
-  CardTitle,
-  CardDescription,
-} from '@/components/ui/card';
-import { Badge } from '@/components/ui/badge';
-import { Button } from '@/components/ui/button';
-import { Copy, RotateCcw } from 'lucide-react';
-import type { WalletSnapshot } from './key-info-details';
-import { toast } from 'sonner';
-
-interface KeyInfoDisplayProps {
-  walletInfo: WalletSnapshot;
-  onResetSpent?: (childKey: string) => Promise<void>;
-  isResetting?: string | null;
-}
-
-const formatSats = (msats: number) =>
-  new Intl.NumberFormat('en-US').format(Math.floor(msats / 1000));
-const formatMsats = (msats: number) =>
-  new Intl.NumberFormat('en-US').format(msats);
-const formatDate = (timestamp: number | null) =>
-  timestamp ? new Date(timestamp * 1000).toLocaleDateString() : 'Never';
-
-export function KeyInfoDisplay({
-  walletInfo,
-  onResetSpent,
-  isResetting,
-}: KeyInfoDisplayProps) {
-  const handleCopy = (value: string) => {
-    navigator.clipboard.writeText(value);
-    toast.success('Copied to clipboard');
-  };
-
-  return (
-    <div className='space-y-4'>
-      <div className='grid gap-4 md:grid-cols-2'>
-        <Card>
-          <CardHeader className='pb-2'>
-            <CardTitle className='text-lg'>Status & Identity</CardTitle>
-          </CardHeader>
-          <CardContent className='space-y-4'>
-            <div className='flex items-center justify-between'>
-              <span className='text-muted-foreground text-sm'>Type</span>
-              <Badge variant={walletInfo.isChild ? 'secondary' : 'default'}>
-                {walletInfo.isChild ? 'Child Key' : 'Parent Key'}
-              </Badge>
-            </div>
-            {walletInfo.parentKey && (
-              <div className='space-y-1'>
-                <span className='text-muted-foreground text-xs tracking-wider'>
-                  Parent Key
-                </span>
-                <div className='flex items-center gap-2'>
-                  <code className='bg-muted flex-1 rounded px-2 py-1 font-mono text-xs break-all'>
-                    {walletInfo.parentKey}
-                  </code>
-                  <Button
-                    variant='ghost'
-                    size='icon'
-                    className='h-8 w-8'
-                    onClick={() => handleCopy(walletInfo.parentKey!)}
-                  >
-                    <Copy className='h-4 w-4' />
-                  </Button>
-                </div>
-              </div>
-            )}
-            <div className='flex items-center justify-between'>
-              <span className='text-muted-foreground text-sm'>Validity</span>
-              <span className='text-sm font-medium'>
-                {formatDate(walletInfo.validityDate)}
-              </span>
-            </div>
-          </CardContent>
-        </Card>
-
-        <Card>
-          <CardHeader className='pb-2'>
-            <CardTitle className='text-lg'>Infos</CardTitle>
-          </CardHeader>
-          <CardContent className='space-y-4'>
-            <div className='flex items-center justify-between'>
-              <span className='text-muted-foreground text-sm'>
-                Spendable Balance
-              </span>
-              <span className='text-primary font-mono text-sm font-medium'>
-                {formatSats(walletInfo.balanceMsats)} sats
-              </span>
-            </div>
-            <div className='flex items-center justify-between'>
-              <span className='text-muted-foreground text-sm'>
-                Total Requests
-              </span>
-              <span className='font-mono text-sm font-medium'>
-                {walletInfo.totalRequests}
-              </span>
-            </div>
-            <div className='flex items-center justify-between'>
-              <span className='text-muted-foreground text-sm'>Total Spent</span>
-              <div className='text-right'>
-                <p className='font-mono text-sm font-medium'>
-                  {formatSats(walletInfo.totalSpent)} sats
-                </p>
-                <p className='text-muted-foreground font-mono text-[0.6rem]'>
-                  {formatMsats(walletInfo.totalSpent)} msats
-                </p>
-              </div>
-            </div>
-            {walletInfo.balanceLimit !== null && (
-              <div className='space-y-2'>
-                <div className='flex items-center justify-between'>
-                  <span className='text-muted-foreground text-sm'>
-                    Spend Limit
-                  </span>
-                  <span className='font-mono text-sm font-medium'>
-                    {formatSats(walletInfo.balanceLimit)} sats
-                  </span>
-                </div>
-                {walletInfo.balanceLimitReset && (
-                  <div className='flex items-center justify-between'>
-                    <span className='text-muted-foreground text-sm'>
-                      Reset Policy
-                    </span>
-                    <Badge variant='outline' className='capitalize'>
-                      {walletInfo.balanceLimitReset}
-                    </Badge>
-                  </div>
-                )}
-              </div>
-            )}
-          </CardContent>
-        </Card>
-      </div>
-
-      {!walletInfo.isChild &&
-        walletInfo.childKeys &&
-        walletInfo.childKeys.length > 0 && (
-          <Card>
-            <CardHeader>
-              <CardTitle className='text-lg'>
-                Child Keys ({walletInfo.childKeys.length})
-              </CardTitle>
-              <CardDescription>
-                Secondary keys using this account&apos;s balance
-              </CardDescription>
-            </CardHeader>
-            <CardContent>
-              <div className='space-y-4'>
-                {walletInfo.childKeys.map((ck) => (
-                  <div
-                    key={ck.api_key}
-                    className='space-y-3 rounded-lg border p-4'
-                  >
-                    <div className='flex items-center justify-between gap-4'>
-                      <code className='bg-muted flex-1 rounded px-2 py-1 font-mono text-xs break-all'>
-                        {ck.api_key}
-                      </code>
-                      <div className='flex gap-1'>
-                        <Button
-                          variant='ghost'
-                          size='icon'
-                          className='h-8 w-8'
-                          onClick={() => handleCopy(ck.api_key)}
-                        >
-                          <Copy className='h-4 w-4' />
-                        </Button>
-                        {onResetSpent && (
-                          <Button
-                            variant='ghost'
-                            size='icon'
-                            className='text-destructive h-8 w-8'
-                            title='Reset consumption'
-                            disabled={isResetting === ck.api_key}
-                            onClick={() => onResetSpent(ck.api_key)}
-                          >
-                            {isResetting === ck.api_key ? (
-                              <RotateCcw className='h-4 w-4 animate-spin' />
-                            ) : (
-                              <RotateCcw className='h-4 w-4' />
-                            )}
-                          </Button>
-                        )}
-                      </div>
-                    </div>
-                  </div>
-                ))}
-              </div>
-            </CardContent>
-          </Card>
-        )}
-    </div>
-  );
-}

ui/hooks/use-wallet-info.ts

@@ -1,38 +0,0 @@
-import { useQuery } from '@tanstack/react-query';
-import type { WalletSnapshot } from '@/components/landing/key-info-details';
-
-export function useWalletInfo(baseUrl: string, apiKey: string) {
-  return useQuery({
-    queryKey: ['walletInfo', baseUrl, apiKey],
-    queryFn: async (): Promise<WalletSnapshot> => {
-      const response = await fetch(`${baseUrl}/v1/balance/info`, {
-        headers: {
-          'Content-Type': 'application/json',
-          Authorization: `Bearer ${apiKey}`,
-        },
-      });
-
-      if (!response.ok) {
-        const errorText = await response.text();
-        throw new Error(errorText || 'Unable to load wallet info');
-      }
-
-      const payload = await response.json();
-      return {
-        apiKey: payload.api_key || apiKey,
-        balanceMsats: payload.balance ?? 0,
-        reservedMsats: payload.reserved ?? 0,
-        isChild: payload.is_child,
-        parentKey: payload.parent_key,
-        totalRequests: payload.total_requests,
-        totalSpent: payload.total_spent,
-        balanceLimit: payload.balance_limit,
-        balanceLimitReset: payload.balance_limit_reset,
-        validityDate: payload.validity_date,
-        childKeys: payload.child_keys,
-      };
-    },
-    enabled: !!baseUrl && !!apiKey,
-    staleTime: 5000, // Consider data stale after 5 seconds
-  });
-}