gloomberb 0.10.0 → 0.10.2

package/src/cli/commands/system.ts

@@ -7,6 +7,7 @@ import { createAlert, deserializeAlerts, serializeAlerts } from "../../plugins/b
 import type { AlertCondition } from "../../plugins/builtin/alerts/types";
 import type { CliCommandDef } from "../../types/plugin";
 import { debugLog, type LogLevel } from "../../utils/debug-log";
+import { withCliServices, withConfigData } from "../context";
 import { parsePositiveInt, requireArg, takeOption } from "./command-utils";
 
 const ALERTS_PLUGIN_ID = "alerts";
@@ -51,9 +52,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
     execute: async (_args, ctx) => {
       let dataDir: string | null = null;
       try {
-        const config = await ctx.initConfigData();
-        dataDir = config.dataDir;
-        config.persistence.close();
+        dataDir = await withConfigData(ctx, ({ dataDir: configuredDataDir }) => configuredDataDir);
       } catch {
         dataDir = null;
       }
@@ -74,17 +73,15 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
     description: "Check local CLI runtime, config, plugins, and capability health",
     execute: async (_args, ctx) => {
       const checks: Array<{ check: string; status: string; detail: string }> = [];
-      let services: Awaited<ReturnType<typeof ctx.initServices>> | null = null;
       try {
-        services = await ctx.initServices();
-        checks.push({ check: "config", status: "ok", detail: services.dataDir });
-        checks.push({ check: "database", status: "ok", detail: join(services.dataDir, ".gloomberb-cache.db") });
-        checks.push({ check: "plugins", status: "ok", detail: String(services.services.pluginRegistry.allPlugins.size) });
-        checks.push({ check: "capabilities", status: "ok", detail: String(services.services.pluginRegistry.capabilities.manifests().length) });
+        await withCliServices(ctx, async (services) => {
+          checks.push({ check: "config", status: "ok", detail: services.dataDir });
+          checks.push({ check: "database", status: "ok", detail: join(services.dataDir, ".gloomberb-cache.db") });
+          checks.push({ check: "plugins", status: "ok", detail: String(services.services.pluginRegistry.allPlugins.size) });
+          checks.push({ check: "capabilities", status: "ok", detail: String(services.services.pluginRegistry.capabilities.manifests().length) });
+        });
       } catch (error) {
         checks.push({ check: "runtime", status: "error", detail: error instanceof Error ? error.message : String(error) });
-      } finally {
-        services?.destroy();
       }
       ctx.printResult({ data: checks }, {
         columns: [
@@ -102,8 +99,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
     help: { usage: ["config list", "config get <key>", "config set <key> <value>"] },
     execute: async (args, ctx) => {
       const action = args[0] ?? "list";
-      const context = await ctx.initConfigData();
-      try {
+      await withConfigData(ctx, async (context) => {
         const safeConfig: Record<string, unknown> = {
           dataDir: context.config.dataDir,
           baseCurrency: context.config.baseCurrency,
@@ -146,9 +142,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
           return;
         }
         ctx.fail("Usage: gloomberb config list|get|set");
-      } finally {
-        context.persistence.close();
-      }
+      });
     },
   };
 
@@ -158,8 +152,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
     help: { usage: ["cache status", "cache clear [namespace]"] },
     execute: async (args, ctx) => {
       const action = args[0] ?? "status";
-      const services = await ctx.initServices();
-      try {
+      await withCliServices(ctx, async (services) => {
         if (action === "status") {
           ctx.printResult({ data: [resourceCacheStats(services)] });
           return;
@@ -173,9 +166,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
           return;
         }
         ctx.fail("Usage: gloomberb cache status|clear");
-      } finally {
-        services.destroy();
-      }
+      });
     },
   };
 
@@ -185,8 +176,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
     description: "Inspect provider/source capability status",
     help: { usage: ["provider status"] },
     execute: async (_args, ctx) => {
-      const services = await ctx.initServices();
-      try {
+      await withCliServices(ctx, async (services) => {
         const disabledSources = new Set(services.config.disabledSources ?? []);
         const rows = services.services.pluginRegistry.capabilities.manifests().map((manifest) => ({
           id: manifest.sourceId ?? manifest.id,
@@ -203,9 +193,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
             { key: "operations", header: "Operations" },
           ],
         });
-      } finally {
-        services.destroy();
-      }
+      });
     },
   };
 
@@ -215,8 +203,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
     help: { usage: ["plugin list", "plugin info <id>", "plugin enable <id>", "plugin disable <id>", "plugin doctor"] },
     execute: async (args, ctx) => {
       const action = args[0] ?? "list";
-      const services = await ctx.initServices();
-      try {
+      await withCliServices(ctx, async (services) => {
         const pluginRows = () => [...services.services.pluginRegistry.allPlugins.values()].map((plugin) => ({
           id: plugin.id,
           name: plugin.name,
@@ -255,9 +242,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
           return;
         }
         ctx.fail("Usage: gloomberb plugin list|info|enable|disable|doctor");
-      } finally {
-        services.destroy();
-      }
+      });
     },
   };
 
@@ -265,8 +250,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
     name: "layout",
     description: "Inspect saved layouts",
     execute: async (_args, ctx) => {
-      const config = await ctx.initConfigData();
-      try {
+      await withConfigData(ctx, async (config) => {
         const rows = config.config.layouts.map((layout, index) => ({
           index,
           active: index === config.config.activeLayoutIndex,
@@ -276,9 +260,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
           detached: layout.layout.detached.length,
         }));
         ctx.printResult({ data: rows });
-      } finally {
-        config.persistence.close();
-      }
+      });
     },
   };
 
@@ -287,8 +269,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
     description: "Inspect pane and pane-template inventory",
     help: { usage: ["pane list"] },
     execute: async (_args, ctx) => {
-      const services = await ctx.initServices();
-      try {
+      await withCliServices(ctx, async (services) => {
         const rows = [
           ...[...services.services.pluginRegistry.panes.entries()].map(([id, pane]) => ({
             kind: "pane",
@@ -311,9 +292,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
             { key: "owner", header: "Owner" },
           ],
         });
-      } finally {
-        services.destroy();
-      }
+      });
     },
   };
 
@@ -322,9 +301,8 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
     description: "Read and mutate ticker or quick notes",
     help: { usage: ["notes show <symbol>", "notes set <symbol> TEXT", "notes delete <symbol>", "notes quick list"] },
     execute: async (args, ctx) => {
-      const config = await ctx.initConfigData();
-      const notes = new NotesFiles(config.dataDir);
-      try {
+      await withConfigData(ctx, async (config) => {
+        const notes = new NotesFiles(config.dataDir);
         const action = args[0] ?? "list";
         if (action === "quick") {
           const subaction = args[1] ?? "list";
@@ -352,9 +330,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
           return;
         }
         ctx.fail("Usage: gloomberb notes show|set|delete|quick");
-      } finally {
-        config.persistence.close();
-      }
+      });
     },
   };
 
@@ -365,8 +341,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
     help: { usage: ["alerts list", "alerts add <symbol> <above|below|crosses> <price>", "alerts delete <id>", "alerts rearm <id>"] },
     execute: async (args, ctx) => {
       const action = args[0] ?? "list";
-      const config = await ctx.initConfigData();
-      try {
+      await withConfigData(ctx, async (config) => {
         const raw = config.config.pluginConfig[ALERTS_PLUGIN_ID]?.[ALERTS_KEY];
         const alerts = deserializeAlerts(typeof raw === "string" ? raw : "[]");
         const saveAlerts = async (nextAlerts: typeof alerts) => {
@@ -413,9 +388,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
           return;
         }
         ctx.fail("Usage: gloomberb alerts list|add|delete|rearm");
-      } finally {
-        config.persistence.close();
-      }
+      });
     },
   };
 
@@ -485,8 +458,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
     description: "Show CLI coverage for panes, templates, capabilities, and deferred exceptions",
     execute: async (_args, ctx) => {
       const commandNames = new Set(allCommands().map((command) => command.name));
-      const services = await ctx.initServices();
-      try {
+      await withCliServices(ctx, async (services) => {
         const rows = [
           ...[...services.services.pluginRegistry.panes.entries()].map(([id, pane]) => ({
             surface: "pane",
@@ -531,9 +503,7 @@ export function createSystemCliCommands(allCommands: () => CliCommandDef[]): Cli
             { key: "label", header: "Label" },
           ],
         });
-      } finally {
-        services.destroy();
-      }
+      });
     },
   };
 

package/src/cli/commands/ticker.ts

@@ -19,8 +19,8 @@ import type { NewsArticle } from "../../news/types";
 import type { TickerRecord } from "../../types/ticker";
 import type { CliCommandContext } from "../../types/plugin";
 import { createBaseConverter } from "../base-converter";
-import { initMarketData } from "../context";
-import { closeAndFail } from "../errors";
+import { initMarketData, withMarketData } from "../context";
+import { fail } from "../errors";
 import type { MarketContext } from "../types";
 import {
   formatBidAsk,
@@ -40,7 +40,7 @@ const SEC_FILING_LIMIT = 5;
 
 interface TickerCommandDependencies {
   initMarketData?: () => Promise<MarketContext>;
-  closeAndFail?: typeof closeAndFail;
+  fail?: (message: string, details?: string) => never;
   printResult?: CliCommandContext["printResult"];
 }
 
@@ -443,74 +443,73 @@ function buildTickerStructuredData({
 
 export async function ticker(symbol: string, dependencies: TickerCommandDependencies = {}) {
   const initMarketDataFn = dependencies.initMarketData ?? initMarketData;
-  const closeAndFailCommand = dependencies.closeAndFail ?? closeAndFail;
-  const { config, store, dataProvider, persistence, dataDir } = await initMarketDataFn();
-  const normalized = symbol.trim().toUpperCase();
-  const tickerFile = await store.loadTicker(normalized);
-  const exchange = tickerFile?.metadata.exchange ?? "";
-  const toBase = createBaseConverter(dataProvider, config.baseCurrency);
-
-  let financials: TickerFinancials | null = null;
-  try {
-    financials = await dataProvider.getTickerFinancials(normalized, exchange);
-  } catch (error) {
-    closeAndFailCommand(
-      persistence,
-      `Failed to fetch data for ${normalized}.`,
-      error instanceof Error ? error.message : String(error),
-    );
-  }
-
-  if (!financials?.quote) {
-    closeAndFailCommand(persistence, `No quote data available for ${normalized}.`);
-  }
-  const resolvedFinancials = financials as TickerFinancials;
-  const quote = resolvedFinancials.quote!;
-
-  const notesFiles = new NotesFiles(dataDir);
-  const [notesResult, newsResult, secFilingsResult] = await Promise.allSettled([
-    notesFiles.load(normalized),
-    dataProvider.getNews({
-      feed: "ticker",
-      scope: "ticker",
-      ticker: normalized,
-      exchange: exchange || quote.exchangeName || "",
-      tickerTier: "primary",
-      limit: NEWS_ITEM_LIMIT,
-    }),
-    shouldFetchSecFilings(tickerFile, resolvedFinancials) && dataProvider.getSecFilings
-      ? dataProvider.getSecFilings(normalized, SEC_FILING_LIMIT, exchange || quote.exchangeName || "")
-      : Promise.resolve([]),
-  ]);
+  const failCommand = dependencies.fail ?? fail;
+  await withMarketData(initMarketDataFn, async ({ config, store, dataProvider, dataDir }) => {
+    const normalized = symbol.trim().toUpperCase();
+    const tickerFile = await store.loadTicker(normalized);
+    const exchange = tickerFile?.metadata.exchange ?? "";
+    const toBase = createBaseConverter(dataProvider, config.baseCurrency);
+
+    let financials: TickerFinancials | null = null;
+    try {
+      financials = await dataProvider.getTickerFinancials(normalized, exchange);
+    } catch (error) {
+      failCommand(
+        `Failed to fetch data for ${normalized}.`,
+        error instanceof Error ? error.message : String(error),
+      );
+    }
 
-  const notes = notesResult.status === "fulfilled" ? notesResult.value : "";
-  const recentNews = newsResult.status === "fulfilled" ? newsResult.value : [];
-  const recentSecFilings = secFilingsResult.status === "fulfilled" ? secFilingsResult.value : [];
+    if (!financials?.quote) {
+      failCommand(`No quote data available for ${normalized}.`);
+    }
+    const resolvedFinancials = financials as TickerFinancials;
+    const quote = resolvedFinancials.quote!;
+
+    const notesFiles = new NotesFiles(dataDir);
+    const [notesResult, newsResult, secFilingsResult] = await Promise.allSettled([
+      notesFiles.load(normalized),
+      dataProvider.getNews({
+        feed: "ticker",
+        scope: "ticker",
+        ticker: normalized,
+        exchange: exchange || quote.exchangeName || "",
+        tickerTier: "primary",
+        limit: NEWS_ITEM_LIMIT,
+      }),
+      shouldFetchSecFilings(tickerFile, resolvedFinancials) && dataProvider.getSecFilings
+        ? dataProvider.getSecFilings(normalized, SEC_FILING_LIMIT, exchange || quote.exchangeName || "")
+        : Promise.resolve([]),
+    ]);
+
+    const notes = notesResult.status === "fulfilled" ? notesResult.value : "";
+    const recentNews = newsResult.status === "fulfilled" ? newsResult.value : [];
+    const recentSecFilings = secFilingsResult.status === "fulfilled" ? secFilingsResult.value : [];
+
+    if (dependencies.printResult) {
+      dependencies.printResult({
+        data: buildTickerStructuredData({
+          symbol: normalized,
+          tickerFile,
+          financials: resolvedFinancials,
+          config,
+          notes,
+          recentNews,
+          recentSecFilings,
+        }),
+      });
+      return;
+    }
 
-  if (dependencies.printResult) {
-    dependencies.printResult({
-      data: buildTickerStructuredData({
+    console.log(await buildTickerReport({
         symbol: normalized,
         tickerFile,
         financials: resolvedFinancials,
         config,
+        toBase,
         notes,
         recentNews,
         recentSecFilings,
-      }),
-    });
-  } else {
-    console.log(await buildTickerReport({
-      symbol: normalized,
-      tickerFile,
-      financials: resolvedFinancials,
-      config,
-      toBase,
-      notes,
-      recentNews,
-      recentSecFilings,
     }));
-  }
-
-  persistence.close();
+  });
 }

package/src/cli/context.test.ts

@@ -0,0 +1,63 @@
+import { describe, expect, test } from "bun:test";
+import type { CliCommandContext } from "../types/plugin";
+import type { CliServicesContext, ConfigContext, MarketContext } from "./types";
+import {
+  ensureCliServicesReady,
+  withCliServices,
+  withConfigData,
+  withMarketData,
+} from "./context";
+
+function initializer<K extends "initConfigData" | "initMarketData" | "initServices", T>(
+  key: K,
+  value: T,
+): Pick<CliCommandContext, K> {
+  return { [key]: async () => value } as Pick<CliCommandContext, K>;
+}
+
+describe("CLI scoped contexts", () => {
+  test("closes config and market persistence after successful operations", async () => {
+    let configCloses = 0;
+    let marketCloses = 0;
+    const config = { persistence: { close: () => configCloses++ } } as unknown as ConfigContext;
+    const market = { persistence: { close: () => marketCloses++ } } as unknown as MarketContext;
+
+    expect(await withConfigData(initializer("initConfigData", config), async () => "config"))
+      .toBe("config");
+    expect(await withMarketData(initializer("initMarketData", market), async () => "market"))
+      .toBe("market");
+    expect(configCloses).toBe(1);
+    expect(marketCloses).toBe(1);
+  });
+
+  test("destroys services when an operation throws", async () => {
+    let destroys = 0;
+    const services = { destroy: () => destroys++ } as unknown as CliServicesContext;
+
+    await expect(withCliServices(initializer("initServices", services), async () => {
+      throw new Error("operation failed");
+    })).rejects.toThrow("operation failed");
+    expect(destroys).toBe(1);
+  });
+
+  test("accepts an injected market initializer and closes it when an operation throws", async () => {
+    let closes = 0;
+    const market = { persistence: { close: () => closes++ } } as unknown as MarketContext;
+
+    await expect(withMarketData(async () => market, async () => {
+      throw new Error("market operation failed");
+    })).rejects.toThrow("market operation failed");
+    expect(closes).toBe(1);
+  });
+
+  test("destroys partially initialized services when plugin setup rejects", async () => {
+    let destroys = 0;
+    const services = {
+      ready: Promise.reject(new Error("plugin setup failed")),
+      destroy: () => destroys++,
+    };
+
+    await expect(ensureCliServicesReady(services)).rejects.toThrow("plugin setup failed");
+    expect(destroys).toBe(1);
+  });
+});

package/src/cli/context.ts

@@ -14,6 +14,8 @@ import type { LoadedExternalPlugin } from "../plugins/loader";
 import { fail } from "./errors";
 import type { ConfigContext, MarketContext } from "./types";
 
+export { withCliServices, withConfigData, withMarketData } from "./scoped-context";
+
 interface CliContextOptions {
   plugins?: GloomPlugin[];
 }
@@ -22,6 +24,18 @@ interface CliServicesOptions {
   externalPlugins?: LoadedExternalPlugin[];
 }
 
+export async function ensureCliServicesReady<T extends { ready: Promise<unknown>; destroy(): void }>(
+  services: T,
+): Promise<T> {
+  try {
+    await services.ready;
+    return services;
+  } catch (error) {
+    services.destroy();
+    throw error;
+  }
+}
+
 function resolveCliCapabilities(config: AppConfig, plugins: GloomPlugin[]): PluginCapability[] {
   const disabledPlugins = new Set(config.disabledPlugins ?? []);
   const disabledSources = new Set(config.disabledSources ?? []);
@@ -83,7 +97,7 @@ export async function initCliServices(options: CliServicesOptions = {}) {
     plugins: getLoadablePlugins(options.externalPlugins ?? []),
   });
   services.providerRouter.setConfigAccessor(() => config);
-  await services.ready;
+  await ensureCliServicesReady(services);
   return {
     config,
     dataDir,

package/src/cli/index.ts

@@ -98,7 +98,7 @@ function createCoreCliCommands(renderHelp: () => string): CliCommandDef[] {
         }
         await ticker(symbol!, {
           initMarketData: ctx.initMarketData,
-          closeAndFail: ctx.closeAndFail,
+          fail: ctx.fail,
           ...(ctx.cliOptions.format === "text" ? {} : { printResult: ctx.printResult }),
         });
       },

package/src/cli/pane-functions/data.ts

@@ -1,7 +1,6 @@
-import type { PricePoint, TickerFinancials } from "../../types/financials";
+import type { TickerFinancials } from "../../types/financials";
 import type { TickerRecord } from "../../types/ticker";
-import type { TimeRange } from "../../components/chart/core/types";
-import { subtractTimeRange } from "../../components/chart/core/date-window";
+export { clipPriceHistoryToRange } from "../../time-series/history-window";
 import { CHART_COMPOSER_PANE_ID, LEGACY_TICKER_DETAIL_PANE_ID } from "../../types/config";
 import { parseChartSpec } from "../../plugins/builtin/chart-composer/chart-spec";
 import { normalizeTickerInput } from "../../tickers/search";
@@ -102,19 +101,3 @@ export function collectShotSymbols(resolved: ResolvedPaneFunction, rawArg: strin
   }
   return [...new Set(symbols.map(cleanTickerInput).filter(Boolean))];
 }
-
-export function clipPriceHistoryToRange(points: PricePoint[], range: TimeRange): PricePoint[] {
-  if (range === "ALL" || points.length === 0) return points;
-  const dated = points.flatMap((point) => {
-    const date = point.date instanceof Date ? point.date : new Date(point.date);
-    return Number.isFinite(date.getTime()) ? [{ point, date }] : [];
-  });
-  const end = dated.reduce<Date | null>((latest, entry) => (
-    !latest || entry.date.getTime() > latest.getTime() ? entry.date : latest
-  ), null);
-  if (!end) return [];
-  const start = subtractTimeRange(end, range);
-  return dated
-    .filter(({ date }) => date.getTime() >= start.getTime() && date.getTime() <= end.getTime())
-    .map(({ point }) => point);
-}

package/src/cli/pane-functions/index.ts

@@ -1,6 +1,7 @@
 import { extname, resolve } from "path";
 import type { CliCommandContext } from "../../types/plugin";
 import type { MarketContext } from "../types";
+import { withMarketData } from "../context";
 import {
   filterPaneCatalogEntries,
   renderPaneCatalogReport,
@@ -38,15 +39,15 @@ async function withPaneRuntime<T>(
   }) => Promise<T>,
 ): Promise<T> {
   const parsed = parsePaneFunctionArgs(args);
-  const context = await ctx.initMarketData();
-  const registry = await createPaneCatalog(context, ctx.plugins);
-  try {
-    const resolved = await resolvePaneFunction(registry, context, parsed);
-    return await run({ parsed, context, registry, resolved });
-  } finally {
-    registry.destroy();
-    context.persistence.close();
-  }
+  return withMarketData(ctx, async (context) => {
+    const registry = await createPaneCatalog(context, ctx.plugins);
+    try {
+      const resolved = await resolvePaneFunction(registry, context, parsed);
+      return await run({ parsed, context, registry, resolved });
+    } finally {
+      registry.destroy();
+    }
+  });
 }
 
 export async function runPaneFunction(args: string[], ctx: CliCommandContext) {
@@ -133,23 +134,23 @@ export async function runPaneCatalog(args: string[], ctx: CliCommandContext) {
       ...parsed,
       limit: ctx.cliOptions.limit ?? parsed.limit,
     };
-    const context = await ctx.initMarketData();
-    const registry = await createPaneCatalog(context, ctx.plugins);
-    try {
-      const entries = await buildPaneCatalogEntries(registry, context);
-      const botSafeEntries = effectiveParsed.botSafeOnly
-        ? entries.filter((entry) => entry.capability.botSafe)
-        : entries;
-      const filtered = filterPaneCatalogEntries(botSafeEntries, parsed.query);
-      if (ctx.cliOptions.format === "text") {
-        console.log(renderPaneCatalogReport(filtered, effectiveParsed));
-      } else {
-        ctx.printResult({ data: filtered.slice(0, effectiveParsed.limit) });
+    await withMarketData(ctx, async (context) => {
+      const registry = await createPaneCatalog(context, ctx.plugins);
+      try {
+        const entries = await buildPaneCatalogEntries(registry, context);
+        const botSafeEntries = effectiveParsed.botSafeOnly
+          ? entries.filter((entry) => entry.capability.botSafe)
+          : entries;
+        const filtered = filterPaneCatalogEntries(botSafeEntries, parsed.query);
+        if (ctx.cliOptions.format === "text") {
+          console.log(renderPaneCatalogReport(filtered, effectiveParsed));
+        } else {
+          ctx.printResult({ data: filtered.slice(0, effectiveParsed.limit) });
+        }
+      } finally {
+        registry.destroy();
       }
-    } finally {
-      registry.destroy();
-      context.persistence.close();
-    }
+    });
   });
 }
 

package/src/cli/pane-functions/report.ts

@@ -1,4 +1,4 @@
-import type { TimeRange } from "../../components/chart/core/types";
+import type { TimeRange } from "../../time-series/range";
 import {
   buildCorrelationMatrix,
   pairKey,

package/src/cli/pane-functions/screenshot.ts

@@ -29,12 +29,12 @@ import type {
   GraphKind,
   GraphMetricKey,
 } from "../../time-series/reporting";
-import type { TimeRange } from "../../components/chart/core/types";
-import { appendLiveQuotePoint } from "../../components/chart/core/data";
+import type { TimeRange } from "../../time-series/range";
+import { appendLiveQuotePoint } from "../../time-series/chart-data";
+import { subtractTimeRange } from "../../time-series/date-window";
 import {
   buildPresetDateWindow,
   getVisibleWindowForDateRange,
-  subtractTimeRange,
 } from "../../components/chart/core/date-window";
 import { parseChartSpec } from "../../plugins/builtin/chart-composer/chart-spec";
 import { publicTickerKey } from "../../utils/exchanges";

package/src/cli/scoped-context.ts

@@ -0,0 +1,41 @@
+import type { CliCommandContext } from "../types/plugin";
+import type { CliServicesContext, ConfigContext, MarketContext } from "./types";
+
+type AsyncInitializer<T> = () => Promise<T>;
+
+async function withCleanup<TContext, TResult>(
+  initialize: AsyncInitializer<TContext>,
+  cleanup: (context: TContext) => void,
+  operation: (context: TContext) => TResult | Promise<TResult>,
+): Promise<TResult> {
+  const context = await initialize();
+  try {
+    return await operation(context);
+  } finally {
+    cleanup(context);
+  }
+}
+
+export async function withConfigData<TResult>(
+  source: Pick<CliCommandContext, "initConfigData"> | AsyncInitializer<ConfigContext>,
+  operation: (context: ConfigContext) => TResult | Promise<TResult>,
+): Promise<TResult> {
+  const initialize = typeof source === "function" ? source : () => source.initConfigData();
+  return withCleanup(initialize, (context) => context.persistence.close(), operation);
+}
+
+export async function withMarketData<TResult>(
+  source: Pick<CliCommandContext, "initMarketData"> | AsyncInitializer<MarketContext>,
+  operation: (context: MarketContext) => TResult | Promise<TResult>,
+): Promise<TResult> {
+  const initialize = typeof source === "function" ? source : () => source.initMarketData();
+  return withCleanup(initialize, (context) => context.persistence.close(), operation);
+}
+
+export async function withCliServices<TResult>(
+  source: Pick<CliCommandContext, "initServices"> | AsyncInitializer<CliServicesContext>,
+  operation: (context: CliServicesContext) => TResult | Promise<TResult>,
+): Promise<TResult> {
+  const initialize = typeof source === "function" ? source : () => source.initServices();
+  return withCleanup(initialize, (context) => context.destroy(), operation);
+}