pi-one-ui 0.1.1

package/extensions/shell/telemetry.ts

@@ -0,0 +1,90 @@
+import {
+  type ExtensionContext,
+  SettingsManager,
+} from "@earendil-works/pi-coding-agent";
+
+export type FooterTelemetry = {
+  subscription?: boolean;
+  autoCompaction?: boolean;
+};
+
+type SettingsManagerLike = {
+  drainErrors?: () => unknown[];
+  getCompactionEnabled?: () => unknown;
+};
+
+type SettingsManagerFactory = {
+  create?: (
+    cwd: string,
+    agentDir?: string,
+    options?: { projectTrusted?: boolean },
+  ) => SettingsManagerLike;
+};
+
+export type TelemetryCapabilities = {
+  settingsManager?: SettingsManagerFactory;
+};
+
+function resolveSubscription(ctx: ExtensionContext): boolean | undefined {
+  const model = ctx.model;
+  if (!model) return undefined;
+  if (model.provider === "kimi-coding") return true;
+
+  try {
+    const registry = ctx.modelRegistry as {
+      isUsingOAuth?: (candidate: typeof model) => unknown;
+    };
+    if (typeof registry?.isUsingOAuth !== "function") return undefined;
+    const result = registry.isUsingOAuth(model);
+    return typeof result === "boolean" ? result : undefined;
+  } catch {
+    return undefined;
+  }
+}
+
+function resolveAutoCompaction(
+  ctx: ExtensionContext,
+  factory: SettingsManagerFactory | undefined,
+): boolean | undefined {
+  try {
+    const isProjectTrusted = (
+      ctx as ExtensionContext & {
+        isProjectTrusted?: () => unknown;
+      }
+    ).isProjectTrusted;
+    if (
+      typeof factory?.create !== "function" ||
+      typeof isProjectTrusted !== "function"
+    ) {
+      return undefined;
+    }
+    const trusted = isProjectTrusted.call(ctx);
+    if (typeof trusted !== "boolean") return undefined;
+    const settings = factory.create(ctx.cwd, undefined, {
+      projectTrusted: trusted,
+    });
+    if (
+      typeof settings?.drainErrors !== "function" ||
+      typeof settings.getCompactionEnabled !== "function"
+    ) {
+      return undefined;
+    }
+    if (settings.drainErrors().length > 0) return undefined;
+    const enabled = settings.getCompactionEnabled();
+    return typeof enabled === "boolean" ? enabled : undefined;
+  } catch {
+    return undefined;
+  }
+}
+
+/** Resolve optional Pi telemetry without depending on private session or TUI fields. */
+export function resolveFooterTelemetry(
+  ctx: ExtensionContext,
+  capabilities: TelemetryCapabilities = {},
+): FooterTelemetry {
+  const settingsManager = capabilities.settingsManager ?? SettingsManager;
+  return {
+    subscription: resolveSubscription(ctx),
+    autoCompaction: resolveAutoCompaction(ctx, settingsManager),
+  };
+}