@aemonge-dev/pi-stardeck 0.5.0 → 0.6.0

package/packages/pi-fancy-footer-full-palette/src/index.ts

@@ -1,479 +0,0 @@
-import { createRequire } from "node:module";
-import {
-  type ExtensionAPI,
-  type ExtensionContext,
-} from "@earendil-works/pi-coding-agent";
-import {
-  DEFAULT_FOOTER_CONFIG,
-  EMPTY_GIT_INFO,
-  MAX_FOOTER_REFRESH_MS,
-  MAX_PROVIDER_STATUS_REFRESH_MS,
-  MIN_FOOTER_REFRESH_MS,
-  MIN_PROVIDER_STATUS_REFRESH_MS,
-  clampInt,
-  type FooterConfigSnapshot,
-  type ProviderStatusSnapshot,
-  type SessionUsageMetrics,
-} from "./shared.ts";
-import {
-  cloneFooterConfig,
-  getFooterConfigPath,
-  hasFooterConfigError,
-  loadFooterConfig,
-  writeFooterConfigSnapshot,
-} from "./config.ts";
-import { collectGitInfo } from "./git.ts";
-import {
-  FANCY_FOOTER_PROTOCOL_VERSION,
-  FANCY_FOOTER_READY_CHANNEL,
-  FANCY_FOOTER_TELEMETRY_CHANNEL,
-  FANCY_FOOTER_WIDGET_CHANNEL,
-  PI_STATUS_SOURCE_CHANNEL,
-  type FancyFooterReadyMessage,
-  type FancyFooterStatusSourceMessage,
-  type FancyFooterTelemetryMessage,
-} from "./api.ts";
-import {
-  createMicrotaskCoalescer,
-  FancyFooterDataWidgetStore,
-  type NormalizedFancyFooterDataWidget,
-} from "./data-widgets.ts";
-import { openFooterConfigEditor } from "./config-editor.ts";
-import { collectSessionUsageMetrics, renderFooterLines } from "./render.ts";
-import {
-  collectProviderStatus,
-  isProviderStatusRelevantToModel,
-  projectProviderStatusForModel,
-  providerStatusQuotaWindows,
-  updateProviderStatusFromHeaders,
-} from "./provider-status.ts";
-
-interface ActiveFooterControls {
-  requestRender: () => void;
-  reschedule: () => void;
-  rescheduleProviderStatus: () => void;
-  updateProviderStatus: (status: ProviderStatusSnapshot) => void;
-}
-
-const PACKAGE_VERSION = (
-  createRequire(import.meta.url)("../package.json") as { version: string }
-).version;
-
-export type FancyFooterSurface = "footer" | "telemetry";
-
-export interface FancyFooterOptions {
-  surface?: FancyFooterSurface;
-}
-
-export default function (pi: ExtensionAPI, options: FancyFooterOptions = {}) {
-  const telemetryOnly = options.surface === "telemetry";
-  let footerConfig: FooterConfigSnapshot = {
-    refreshMs: DEFAULT_FOOTER_CONFIG.refreshMs,
-    iconFamily: DEFAULT_FOOTER_CONFIG.iconFamily,
-    gaugeStyle: DEFAULT_FOOTER_CONFIG.gaugeStyle,
-    gaugeWidth: DEFAULT_FOOTER_CONFIG.gaugeWidth,
-    gaugeColors: { ...DEFAULT_FOOTER_CONFIG.gaugeColors },
-    defaultTextColor: DEFAULT_FOOTER_CONFIG.defaultTextColor,
-    defaultIconColor: DEFAULT_FOOTER_CONFIG.defaultIconColor,
-    providerStatus: { ...DEFAULT_FOOTER_CONFIG.providerStatus },
-    widgets: { ...DEFAULT_FOOTER_CONFIG.widgets },
-    extensionWidgets: { ...DEFAULT_FOOTER_CONFIG.extensionWidgets },
-  };
-  const dataWidgets = new FancyFooterDataWidgetStore();
-  let extensionWidgets: NormalizedFancyFooterDataWidget[] = [];
-
-  let activeFooterControls: ActiveFooterControls | undefined;
-  let footerInstanceId = 0;
-  let telemetryContext: ExtensionContext | undefined;
-  let telemetryProviderStatuses = new Map<string, ProviderStatusSnapshot>();
-  let telemetryRefreshTimer: ReturnType<typeof setTimeout> | undefined;
-  let telemetryGeneration = 0;
-
-  const invalidateActiveFooter = () => {
-    footerInstanceId += 1;
-    activeFooterControls = undefined;
-  };
-
-  const requestDataWidgetRender = createMicrotaskCoalescer(() => {
-    activeFooterControls?.requestRender();
-  });
-  const refreshDataWidgets = () => {
-    extensionWidgets = dataWidgets.values();
-    requestDataWidgetRender();
-  };
-  const publishReady = () => {
-    const message: FancyFooterReadyMessage = {
-      protocol: FANCY_FOOTER_PROTOCOL_VERSION,
-      version: PACKAGE_VERSION,
-    };
-    pi.events.emit(FANCY_FOOTER_READY_CHANNEL, message);
-  };
-
-  const telemetryQuotaPercent = (): number | undefined => {
-    const model = telemetryContext?.model;
-    const percentages: number[] = [];
-    for (const snapshot of telemetryProviderStatuses.values()) {
-      if (!isProviderStatusRelevantToModel(snapshot.provider, model)) continue;
-      const projected = projectProviderStatusForModel(snapshot, model);
-      for (const window of [projected.primary, projected.secondary]) {
-        if (!window || window.usageUnknown) continue;
-        percentages.push(window.usedPercent);
-      }
-    }
-    return percentages.length > 0 ? Math.max(...percentages) : undefined;
-  };
-
-  const publishTelemetry = () => {
-    if (!telemetryContext) return;
-    const totalCost = collectSessionUsageMetrics(telemetryContext).totalCost;
-    const quotaPercent = telemetryQuotaPercent();
-    const quotaWindows = providerStatusQuotaWindows(
-      telemetryProviderStatuses.values(),
-      telemetryContext?.model,
-    );
-    const message: FancyFooterTelemetryMessage = {
-      protocol: FANCY_FOOTER_PROTOCOL_VERSION,
-      type: "snapshot",
-      totalCost,
-      ...(quotaPercent === undefined ? {} : { quotaPercent }),
-      ...(quotaWindows.length > 0 ? { quotaWindows } : {}),
-    };
-    pi.events.emit(FANCY_FOOTER_TELEMETRY_CHANNEL, message);
-
-    const conditions: FancyFooterStatusSourceMessage["conditions"] = [];
-    if (hasFooterConfigError()) {
-      conditions.push({
-        id: "config",
-        severity: "error",
-        summary: "Footer configuration is invalid; defaults are active",
-      });
-    }
-    for (const snapshot of telemetryProviderStatuses.values()) {
-      if (!snapshot.error) continue;
-      const retained =
-        snapshot.source === "cache" &&
-        (snapshot.primary !== undefined || snapshot.secondary !== undefined);
-      conditions.push({
-        id: `provider-status.${snapshot.provider}`,
-        severity: retained ? "warning" : "error",
-        summary: retained
-          ? `${snapshot.provider} quota refresh failed; cached status is active`
-          : `${snapshot.provider} quota status is unavailable`,
-      });
-    }
-    const health: FancyFooterStatusSourceMessage = {
-      protocol: FANCY_FOOTER_PROTOCOL_VERSION,
-      type: "snapshot",
-      source: "fancy-footer",
-      conditions,
-    };
-    pi.events.emit(PI_STATUS_SOURCE_CHANNEL, health);
-  };
-
-  const stopTelemetryProvider = () => {
-    telemetryGeneration += 1;
-    telemetryContext = undefined;
-    telemetryProviderStatuses.clear();
-    if (telemetryRefreshTimer) clearTimeout(telemetryRefreshTimer);
-    telemetryRefreshTimer = undefined;
-  };
-
-  const installTelemetryProvider = (ctx: ExtensionContext) => {
-    stopTelemetryProvider();
-    telemetryContext = ctx;
-    footerConfig = loadFooterConfig();
-    const generation = telemetryGeneration;
-
-    const refresh = async () => {
-      if (!telemetryContext || generation !== telemetryGeneration) return;
-      const next = await collectProviderStatus(pi, footerConfig.providerStatus);
-      if (!telemetryContext || generation !== telemetryGeneration) return;
-      telemetryProviderStatuses = new Map(
-        next.map((snapshot) => [snapshot.provider, snapshot]),
-      );
-      publishTelemetry();
-    };
-    const schedule = () => {
-      if (!telemetryContext || generation !== telemetryGeneration) return;
-      if (telemetryRefreshTimer) clearTimeout(telemetryRefreshTimer);
-      const refreshMs = clampInt(
-        footerConfig.providerStatus.refreshMs,
-        MIN_PROVIDER_STATUS_REFRESH_MS,
-        MAX_PROVIDER_STATUS_REFRESH_MS,
-      );
-      telemetryRefreshTimer = setTimeout(() => {
-        void refresh().finally(schedule);
-      }, refreshMs);
-    };
-
-    publishTelemetry();
-    void refresh();
-    schedule();
-  };
-
-  const stopDataWidgetListener = pi.events.on(FANCY_FOOTER_WIDGET_CHANNEL, (raw) => {
-    if (dataWidgets.apply(raw)) refreshDataWidgets();
-  });
-
-  const installFooter = (ctx: ExtensionContext) => {
-    if (!ctx.hasUI) return;
-
-    footerConfig = loadFooterConfig();
-
-    ctx.ui.setFooter((tui, theme, footerData) => {
-      const instanceId = ++footerInstanceId;
-      const fallbackThinkingLevel = pi.getThinkingLevel();
-      let currentGit = { ...EMPTY_GIT_INFO };
-      let providerStatuses = new Map<string, ProviderStatusSnapshot>();
-      let usageMetrics: SessionUsageMetrics = collectSessionUsageMetrics(ctx);
-      let refreshing = false;
-      let refreshQueued = false;
-      let providerStatusRefreshing = false;
-      let providerStatusRefreshQueued = false;
-      let providerStatusRefreshAt = 0;
-      let disposed = false;
-      let refreshTimer: ReturnType<typeof setTimeout> | undefined;
-      let providerStatusTimer: ReturnType<typeof setTimeout> | undefined;
-
-      const isActiveFooter = () => !disposed && instanceId === footerInstanceId;
-
-      const requestRender = () => {
-        if (!isActiveFooter()) return;
-        tui.requestRender();
-      };
-
-      const isProviderStatusWidgetEnabled = () =>
-        footerConfig.widgets["provider-status"]?.enabled !== false;
-
-      const refreshProviderStatus = async (force = false) => {
-        if (!isActiveFooter() || !isProviderStatusWidgetEnabled()) return;
-        const now = Date.now();
-        const refreshMs = clampInt(
-          footerConfig.providerStatus.refreshMs,
-          MIN_PROVIDER_STATUS_REFRESH_MS,
-          MAX_PROVIDER_STATUS_REFRESH_MS,
-        );
-        if (!force && providerStatusRefreshAt + refreshMs > now) return;
-        if (providerStatusRefreshing) {
-          providerStatusRefreshQueued = true;
-          return;
-        }
-
-        providerStatusRefreshing = true;
-        try {
-          do {
-            providerStatusRefreshQueued = false;
-            if (!isActiveFooter() || !isProviderStatusWidgetEnabled()) {
-              continue;
-            }
-
-            providerStatusRefreshAt = Date.now();
-            const next = await collectProviderStatus(pi, footerConfig.providerStatus);
-            if (!isActiveFooter()) return;
-            providerStatuses = new Map(
-              next.map((snapshot) => [snapshot.provider, snapshot]),
-            );
-            requestRender();
-          } while (isActiveFooter() && providerStatusRefreshQueued);
-        } finally {
-          providerStatusRefreshing = false;
-        }
-      };
-
-      const refreshGit = async () => {
-        if (!isActiveFooter()) return;
-        if (refreshing) {
-          refreshQueued = true;
-          return;
-        }
-
-        refreshing = true;
-        try {
-          do {
-            refreshQueued = false;
-            if (!isActiveFooter()) continue;
-
-            footerConfig = loadFooterConfig();
-            usageMetrics = collectSessionUsageMetrics(ctx);
-
-            const git = await collectGitInfo(pi, ctx.cwd);
-            if (!isActiveFooter()) return;
-            currentGit = git;
-            requestRender();
-          } while (isActiveFooter() && refreshQueued);
-        } finally {
-          refreshing = false;
-        }
-      };
-
-      const scheduleRefresh = () => {
-        if (!isActiveFooter()) return;
-        if (refreshTimer) clearTimeout(refreshTimer);
-
-        const refreshMs = clampInt(
-          footerConfig.refreshMs,
-          MIN_FOOTER_REFRESH_MS,
-          MAX_FOOTER_REFRESH_MS,
-        );
-        refreshTimer = setTimeout(() => {
-          if (!isActiveFooter()) return;
-          void refreshGit().finally(() => {
-            scheduleRefresh();
-          });
-        }, refreshMs);
-      };
-
-      const scheduleProviderStatusRefresh = () => {
-        if (!isActiveFooter()) return;
-        if (providerStatusTimer) clearTimeout(providerStatusTimer);
-
-        const refreshMs = clampInt(
-          footerConfig.providerStatus.refreshMs,
-          MIN_PROVIDER_STATUS_REFRESH_MS,
-          MAX_PROVIDER_STATUS_REFRESH_MS,
-        );
-        providerStatusTimer = setTimeout(() => {
-          if (!isActiveFooter()) return;
-          void refreshProviderStatus().finally(() => {
-            scheduleProviderStatusRefresh();
-          });
-        }, refreshMs);
-      };
-
-      const onBranchChange = footerData.onBranchChange(() => {
-        if (!isActiveFooter()) return;
-        usageMetrics = collectSessionUsageMetrics(ctx);
-        requestRender();
-        void refreshGit();
-      });
-
-      activeFooterControls = {
-        requestRender,
-        reschedule: scheduleRefresh,
-        rescheduleProviderStatus: scheduleProviderStatusRefresh,
-        updateProviderStatus: (status) => {
-          if (!isActiveFooter()) return;
-          providerStatuses.set(status.provider, status);
-          requestRender();
-        },
-      };
-
-      void refreshGit();
-      void refreshProviderStatus(true);
-      scheduleRefresh();
-      scheduleProviderStatusRefresh();
-
-      return {
-        invalidate() {},
-        dispose() {
-          disposed = true;
-          onBranchChange();
-          if (refreshTimer) clearTimeout(refreshTimer);
-          if (providerStatusTimer) clearTimeout(providerStatusTimer);
-          if (activeFooterControls?.requestRender === requestRender) {
-            activeFooterControls = undefined;
-          }
-        },
-        render(width: number): string[] {
-          if (!isActiveFooter()) return ["", ""];
-
-          return renderFooterLines(
-            width,
-            ctx,
-            currentGit,
-            fallbackThinkingLevel,
-            theme,
-            usageMetrics,
-            footerConfig,
-            extensionWidgets,
-            Array.from(providerStatuses.values()).sort((a, b) =>
-              a.provider.localeCompare(b.provider),
-            ),
-          );
-        },
-      };
-    });
-  };
-
-  pi.registerCommand("fancy-footer", {
-    description: "Configure the fancy footer.",
-    handler: async (_args, ctx) => {
-      const configPath = getFooterConfigPath();
-
-      if (!ctx.hasUI) {
-        ctx.ui.notify("/fancy-footer requires interactive UI mode", "warning");
-        return;
-      }
-
-      const draft = cloneFooterConfig(loadFooterConfig());
-      const applyDraft = () => {
-        try {
-          writeFooterConfigSnapshot(draft);
-          footerConfig = loadFooterConfig();
-          activeFooterControls?.reschedule();
-          activeFooterControls?.rescheduleProviderStatus();
-          activeFooterControls?.requestRender();
-        } catch (error) {
-          const msg = error instanceof Error ? error.message : String(error);
-          ctx.ui.notify(`Failed to save config: ${msg}`, "error");
-        }
-      };
-
-      await openFooterConfigEditor({
-        ctx,
-        configPath,
-        draft,
-        extensionWidgets,
-        applyDraft,
-      });
-    },
-  });
-
-  pi.on("after_provider_response", async (event) => {
-    if (footerConfig.widgets["provider-status"]?.enabled === false) return;
-
-    const updated = await updateProviderStatusFromHeaders(
-      event.headers ?? {},
-      footerConfig.providerStatus,
-    );
-    for (const snapshot of updated) {
-      activeFooterControls?.updateProviderStatus(snapshot);
-      if (telemetryOnly) telemetryProviderStatuses.set(snapshot.provider, snapshot);
-    }
-    if (telemetryOnly && updated.length > 0) publishTelemetry();
-  });
-
-  pi.on("model_select", () => {
-    activeFooterControls?.requestRender();
-    if (telemetryOnly) publishTelemetry();
-  });
-
-  pi.on("thinking_level_select", () => {
-    activeFooterControls?.requestRender();
-  });
-
-  pi.on("session_compact", () => {
-    activeFooterControls?.requestRender();
-    if (telemetryOnly) publishTelemetry();
-  });
-
-  pi.on("message_end", () => {
-    if (telemetryOnly) publishTelemetry();
-  });
-
-  pi.on("session_shutdown", async () => {
-    stopDataWidgetListener();
-    stopTelemetryProvider();
-    invalidateActiveFooter();
-    if (dataWidgets.clear()) extensionWidgets = [];
-  });
-
-  pi.on("session_start", async (_event, ctx) => {
-    if (dataWidgets.clear()) extensionWidgets = [];
-    if (telemetryOnly) installTelemetryProvider(ctx);
-    else installFooter(ctx);
-    publishReady();
-  });
-
-  publishReady();
-}