gloomberb 0.1.0 → 0.3.1

package/src/plugins/ibkr/config.ts

@@ -0,0 +1,262 @@
+import type { BrokerConfigField } from "../../types/broker";
+import type { IbkrGatewayConfig, ResolvedIbkrGatewayConnection } from "./gateway-service";
+
+export const IBKR_STATEMENT_URL = "https://gdcdyn.interactivebrokers.com/Universal/servlet/FlexStatementService.SendRequest";
+
+export type IbkrConnectionMode = "flex" | "gateway";
+export type IbkrGatewaySetupMode = "auto" | "manual";
+
+export interface FlexQueryConfig {
+  token: string;
+  queryId: string;
+  endpoint?: string;
+}
+
+export interface IbkrConfig {
+  connectionMode: IbkrConnectionMode;
+  gatewaySetupMode: IbkrGatewaySetupMode;
+  flex: FlexQueryConfig;
+  gateway: IbkrGatewayConfig;
+}
+
+const DEFAULT_FLEX_CONFIG: FlexQueryConfig = {
+  token: "",
+  queryId: "",
+  endpoint: IBKR_STATEMENT_URL,
+};
+
+export const DEFAULT_GATEWAY_HOST = "127.0.0.1";
+
+export const DEFAULT_GATEWAY_CONFIG: IbkrGatewayConfig = {
+  host: DEFAULT_GATEWAY_HOST,
+  marketDataType: "auto",
+};
+
+export const IBKR_CONFIG_FIELDS: BrokerConfigField[] = [
+  {
+    key: "connectionMode",
+    label: "Connection Mode",
+    type: "select",
+    required: true,
+    options: [
+      { label: "Flex (read-only)", value: "flex", description: "Syncs via IBKR's Flex API. No software needed." },
+      { label: "Gateway / TWS", value: "gateway", description: "Use IB Gateway or TWS for live data and trading." },
+    ],
+  },
+  {
+    key: "token",
+    label: "Flex Token",
+    type: "password",
+    required: true,
+    placeholder: "Your Flex Web Service token",
+    dependsOn: { key: "connectionMode", value: "flex" },
+  },
+  {
+    key: "queryId",
+    label: "Query ID",
+    type: "text",
+    required: true,
+    placeholder: "Numeric Flex Query ID",
+    dependsOn: { key: "connectionMode", value: "flex" },
+  },
+  {
+    key: "endpoint",
+    label: "Endpoint",
+    type: "text",
+    required: false,
+    placeholder: IBKR_STATEMENT_URL,
+    dependsOn: { key: "connectionMode", value: "flex" },
+  },
+  {
+    key: "gatewaySetupMode",
+    label: "Connection Setup",
+    type: "select",
+    required: true,
+    defaultValue: "auto",
+    options: [
+      {
+        label: "Automatic",
+        value: "auto",
+        description: "Detect the local IBKR API port and choose a client ID automatically.",
+      },
+      {
+        label: "Manual",
+        value: "manual",
+        description: "Specify the host and socket port yourself.",
+      },
+    ],
+    dependsOn: { key: "connectionMode", value: "gateway" },
+  },
+  {
+    key: "host",
+    label: "Gateway Host",
+    type: "text",
+    required: true,
+    placeholder: DEFAULT_GATEWAY_HOST,
+    defaultValue: DEFAULT_GATEWAY_HOST,
+    dependsOn: { key: "gatewaySetupMode", value: "manual" },
+  },
+  {
+    key: "port",
+    label: "Gateway Port",
+    type: "number",
+    required: true,
+    placeholder: "4001, 4002, 7496, or 7497",
+    dependsOn: { key: "gatewaySetupMode", value: "manual" },
+  },
+];
+
+function coerceOptionalNumber(value: unknown): number | undefined {
+  if (typeof value === "number" && Number.isFinite(value)) return value;
+  if (typeof value === "string" && value.trim()) {
+    const parsed = Number(value);
+    if (Number.isFinite(parsed)) return parsed;
+  }
+  return undefined;
+}
+
+function coerceString(value: unknown, fallback = ""): string {
+  return typeof value === "string" ? value : fallback;
+}
+
+function getMode(value: unknown): IbkrConnectionMode {
+  return value === "gateway" ? "gateway" : "flex";
+}
+
+function getGatewaySetupMode(value: unknown): IbkrGatewaySetupMode | undefined {
+  return value === "manual" || value === "auto" ? value : undefined;
+}
+
+function inferGatewaySetupMode(
+  connectionMode: IbkrConnectionMode,
+  input: Record<string, unknown>,
+  nestedGateway: Record<string, unknown>,
+): IbkrGatewaySetupMode {
+  if (connectionMode !== "gateway") return "auto";
+
+  const explicit = getGatewaySetupMode(input.gatewaySetupMode ?? nestedGateway.setupMode);
+  if (explicit) return explicit;
+
+  const rawHost = coerceString(nestedGateway.host ?? input.host, DEFAULT_GATEWAY_HOST);
+  const rawPort = coerceOptionalNumber(nestedGateway.port ?? input.port);
+  const rawClientId = coerceOptionalNumber(nestedGateway.clientId ?? input.clientId);
+
+  return rawHost !== DEFAULT_GATEWAY_HOST || rawPort != null || rawClientId != null
+    ? "manual"
+    : "auto";
+}
+
+export function normalizeIbkrConfig(raw?: Record<string, unknown>): IbkrConfig {
+  const input = raw ?? {};
+  const nestedFlex = typeof input.flex === "object" && input.flex ? input.flex as Record<string, unknown> : {};
+  const nestedGateway = typeof input.gateway === "object" && input.gateway ? input.gateway as Record<string, unknown> : {};
+  const inferredMode = input.connectionMode
+    ?? (input.host || input.port || input.clientId || nestedGateway.host || nestedGateway.port || nestedGateway.clientId
+      ? "gateway"
+      : "flex");
+  const connectionMode = getMode(inferredMode);
+  const gatewaySetupMode = inferGatewaySetupMode(connectionMode, input, nestedGateway);
+  const host = coerceString(nestedGateway.host ?? input.host, DEFAULT_GATEWAY_HOST);
+  const port = coerceOptionalNumber(nestedGateway.port ?? input.port);
+  const clientId = coerceOptionalNumber(nestedGateway.clientId ?? input.clientId);
+
+  return {
+    connectionMode,
+    gatewaySetupMode,
+    flex: {
+      token: coerceString(nestedFlex.token ?? input.token),
+      queryId: coerceString(nestedFlex.queryId ?? input.queryId),
+      endpoint: coerceString(nestedFlex.endpoint ?? input.endpoint, DEFAULT_FLEX_CONFIG.endpoint),
+    },
+    gateway: {
+      host,
+      port: gatewaySetupMode === "manual" ? port : undefined,
+      clientId: gatewaySetupMode === "manual" ? clientId : undefined,
+      lastSuccessfulPort: coerceOptionalNumber(nestedGateway.lastSuccessfulPort ?? input.lastSuccessfulPort),
+      lastSuccessfulClientId: coerceOptionalNumber(nestedGateway.lastSuccessfulClientId ?? input.lastSuccessfulClientId),
+      marketDataType: coerceString(
+        nestedGateway.marketDataType ?? input.marketDataType,
+        DEFAULT_GATEWAY_CONFIG.marketDataType,
+      ) as IbkrGatewayConfig["marketDataType"],
+    },
+  };
+}
+
+export function buildIbkrConfigFromValues(values: Record<string, unknown>): IbkrConfig {
+  const normalized = normalizeIbkrConfig(values);
+  return {
+    connectionMode: normalized.connectionMode,
+    gatewaySetupMode: normalized.gatewaySetupMode,
+    flex: normalized.flex,
+    gateway: normalized.gateway,
+  };
+}
+
+export function getGatewayConfig(raw?: Record<string, unknown>): IbkrGatewayConfig {
+  return normalizeIbkrConfig(raw).gateway;
+}
+
+export function getFlexConfig(raw?: Record<string, unknown>): FlexQueryConfig {
+  return normalizeIbkrConfig(raw).flex;
+}
+
+export function isGatewayConfigured(raw?: Record<string, unknown>): boolean {
+  const config = normalizeIbkrConfig(raw);
+  return config.connectionMode === "gateway"
+    && !!config.gateway.host
+    && (config.gatewaySetupMode !== "manual" || Number.isFinite(config.gateway.port));
+}
+
+export function isFlexConfigured(raw?: Record<string, unknown>): boolean {
+  const config = normalizeIbkrConfig(raw);
+  return config.connectionMode === "flex"
+    && !!config.flex.token
+    && !!config.flex.queryId;
+}
+
+export function buildPersistedIbkrGatewayConfig(
+  raw: Record<string, unknown> | undefined,
+  resolved: ResolvedIbkrGatewayConnection,
+): Record<string, unknown> | null {
+  const normalized = normalizeIbkrConfig(raw);
+  if (normalized.connectionMode !== "gateway") return null;
+
+  if (
+    normalized.gateway.lastSuccessfulPort === resolved.port
+    && normalized.gateway.lastSuccessfulClientId === resolved.clientId
+  ) {
+    return null;
+  }
+
+  const gateway: Record<string, unknown> = {
+    host: normalized.gateway.host,
+    marketDataType: normalized.gateway.marketDataType,
+    lastSuccessfulPort: resolved.port,
+    lastSuccessfulClientId: resolved.clientId,
+  };
+
+  if (normalized.gateway.port != null) gateway.port = normalized.gateway.port;
+  if (normalized.gateway.clientId != null) gateway.clientId = normalized.gateway.clientId;
+
+  return {
+    connectionMode: normalized.connectionMode,
+    gatewaySetupMode: normalized.gatewaySetupMode,
+    flex: normalized.flex,
+    gateway,
+  };
+}
+
+export function getIbkrConfigIdentity(raw?: Record<string, unknown>): string {
+  const normalized = normalizeIbkrConfig(raw);
+  return JSON.stringify({
+    connectionMode: normalized.connectionMode,
+    gatewaySetupMode: normalized.gatewaySetupMode,
+    flex: normalized.flex,
+    gateway: {
+      host: normalized.gateway.host,
+      port: normalized.gateway.port ?? null,
+      clientId: normalized.gateway.clientId ?? null,
+      marketDataType: normalized.gateway.marketDataType ?? "auto",
+    },
+  });
+}

package/src/plugins/ibkr/dialogs.tsx

@@ -0,0 +1,83 @@
+import { TextAttributes } from "@opentui/core";
+import type { InputRenderable } from "@opentui/core";
+import { useEffect, useRef, useState } from "react";
+import { useDialogKeyboard, type PromptContext } from "@opentui-ui/dialog/react";
+import type { WizardStep } from "../../types/plugin";
+import { colors } from "../../theme/colors";
+
+export function InputDialog({ resolve, step }: PromptContext<string> & { step: WizardStep }) {
+  const inputRef = useRef<InputRenderable>(null);
+  const [value, setValue] = useState("");
+
+  useEffect(() => {
+    inputRef.current?.focus();
+  }, []);
+
+  return (
+    <box flexDirection="column">
+      <text attributes={TextAttributes.BOLD} fg={colors.text}>{step.label}</text>
+      <box height={1} />
+      {step.body?.map((line, index) => (
+        <text key={index} fg={colors.textDim}>{line || " "}</text>
+      ))}
+      <box height={1} />
+      <input
+        ref={inputRef}
+        focused
+        placeholder={step.placeholder || ""}
+        textColor={colors.text}
+        placeholderColor={colors.textDim}
+        backgroundColor={colors.bg}
+        onInput={(nextValue) => setValue(nextValue)}
+        onChange={(nextValue) => setValue(nextValue)}
+        onSubmit={() => resolve(value.trim())}
+      />
+    </box>
+  );
+}
+
+export function ChoiceDialog({
+  resolve,
+  dialogId,
+  title,
+  choices,
+}: PromptContext<string> & { title: string; choices: Array<{ id: string; label: string; desc: string }> }) {
+  const [index, setIndex] = useState(0);
+
+  useDialogKeyboard((event) => {
+    event.stopPropagation();
+    if (event.name === "up" || event.name === "k") setIndex((current) => Math.max(0, current - 1));
+    else if (event.name === "down" || event.name === "j") setIndex((current) => Math.min(choices.length - 1, current + 1));
+    else if (event.name === "return") resolve(choices[index]!.id);
+    else if (event.name === "escape") resolve("");
+  }, dialogId);
+
+  return (
+    <box flexDirection="column">
+      <text attributes={TextAttributes.BOLD} fg={colors.text}>{title}</text>
+      <box height={1} />
+      {choices.map((choice, choiceIndex) => {
+        const selected = choiceIndex === index;
+        return (
+          <box
+            key={choice.id}
+            flexDirection="row"
+            height={1}
+            backgroundColor={selected ? colors.selected : colors.bg}
+            onMouseMove={() => setIndex(choiceIndex)}
+            onMouseDown={() => resolve(choice.id)}
+          >
+            <text fg={selected ? colors.selectedText : colors.textDim}>{selected ? "▸ " : "  "}</text>
+            <text fg={selected ? colors.text : colors.textDim} attributes={selected ? TextAttributes.BOLD : 0}>
+              {choice.label}
+            </text>
+          </box>
+        );
+      })}
+      <box height={1} />
+      <text fg={colors.textDim}>{choices[index]?.desc || ""}</text>
+      <box height={1} />
+      <text fg={colors.textMuted}>↑↓ choose · Enter/click select · Esc cancel</text>
+    </box>
+  );
+}

package/src/plugins/ibkr/flex.test.ts

@@ -0,0 +1,107 @@
+import { describe, expect, test } from "bun:test";
+import { parseFlexAccounts, parseFlexPositions } from "./flex";
+
+describe("parseFlexPositions", () => {
+  test("parses option positions with broker contract metadata", () => {
+    const xml = `
+      <FlexQueryResponse>
+        <OpenPositions>
+          <OpenPosition accountId="DU12345" symbol="SPY  260619C00500000" description="SPY Jun19'26 500 Call" assetCategory="OPT" position="2" costBasisPrice="4.25" currency="USD" exchange="SMART" conid="123456" listingExchange="SMART" multiplier="100" expiry="20260619" strike="500" putCall="CALL" localSymbol="SPY  260619C00500000" tradingClass="SPY" />
+        </OpenPositions>
+      </FlexQueryResponse>
+    `;
+
+    const positions = parseFlexPositions(xml);
+    expect(positions).toHaveLength(1);
+    expect(positions[0]?.ticker).toBe("SPY  260619C00500000");
+    expect(positions[0]?.brokerContract).toEqual({
+      brokerId: "ibkr",
+      conId: 123456,
+      symbol: "SPY  260619C00500000",
+      localSymbol: "SPY  260619C00500000",
+      secType: "OPT",
+      exchange: "SMART",
+      primaryExchange: "SMART",
+      currency: "USD",
+      lastTradeDateOrContractMonth: "20260619",
+      right: "C",
+      strike: 500,
+      multiplier: "100",
+      tradingClass: "SPY",
+    });
+  });
+});
+
+describe("parseFlexAccounts", () => {
+  test("parses cash balances and summary values from a flex statement", () => {
+    const xml = `
+      <FlexQueryResponse>
+        <FlexStatements count="1">
+          <FlexStatement accountId="DU12345" fromDate="20260327" toDate="20260327" whenGenerated="20260328;102707">
+            <ChangeInNAV accountId="DU12345" acctAlias="Main" currency="USD" endingValue="764713.626876249" />
+            <CashReport>
+              <CashReportCurrency accountId="DU12345" currency="BASE_SUMMARY" endingCash="-1050953.720462251" endingSettledCash="-917604.448220862" />
+            </CashReport>
+            <FxPositions>
+              <FxPosition accountId="DU12345" assetCategory="CASH" functionalCurrency="USD" fxCurrency="USD" quantity="-303029.144938754" value="-303029.144938754" />
+              <FxPosition accountId="DU12345" assetCategory="CASH" functionalCurrency="USD" fxCurrency="EUR" quantity="-351957.025" value="-405102.535775" />
+            </FxPositions>
+          </FlexStatement>
+        </FlexStatements>
+      </FlexQueryResponse>
+    `;
+
+    const accounts = parseFlexAccounts(xml);
+    expect(accounts).toHaveLength(1);
+    expect(accounts[0]).toMatchObject({
+      accountId: "DU12345",
+      name: "Main",
+      currency: "USD",
+      source: "flex",
+      netLiquidation: 764713.626876249,
+      totalCashValue: -1050953.720462251,
+      settledCash: -917604.448220862,
+      cashBalances: [
+        {
+          currency: "USD",
+          quantity: -303029.144938754,
+          baseValue: -303029.144938754,
+          baseCurrency: "USD",
+        },
+        {
+          currency: "EUR",
+          quantity: -351957.025,
+          baseValue: -405102.535775,
+          baseCurrency: "USD",
+        },
+      ],
+    });
+    expect(accounts[0]?.updatedAt).toBe(new Date(2026, 2, 28, 10, 27, 7).getTime());
+  });
+
+  test("handles missing cash sections gracefully", () => {
+    const xml = `
+      <FlexQueryResponse>
+        <FlexStatements count="1">
+          <FlexStatement accountId="DU12345" fromDate="20260327" toDate="20260327">
+            <ChangeInNAV accountId="DU12345" currency="USD" endingValue="12345.67" />
+          </FlexStatement>
+        </FlexStatements>
+      </FlexQueryResponse>
+    `;
+
+    expect(parseFlexAccounts(xml)).toEqual([
+      {
+        accountId: "DU12345",
+        name: "DU12345",
+        currency: "USD",
+        source: "flex",
+        updatedAt: new Date(2026, 2, 27).getTime(),
+        netLiquidation: 12345.67,
+        totalCashValue: undefined,
+        settledCash: undefined,
+        cashBalances: undefined,
+      },
+    ]);
+  });
+});