@hieplp/pi-account-switcher 0.2.0 → 0.2.2

package/src/utils/models.ts

@@ -1,4 +1,4 @@
-import type { Api, Model } from "@mariozechner/pi-ai";
+import type { Api, Model } from "@earendil-works/pi-ai";
 import type { AccountConfig, AccountSwitcherContext, ProviderConfig } from "@/types";
 import { providerUtil } from "./providers";
 import { uiUtil } from "./ui";

package/src/utils/ui.ts

@@ -1,6 +1,6 @@
-import type { ExtensionUIContext } from "@mariozechner/pi-coding-agent";
+import type { ExtensionUIContext } from "@earendil-works/pi-coding-agent";
 import { commonUtil } from "./common.js";
-import { FilterableExtensionSelectorComponent } from "./filterable-selector.js";
+import { FilterableExtensionSelectorComponent, FilterableMultiSelectComponent } from "./filterable-selector.js";
 
 function deduplicateLabels(labels: string[]): string[] {
   const seen = new Map<string, number>();
@@ -46,4 +46,38 @@ export const uiUtil = {
       (_tui, theme, _keybindings, done) => new FilterableExtensionSelectorComponent(title, options, done, theme),
     );
   },
+
+  /** Show a checkbox-style multi-select component. */
+  multiSelect: (
+    ui: ExtensionUIContext,
+    title: string,
+    options: string[],
+    initialChecked: boolean[] = [],
+    disabled: boolean[] = [],
+  ): Promise<string[] | undefined> => {
+    return ui.custom<string[] | undefined>(
+      (_tui, theme, _keybindings, done) =>
+        new FilterableMultiSelectComponent(title, options, initialChecked, done, theme, disabled),
+    );
+  },
+
+  /** Like multiSelect but skips items where the corresponding value is null (used for group headers). */
+  multiGroupedSelect: async <T>(
+    ui: ExtensionUIContext,
+    title: string,
+    labels: string[],
+    values: Array<T | null>,
+    initialChecked: boolean[] = [],
+  ): Promise<T[] | undefined> => {
+    const deduped = deduplicateLabels(labels);
+    const selected = await uiUtil.multiSelect(
+      ui,
+      title,
+      deduped,
+      initialChecked,
+      values.map((value) => value === null),
+    );
+    if (!selected) return undefined;
+    return selected.map((label) => values[deduped.indexOf(label)]).filter((value): value is T => value !== null);
+  },
 };