@ai-matrx/associations 0.6.2 → 0.7.1

package/dist/react/index.cjs

@@ -1109,6 +1109,12 @@ var useEntityRelationships = useAssociations;
 
 // src/react/hooks/useContainerLinks.ts
 var import_react3 = require("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();
@@ -1230,6 +1236,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(),
@@ -1247,7 +1278,8 @@ function useContainerLinks(args) {
     attachedIdsFor,
     linksFor,
     attach,
-    detach
+    detach,
+    setPinned
   };
 }
 
@@ -1414,7 +1446,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({
@@ -2452,7 +2484,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] = (0, import_react14.useState)("");
   const [browseToken, setBrowseToken] = (0, import_react14.useState)(null);
   const [busyKey, setBusyKey] = (0, import_react14.useState)(null);
@@ -2628,6 +2660,9 @@ function UniversalAssociationPicker(props) {
 var import_react15 = require("react");
 var import_design_system5 = require("@ai-matrx/design-system");
 var import_jsx_runtime9 = require("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);
@@ -2663,7 +2698,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 } : {}
         }
       )
     }
@@ -2674,7 +2710,8 @@ function AttachedItemsBody({
   enabled,
   links,
   onAdd,
-  onDetach
+  onDetach,
+  onSetPinned
 }) {
   const store = useAssociationsStore();
   const notifier = useNotifier();
@@ -2716,11 +2753,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}` : ""}`);
       }
@@ -2728,6 +2766,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__ */ (0, import_jsx_runtime9.jsxs)(import_jsx_runtime9.Fragment, { children: [
     titleError ? /* @__PURE__ */ (0, import_jsx_runtime9.jsxs)("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.",
@@ -2742,6 +2794,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__ */ (0, import_jsx_runtime9.jsxs)(
         "li",
         {
@@ -2761,12 +2814,35 @@ function AttachedItemsBody({
                 className: "text-foreground"
               }
             ) }),
+            onSetPinned ? /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(
+              "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: (0, import_design_system5.cn)(
+                  "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__ */ (0, import_jsx_runtime9.jsx)(
+                  PinIcon,
+                  {
+                    className: (0, import_design_system5.cn)(
+                      "h-3.5 w-3.5",
+                      pinned && "fill-primary/20"
+                    )
+                  }
+                )
+              }
+            ) : null,
             /* @__PURE__ */ (0, import_jsx_runtime9.jsx)(
               "button",
               {
                 type: "button",
                 disabled: busy,
-                onClick: () => void detach(link.resourceId),
+                onClick: () => void detach(link),
                 title: "Detach",
                 className: (0, import_design_system5.cn)(
                   "flex h-6 w-6 shrink-0 items-center justify-center rounded-md text-muted-foreground transition-colors",
@@ -2862,7 +2938,15 @@ function AssociationCard({
   const role = getContentRoleMeta(info.contentRole);
   const Icon = info.Icon ?? DefaultEntityIcon;
   const [listOpen, setListOpen] = (0, import_react16.useState)(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
@@ -2959,7 +3043,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__ */ (0, import_jsx_runtime10.jsx)(
@@ -3018,6 +3103,9 @@ function AssociationCardGrid({
 var import_react17 = require("react");
 var import_design_system8 = require("@ai-matrx/design-system");
 var import_jsx_runtime12 = require("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({
@@ -3032,6 +3120,7 @@ function useContainerLinksAdapter(container, tokens) {
       resourceId: l.resourceId,
       label: l.label,
       role: l.role,
+      pinned: metadataIsPinned2(l.metadata),
       removable: true
     }))
   ) : [];
@@ -3041,11 +3130,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(
@@ -3058,6 +3149,7 @@ function AssociationList(props) {
   const [pickerToken, setPickerToken] = (0, import_react17.useState)(null);
   const [showUniversal, setShowUniversal] = (0, import_react17.useState)(false);
   const [removingKeys, setRemovingKeys] = (0, import_react17.useState)(/* @__PURE__ */ new Set());
+  const [pinningKeys, setPinningKeys] = (0, import_react17.useState)(/* @__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(
@@ -3081,6 +3173,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);
@@ -3094,7 +3211,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__ */ (0, import_jsx_runtime12.jsxs)("div", { className: (0, import_design_system8.cn)("flex flex-col gap-2 text-foreground", props.className), children: [
     /* @__PURE__ */ (0, import_jsx_runtime12.jsxs)("div", { className: "flex items-center justify-between gap-2 px-1", children: [
       /* @__PURE__ */ (0, import_jsx_runtime12.jsxs)("h3", { className: "flex items-center gap-1.5 text-xs font-semibold uppercase tracking-wide text-muted-foreground", children: [
@@ -3271,18 +3388,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: (0, import_design_system8.cn)(
                           "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__ */ (0, import_jsx_runtime12.jsx)(
+                        children: pinningKeys.has(row.key) ? /* @__PURE__ */ (0, import_jsx_runtime12.jsx)(SpinnerIcon, { className: "h-3 w-3 animate-spin" }) : /* @__PURE__ */ (0, import_jsx_runtime12.jsx)(
                           PinIcon,
                           {
                             className: (0, import_design_system8.cn)(