@dxos/plugin-assistant 0.8.3-staging.0fa589b → 0.8.4-main.03d5cd7b56

package/dist/lib/neutral/companion-chat-provisioner-5N4HWNVJ.mjs

@@ -0,0 +1,116 @@
+import "./chunk-J5LGTIGS.mjs";
+
+// src/capabilities/companion-chat-provisioner.ts
+import * as Effect from "effect/Effect";
+import * as Option from "effect/Option";
+import { Capabilities, Capability } from "@dxos/app-framework";
+import { Graph } from "@dxos/app-graph";
+import { AppCapabilities } from "@dxos/app-toolkit";
+import { Chat } from "@dxos/assistant-toolkit";
+import { Obj } from "@dxos/echo";
+import { log } from "@dxos/log";
+import { DeckCapabilities, PLANK_COMPANION_TYPE } from "@dxos/plugin-deck";
+import { getLinkedVariant } from "@dxos/react-ui-attention";
+import { byPosition } from "@dxos/util";
+import { ASSISTANT_COMPANION_VARIANT } from "#meta";
+import { AssistantCapabilities, AssistantOperation } from "#types";
+var __dxlog_file = "/__w/dxos/dxos/packages/plugins/plugin-assistant/src/capabilities/companion-chat-provisioner.ts";
+var companion_chat_provisioner_default = Capability.makeModule(Effect.fnUntraced(function* () {
+  const operationInvoker = yield* Capability.get(Capabilities.OperationInvoker);
+  const { graph } = yield* Capability.get(AppCapabilities.AppGraph);
+  const registry = yield* Capability.get(Capabilities.AtomRegistry);
+  const deckStateAtom = yield* Capability.get(DeckCapabilities.State);
+  const cacheAtom = yield* Capability.get(AssistantCapabilities.CompanionChatCache);
+  const stateAtom = yield* Capability.get(AssistantCapabilities.State);
+  const plankSubs = /* @__PURE__ */ new Map();
+  const unsubPlank = (plankId) => {
+    plankSubs.get(plankId)?.();
+    plankSubs.delete(plankId);
+  };
+  const unsubAllPlanks = () => {
+    for (const unsub of plankSubs.values()) {
+      unsub();
+    }
+    plankSubs.clear();
+  };
+  const provisionForPlank = (plankId, companionVariant) => {
+    const node = Graph.getNode(graph, plankId).pipe(Option.getOrNull);
+    if (!node || !Obj.isObject(node.data) || Obj.instanceOf(Chat.Chat, node.data)) {
+      return false;
+    }
+    if (resolveEffectiveVariant(graph, plankId, companionVariant) !== ASSISTANT_COMPANION_VARIANT) {
+      return false;
+    }
+    const object = node.data;
+    const companionDxn = Obj.getDXN(object).toString();
+    const cache = registry.get(cacheAtom);
+    if (cache[companionDxn]) {
+      return true;
+    }
+    const db = Obj.getDatabase(object);
+    if (!db) {
+      return false;
+    }
+    void operationInvoker.invokePromise(AssistantOperation.EnsureCompanionChat, {
+      db,
+      companionTo: object
+    }).catch((error) => log.warn("Failed to provision companion chat", {
+      plankId,
+      error
+    }, { "~LogMeta": "~LogMeta", F: __dxlog_file, L: 63, S: this }));
+    return false;
+  };
+  const provision = () => {
+    const deckState = registry.get(deckStateAtom);
+    const deck = deckState.decks[deckState.activeDeck];
+    if (!deck?.companionOpen) {
+      unsubAllPlanks();
+      return;
+    }
+    const plankIds = new Set(deck.solo ? [
+      deck.solo
+    ] : deck.active);
+    for (const trackedId of plankSubs.keys()) {
+      if (!plankIds.has(trackedId)) {
+        unsubPlank(trackedId);
+      }
+    }
+    for (const plankId of plankIds) {
+      const resolved = provisionForPlank(plankId, deck.companionVariant);
+      if (resolved) {
+        unsubPlank(plankId);
+      } else if (!plankSubs.has(plankId)) {
+        plankSubs.set(plankId, registry.subscribe(graph.connections(plankId, "child"), () => {
+          if (provisionForPlank(plankId, deck.companionVariant)) {
+            unsubPlank(plankId);
+          }
+        }));
+      }
+    }
+  };
+  provision();
+  const unsub1 = registry.subscribe(deckStateAtom, provision);
+  const unsub2 = registry.subscribe(stateAtom, provision);
+  return Capability.contributes(Capabilities.Null, null, () => Effect.sync(() => {
+    unsub1();
+    unsub2();
+    unsubAllPlanks();
+  }));
+}));
+var resolveEffectiveVariant = (graph, plankId, preferredVariant) => {
+  const companions = Graph.getConnections(graph, plankId, "child").filter((node) => node.type === PLANK_COMPANION_TYPE).toSorted((a, b) => byPosition(a.properties, b.properties));
+  if (companions.length === 0) {
+    return void 0;
+  }
+  if (preferredVariant) {
+    const preferred = companions.find((companion) => getLinkedVariant(companion.id) === preferredVariant);
+    if (preferred) {
+      return getLinkedVariant(preferred.id);
+    }
+  }
+  return getLinkedVariant(companions[0].id);
+};
+export {
+  companion_chat_provisioner_default as default
+};
+//# sourceMappingURL=companion-chat-provisioner-5N4HWNVJ.mjs.map

package/dist/lib/neutral/companion-chat-provisioner-5N4HWNVJ.mjs.map

@@ -0,0 +1,7 @@
+{
+  "version": 3,
+  "sources": ["../../../src/capabilities/companion-chat-provisioner.ts"],
+  "sourcesContent": ["//\n// Copyright 2025 DXOS.org\n//\n\nimport { type Registry } from '@effect-atom/atom-react';\nimport * as Effect from 'effect/Effect';\nimport * as Option from 'effect/Option';\n\nimport { Capabilities, Capability } from '@dxos/app-framework';\nimport { Graph, type Node } from '@dxos/app-graph';\nimport { AppCapabilities } from '@dxos/app-toolkit';\nimport { Chat } from '@dxos/assistant-toolkit';\nimport { Obj } from '@dxos/echo';\nimport { log } from '@dxos/log';\nimport { DeckCapabilities, PLANK_COMPANION_TYPE, type StoredDeckState } from '@dxos/plugin-deck';\nimport { getLinkedVariant } from '@dxos/react-ui-attention';\nimport { byPosition } from '@dxos/util';\n\nimport { ASSISTANT_COMPANION_VARIANT } from '#meta';\nimport { AssistantCapabilities, AssistantOperation } from '#types';\n\n/**\n * Non-React capability that watches deck companion state and provisions transient chats\n * for active planks when the assistant companion is selected.\n */\nexport default Capability.makeModule(\n  Effect.fnUntraced(function* () {\n    const operationInvoker = yield* Capability.get(Capabilities.OperationInvoker);\n    const { graph } = yield* Capability.get(AppCapabilities.AppGraph);\n    const registry: Registry.Registry = yield* Capability.get(Capabilities.AtomRegistry);\n    const deckStateAtom = yield* Capability.get(DeckCapabilities.State);\n    const cacheAtom = yield* Capability.get(AssistantCapabilities.CompanionChatCache);\n    const stateAtom = yield* Capability.get(AssistantCapabilities.State);\n\n    const plankSubs = new Map<string, () => void>();\n\n    /** Unsubscribe a single plank and remove it from the map. */\n    const unsubPlank = (plankId: string) => {\n      plankSubs.get(plankId)?.();\n      plankSubs.delete(plankId);\n    };\n\n    /** Unsubscribe all per-plank subscriptions. */\n    const unsubAllPlanks = () => {\n      for (const unsub of plankSubs.values()) {\n        unsub();\n      }\n      plankSubs.clear();\n    };\n\n    /**\n     * Attempt to provision for a single plank.\n     * Returns true when the plank is resolved (provisioned or already cached)\n     * so the caller can tear down the connection subscription.\n     */\n    const provisionForPlank = (plankId: string, companionVariant: string | undefined): boolean => {\n      const node: Node.Node | null = Graph.getNode(graph, plankId).pipe(Option.getOrNull);\n      if (!node || !Obj.isObject(node.data) || Obj.instanceOf(Chat.Chat, node.data)) {\n        return false;\n      }\n\n      if (resolveEffectiveVariant(graph, plankId, companionVariant) !== ASSISTANT_COMPANION_VARIANT) {\n        return false;\n      }\n\n      const object = node.data;\n      const companionDxn = Obj.getDXN(object).toString();\n      const cache = registry.get(cacheAtom);\n      if (cache[companionDxn]) {\n        return true;\n      }\n\n      const db = Obj.getDatabase(object);\n      if (!db) {\n        return false;\n      }\n\n      void operationInvoker\n        .invokePromise(AssistantOperation.EnsureCompanionChat, { db, companionTo: object })\n        .catch((error) => log.warn('Failed to provision companion chat', { plankId, error }));\n\n      return false;\n    };\n\n    const provision = () => {\n      const deckState: StoredDeckState = registry.get(deckStateAtom);\n      const deck = deckState.decks[deckState.activeDeck];\n      if (!deck?.companionOpen) {\n        unsubAllPlanks();\n        return;\n      }\n\n      const plankIds = new Set(deck.solo ? [deck.solo] : deck.active);\n\n      // Remove subscriptions for planks that are no longer active.\n      for (const trackedId of plankSubs.keys()) {\n        if (!plankIds.has(trackedId)) {\n          unsubPlank(trackedId);\n        }\n      }\n\n      for (const plankId of plankIds) {\n        const resolved = provisionForPlank(plankId, deck.companionVariant);\n\n        if (resolved) {\n          // Already provisioned — no need to watch connections.\n          unsubPlank(plankId);\n        } else if (!plankSubs.has(plankId)) {\n          // Not yet resolved — subscribe to child connections so we re-try\n          // when graph builder extensions add companion nodes (after expand).\n          plankSubs.set(\n            plankId,\n            registry.subscribe(graph.connections(plankId, 'child'), () => {\n              if (provisionForPlank(plankId, deck.companionVariant)) {\n                unsubPlank(plankId);\n              }\n            }),\n          );\n        }\n      }\n    };\n\n    provision();\n\n    const unsub1 = registry.subscribe(deckStateAtom, provision);\n    const unsub2 = registry.subscribe(stateAtom, provision);\n\n    return Capability.contributes(Capabilities.Null, null, () =>\n      Effect.sync(() => {\n        unsub1();\n        unsub2();\n        unsubAllPlanks();\n      }),\n    );\n  }),\n);\n\n/**\n * Mirrors useSelectedCompanion fallback logic outside of React.\n * Returns the variant that would actually be rendered for a given plank.\n */\nconst resolveEffectiveVariant = (\n  graph: Graph.BaseGraph,\n  plankId: string,\n  preferredVariant: string | undefined,\n): string | undefined => {\n  const companions = Graph.getConnections(graph, plankId, 'child')\n    .filter((node) => node.type === PLANK_COMPANION_TYPE)\n    .toSorted((a, b) => byPosition(a.properties, b.properties));\n\n  if (companions.length === 0) {\n    return undefined;\n  }\n\n  if (preferredVariant) {\n    const preferred = companions.find((companion) => getLinkedVariant(companion.id) === preferredVariant);\n    if (preferred) {\n      return getLinkedVariant(preferred.id);\n    }\n  }\n\n  return getLinkedVariant(companions[0].id);\n};\n"],
+  "mappings": ";;;AAKA,YAAYA,YAAY;AACxB,YAAYC,YAAY;AAExB,SAASC,cAAcC,kBAAkB;AACzC,SAASC,aAAwB;AACjC,SAASC,uBAAuB;AAChC,SAASC,YAAY;AACrB,SAASC,WAAW;AACpB,SAASC,WAAW;AACpB,SAASC,kBAAkBC,4BAAkD;AAC7E,SAASC,wBAAwB;AACjC,SAASC,kBAAkB;AAE3B,SAASC,mCAAmC;AAC5C,SAASC,uBAAuBC,0BAA0B;AAE1D,IAAA,eAAA;AAOI,IAAM,qCAAS,WAAUZ,WAAeE,kBAAAA,aAAwB;AAChE,QAAMW,mBAAqCb,OAAAA,WAAeD,IAAAA,aAAae,gBAAY;AACnF,QAAMC,EAAAA,MAAAA,IAAAA,OAAgB,WAAOf,IAAWgB,gBAAIV,QAAiBW;AAC7D,QAAMC,WAAAA,OAAY,WAAOlB,IAAc,aAACW,YAAsBQ;AAC9D,QAAMC,gBAAY,OAAOpB,WAAeW,IAAAA,iBAAAA,KAAsBM;AAE9D,QAAMI,YAAY,OAAIC,WAAAA,IAAAA,sBAAAA,kBAAAA;AAEtB,QAAA,YAAA,OAAA,WAAA,IAAA,sBACA,KAAA;QACED,YAAcE,oBAAAA,IAAAA;AACGA,QAAAA,aAAAA,CAAAA,YAAAA;AACnB,cAAA,IAAA,OAAA,IAAA;AAEA,cAAA,OAAA,OAAA;;AAGIC,QAAAA,iBAAAA,MAAAA;AACF,eAAA,SAAA,UAAA,OAAA,GAAA;AACAH,YAAUI;IACZ;AAEA,cAAA,MAAA;;AAOE,QAAKC,oBAAqB,CAACA,SAAS,qBAAmB;UACrD,OAAO,MAAA,QAAA,OAAA,OAAA,EAAA,KAAA,gBAAA;AACT,QAAA,CAAA,QAAA,CAAA,IAAA,SAAA,KAAA,IAAA,KAAA,IAAA,WAAA,KAAA,MAAA,KAAA,IAAA,GAAA;AAEIC,aAAAA;;AAEJ,QAAA,wBAAA,OAAA,SAAA,gBAAA,MAAA,6BAAA;AAEA,aAAMC;IACN;AACA,UAAMC,SAAQhB,KAAAA;AACd,UAAIgB,eAAMC,IAAe,OAAA,MAAA,EAAA,SAAA;UACvB,QAAO,SAAA,IAAA,SAAA;AACT,QAAA,MAAA,YAAA,GAAA;AAEA,aAAW1B;IACX;UACE,KAAO,IAAA,YAAA,MAAA;AACT,QAAA,CAAA,IAAA;AAEA,aAAK2B;;SAC0DC,iBAAaJ,cAAAA,mBAAAA,qBAAAA;MACzEK;mBAAkEV;aAASW,CAAAA,UAAAA,IAAAA,KAAAA,sCAAAA;MAAM;MAE7E;IACT,GAAA,EAAA,YAAA,YAAA,GAAA,cAAA,GAAA,IAAA,GAAA,KAAA,CAAA,CAAA;AAEA,WAAMC;;QAEJ,YAAaC,MAAAA;AACb,UAAKC,YAAMC,SAAe,IAAA,aAAA;UACxBC,OAAAA,UAAAA,MAAAA,UAAAA,UAAAA;QACA,CAAA,MAAA,eAAA;AACF,qBAAA;AAEA;;UAAmDF,WAAW,IAAA,IAAA,KAAA,OAAA;MAE9D,KAAA;IACA,IAAA,KAAK,MAAMG;eAEPC,aAAWD,UAAAA,KAAAA,GAAAA;AACb,UAAA,CAAA,SAAA,IAAA,SAAA,GAAA;AACF,mBAAA,SAAA;MAEA;;eAGME,WAAU,UAAA;YACZ,WAAA,kBAAA,SAAA,KAAA,gBAAsD;UACtDD,UAAAA;AAEA,mBAAA,OAAA;iBACA,CAAA,UAAA,IAAA,OAAA,GAAA;kBAKMA,IAAAA,SAAWlB,SAAAA,UAAAA,MAAAA,YAAAA,SAAAA,OAAAA,GAAAA,MAAAA;AACb,cAAA,kBAAA,SAAA,KAAA,gBAAA,GAAA;AACF,uBAAA,OAAA;UAEJ;QACF,CAAA,CAAA;MACF;IAEAY;EAEA;AACA,YAAMQ;AAEN,QAAA,SAAO3C,SAAW4C,UAAY7C,eAAmB,SAAM;iBAEnD8C,SAAAA,UAAAA,WAAAA,SAAAA;oBACAF,YAAAA,aAAAA,MAAAA,MAAAA,MAAAA,YAAAA,MAAAA;AACAJ,WAAAA;AACF,WAAA;AAGJ,mBAAA;EAEF,CAAA,CAAA;;AAaE,IAAIO,0BAAyB,CAAA,OAAA,SAAA,qBAAA;QAC3B,aAAOC,MAAAA,eAAAA,OAAAA,SAAAA,OAAAA,EAAAA,OAAAA,CAAAA,SAAAA,KAAAA,SAAAA,oBAAAA,EAAAA,SAAAA,CAAAA,GAAAA,MAAAA,WAAAA,EAAAA,YAAAA,EAAAA,UAAAA,CAAAA;AACT,MAAA,WAAA,WAAA,GAAA;AAEIC,WAAAA;;MAEF,kBAAe;UACb,YAAOxC,WAAiByC,KAAAA,CAAAA,cAAY,iBAAA,UAAA,EAAA,MAAA,gBAAA;AACtC,QAAA,WAAA;AACF,aAAA,iBAAA,UAAA,EAAA;IAEA;EACF;;;",
+  "names": ["Effect", "Option", "Capabilities", "Capability", "Graph", "AppCapabilities", "Chat", "Obj", "log", "DeckCapabilities", "PLANK_COMPANION_TYPE", "getLinkedVariant", "byPosition", "ASSISTANT_COMPANION_VARIANT", "AssistantCapabilities", "AssistantOperation", "registry", "AtomRegistry", "deckStateAtom", "get", "State", "cacheAtom", "CompanionChatCache", "stateAtom", "plankSubs", "Map", "plankId", "unsub", "clear", "node", "resolveEffectiveVariant", "object", "cache", "companionDxn", "operationInvoker", "companionTo", "catch", "error", "provision", "deckState", "deck", "companionOpen", "unsubAllPlanks", "trackedId", "unsubPlank", "resolved", "unsub2", "contributes", "unsub1", "companions", "undefined", "preferredVariant", "preferred"]
+}