gloomberb 0.7.4 → 0.8.0

package/src/plugins/builtin/alerts/index.tsx

@@ -1,483 +1,29 @@
-import { useCallback, useEffect, useMemo, useState } from "react";
-import { TextAttributes } from "../../../ui";
-import {
-  DataTableView,
-  usePaneFooter,
-  type DataTableCell,
-  type DataTableColumn,
-  type DataTableKeyEvent,
-} from "../../../components";
-import type { GloomPlugin, GloomPluginContext, PaneProps } from "../../../types/plugin";
-import type { AlertCondition, AlertRule } from "./types";
-import { colors } from "../../../theme/colors";
-import { useMarketData, usePluginAppActions, usePluginConfigState } from "../../plugin-runtime";
+import type { GloomPlugin } from "../../../types/plugin";
+import { formatMarketPrice } from "../../../market-data/market/format";
 import {
   createAlert,
   evaluateAlert,
   formatAlertDescription,
-  serializeAlerts,
-  deserializeAlerts,
 } from "./alert-engine";
-import type { DataProvider } from "../../../types/data-provider";
-import type { Quote } from "../../../types/financials";
-import { formatMarketPrice } from "../../../utils/market-format";
-import { formatQuoteAgeWithSource } from "../../../utils/quote-time";
-
-const ALERTS_KEY = "alerts";
-const POLL_INTERVAL_MS = 30_000;
-const PANE_QUOTE_REFRESH_MS = 30_000;
-
-function relativeTime(ts: number): string {
-  const diff = Date.now() - ts;
-  const seconds = Math.floor(diff / 1000);
-  if (seconds < 60) return `${seconds}s ago`;
-  const minutes = Math.floor(seconds / 60);
-  if (minutes < 60) return `${minutes}m ago`;
-  const hours = Math.floor(minutes / 60);
-  if (hours < 24) return `${hours}h ago`;
-  const days = Math.floor(hours / 24);
-  return `${days}d ago`;
-}
-
-function normalizeAlertSymbol(value: string | null | undefined): string {
-  return value?.trim().toUpperCase() ?? "";
-}
-
-function createQuoteErrorMessage(symbol: string, error?: unknown): string {
-  if (error instanceof Error && error.message.trim()) return error.message;
-  return `No quote found for "${symbol}".`;
-}
-
-async function resolveAlertQuote(marketData: Pick<DataProvider, "getQuote">, symbol: string): Promise<Quote> {
-  const quote = await marketData.getQuote(symbol, "");
-  if (!quote || typeof quote.price !== "number" || !Number.isFinite(quote.price)) {
-    throw new Error(`No quote found for "${symbol}".`);
-  }
-  return quote;
-}
-
-function quoteAlertFields(quote: Quote, checkedAt = Date.now()): Pick<
-  AlertRule,
-  "lastCheckedPrice" | "lastCheckedAt" | "lastQuoteUpdatedAt" | "lastQuoteSource" | "lastQuoteProviderId"
-> & { lastCheckError?: undefined } {
-  return {
-    lastCheckedPrice: quote.price,
-    lastCheckedAt: checkedAt,
-    lastQuoteUpdatedAt: quote.lastUpdated,
-    lastQuoteSource: quote.dataSource,
-    lastQuoteProviderId: quote.providerId,
-    lastCheckError: undefined,
-  };
-}
-
-function quoteErrorAlertFields(error: string, checkedAt = Date.now()): Pick<
-  AlertRule,
-  "lastCheckedAt" | "lastCheckError"
-> {
-  return {
-    lastCheckedAt: checkedAt,
-    lastCheckError: error,
-  };
-}
-
-function formatCurrentPrice(alert: AlertRule, maxWidth = 9): string {
-  if (alert.lastCheckError) return "No quote";
-  return alert.lastCheckedPrice == null
-    ? "-"
-    : formatMarketPrice(alert.lastCheckedPrice, { maxWidth, minimumFractionDigits: 2 });
-}
-
-function formatAlertTargetPrice(alert: AlertRule, maxWidth = 9): string {
-  return formatMarketPrice(alert.targetPrice, { maxWidth, minimumFractionDigits: 2 });
-}
-
-function formatAlertDistance(alert: AlertRule): string {
-  const currentPrice = alert.lastCheckedPrice;
-  if (alert.lastCheckError) return "-";
-  if (currentPrice == null || !Number.isFinite(currentPrice) || currentPrice === 0) return "-";
-  const percent = ((alert.targetPrice - currentPrice) / currentPrice) * 100;
-  if (!Number.isFinite(percent)) return "-";
-  const abs = Math.abs(percent);
-  const decimals = abs < 10 ? 1 : 0;
-  return `${percent >= 0 ? "+" : ""}${percent.toFixed(decimals)}%`;
-}
-
-function formatQuoteChecked(alert: AlertRule): string {
-  if (alert.lastCheckError) return "No quote";
-  if (alert.lastQuoteUpdatedAt) {
-    return formatQuoteAgeWithSource({
-      lastUpdated: alert.lastQuoteUpdatedAt,
-      dataSource: alert.lastQuoteSource,
-    });
-  }
-  if (alert.lastCheckedAt) return relativeTime(alert.lastCheckedAt);
-  return "-";
-}
-
-function conditionLabel(condition: AlertCondition): string {
-  switch (condition) {
-    case "above":
-      return "Above";
-    case "below":
-      return "Below";
-    case "crosses":
-      return "Cross";
-  }
-}
-
-interface AlertCommandInput {
-  symbol: string;
-  condition: AlertCondition;
-  price: number;
-}
-
-function parseAlertCondition(value: string | undefined): AlertCondition | null {
-  const normalized = value?.trim().toLowerCase();
-  switch (normalized) {
-    case ">":
-    case "above":
-    case "over":
-    case "gt":
-      return "above";
-    case "<":
-    case "below":
-    case "under":
-    case "lt":
-      return "below";
-    case "x":
-    case "cross":
-    case "crosses":
-      return "crosses";
-    default:
-      return null;
-  }
-}
-
-function parseAlertShortcutValues(
-  input: string,
-  context?: { activeTicker: string | null },
-): Record<string, string> {
-  const parts = input.trim().split(/\s+/).filter(Boolean);
-  if (parts.length === 0) {
-    const activeTicker = normalizeAlertSymbol(context?.activeTicker);
-    return activeTicker ? { symbol: activeTicker } : {};
-  }
-  if (parts.length > 3) {
-    throw new Error("Use SA SYMBOL above|below|crosses PRICE.");
-  }
-
-  const values: Record<string, string> = {
-    symbol: normalizeAlertSymbol(parts[0]),
-  };
-
-  if (parts[1]) {
-    const condition = parseAlertCondition(parts[1]);
-    if (!condition) {
-      throw new Error("Use above, below, crosses, >, <, or x.");
-    }
-    values.condition = condition;
-  }
-
-  if (parts[2]) {
-    const price = Number.parseFloat(parts[2]!.replace(/^\$/, ""));
-    if (!Number.isFinite(price)) {
-      throw new Error("Use a numeric target price.");
-    }
-    values.price = String(price);
-  }
-
-  return values;
-}
-
-function parseAlertCommandValues(values?: Record<string, string>): AlertCommandInput | null {
-  const shortcut = values?.shortcut?.trim();
-  if (shortcut) {
-    values = {
-      ...values,
-      ...parseAlertShortcutValues(shortcut),
-    };
-  }
-
-  const symbol = values?.symbol?.trim().toUpperCase();
-  const condition = parseAlertCondition(values?.condition);
-  const priceStr = values?.price?.trim();
-  if (!symbol || !condition || !priceStr) return null;
-  const price = Number.parseFloat(priceStr);
-  if (!Number.isFinite(price)) return null;
-  return { symbol, condition, price };
-}
-
-type AlertColumnId =
-  | "status"
-  | "symbol"
-  | "current"
-  | "target"
-  | "away"
-  | "condition"
-  | "quote"
-  | "triggered"
-  | "rearm";
-
-type AlertColumn = DataTableColumn & { id: AlertColumnId };
-
-const ALERT_COLUMNS: AlertColumn[] = [
-  { id: "status", label: "State", width: 6, align: "left" },
-  { id: "symbol", label: "Symbol", width: 7, align: "left" },
-  { id: "current", label: "Current", width: 9, align: "right" },
-  { id: "target", label: "Target", width: 9, align: "right" },
-  { id: "away", label: "Away", width: 8, align: "right" },
-  { id: "condition", label: "Trigger", width: 7, align: "left" },
-  { id: "quote", label: "Quote", width: 8, align: "left" },
-  { id: "triggered", label: "Alerted", width: 8, align: "left" },
-  { id: "rearm", label: "", width: 6, align: "left" },
-];
-
-const ALERT_TABLE_CONTENT_WIDTH = ALERT_COLUMNS.reduce(
-  (sum, column) => sum + column.width + 1,
-  2,
-);
-
-export function AlertsPane({ focused, width, height, close }: PaneProps) {
-  const [alertsJson, setAlertsJson] = usePluginConfigState<string>(ALERTS_KEY, "[]");
-  const marketData = useMarketData();
-  const { openPluginCommandWorkflow } = usePluginAppActions();
-  const [selectedIdx, setSelectedIdx] = useState(0);
-  const marketDataId = marketData?.id ?? null;
-  const showHorizontalScrollbar = ALERT_TABLE_CONTENT_WIDTH > width;
-
-  const { alerts, rows, activeCount, triggeredCount } = useMemo(() => {
-    const parsed = deserializeAlerts(alertsJson);
-    const activeAlerts = parsed.filter((a) => a.status === "active");
-    const triggeredAlerts = parsed
-      .filter((a) => a.status === "triggered")
-      .sort((a, b) => (b.triggeredAt ?? 0) - (a.triggeredAt ?? 0));
-
-    return {
-      alerts: parsed,
-      rows: [...activeAlerts, ...triggeredAlerts],
-      activeCount: activeAlerts.length,
-      triggeredCount: triggeredAlerts.length,
-    };
-  }, [alertsJson]);
-
-  const saveAlerts = useCallback((next: AlertRule[] | ((current: AlertRule[]) => AlertRule[])) => {
-    setAlertsJson((currentJson) => {
-      const current = deserializeAlerts(currentJson);
-      const resolved = typeof next === "function" ? next(current) : next;
-      return serializeAlerts(resolved);
-    });
-  }, [setAlertsJson]);
-
-  const deleteAlert = useCallback((id: string) => {
-    saveAlerts(alerts.filter((a) => a.id !== id));
-    setSelectedIdx((prev) => Math.max(0, Math.min(prev, rows.length - 2)));
-  }, [alerts, rows.length, saveAlerts]);
-
-  const rearmAlert = useCallback((id: string) => {
-    saveAlerts(
-      alerts.map((a) =>
-        a.id === id ? { ...a, status: "active" as const, triggeredAt: undefined, lastCheckError: undefined } : a,
-      ),
-    );
-  }, [alerts, saveAlerts]);
-
-  const startAddAlert = useCallback(() => {
-    openPluginCommandWorkflow("set-alert");
-  }, [openPluginCommandWorkflow]);
-
-  const deleteSelectedAlert = useCallback(() => {
-    const selected = rows[selectedIdx];
-    if (selected) deleteAlert(selected.id);
-  }, [deleteAlert, rows, selectedIdx]);
-
-  useEffect(() => {
-    if (!marketData || rows.length === 0) return;
-    const now = Date.now();
-    const dueAlerts = rows.filter((alert) => (
-      !alert.lastCheckedAt || now - alert.lastCheckedAt > PANE_QUOTE_REFRESH_MS
-    ));
-    if (dueAlerts.length === 0) return;
-
-    let cancelled = false;
-    void Promise.all(dueAlerts.map(async (alert) => {
-      try {
-        const quote = await resolveAlertQuote(marketData, alert.symbol);
-        return { id: alert.id, patch: quoteAlertFields(quote) };
-      } catch (error) {
-        return {
-          id: alert.id,
-          patch: quoteErrorAlertFields(createQuoteErrorMessage(alert.symbol, error)),
-        };
-      }
-    })).then((updates) => {
-      if (cancelled || updates.length === 0) return;
-      const patches = new Map(updates.map((update) => [update.id, update.patch]));
-      saveAlerts((current) => current.map((alert) => {
-        const patch = patches.get(alert.id);
-        return patch ? { ...alert, ...patch } : alert;
-      }));
-    });
-
-    return () => {
-      cancelled = true;
-    };
-  }, [marketDataId, rows, saveAlerts]);
-
-  usePaneFooter("alerts", () => ({
-    info: [
-      {
-        id: "active",
-        parts: [
-          { text: String(activeCount), tone: "value", bold: true },
-          { text: "active", tone: "label" },
-        ],
-      },
-      ...(triggeredCount > 0 ? [{
-        id: "triggered",
-        parts: [
-          { text: String(triggeredCount), tone: "warning" as const, bold: true },
-          { text: "triggered", tone: "label" as const },
-        ],
-      }] : []),
-    ],
-    hints: [
-      { id: "add", key: "a", label: "dd alert", onPress: startAddAlert },
-      { id: "delete", key: "d", label: "elete", onPress: deleteSelectedAlert, disabled: rows.length === 0 },
-    ],
-  }), [
-    activeCount,
-    deleteSelectedAlert,
-    rows.length,
-    startAddAlert,
-    triggeredCount,
-  ]);
-
-  useEffect(() => {
-    setSelectedIdx((prev) => (rows.length === 0 ? 0 : Math.min(prev, rows.length - 1)));
-  }, [rows.length]);
-
-  const handleTableKeyDown = useCallback((event: DataTableKeyEvent) => {
-    if (event.name === "d") {
-      event.preventDefault?.();
-      deleteSelectedAlert();
-      return true;
-    }
-    if (event.name === "a" || event.name === "n") {
-      event.preventDefault?.();
-      startAddAlert();
-      return true;
-    }
-    if (event.name === "escape") {
-      event.preventDefault?.();
-      close?.();
-      return true;
-    }
-    return false;
-  }, [close, deleteSelectedAlert, startAddAlert]);
-
-  const renderCell = useCallback((
-    alert: AlertRule,
-    column: AlertColumn,
-    _index: number,
-    rowState: { selected: boolean; hovered: boolean },
-  ): DataTableCell => {
-    const selectedColor = rowState.selected ? colors.selectedText : undefined;
-    const actionMouseDown = (handler: () => void) => (event: any) => {
-      event.preventDefault?.();
-      event.stopPropagation?.();
-      handler();
-    };
-
-    switch (column.id) {
-      case "status":
-        return {
-          text: alert.status === "triggered" ? "Trig" : "Active",
-          color: selectedColor ?? (alert.status === "triggered" ? colors.positive : colors.textDim),
-          attributes: alert.status === "triggered" ? TextAttributes.BOLD : TextAttributes.NONE,
-        };
-      case "symbol":
-        return {
-          text: alert.symbol,
-          color: selectedColor ?? colors.textBright,
-          attributes: TextAttributes.BOLD,
-        };
-      case "current":
-        return {
-          text: formatCurrentPrice(alert, column.width),
-          color: selectedColor ?? (alert.lastCheckError ? colors.negative : colors.text),
-        };
-      case "target":
-        return { text: formatAlertTargetPrice(alert, column.width), color: selectedColor };
-      case "away":
-        return {
-          text: formatAlertDistance(alert),
-          color: selectedColor ?? colors.textDim,
-        };
-      case "condition":
-        return {
-          text: conditionLabel(alert.condition),
-          color: selectedColor,
-        };
-      case "quote":
-        return {
-          text: formatQuoteChecked(alert),
-          color: selectedColor ?? colors.textDim,
-        };
-      case "triggered":
-        return {
-          text: alert.triggeredAt ? relativeTime(alert.triggeredAt) : "-",
-          color: selectedColor ?? colors.textDim,
-        };
-      case "rearm":
-        return alert.status === "triggered"
-          ? {
-              text: "Re-arm",
-              color: selectedColor ?? colors.textBright,
-              onMouseDown: actionMouseDown(() => rearmAlert(alert.id)),
-            }
-          : { text: "-", color: selectedColor ?? colors.textDim };
-    }
-  }, [rearmAlert]);
-
-  return (
-    <DataTableView<AlertRule, AlertColumn>
-      focused={focused}
-      selectedIndex={selectedIdx}
-      onRootKeyDown={handleTableKeyDown}
-      rootWidth={width}
-      rootHeight={height}
-      rootBackgroundColor={colors.bg}
-      columns={ALERT_COLUMNS}
-      items={rows}
-      sortColumnId={null}
-      sortDirection="asc"
-      onHeaderClick={() => {}}
-      getItemKey={(alert) => alert.id}
-      isSelected={(_alert, index) => index === selectedIdx}
-      onSelect={(_alert, index) => setSelectedIdx(index)}
-      onActivate={(alert) => {
-        if (alert.status === "triggered") rearmAlert(alert.id);
-      }}
-      renderCell={renderCell}
-      emptyStateTitle="No alerts"
-      emptyStateHint="Use the action bar to create one."
-      showHorizontalScrollbar={showHorizontalScrollbar}
-    />
-  );
-}
+import {
+  parseAlertCommandValues,
+  parseAlertShortcutValues,
+} from "./command";
+import { POLL_INTERVAL_MS } from "./constants";
+import { AlertsPane } from "./pane";
+import {
+  createQuoteErrorMessage,
+  quoteAlertFields,
+  quoteErrorAlertFields,
+  resolveAlertQuote,
+} from "./quotes";
+import {
+  loadAlerts,
+  saveAlerts,
+} from "./storage";
 
 let pollInterval: ReturnType<typeof setInterval> | null = null;
 
-function loadAlerts(ctx: GloomPluginContext): AlertRule[] {
-  const json = ctx.configState.get<string>(ALERTS_KEY);
-  if (!json) return [];
-  return deserializeAlerts(json);
-}
-
-function saveAlerts(ctx: GloomPluginContext, alerts: AlertRule[]): void {
-  ctx.configState.set(ALERTS_KEY, serializeAlerts(alerts));
-}
-
 export const alertsPlugin: GloomPlugin = {
   id: "alerts",
   name: "Alerts",
@@ -544,7 +90,6 @@ export const alertsPlugin: GloomPlugin = {
       },
     });
 
-    // Polling engine
     const poll = async () => {
       const alerts = loadAlerts(ctx);
       if (alerts.length === 0) return;