@dxos/plugin-assistant 0.8.4-main.c85a9c8dae → 0.8.4-main.dfabb4ec29

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"]
+}