mirror of
https://github.com/wavetermdev/waveterm.git
synced 2024-12-22 16:48:23 +01:00
54 lines
2.2 KiB
TypeScript
54 lines
2.2 KiB
TypeScript
// Copyright 2024, Command Line Inc.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
import { WOS } from "@/app/store/global";
|
|
import { Atom, atom, Getter } from "jotai";
|
|
import { LayoutTreeState, WritableLayoutTreeStateAtom } from "./types";
|
|
|
|
const layoutStateAtomMap: WeakMap<Atom<Tab>, WritableLayoutTreeStateAtom> = new WeakMap();
|
|
|
|
function getLayoutStateAtomFromTab(tabAtom: Atom<Tab>, get: Getter): WritableWaveObjectAtom<LayoutState> {
|
|
const tabData = get(tabAtom);
|
|
if (!tabData) return;
|
|
const layoutStateOref = WOS.makeORef("layout", tabData.layoutstate);
|
|
const layoutStateAtom = WOS.getWaveObjectAtom<LayoutState>(layoutStateOref);
|
|
return layoutStateAtom;
|
|
}
|
|
|
|
export function withLayoutTreeStateAtomFromTab(tabAtom: Atom<Tab>): WritableLayoutTreeStateAtom {
|
|
if (layoutStateAtomMap.has(tabAtom)) {
|
|
// console.log("found atom");
|
|
return layoutStateAtomMap.get(tabAtom);
|
|
}
|
|
const generationAtom = atom(1);
|
|
const treeStateAtom: WritableLayoutTreeStateAtom = atom(
|
|
(get) => {
|
|
const stateAtom = getLayoutStateAtomFromTab(tabAtom, get);
|
|
if (!stateAtom) return;
|
|
const layoutStateData = get(stateAtom);
|
|
// console.log("layoutStateData", layoutStateData);
|
|
const layoutTreeState: LayoutTreeState = {
|
|
rootNode: layoutStateData?.rootnode,
|
|
magnifiedNodeId: layoutStateData?.magnifiednodeid,
|
|
generation: get(generationAtom),
|
|
};
|
|
return layoutTreeState;
|
|
},
|
|
(get, set, value) => {
|
|
if (get(generationAtom) < value.generation) {
|
|
const stateAtom = getLayoutStateAtomFromTab(tabAtom, get);
|
|
// console.log("setting new atom val", value);
|
|
if (!stateAtom) return;
|
|
const waveObjVal = get(stateAtom);
|
|
// console.log("waveObjVal", waveObjVal);
|
|
waveObjVal.rootnode = value.rootNode;
|
|
waveObjVal.magnifiednodeid = value.magnifiedNodeId;
|
|
set(generationAtom, value.generation);
|
|
set(stateAtom, waveObjVal);
|
|
}
|
|
}
|
|
);
|
|
layoutStateAtomMap.set(tabAtom, treeStateAtom);
|
|
return treeStateAtom;
|
|
}
|