import type {  EnvironmentId,  OrchestrationV2ShellSnapshot,  OrchestrationV2ThreadShell,  ProjectId,  ScopedProjectRef,  ScopedThreadRef,  ThreadId,} from "@t3tools/contracts";import { Atom } from "effect/unstable/reactivity"; import type { EnvironmentThreadShell } from "./models.ts";import { presentThreadShell } from "./models.ts";import type { EnvironmentCatalogState } from "./connections.ts";import {  arrayElementsEqual,  parseProjectRefCollectionKey,  parseThreadKey,  projectRefCollectionKey,  threadKey,  threadRefsEqual,} from "./entities.ts"; const EMPTY_THREADS: ReadonlyArray<OrchestrationV2ThreadShell> = Object.freeze([]);const EMPTY_SCOPED_THREAD_REFS: ReadonlyArray<ScopedThreadRef> = Object.freeze([]);const EMPTY_THREAD_INDEX: ReadonlyMap<ThreadId, OrchestrationV2ThreadShell> = new Map();const EMPTY_THREAD_REFS_BY_PROJECT: ReadonlyMap<  ProjectId,  ReadonlyArray<ScopedThreadRef>> = new Map(); export function createEnvironmentThreadShellAtoms(input: {  readonly catalogValueAtom: Atom.Atom<EnvironmentCatalogState>;  readonly snapshotAtom: (    environmentId: EnvironmentId,  ) => Atom.Atom<OrchestrationV2ShellSnapshot | null>;}) {  // Point reads and aggregate lists share values without keeping an atom alive  // for every listed thread. Replaced source objects can be collected.  const scopedThreads = new WeakMap<    OrchestrationV2ThreadShell,    Map<EnvironmentId, EnvironmentThreadShell>  >();  const scopedThread = (environmentId: EnvironmentId, thread: OrchestrationV2ThreadShell) => {    let byEnvironment = scopedThreads.get(thread);    if (byEnvironment === undefined) {      byEnvironment = new Map();      scopedThreads.set(thread, byEnvironment);    }    let value = byEnvironment.get(environmentId);    if (value === undefined) {      value = presentThreadShell(environmentId, thread);      byEnvironment.set(environmentId, value);    }    return value;  };   const environmentThreadsAtom = Atom.family((environmentId: EnvironmentId) =>    Atom.make(      (get): ReadonlyArray<OrchestrationV2ThreadShell> =>        get(input.snapshotAtom(environmentId))?.threads ?? EMPTY_THREADS,    ).pipe(Atom.withLabel(`environment-threads:${environmentId}`)),  );   const environmentThreadIndexAtom = Atom.family((environmentId: EnvironmentId) =>    Atom.make((get): ReadonlyMap<ThreadId, OrchestrationV2ThreadShell> => {      const threads = get(environmentThreadsAtom(environmentId));      if (threads.length === 0) {        return EMPTY_THREAD_INDEX;      }      return new Map(threads.map((thread) => [thread.id, thread] as const));    }).pipe(Atom.withLabel(`environment-thread-index:${environmentId}`)),  );   const environmentThreadRefsAtom = Atom.family((environmentId: EnvironmentId) => {    let previous: ReadonlyArray<ScopedThreadRef> = [];    return Atom.make((get) => {      const next = get(environmentThreadsAtom(environmentId)).map((thread) => ({        environmentId,        threadId: thread.id,      }));      if (threadRefsEqual(previous, next)) {        return previous;      }      previous = next;      return next;    }).pipe(Atom.withLabel(`environment-thread-refs:${environmentId}`));  });   const environmentThreadRefsByProjectAtom = Atom.family((environmentId: EnvironmentId) => {    let previous: ReadonlyMap<      ProjectId,      ReadonlyArray<ScopedThreadRef>    > = EMPTY_THREAD_REFS_BY_PROJECT;    return Atom.make((get) => {      const grouped = new Map<ProjectId, ScopedThreadRef[]>();      for (const thread of get(environmentThreadsAtom(environmentId))) {        const refs = grouped.get(thread.projectId);        const ref = { environmentId, threadId: thread.id };        if (refs === undefined) {          grouped.set(thread.projectId, [ref]);        } else {          refs.push(ref);        }      }      if (grouped.size === 0) {        previous = EMPTY_THREAD_REFS_BY_PROJECT;        return previous;      }      const next = new Map<ProjectId, ReadonlyArray<ScopedThreadRef>>();      for (const [projectId, refs] of grouped) {        const previousRefs = previous.get(projectId);        next.set(          projectId,          previousRefs !== undefined && threadRefsEqual(previousRefs, refs) ? previousRefs : refs,        );      }      const previousProjectIds = [...previous.keys()];      if (        next.size === previous.size &&        [...next].every(          ([projectId, refs], index) =>            previousProjectIds[index] === projectId && previous.get(projectId) === refs,        )      ) {        return previous;      }      previous = next;      return previous;    }).pipe(Atom.withLabel(`environment-thread-refs-by-project:${environmentId}`));  });   const threadShellAtomFamily = Atom.family((key: string) => {    const ref = parseThreadKey(key);    return Atom.make((get) => {      const source = get(environmentThreadIndexAtom(ref.environmentId)).get(ref.threadId) ?? null;      return source === null ? null : scopedThread(ref.environmentId, source);    }).pipe(Atom.withLabel(`environment-thread-shell:${key}`));  });   const threadShellsForProjectRefsAtomFamily = Atom.family((key: string) => {    const projectRefs = parseProjectRefCollectionKey(key);    let previous: ReadonlyArray<EnvironmentThreadShell> = [];    return Atom.make((get) => {      const next: EnvironmentThreadShell[] = [];      const seen = new Set<string>();      for (const projectRef of projectRefs) {        const refs =          get(environmentThreadRefsByProjectAtom(projectRef.environmentId)).get(            projectRef.projectId,          ) ?? EMPTY_SCOPED_THREAD_REFS;        if (refs.length === 0) continue;        const threads = get(environmentThreadIndexAtom(projectRef.environmentId));        for (const ref of refs) {          const key = threadKey(ref);          if (seen.has(key)) {            continue;          }          seen.add(key);          const thread = threads.get(ref.threadId);          if (thread !== undefined) {            next.push(scopedThread(ref.environmentId, thread));          }        }      }      if (arrayElementsEqual(previous, next)) {        return previous;      }      previous = next;      return previous;    }).pipe(Atom.withLabel(`environment-thread-shells-for-projects:${key}`));  });   let previousThreadRefs: ReadonlyArray<ScopedThreadRef> = [];  const threadRefsAtom = Atom.make((get) => {    const refs: ScopedThreadRef[] = [];    for (const environmentId of get(input.catalogValueAtom).entries.keys()) {      refs.push(...get(environmentThreadRefsAtom(environmentId)));    }    if (threadRefsEqual(previousThreadRefs, refs)) {      return previousThreadRefs;    }    previousThreadRefs = refs;    return refs;  }).pipe(Atom.withLabel("environment-thread-refs"));   let previousThreadShells: ReadonlyArray<EnvironmentThreadShell> = [];  const threadShellsAtom = Atom.make((get) => {    const next: EnvironmentThreadShell[] = [];    for (const environmentId of get(input.catalogValueAtom).entries.keys()) {      for (const thread of get(environmentThreadsAtom(environmentId))) {        next.push(scopedThread(environmentId, thread));      }    }    if (arrayElementsEqual(previousThreadShells, next)) {      return previousThreadShells;    }    previousThreadShells = next;    return previousThreadShells;  }).pipe(Atom.withLabel("environment-thread-shell-list"));   return {    environmentThreadsAtom,    environmentThreadIndexAtom,    environmentThreadRefsAtom,    environmentThreadRefsByProjectAtom,    threadRefsAtom,    threadShellsAtom,    threadShellsForProjectRefsAtom: (refs: ReadonlyArray<ScopedProjectRef>) =>      threadShellsForProjectRefsAtomFamily(projectRefCollectionKey(refs)),    threadShellAtom: (ref: ScopedThreadRef) => threadShellAtomFamily(threadKey(ref)),  };}