@ai-matrx/associations 0.6.2 → 0.7.1

package/dist/react/index.d.cts

@@ -1,7 +1,7 @@
 import * as react from 'react';
 import { ReactNode, ComponentType } from 'react';
-import { j as AssociationsNotifier, W as WindowShellPort, k as CapturePort, P as PickerOverrideMap, E as EntityDoorsPort, l as AuthorDisplayPort, m as AssociationsStore, n as CategoryMutationResult, o as CategoryDimension, p as PlatformCategory, q as CategoriesEntry, r as EntityTypeToken, s as UniversalCandidate, t as CandidateRecord, u as PlatformComment, v as CommentsEntry, w as CommentMutationResult, e as AssociationPickerProps, x as EntityDoorProps, y as EntityRefProps, z as ContentRole } from '../CommentThread-CqZHMTsH.cjs';
-export { B as AssociationCaptureToolbar, A as AssociationCaptureToolbarProps, D as AssociationCard, F as AssociationCardGrid, b as AssociationCardGridProps, a as AssociationCardProps, G as AssociationEntitySelect, H as AssociationEntitySelectAdapter, I as AssociationEntitySelectItem, c as AssociationEntitySelectProps, J as AssociationList, d as AssociationListProps, K as AssociationWindow, f as AssociationWindowProps, L as AssociationWriteResult, M as AttachedItemsSheet, g as AttachedItemsSheetProps, N as CaptureAttach, O as CaptureAttachResult, Q as CaptureToolbarAction, R as CategorySelect, C as CategorySelectProps, S as CategoryTagPicker, h as CategoryTagPickerProps, T as CommentComposer, V as CommentComposerProps, X as CommentThread, i as CommentThreadProps, Y as ContainerLink, Z as ContainerResourceRow, _ as ContainerResourcesAdapter, $ as DefaultEntityIcon, a0 as PrimaryEntity, a1 as PrimaryEntityProvider, a2 as TagPickerIcon, a3 as UniversalAssociationPicker, U as UniversalAssociationPickerProps, a4 as UseAssociationEntitySelectAdapterArgs, a5 as UseAssociationsArgs, a6 as UseAssociationsReturn, a7 as UseContainerLinksArgs, a8 as UseContainerLinksReturn, a9 as attachedKey, aa as buildCommentTree, ab as useAssociationEntitySelectAdapter, ac as useAssociations, ad as useContainerLinks, ae as useContainerLinksAdapter, af as useEntityRelationships, ag as usePrimaryEntity } from '../CommentThread-CqZHMTsH.cjs';
+import { j as AssociationsNotifier, W as WindowShellPort, k as CapturePort, P as PickerOverrideMap, E as EntityDoorsPort, l as AuthorDisplayPort, m as AssociationsStore, n as CategoryMutationResult, o as CategoryDimension, p as PlatformCategory, q as CategoriesEntry, r as EntityTypeToken, s as UniversalCandidate, t as CandidateRecord, u as PlatformComment, v as CommentsEntry, w as CommentMutationResult, e as AssociationPickerProps, x as EntityDoorProps, y as EntityRefProps, z as ContentRole } from '../CommentThread-HW4hLdsd.cjs';
+export { B as AssociationCaptureToolbar, A as AssociationCaptureToolbarProps, D as AssociationCard, F as AssociationCardGrid, b as AssociationCardGridProps, a as AssociationCardProps, G as AssociationEntitySelect, H as AssociationEntitySelectAdapter, I as AssociationEntitySelectItem, c as AssociationEntitySelectProps, J as AssociationList, d as AssociationListProps, K as AssociationWindow, f as AssociationWindowProps, L as AssociationWriteResult, M as AttachedItemsSheet, g as AttachedItemsSheetProps, N as CaptureAttach, O as CaptureAttachResult, Q as CaptureToolbarAction, R as CategorySelect, C as CategorySelectProps, S as CategoryTagPicker, h as CategoryTagPickerProps, T as CommentComposer, V as CommentComposerProps, X as CommentThread, i as CommentThreadProps, Y as ContainerLink, Z as ContainerResourceRow, _ as ContainerResourcesAdapter, $ as DefaultEntityIcon, a0 as PrimaryEntity, a1 as PrimaryEntityProvider, a2 as TagPickerIcon, a3 as UniversalAssociationPicker, U as UniversalAssociationPickerProps, a4 as UseAssociationEntitySelectAdapterArgs, a5 as UseAssociationsArgs, a6 as UseAssociationsReturn, a7 as UseContainerLinksArgs, a8 as UseContainerLinksReturn, a9 as attachedKey, aa as buildCommentTree, ab as useAssociationEntitySelectAdapter, ac as useAssociations, ad as useContainerLinks, ae as useContainerLinksAdapter, af as useEntityRelationships, ag as usePrimaryEntity } from '../CommentThread-HW4hLdsd.cjs';
 
 interface AssociationsUiPorts {
     notifier?: AssociationsNotifier;

package/dist/react/index.d.ts

@@ -1,7 +1,7 @@
 import * as react from 'react';
 import { ReactNode, ComponentType } from 'react';
-import { j as AssociationsNotifier, W as WindowShellPort, k as CapturePort, P as PickerOverrideMap, E as EntityDoorsPort, l as AuthorDisplayPort, m as AssociationsStore, n as CategoryMutationResult, o as CategoryDimension, p as PlatformCategory, q as CategoriesEntry, r as EntityTypeToken, s as UniversalCandidate, t as CandidateRecord, u as PlatformComment, v as CommentsEntry, w as CommentMutationResult, e as AssociationPickerProps, x as EntityDoorProps, y as EntityRefProps, z as ContentRole } from '../CommentThread-CqZHMTsH.js';
-export { B as AssociationCaptureToolbar, A as AssociationCaptureToolbarProps, D as AssociationCard, F as AssociationCardGrid, b as AssociationCardGridProps, a as AssociationCardProps, G as AssociationEntitySelect, H as AssociationEntitySelectAdapter, I as AssociationEntitySelectItem, c as AssociationEntitySelectProps, J as AssociationList, d as AssociationListProps, K as AssociationWindow, f as AssociationWindowProps, L as AssociationWriteResult, M as AttachedItemsSheet, g as AttachedItemsSheetProps, N as CaptureAttach, O as CaptureAttachResult, Q as CaptureToolbarAction, R as CategorySelect, C as CategorySelectProps, S as CategoryTagPicker, h as CategoryTagPickerProps, T as CommentComposer, V as CommentComposerProps, X as CommentThread, i as CommentThreadProps, Y as ContainerLink, Z as ContainerResourceRow, _ as ContainerResourcesAdapter, $ as DefaultEntityIcon, a0 as PrimaryEntity, a1 as PrimaryEntityProvider, a2 as TagPickerIcon, a3 as UniversalAssociationPicker, U as UniversalAssociationPickerProps, a4 as UseAssociationEntitySelectAdapterArgs, a5 as UseAssociationsArgs, a6 as UseAssociationsReturn, a7 as UseContainerLinksArgs, a8 as UseContainerLinksReturn, a9 as attachedKey, aa as buildCommentTree, ab as useAssociationEntitySelectAdapter, ac as useAssociations, ad as useContainerLinks, ae as useContainerLinksAdapter, af as useEntityRelationships, ag as usePrimaryEntity } from '../CommentThread-CqZHMTsH.js';
+import { j as AssociationsNotifier, W as WindowShellPort, k as CapturePort, P as PickerOverrideMap, E as EntityDoorsPort, l as AuthorDisplayPort, m as AssociationsStore, n as CategoryMutationResult, o as CategoryDimension, p as PlatformCategory, q as CategoriesEntry, r as EntityTypeToken, s as UniversalCandidate, t as CandidateRecord, u as PlatformComment, v as CommentsEntry, w as CommentMutationResult, e as AssociationPickerProps, x as EntityDoorProps, y as EntityRefProps, z as ContentRole } from '../CommentThread-HW4hLdsd.js';
+export { B as AssociationCaptureToolbar, A as AssociationCaptureToolbarProps, D as AssociationCard, F as AssociationCardGrid, b as AssociationCardGridProps, a as AssociationCardProps, G as AssociationEntitySelect, H as AssociationEntitySelectAdapter, I as AssociationEntitySelectItem, c as AssociationEntitySelectProps, J as AssociationList, d as AssociationListProps, K as AssociationWindow, f as AssociationWindowProps, L as AssociationWriteResult, M as AttachedItemsSheet, g as AttachedItemsSheetProps, N as CaptureAttach, O as CaptureAttachResult, Q as CaptureToolbarAction, R as CategorySelect, C as CategorySelectProps, S as CategoryTagPicker, h as CategoryTagPickerProps, T as CommentComposer, V as CommentComposerProps, X as CommentThread, i as CommentThreadProps, Y as ContainerLink, Z as ContainerResourceRow, _ as ContainerResourcesAdapter, $ as DefaultEntityIcon, a0 as PrimaryEntity, a1 as PrimaryEntityProvider, a2 as TagPickerIcon, a3 as UniversalAssociationPicker, U as UniversalAssociationPickerProps, a4 as UseAssociationEntitySelectAdapterArgs, a5 as UseAssociationsArgs, a6 as UseAssociationsReturn, a7 as UseContainerLinksArgs, a8 as UseContainerLinksReturn, a9 as attachedKey, aa as buildCommentTree, ab as useAssociationEntitySelectAdapter, ac as useAssociations, ad as useContainerLinks, ae as useContainerLinksAdapter, af as useEntityRelationships, ag as usePrimaryEntity } from '../CommentThread-HW4hLdsd.js';
 
 interface AssociationsUiPorts {
     notifier?: AssociationsNotifier;

package/dist/react/index.js

@@ -1052,6 +1052,12 @@ import {
   useRef as useRef3,
   useState
 } from "react";
+function metadataRecord(metadata) {
+  if (metadata === null || typeof metadata !== "object" || Array.isArray(metadata)) {
+    return metadata == null ? {} : null;
+  }
+  return Object.fromEntries(Object.entries(metadata));
+}
 function useContainerLinks(args) {
   const { containerType, containerId, orgId } = args;
   const store = useAssociationsStore();
@@ -1173,6 +1179,31 @@ function useContainerLinks(args) {
     }
     return result;
   };
+  const setPinned = async (token, resourceId, pinned, role) => {
+    if (!containerId) return { ok: false, error: "Missing container id" };
+    const exactRole = role ?? null;
+    const link = incoming.find(
+      (candidate) => candidate.token === token && candidate.resourceId === resourceId && candidate.role === exactRole
+    );
+    if (!link) return { ok: false, error: "Association edge not found" };
+    const currentMetadata = metadataRecord(link.metadata);
+    if (!currentMetadata) {
+      return {
+        ok: false,
+        error: "Association metadata is not an object and cannot be updated safely"
+      };
+    }
+    return store.add({
+      sourceType: token,
+      sourceId: resourceId,
+      targetType: containerType,
+      targetId: containerId,
+      ...orgId != null ? { orgId } : {},
+      ...link.label != null ? { label: link.label } : {},
+      ...role !== void 0 ? { role } : {},
+      metadata: { ...currentMetadata, pinned }
+    });
+  };
   const reload = async () => {
     await Promise.all([
       reloadGeneric(),
@@ -1190,7 +1221,8 @@ function useContainerLinks(args) {
     attachedIdsFor,
     linksFor,
     attach,
-    detach
+    detach,
+    setPinned
   };
 }
 
@@ -1357,7 +1389,7 @@ function useUniversalEntitySearch(args) {
     const timer = setTimeout(async () => {
       if (runRef.current !== run) return;
       setLoading(true);
-      const activeTokens = tokensKey ? tokensKey.split(",") : store.registry.curatedTokens();
+      const activeTokens = tokensKey ? tokensKey.split(",") : store.registry.listableTokens();
       let next = [];
       if (trimmed || emptyQueryMode === "candidates") {
         next = await store.candidates.searchCandidatesAcrossTokens({
@@ -2401,7 +2433,7 @@ function UniversalAssociationPicker(props) {
   const store = useAssociationsStore();
   const { pickerOverrides } = useAssociationsUiPorts();
   const notifier = useNotifier();
-  const tokens = props.tokens ?? store.registry.curatedTokens();
+  const tokens = props.tokens ?? store.registry.listableTokens();
   const [query, setQuery] = useState8("");
   const [browseToken, setBrowseToken] = useState8(null);
   const [busyKey, setBusyKey] = useState8(null);
@@ -2577,6 +2609,9 @@ function UniversalAssociationPicker(props) {
 import { useEffect as useEffect10, useState as useState9 } from "react";
 import { cn as cn5 } from "@ai-matrx/design-system";
 import { Fragment as Fragment2, jsx as jsx9, jsxs as jsxs7 } from "react/jsx-runtime";
+function metadataIsPinned(metadata) {
+  return metadata !== null && typeof metadata === "object" && !Array.isArray(metadata) && Reflect.get(metadata, "pinned") === true;
+}
 function AttachedItemsSheet(props) {
   const store = useAssociationsStore();
   const info = store.registry.getEntityInfo(props.token);
@@ -2612,7 +2647,8 @@ function AttachedItemsSheet(props) {
           enabled: props.open,
           links: props.links,
           ...props.onAdd ? { onAdd: props.onAdd } : {},
-          onDetach: props.onDetach
+          onDetach: props.onDetach,
+          ...props.onSetPinned ? { onSetPinned: props.onSetPinned } : {}
         }
       )
     }
@@ -2623,7 +2659,8 @@ function AttachedItemsBody({
   enabled,
   links,
   onAdd,
-  onDetach
+  onDetach,
+  onSetPinned
 }) {
   const store = useAssociationsStore();
   const notifier = useNotifier();
@@ -2665,11 +2702,12 @@ function AttachedItemsBody({
       cancelled = true;
     };
   }, [token, idKey, enabled, store]);
-  const detach = async (resourceId) => {
+  const detach = async (link) => {
+    const resourceId = link.resourceId;
     if (busyId) return;
     setBusyId(resourceId);
     try {
-      const res = await onDetach(resourceId);
+      const res = await onDetach(resourceId, link.role);
       if (!res.ok) {
         notifier.error(`Couldn't detach${res.error ? `: ${res.error}` : ""}`);
       }
@@ -2677,6 +2715,20 @@ function AttachedItemsBody({
       setBusyId(null);
     }
   };
+  const setPinned = async (link, pinned) => {
+    if (!onSetPinned || busyId) return;
+    setBusyId(link.resourceId);
+    try {
+      const res = await onSetPinned(link.resourceId, pinned, link.role);
+      if (!res.ok) {
+        notifier.error(
+          `Couldn't ${pinned ? "pin" : "unpin"}${res.error ? `: ${res.error}` : ""}`
+        );
+      }
+    } finally {
+      setBusyId(null);
+    }
+  };
   return /* @__PURE__ */ jsxs7(Fragment2, { children: [
     titleError ? /* @__PURE__ */ jsxs7("div", { className: "mb-2 rounded-md border border-destructive/40 bg-destructive/10 px-3 py-2 text-[12px] text-destructive", children: [
       "Showing names recorded at attach time \u2014 they may be out of date.",
@@ -2691,6 +2743,7 @@ function AttachedItemsBody({
       const missing = !titles.has(link.resourceId);
       const name = resolved ?? link.label ?? (missing ? null : "Untitled");
       const busy = busyId === link.resourceId;
+      const pinned = metadataIsPinned(link.metadata);
       return /* @__PURE__ */ jsxs7(
         "li",
         {
@@ -2710,12 +2763,35 @@ function AttachedItemsBody({
                 className: "text-foreground"
               }
             ) }),
+            onSetPinned ? /* @__PURE__ */ jsx9(
+              "button",
+              {
+                type: "button",
+                disabled: busy,
+                onClick: () => void setPinned(link, !pinned),
+                title: pinned ? "Unpin (stops staying in agent context)" : "Pin \u2014 always in agent context",
+                "aria-label": pinned ? "Unpin" : "Pin",
+                className: cn5(
+                  "flex h-6 w-6 shrink-0 items-center justify-center rounded-md transition-colors hover:bg-accent hover:text-foreground disabled:opacity-50",
+                  pinned ? "text-primary" : "text-muted-foreground"
+                ),
+                children: /* @__PURE__ */ jsx9(
+                  PinIcon,
+                  {
+                    className: cn5(
+                      "h-3.5 w-3.5",
+                      pinned && "fill-primary/20"
+                    )
+                  }
+                )
+              }
+            ) : null,
             /* @__PURE__ */ jsx9(
               "button",
               {
                 type: "button",
                 disabled: busy,
-                onClick: () => void detach(link.resourceId),
+                onClick: () => void detach(link),
                 title: "Detach",
                 className: cn5(
                   "flex h-6 w-6 shrink-0 items-center justify-center rounded-md text-muted-foreground transition-colors",
@@ -2811,7 +2887,15 @@ function AssociationCard({
   const role = getContentRoleMeta(info.contentRole);
   const Icon = info.Icon ?? DefaultEntityIcon;
   const [listOpen, setListOpen] = useState10(false);
-  const { status, countFor, attachedIdsFor, linksFor, attach, detach } = useContainerLinks({
+  const {
+    status,
+    countFor,
+    attachedIdsFor,
+    linksFor,
+    attach,
+    detach,
+    setPinned
+  } = useContainerLinks({
     containerType: container?.type ?? "organization",
     containerId: container?.id ?? null,
     orgId: container?.orgId ?? null
@@ -2908,7 +2992,8 @@ function AssociationCard({
             setOpen(true);
           }
         } : {},
-        onDetach: (resourceId) => detach(token, resourceId)
+        onDetach: (resourceId, role2) => detach(token, resourceId, role2 ?? void 0),
+        onSetPinned: (resourceId, pinned, role2) => setPinned(token, resourceId, pinned, role2 ?? void 0)
       }
     ),
     canAttach && /* @__PURE__ */ jsx10(
@@ -2967,6 +3052,9 @@ function AssociationCardGrid({
 import { useState as useState11 } from "react";
 import { cn as cn8, Skeleton } from "@ai-matrx/design-system";
 import { jsx as jsx12, jsxs as jsxs10 } from "react/jsx-runtime";
+function metadataIsPinned2(metadata) {
+  return metadata !== null && typeof metadata === "object" && !Array.isArray(metadata) && Reflect.get(metadata, "pinned") === true;
+}
 function useContainerLinksAdapter(container, tokens) {
   const store = useAssociationsStore();
   const links = useContainerLinks({
@@ -2981,6 +3069,7 @@ function useContainerLinksAdapter(container, tokens) {
       resourceId: l.resourceId,
       label: l.label,
       role: l.role,
+      pinned: metadataIsPinned2(l.metadata),
       removable: true
     }))
   ) : [];
@@ -2990,11 +3079,13 @@ function useContainerLinksAdapter(container, tokens) {
     reload: links.reload,
     rows,
     attach: async (token, id, title) => links.attach(token, id, title),
-    detach: async (token, id, role) => links.detach(token, id, role)
+    detach: async (token, id, role) => links.detach(token, id, role),
+    setPinned: async (token, id, pinned, role) => links.setPinned(token, id, pinned, role)
   };
 }
 function AssociationList(props) {
   const store = useAssociationsStore();
+  const notifier = useNotifier();
   const ctxEntity = usePrimaryEntity();
   const container = props.container ?? ctxEntity ?? null;
   const defaultAdapter = useContainerLinksAdapter(
@@ -3007,6 +3098,7 @@ function AssociationList(props) {
   const [pickerToken, setPickerToken] = useState11(null);
   const [showUniversal, setShowUniversal] = useState11(false);
   const [removingKeys, setRemovingKeys] = useState11(/* @__PURE__ */ new Set());
+  const [pinningKeys, setPinningKeys] = useState11(/* @__PURE__ */ new Set());
   const rows = tokenFilter ? adapter.rows.filter((r) => tokenFilter.includes(r.token)) : adapter.rows;
   const visibleRows = rows.filter((r) => !removingKeys.has(r.key));
   const { titleFor } = useEntityTitles(
@@ -3030,6 +3122,31 @@ function AssociationList(props) {
     });
     return res;
   };
+  const handlePin = async (row, pinned) => {
+    if (!adapter.setPinned) return;
+    const info = store.registry.tryGetEntityInfo(row.token);
+    if (!info) return;
+    setPinningKeys((prev) => new Set(prev).add(row.key));
+    try {
+      const res = await adapter.setPinned(
+        info.token,
+        row.resourceId,
+        pinned,
+        row.role ?? void 0
+      );
+      if (!res.ok) {
+        notifier.error(
+          `Couldn't ${pinned ? "pin" : "unpin"} resource${res.error ? `: ${res.error}` : ""}`
+        );
+      }
+    } finally {
+      setPinningKeys((prev) => {
+        const next = new Set(prev);
+        next.delete(row.key);
+        return next;
+      });
+    }
+  };
   const openRow = (row) => {
     if (props.openEntity) {
       props.openEntity(row.token, row.resourceId);
@@ -3043,7 +3160,7 @@ function AssociationList(props) {
   };
   const grouped = groupRows(store, visibleRows);
   const isLoading = adapter.status === "loading" || adapter.status === "idle";
-  const attachableTokens = tokenFilter ?? store.registry.curatedTokens();
+  const attachableTokens = tokenFilter ?? store.registry.listableTokens();
   return /* @__PURE__ */ jsxs10("div", { className: cn8("flex flex-col gap-2 text-foreground", props.className), children: [
     /* @__PURE__ */ jsxs10("div", { className: "flex items-center justify-between gap-2 px-1", children: [
       /* @__PURE__ */ jsxs10("h3", { className: "flex items-center gap-1.5 text-xs font-semibold uppercase tracking-wide text-muted-foreground", children: [
@@ -3220,18 +3337,15 @@ function AssociationList(props) {
                       "button",
                       {
                         type: "button",
-                        onClick: () => void adapter.setPinned?.(
-                          info.token,
-                          row.resourceId,
-                          !(row.pinned ?? false)
-                        ),
+                        disabled: pinningKeys.has(row.key),
+                        onClick: () => void handlePin(row, !(row.pinned ?? false)),
                         title: row.pinned ? "Unpin (stops staying in agent context)" : "Pin \u2014 always in agent context",
                         "aria-label": row.pinned ? "Unpin" : "Pin",
                         className: cn8(
                           "shrink-0 rounded p-0.5 transition-colors hover:!text-foreground",
                           row.pinned ? "text-primary" : "text-muted-foreground/0 group-hover:text-muted-foreground"
                         ),
-                        children: /* @__PURE__ */ jsx12(
+                        children: pinningKeys.has(row.key) ? /* @__PURE__ */ jsx12(SpinnerIcon, { className: "h-3 w-3 animate-spin" }) : /* @__PURE__ */ jsx12(
                           PinIcon,
                           {
                             className: cn8(