2024-08-21 02:01:29 +02:00
|
|
|
// Copyright 2024, Command Line Inc.
|
|
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
|
2024-08-26 20:56:00 +02:00
|
|
|
import { atoms, globalStore, WOS } from "@/app/store/global";
|
2024-08-15 03:40:41 +02:00
|
|
|
import useResizeObserver from "@react-hook/resize-observer";
|
2024-08-15 22:45:45 +02:00
|
|
|
import { Atom, useAtomValue } from "jotai";
|
2024-08-26 20:56:00 +02:00
|
|
|
import { useEffect } from "react";
|
2024-08-15 03:40:41 +02:00
|
|
|
import { withLayoutTreeStateAtomFromTab } from "./layoutAtom";
|
2024-08-15 04:43:25 +02:00
|
|
|
import { LayoutModel } from "./layoutModel";
|
2024-08-26 20:56:00 +02:00
|
|
|
import { LayoutNode, NodeModel, TileLayoutContents } from "./types";
|
2024-08-15 03:40:41 +02:00
|
|
|
|
|
|
|
const layoutModelMap: Map<string, LayoutModel> = new Map();
|
|
|
|
|
|
|
|
export function getLayoutModelForTab(tabAtom: Atom<Tab>): LayoutModel {
|
|
|
|
const tabData = globalStore.get(tabAtom);
|
|
|
|
if (!tabData) return;
|
|
|
|
const tabId = tabData.oid;
|
|
|
|
if (layoutModelMap.has(tabId)) {
|
|
|
|
const layoutModel = layoutModelMap.get(tabData.oid);
|
|
|
|
if (layoutModel) {
|
|
|
|
return layoutModel;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
const layoutTreeStateAtom = withLayoutTreeStateAtomFromTab(tabAtom);
|
|
|
|
const layoutModel = new LayoutModel(layoutTreeStateAtom, globalStore.get, globalStore.set);
|
|
|
|
globalStore.sub(layoutTreeStateAtom, () => layoutModel.updateTreeState());
|
|
|
|
layoutModelMap.set(tabId, layoutModel);
|
|
|
|
return layoutModel;
|
|
|
|
}
|
|
|
|
|
|
|
|
export function getLayoutModelForTabById(tabId: string) {
|
|
|
|
const tabOref = WOS.makeORef("tab", tabId);
|
|
|
|
const tabAtom = WOS.getWaveObjectAtom<Tab>(tabOref);
|
|
|
|
return getLayoutModelForTab(tabAtom);
|
|
|
|
}
|
|
|
|
|
2024-08-26 20:56:00 +02:00
|
|
|
export function getLayoutModelForActiveTab() {
|
|
|
|
const tabId = globalStore.get(atoms.activeTabId);
|
|
|
|
return getLayoutModelForTabById(tabId);
|
|
|
|
}
|
|
|
|
|
2024-08-15 03:40:41 +02:00
|
|
|
export function deleteLayoutModelForTab(tabId: string) {
|
|
|
|
if (layoutModelMap.has(tabId)) layoutModelMap.delete(tabId);
|
|
|
|
}
|
|
|
|
|
|
|
|
export function useLayoutModel(tabAtom: Atom<Tab>): LayoutModel {
|
|
|
|
return getLayoutModelForTab(tabAtom);
|
|
|
|
}
|
|
|
|
|
|
|
|
export function useTileLayout(tabAtom: Atom<Tab>, tileContent: TileLayoutContents): LayoutModel {
|
2024-08-21 05:14:14 +02:00
|
|
|
// Use tab data to ensure we can reload if the tab is disposed and remade (such as during Hot Module Reloading)
|
|
|
|
useAtomValue(tabAtom);
|
2024-08-15 03:40:41 +02:00
|
|
|
const layoutModel = useLayoutModel(tabAtom);
|
|
|
|
useResizeObserver(layoutModel?.displayContainerRef, layoutModel?.onContainerResize);
|
|
|
|
useEffect(() => layoutModel.registerTileLayout(tileContent), [tileContent]);
|
|
|
|
return layoutModel;
|
|
|
|
}
|
|
|
|
|
2024-08-26 20:56:00 +02:00
|
|
|
export function useNodeModel(layoutModel: LayoutModel, layoutNode: LayoutNode): NodeModel {
|
|
|
|
return layoutModel.getNodeModel(layoutNode);
|
2024-08-15 03:40:41 +02:00
|
|
|
}
|