@schlessera/brain-ui-react 0.5.0 → 0.6.0

package/src/lib/api-client.ts

@@ -15,6 +15,26 @@ import type {
   PublicKeyCredentialRequestOptionsJSON,
 } from "@simplewebauthn/browser";
 
+/**
+ * One hit from `brain search`. `snippet` carries the CLI's FTS highlight
+ * markers (`>>>term<<<`) — render it through `renderSnippet()` in the search
+ * panel rather than printing it raw.
+ */
+export interface BrainSearchHit {
+  path: string;
+  title: string;
+  type: string;
+  relevance: string;
+  score: number;
+  snippet: string;
+}
+
+export interface BrainSearchResponse {
+  results: BrainSearchHit[];
+  /** Degraded-mode notices, e.g. vector search unavailable so results are FTS-only. */
+  warnings: string[];
+}
+
 /** Backend id + capability flags the client renders behavior from. */
 export interface BackendInfo {
   id: string;
@@ -58,12 +78,26 @@ export const api = {
       activeSession: boolean;
     }>("/status"),
 
-  brainSearch: (q: string, opts?: { type?: string; limit?: number }) =>
-    fetchJson<{
-      results: Array<{ path: string; title: string; snippet: string; score: number }>;
-      warnings: string[];
-    }>(
-      `/brain/search?q=${encodeURIComponent(q)}${opts?.type ? `&type=${opts.type}` : ""}${opts?.limit ? `&limit=${opts.limit}` : ""}`
+  brainSearch: (
+    q: string,
+    opts?: {
+      type?: string;
+      tag?: string;
+      limit?: number;
+      mode?: "fts" | "vector" | "hybrid";
+      signal?: AbortSignal;
+    }
+  ) =>
+    // Built with encodeURIComponent rather than URLSearchParams: the latter
+    // form-encodes spaces as "+", which not every query parser turns back into
+    // a space. %20 is unambiguous.
+    fetchJson<BrainSearchResponse>(
+      `/brain/search?q=${encodeURIComponent(q)}` +
+        (opts?.type ? `&type=${encodeURIComponent(opts.type)}` : "") +
+        (opts?.tag ? `&tag=${encodeURIComponent(opts.tag)}` : "") +
+        (opts?.limit ? `&limit=${opts.limit}` : "") +
+        (opts?.mode ? `&mode=${opts.mode}` : ""),
+      { signal: opts?.signal }
     ),
 
   brainBriefing: () =>

package/src/lib/search-snippet.ts

@@ -0,0 +1,64 @@
+import { stripMarkdown } from "./strip-markdown.js";
+
+/**
+ * `brain search` returns FTS snippets with the matched terms wrapped in
+ * `>>>` / `<<<` markers (sqlite's snippet() delimiters, chosen by the CLI so
+ * they survive JSON without colliding with markdown). Vector hits come back
+ * unmarked, as a raw slice of chunk text — headings, emphasis, blank lines and
+ * all. Split a snippet into plain and matched segments, with the markdown
+ * syntax stripped and the whitespace collapsed, so a result row is one legible
+ * line with its hits highlighted.
+ */
+export interface SnippetSegment {
+  text: string;
+  hit: boolean;
+}
+
+const OPEN = ">>>";
+const CLOSE = "<<<";
+// Control characters stand in for the markers while stripMarkdown runs: a
+// snippet starting with ">>>" would otherwise lose a ">" to the blockquote
+// rule, and emphasis spanning a highlight (`**a >>>b<<<**`) has to be
+// unwrapped as one string, not per segment.
+const OPEN_MARK = "\u0001";
+const CLOSE_MARK = "\u0002";
+
+export function parseSnippet(snippet: string): SnippetSegment[] {
+  const cleaned = stripMarkdown(
+    snippet.split(OPEN).join(OPEN_MARK).split(CLOSE).join(CLOSE_MARK)
+  )
+    .replace(/\s+/g, " ")
+    .trim();
+
+  const segments: SnippetSegment[] = [];
+  let rest = cleaned;
+
+  while (rest.length > 0) {
+    const start = rest.indexOf(OPEN_MARK);
+    if (start === -1) break;
+
+    const end = rest.indexOf(CLOSE_MARK, start + 1);
+    // An unclosed marker means the snippet was clipped mid-highlight; keep the
+    // remainder as plain text rather than dropping it.
+    if (end === -1) break;
+
+    if (start > 0) push(segments, rest.slice(0, start), false);
+    push(segments, rest.slice(start + 1, end), true);
+    rest = rest.slice(end + 1);
+  }
+
+  push(segments, rest, false);
+  return segments;
+}
+
+function push(segments: SnippetSegment[], text: string, hit: boolean): void {
+  // Any stray marker left by a clipped snippet is noise, not content —
+  // squeeze the gap it leaves behind so the row reads as one sentence.
+  const clean = text
+    .split(OPEN_MARK)
+    .join("")
+    .split(CLOSE_MARK)
+    .join("")
+    .replace(/ {2,}/g, " ");
+  if (clean) segments.push({ text: clean, hit });
+}

package/src/stores/provider-store.ts

@@ -51,12 +51,23 @@ export const useProviderStore = create<ProviderState>((set, get) => ({
       const { providers, backends } = await api.providers();
       set((s) => {
         // Keep the persisted choice if it's still on offer; otherwise fall
-        // back to the first available combo.
+        // back to the first available combo. This runs on every reload of the
+        // roster, so hiding the selected model in settings moves the selection
+        // instead of leaving the composer pointing at a profile the picker no
+        // longer lists.
         const stillValid = providers.some((p) => p.id === s.selectedId);
         const selectedId =
           stillValid && s.selectedId
             ? s.selectedId
             : providers[0]?.id ?? "";
+        // Persist the fallback too — otherwise the stale id sits in
+        // localStorage and has to be re-resolved on every boot.
+        if (
+          selectedId !== s.selectedId &&
+          typeof localStorage !== "undefined"
+        ) {
+          localStorage.setItem(PROVIDER_ID_KEY, selectedId);
+        }
         return {
           available: providers,
           selectedId,

package/src/stores/ui-store.ts

@@ -7,17 +7,23 @@ interface UIState {
   sessionPanelOpen: boolean;
   syncPanelOpen: boolean;
   whatsupPanelOpen: boolean;
+  searchPanelOpen: boolean;
+  addPanelOpen: boolean;
   filePanelOpen: boolean;
   settingsPanelOpen: boolean;
   settingsTab: SettingsTab;
   toggleSessionPanel: () => void;
   toggleSyncPanel: () => void;
   toggleWhatsupPanel: () => void;
+  toggleSearchPanel: () => void;
+  toggleAddPanel: () => void;
   toggleFilePanel: () => void;
   toggleSettingsPanel: () => void;
   closeAllPanels: () => void;
   setSyncPanelOpen: (open: boolean) => void;
   setWhatsupPanelOpen: (open: boolean) => void;
+  setSearchPanelOpen: (open: boolean) => void;
+  setAddPanelOpen: (open: boolean) => void;
   setSessionPanelOpen: (open: boolean) => void;
   setFilePanelOpen: (open: boolean) => void;
   setSettingsPanelOpen: (open: boolean) => void;
@@ -30,6 +36,8 @@ const CLOSED = {
   sessionPanelOpen: false,
   syncPanelOpen: false,
   whatsupPanelOpen: false,
+  searchPanelOpen: false,
+  addPanelOpen: false,
   filePanelOpen: false,
   settingsPanelOpen: false,
 };
@@ -43,6 +51,10 @@ export const useUIStore = create<UIState>((set) => ({
     set((s) => ({ ...CLOSED, syncPanelOpen: !s.syncPanelOpen })),
   toggleWhatsupPanel: () =>
     set((s) => ({ ...CLOSED, whatsupPanelOpen: !s.whatsupPanelOpen })),
+  toggleSearchPanel: () =>
+    set((s) => ({ ...CLOSED, searchPanelOpen: !s.searchPanelOpen })),
+  toggleAddPanel: () =>
+    set((s) => ({ ...CLOSED, addPanelOpen: !s.addPanelOpen })),
   toggleFilePanel: () =>
     set((s) => ({ ...CLOSED, filePanelOpen: !s.filePanelOpen })),
   toggleSettingsPanel: () =>
@@ -50,6 +62,8 @@ export const useUIStore = create<UIState>((set) => ({
   closeAllPanels: () => set({ ...CLOSED }),
   setSyncPanelOpen: (open) => set({ syncPanelOpen: open }),
   setWhatsupPanelOpen: (open) => set({ whatsupPanelOpen: open }),
+  setSearchPanelOpen: (open) => set({ searchPanelOpen: open }),
+  setAddPanelOpen: (open) => set({ addPanelOpen: open }),
   setSessionPanelOpen: (open) => set({ sessionPanelOpen: open }),
   setFilePanelOpen: (open) => set({ filePanelOpen: open }),
   setSettingsPanelOpen: (open) => set({ settingsPanelOpen: open }),